io.scalecube.transport.netty.websocket.WebsocketReceiver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalecube-transport-netty Show documentation

There is a newer version: 2.7.0.rc

Show newest version

package io.scalecube.transport.netty.websocket;

import io.netty.channel.ChannelOption;
import io.scalecube.cluster.transport.api.TransportConfig;
import io.scalecube.transport.netty.Receiver;
import io.scalecube.transport.netty.TransportImpl.ReceiverContext;
import reactor.core.publisher.Mono;
import reactor.netty.DisposableServer;
import reactor.netty.http.server.HttpServer;
import reactor.netty.http.server.HttpServerRequest;
import reactor.netty.http.server.HttpServerResponse;
import reactor.netty.http.websocket.WebsocketInbound;
import reactor.netty.http.websocket.WebsocketOutbound;

public final class WebsocketReceiver implements Receiver {

  private final TransportConfig config;

  public WebsocketReceiver(TransportConfig config) {
    this.config = config;
  }

  @Override
  public Mono bind() {
    return Mono.deferContextual(context -> Mono.just(context.get(ReceiverContext.class)))
        .flatMap(
            context ->
                newHttpServer(context)
                    .handle((request, response) -> onMessage(context, request, response))
                    .bind()
                    .cast(DisposableServer.class));
  }

  private HttpServer newHttpServer(ReceiverContext context) {
    return HttpServer.create()
        .runOn(context.loopResources())
        .port(config.port())
        .childOption(ChannelOption.TCP_NODELAY, true)
        .childOption(ChannelOption.SO_KEEPALIVE, true)
        .childOption(ChannelOption.SO_REUSEADDR, true);
  }

  private Mono onMessage(
      ReceiverContext context,
      @SuppressWarnings("unused") HttpServerRequest request,
      HttpServerResponse response) {
    return response.sendWebsocket(
        (WebsocketInbound inbound, WebsocketOutbound outbound) ->
            inbound.receive().retain().doOnNext(context::onMessage).then());
  }
}