Play Framework - 在 Web 套接字响应头中编辑 Sec-WebSocket-Protocol答案

【问题标题】：Play Framework - edit Sec-WebSocket-Protocol in Web socket Response HeaderPlay Framework - 在 Web 套接字响应头中编辑 Sec-WebSocket-Protocol
【发布时间】：2019-07-17 06:50:32
【问题描述】：

编辑从服务器发送到客户端的 Web 套接字头响应。

我正在使用 playframework 创建一个 websocket 服务器应用程序。现在来自服务器的 websocket 响应是由 Play 照顾。以下是响应头，

Request Header:
  (UpgradeToWebSocket,),
  (Host,localhost:8083), 
  (Connection,Upgrade), 
  (Upgrade,websocket),
  (Sec-WebSocket-Version,13), 
  (Accept-Encoding,gzip, deflate, br), 
  (Accept-Language,en-US,en;q=0.9),
  (Sec-WebSocket-Key,ZvfzpVo3EX4DFA4BRcgRIA==)

def chatSystem(): WebSocket = WebSocket.acceptOrResult[String, String] { request =>
    Future.successful{
      AuthenticationService.doBasicAuthentication(request.headers) match {
        case Results.Ok => Right(ActorFlow.actorRef { out => ChatServiceActor.props(out) })
        case _ => Left(Unauthorized)
      }
    }
 }

我想验证 Sec-WebSocket-Protocol 是否存在于请求标头中，或者如果它不存在，则在服务器响应中添加与值相同的值。

【问题讨论】：

从看了一下播放源，我不认为这是应该的，我想您可以尝试克隆 WebSocket 类，看看您是否可以支持看起来并不简单的更改.

标签： scala websocket playframework akka

【解决方案1】：

我使用了以下代码：

// Defined at http://tools.ietf.org/html/rfc6455#section-4.2.2
val MagicGuid = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"

def websocketAcceptForKey(key: String): String = {
  val sha1 = MessageDigest.getInstance("sha1")
  val salted = key + MagicGuid
  val hash = sha1.digest(salted.asciiBytes)
  val acceptKey: String = Base64.rfc2045().encodeToString(hash, false)
  acceptKey
}

像下面这样使用它：

val wsKey: Optional[HttpHeader] = request.getHeader("Sec-WebSocket-Key")
val wsAccept = if (wsKey.isPresent) Some(RawHeader("Sec-WebSocket-Accept", websocketAcceptForKey(wsKey.get.value()))) else None

【讨论】：