• Home
• io.github.cactacea
• finasocket_2.12
• 19.11.0
• source code
• WebSocketClientHandler.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.github.cactacea.finasocket.WebSocketClientHandler.scala Maven / Gradle / Ivy

package io.github.cactacea.finasocket

import io.netty.channel._
import io.netty.handler.codec.http.FullHttpResponse
import io.netty.handler.codec.http.websocketx._

class WebSocketClientHandler(handshaker:WebSocketClientHandshaker)
  extends ChannelInboundHandlerAdapter {

  import io.netty.channel.ChannelHandlerContext

  var handshakeFuture:ChannelPromise = _

  override def channelActive(ctx: ChannelHandlerContext): Unit = {
    handshaker.handshake(ctx.channel())
  }

  override def channelInactive(ctx: ChannelHandlerContext): Unit = {
  }

  override def handlerAdded(ctx:ChannelHandlerContext):Unit = {
    handshakeFuture = ctx.newPromise()
  }

  override def channelRead(ctx: ChannelHandlerContext, msg: AnyRef): Unit = {
    val ch = ctx.channel()
    if(! handshaker.isHandshakeComplete){
      handshaker.finishHandshake(ch, msg.asInstanceOf[FullHttpResponse])
      handshakeFuture.setSuccess()
    }
    super.channelRead(ctx, msg)
  }

  override def exceptionCaught(ctx:ChannelHandlerContext, cause:Throwable):Unit = {
    cause.printStackTrace()
    ctx.close()
  }

}