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.opendaylight.controller.sal.restconf.impl.websockets.client;
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;
27 import java.io.BufferedReader;
28 import java.io.InputStreamReader;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
34 public class WebSocketClient {
36 private final URI uri;
37 private Bootstrap bootstrap = new Bootstrap();;
38 private final WebSocketClientHandler clientHandler;
39 private static final Logger logger = LoggerFactory.getLogger(WebSocketClient.class);
40 private Channel clientChannel;
41 private final EventLoopGroup group = new NioEventLoopGroup();
43 public WebSocketClient(URI uri,IClientMessageCallback clientMessageCallback) {
45 clientHandler = new WebSocketClientHandler(
46 WebSocketClientHandshakerFactory.newHandshaker(
47 uri, WebSocketVersion.V13, null, false,null),clientMessageCallback); // last null could be replaced with DefaultHttpHeaders
50 private void initialize(){
52 String protocol = uri.getScheme();
53 if (!"http".equals(protocol)) {
54 throw new IllegalArgumentException("Unsupported protocol: " + protocol);
57 bootstrap.group(group)
58 .channel(NioSocketChannel.class)
59 .handler(new ChannelInitializer<SocketChannel>() {
61 public void initChannel(SocketChannel ch) throws Exception {
62 ChannelPipeline pipeline = ch.pipeline();
63 pipeline.addLast("http-codec", new HttpClientCodec());
64 pipeline.addLast("aggregator", new HttpObjectAggregator(8192));
65 pipeline.addLast("ws-handler", clientHandler);
69 public void connect() throws InterruptedException{
70 System.out.println("WebSocket Client connecting");
71 clientChannel = bootstrap.connect(uri.getHost(), uri.getPort()).sync().channel();
72 clientHandler.handshakeFuture().sync();
75 public void writeAndFlush(String message){
76 clientChannel.writeAndFlush(new TextWebSocketFrame(message));
78 public void writeAndFlush(Object message){
79 clientChannel.writeAndFlush(message);
83 clientChannel.writeAndFlush(new PingWebSocketFrame(Unpooled.copiedBuffer(new byte[]{1, 2, 3, 4, 5, 6})));
86 public void close(String reasonText) throws InterruptedException {
87 CloseWebSocketFrame closeWebSocketFrame = new CloseWebSocketFrame(1000,reasonText);
88 clientChannel.writeAndFlush(closeWebSocketFrame);
90 // WebSocketClientHandler will close the connection when the server
91 // responds to the CloseWebSocketFrame.
92 clientChannel.closeFuture().sync();
93 group.shutdownGracefully();
96 public static void main(String[] args) throws Exception {
98 if (args.length > 0) {
99 uri = new URI(args[0]);
101 uri = new URI("http://192.168.1.101:8181/opendaylight-inventory:nodes");
103 IClientMessageCallback messageCallback = new ClientMessageCallback();
104 WebSocketClient webSocketClient = new WebSocketClient(uri, messageCallback);
105 webSocketClient.connect();
108 BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
109 String input = br.readLine();
110 if (input.equals("q")) {
111 System.out.print("Would you like to close stream? (Y = yes, empty = yes)\n");
112 input = br.readLine();
113 if (input.equals("yes") || input.isEmpty()) {
114 webSocketClient.close("opendaylight-inventory:nodes");
121 private static class ClientMessageCallback implements IClientMessageCallback {
123 public void onMessageReceived(Object message) {
124 if (message instanceof TextWebSocketFrame) {
125 logger.info("received message {}"+ ((TextWebSocketFrame)message).text());