opendaylight/md-sal/sal-rest-connector/src/test/java/org/opendaylight/controller/sal/restconf/impl/websockets/client/WebSocketClient.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.sal.restconf.impl.websockets.client;
   9
  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
  27 import java.net.URI;
  28
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31
  32 public class WebSocketClient  {
  33
  34     private final URI uri;
  35     private Bootstrap bootstrap = new Bootstrap();;
  36     private final WebSocketClientHandler clientHandler;
  37     private static final Logger logger = LoggerFactory.getLogger(WebSocketClient.class);
  38     private Channel clientChannel;
  39     private final EventLoopGroup group = new NioEventLoopGroup();
  40
  41     public WebSocketClient(URI uri,IClientMessageCallback clientMessageCallback) {
  42         this.uri = uri;
  43         clientHandler = new WebSocketClientHandler(
  44                 WebSocketClientHandshakerFactory.newHandshaker(
  45                         uri, WebSocketVersion.V13, null, false,null),clientMessageCallback); // last null could be replaced with DefaultHttpHeaders
  46         initialize();
  47     }
  48     private void initialize(){
  49
  50         String protocol = uri.getScheme();
  51         if (!"http".equals(protocol)) {
  52             throw new IllegalArgumentException("Unsupported protocol: " + protocol);
  53         }
  54
  55         bootstrap.group(group)
  56                 .channel(NioSocketChannel.class)
  57                 .handler(new ChannelInitializer<SocketChannel>() {
  58                     @Override
  59                     public void initChannel(SocketChannel ch) throws Exception {
  60                         ChannelPipeline pipeline = ch.pipeline();
  61                         pipeline.addLast("http-codec", new HttpClientCodec());
  62                         pipeline.addLast("aggregator", new HttpObjectAggregator(8192));
  63                         pipeline.addLast("ws-handler", clientHandler);
  64                     }
  65                 });
  66     }
  67     public void connect() throws InterruptedException{
  68         System.out.println("WebSocket Client connecting");
  69         clientChannel  = bootstrap.connect(uri.getHost(), uri.getPort()).sync().channel();
  70         clientHandler.handshakeFuture().sync();
  71     }
  72
  73     public void writeAndFlush(String message){
  74         clientChannel.writeAndFlush(new TextWebSocketFrame(message));
  75     }
  76     public void writeAndFlush(Object message){
  77         clientChannel.writeAndFlush(message);
  78     }
  79
  80     public void ping(){
  81         clientChannel.writeAndFlush(new PingWebSocketFrame(Unpooled.copiedBuffer(new byte[]{1, 2, 3, 4, 5, 6})));
  82     }
  83
  84     public void close() throws InterruptedException {
  85         clientChannel.writeAndFlush(new CloseWebSocketFrame());
  86
  87         // WebSocketClientHandler will close the connection when the server
  88         // responds to the CloseWebSocketFrame.
  89         clientChannel.closeFuture().sync();
  90         group.shutdownGracefully();
  91     }
  92
  93     public static void main(String[] args) throws Exception {
  94         URI uri;
  95         if (args.length > 0) {
  96             uri = new URI(args[0]);
  97         } else {
  98             uri = new URI("http://192.168.11.1:8181/opendaylight-inventory:nodes");
  99         }
 100         IClientMessageCallback messageCallback = new ClientMessageCallback();
 101         WebSocketClient webSocketClient = new WebSocketClient(uri, messageCallback);
 102         webSocketClient.connect();
 103     }
 104
 105     private static class ClientMessageCallback implements IClientMessageCallback {
 106         @Override
 107         public void onMessageReceived(Object message) {
 108             logger.info("received message {}", ((TextWebSocketFrame)message).text());
 109         }
 110     }
 111
 112 }