opendaylight/md-sal/sal-rest-connector/src/main/java/org/opendaylight/controller/sal/streams/websockets/WebSocketServer.java

   1 package org.opendaylight.controller.sal.streams.websockets;
   2
   3 import io.netty.bootstrap.ServerBootstrap;
   4 import io.netty.channel.Channel;
   5 import io.netty.channel.EventLoopGroup;
   6 import io.netty.channel.nio.NioEventLoopGroup;
   7 import io.netty.channel.socket.nio.NioServerSocketChannel;
   8
   9 import org.opendaylight.controller.sal.streams.listeners.Notificator;
  10 import org.slf4j.Logger;
  11 import org.slf4j.LoggerFactory;
  12
  13 /**
  14  * {@link WebSocketServer} is responsible to start and stop web socket server at
  15  * {@link #PORT}.
  16  */
  17 public class WebSocketServer implements Runnable {
  18
  19         private static final Logger logger = LoggerFactory
  20                         .getLogger(WebSocketServer.class);
  21
  22         public static final int PORT = 8181;
  23         private EventLoopGroup bossGroup;
  24         private EventLoopGroup workerGroup;
  25
  26         @Override
  27         public void run() {
  28                 bossGroup = new NioEventLoopGroup();
  29                 workerGroup = new NioEventLoopGroup();
  30                 try {
  31                         ServerBootstrap b = new ServerBootstrap();
  32                         b.group(bossGroup, workerGroup)
  33                                         .channel(NioServerSocketChannel.class)
  34                                         .childHandler(new WebSocketServerInitializer());
  35
  36                         Channel ch = b.bind(PORT).sync().channel();
  37                         logger.info("Web socket server started at port {}.", PORT);
  38
  39                         ch.closeFuture().sync();
  40                 } catch (InterruptedException e) {
  41                         // NOOP
  42                 } finally {
  43                         stop();
  44                 }
  45         }
  46
  47         /**
  48          * Stops the web socket server and removes all listeners.
  49          */
  50         private void stop() {
  51                 Notificator.removeAllListeners();
  52                 if (bossGroup != null) {
  53                         bossGroup.shutdownGracefully();
  54                 }
  55                 if (workerGroup != null) {
  56                         workerGroup.shutdownGracefully();
  57                 }
  58         }
  59
  60 }