2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.onap.ccsdk.sli.plugins.fabricdiscovery;
10 import io.netty.bootstrap.Bootstrap;
11 import io.netty.buffer.Unpooled;
12 import io.netty.channel.Channel;
13 import io.netty.channel.ChannelInitializer;
14 import io.netty.channel.ChannelPipeline;
15 import io.netty.channel.EventLoopGroup;
16 import io.netty.channel.nio.NioEventLoopGroup;
17 import io.netty.channel.socket.SocketChannel;
18 import io.netty.channel.socket.nio.NioSocketChannel;
19 import io.netty.handler.codec.http.HttpClientCodec;
20 import io.netty.handler.codec.http.HttpObjectAggregator;
21 import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame;
22 import io.netty.handler.codec.http.websocketx.PingWebSocketFrame;
23 import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
24 import io.netty.handler.codec.http.websocketx.WebSocketClientHandshakerFactory;
25 import io.netty.handler.codec.http.websocketx.WebSocketVersion;
26 import java.io.BufferedReader;
27 import java.io.IOException;
28 import java.io.InputStreamReader;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
33 public class WebSocketClient {
35 private static final Logger LOG = LoggerFactory.getLogger(WebSocketClient.class);
37 private final URI uri;
38 private final Bootstrap bootstrap = new Bootstrap();
39 private final WebSocketClientHandler clientHandler;
40 private Channel clientChannel;
41 private final EventLoopGroup group = new NioEventLoopGroup();
43 public WebSocketClient(final URI uri, final IClientMessageCallback clientMessageCallback) {
45 clientHandler = new WebSocketClientHandler(WebSocketClientHandshakerFactory.newHandshaker(uri,
46 WebSocketVersion.V13, null, false, null), clientMessageCallback);
47 // last null could be replaced with DefaultHttpHeaders
51 public void initialize() {
53 String protocol = uri.getScheme();
54 if (!"ws".equals(protocol)) {
55 throw new IllegalArgumentException("Unsupported protocol: " + protocol);
58 bootstrap.group(group).channel(NioSocketChannel.class).handler(new ChannelInitializer<SocketChannel>() {
60 public void initChannel(final SocketChannel ch) throws Exception {
61 ChannelPipeline pipeline = ch.pipeline();
62 pipeline.addLast("http-codec", new HttpClientCodec());
63 pipeline.addLast("aggregator", new HttpObjectAggregator(8192));
64 pipeline.addLast("ws-handler", clientHandler);
69 public void connect() throws InterruptedException {
70 LOG.info("WebSocket Client connecting");
71 clientChannel = bootstrap.connect(uri.getHost(), uri.getPort()).sync().channel();
72 clientHandler.handshakeFuture().sync();
75 public void writeAndFlush(final String message) {
76 clientChannel.writeAndFlush(new TextWebSocketFrame(message));
79 public void writeAndFlush(final Object message) {
80 clientChannel.writeAndFlush(message);
84 clientChannel.writeAndFlush(new PingWebSocketFrame(Unpooled.copiedBuffer(new byte[] { 1, 2, 3, 4, 5, 6 })));
87 public void close(final String reasonText) throws InterruptedException {
88 CloseWebSocketFrame closeWebSocketFrame = new CloseWebSocketFrame(1000, reasonText);
89 clientChannel.writeAndFlush(closeWebSocketFrame);
91 // WebSocketClientHandler will close the connection when the server
92 // responds to the CloseWebSocketFrame.
93 clientChannel.closeFuture().sync();
94 group.shutdownGracefully();