com.google.firebase.database.connection.NettyWebSocketClient Maven / Gradle / Ivy
Show all versions of firebase-admin Show documentation
package com.google.firebase.database.connection;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
import com.google.common.base.Strings;
import com.google.firebase.internal.FirebaseScheduledExecutor;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.http.DefaultHttpHeaders;
import io.netty.handler.codec.http.FullHttpResponse;
import io.netty.handler.codec.http.HttpClientCodec;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketClientHandshaker;
import io.netty.handler.codec.http.websocketx.WebSocketClientHandshakerFactory;
import io.netty.handler.codec.http.websocketx.WebSocketHandshakeException;
import io.netty.handler.codec.http.websocketx.WebSocketVersion;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.SslContextBuilder;
import java.io.EOFException;
import java.net.URI;
import java.security.KeyStore;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ThreadFactory;
import javax.net.ssl.TrustManagerFactory;
/**
* A {@link WebsocketConnection.WSClient} implementation based on the Netty framework. Uses
* a single-threaded NIO event loop to read and write bytes from a WebSocket connection. Netty
* handles all the low-level IO, SSL and WebSocket handshake, and other protocol-specific details.
*
* This implementation does not initiate connection close on its own. In case of errors or loss
* of connectivity, it notifies the higher layer ({@link WebsocketConnection}), which then decides
* whether to initiate a connection tear down.
*/
class NettyWebSocketClient implements WebsocketConnection.WSClient {
private static final int DEFAULT_WSS_PORT = 443;
private final URI uri;
private final WebsocketConnection.WSClientEventHandler eventHandler;
private final ChannelHandler channelHandler;
private final ExecutorService executorService;
private final EventLoopGroup group;
private Channel channel;
NettyWebSocketClient(
URI uri, String userAgent, ThreadFactory threadFactory,
WebsocketConnection.WSClientEventHandler eventHandler) {
this.uri = checkNotNull(uri, "uri must not be null");
this.eventHandler = checkNotNull(eventHandler, "event handler must not be null");
this.channelHandler = new WebSocketClientHandler(uri, userAgent, eventHandler);
this.executorService = new FirebaseScheduledExecutor(threadFactory,
"firebase-websocket-worker");
this.group = new NioEventLoopGroup(1, this.executorService);
}
@Override
public void connect() {
checkState(channel == null, "channel already initialized");
try {
TrustManagerFactory trustFactory = TrustManagerFactory.getInstance(
TrustManagerFactory.getDefaultAlgorithm());
trustFactory.init((KeyStore) null);
final SslContext sslContext = SslContextBuilder.forClient()
.trustManager(trustFactory).build();
Bootstrap bootstrap = new Bootstrap();
final int port = uri.getPort() != -1 ? uri.getPort() : DEFAULT_WSS_PORT;
bootstrap.group(group)
.channel(NioSocketChannel.class)
.handler(new ChannelInitializer() {
@Override
protected void initChannel(SocketChannel ch) {
ChannelPipeline p = ch.pipeline();
p.addLast(sslContext.newHandler(ch.alloc(), uri.getHost(), port));
p.addLast(
new HttpClientCodec(),
// Set the max size for the HTTP responses. This only applies to the WebSocket
// handshake response from the server.
new HttpObjectAggregator(32 * 1024),
channelHandler);
}
});
ChannelFuture channelFuture = bootstrap.connect(uri.getHost(), port);
this.channel = channelFuture.channel();
channelFuture.addListener(
new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) {
if (!future.isSuccess()) {
eventHandler.onError(future.cause());
}
}
}
);
} catch (Exception e) {
eventHandler.onError(e);
}
}
@Override
public void close() {
checkState(channel != null, "channel not initialized");
try {
channel.close();
} finally {
// The following may leave an active threadDeathWatcher daemon behind. That can be cleaned
// up at a higher level if necessary. See https://github.com/netty/netty/issues/7310.
group.shutdownGracefully();
executorService.shutdown();
}
}
@Override
public void send(String msg) {
checkState(channel != null, "Channel not initialized");
if (!channel.isActive()) {
eventHandler.onError(new EOFException("WebSocket channel became inactive"));
} else {
channel.writeAndFlush(new TextWebSocketFrame(msg));
}
}
/**
* Handles low-level IO events. These events fire on the firebase-websocket-worker thread. We
* notify the {@link WebsocketConnection} on all events, which then hands them off to the
* RunLoop for further processing.
*/
private static class WebSocketClientHandler extends SimpleChannelInboundHandler