netconf/mdsal-netconf-ssh/src/test/java/org/opendaylight/netconf/ssh/EchoServer.java

   1 /*
   2  * Copyright (c) 2014 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.netconf.ssh;
   9
  10 import io.netty.bootstrap.ServerBootstrap;
  11 import io.netty.channel.ChannelFuture;
  12 import io.netty.channel.ChannelInitializer;
  13 import io.netty.channel.ChannelOption;
  14 import io.netty.channel.EventLoopGroup;
  15 import io.netty.channel.local.LocalAddress;
  16 import io.netty.channel.local.LocalChannel;
  17 import io.netty.channel.local.LocalServerChannel;
  18 import io.netty.channel.nio.NioEventLoopGroup;
  19 import io.netty.handler.logging.LogLevel;
  20 import io.netty.handler.logging.LoggingHandler;
  21 import java.io.BufferedReader;
  22 import java.io.IOException;
  23 import java.io.InputStreamReader;
  24 import org.slf4j.Logger;
  25 import org.slf4j.LoggerFactory;
  26
  27 /**
  28  * Echoes back any received data from a client.
  29  */
  30 public class EchoServer implements Runnable {
  31     private static final Logger LOG = LoggerFactory.getLogger(EchoServer.class);
  32
  33     @Override
  34     public void run() {
  35         // Configure the server.
  36         EventLoopGroup bossGroup = new NioEventLoopGroup(1);
  37         EventLoopGroup workerGroup = new NioEventLoopGroup();
  38         try {
  39             ServerBootstrap bootstrap = new ServerBootstrap();
  40             bootstrap.group(bossGroup, workerGroup)
  41                     .channel(LocalServerChannel.class)
  42                     .option(ChannelOption.SO_BACKLOG, 100)
  43                     .handler(new LoggingHandler(LogLevel.INFO))
  44                     .childHandler(new ChannelInitializer<LocalChannel>() {
  45                         @Override
  46                         public void initChannel(final LocalChannel ch) {
  47                             ch.pipeline().addLast(new EchoServerHandler());
  48                         }
  49                     });
  50
  51             // Start the server.
  52             LocalAddress localAddress = new LocalAddress("netconf");
  53             ChannelFuture future = bootstrap.bind(localAddress).sync();
  54
  55             // Wait until the server socket is closed.
  56             future.channel().closeFuture().sync();
  57         } catch (InterruptedException e) {
  58             throw new RuntimeException(e);
  59         } finally {
  60             // Shut down all event loops to terminate all threads.
  61             bossGroup.shutdownGracefully();
  62             workerGroup.shutdownGracefully();
  63         }
  64     }
  65
  66     public static void main(final String[] args) throws InterruptedException, IOException {
  67         new Thread(new EchoServer()).start();
  68         Thread.sleep(1000);
  69         EchoClientHandler clientHandler = new EchoClientHandler();
  70         EchoClient echoClient = new EchoClient(clientHandler);
  71         new Thread(echoClient).start();
  72
  73         BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
  74         do {
  75             String message = reader.readLine();
  76             if (message == null ||  "exit".equalsIgnoreCase(message)) {
  77                 break;
  78             }
  79             LOG.debug("Got '{}'", message);
  80             clientHandler.write(message);
  81         } while (true);
  82         System.exit(0);
  83     }
  84 }