Java tutorial
/* * Copyright (c) 2015 Huawei, Inc and others. All rights reserved. * * This program and the accompanying materials are made available under the * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/epl-v10.html */ package org.opendaylight.usc.agent; import io.netty.bootstrap.Bootstrap; import io.netty.bootstrap.ServerBootstrap; import io.netty.channel.Channel; import io.netty.channel.ChannelFuture; import io.netty.channel.ChannelInitializer; import io.netty.channel.ChannelPipeline; import io.netty.channel.EventLoopGroup; import io.netty.channel.nio.NioEventLoopGroup; import io.netty.channel.socket.nio.NioServerSocketChannel; import io.netty.channel.socket.nio.NioSocketChannel; import io.netty.handler.logging.LogLevel; import io.netty.handler.logging.LoggingHandler; import java.net.InetAddress; import java.net.UnknownHostException; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentMap; import org.opendaylight.usc.manager.UscConfigurationServiceImpl; import org.opendaylight.usc.manager.api.UscSecureService; import org.opendaylight.usc.plugin.UscFrameDecoderTcp; import org.opendaylight.usc.plugin.UscFrameEncoderTcp; import org.opendaylight.usc.plugin.UscPlugin; import org.opendaylight.usc.plugin.model.UscSessionImpl; import org.opendaylight.usc.protocol.UscControl; import org.opendaylight.usc.util.UscServiceUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import com.google.common.util.concurrent.SettableFuture; public class UscAgentTcp implements Runnable, AutoCloseable { private static final Logger LOG = LoggerFactory.getLogger(UscAgentTcp.class); static final int PORT = Integer.parseInt(System.getProperty("port", "1068")); final EventLoopGroup bossGroup = new NioEventLoopGroup(1); final EventLoopGroup workerGroup = new NioEventLoopGroup(); final ServerBootstrap b = new ServerBootstrap(); final EventLoopGroup callHomeGroup = new NioEventLoopGroup(); final Bootstrap cb = new Bootstrap(); private Channel agentServerChannel = null; private ConcurrentMap<Integer, SettableFuture<Boolean>> closeFuture = new ConcurrentHashMap<>(); private UscSecureService secureService = null; public UscAgentTcp(boolean callHome) { this(callHome, InetAddress.getLoopbackAddress()); } public UscAgentTcp(boolean callHome, InetAddress host) { this(callHome, host, "src/test/resources/etc/usc/usc.properties"); } public UscAgentTcp(boolean callHome, InetAddress host, String propertyFile) { final UscAgentTcp agent = this; UscConfigurationServiceImpl.setDefaultPropertyFilePath(propertyFile); secureService = UscServiceUtils.getService(UscSecureService.class); b.group(bossGroup, workerGroup); b.channel(NioServerSocketChannel.class); b.handler(new LoggingHandler("UscAgentTcp server handler", LogLevel.TRACE)); b.childHandler(new ChannelInitializer<NioSocketChannel>() { @Override public void initChannel(NioSocketChannel ch) throws Exception { if (secureService == null) { LOG.error("UscSecureService is not initialized!"); return; } ChannelPipeline p = ch.pipeline(); agentServerChannel = ch; p.addLast(new LoggingHandler("UscAgentTcp PLUGIN5", LogLevel.TRACE)); p.addLast(secureService.getTcpServerHandler(ch)); p.addLast(new LoggingHandler("UscAgentTcp PLUGIN4", LogLevel.TRACE)); p.addLast(new UscFrameEncoderTcp()); p.addLast(new LoggingHandler("UscAgentTcp PLUGIN3", LogLevel.TRACE)); p.addLast(new UscFrameDecoderTcp()); p.addLast(new LoggingHandler("UscAgentTcp PLUGIN2", LogLevel.TRACE)); p.addLast(new UscAgentTcpHandler(agent, ch)); p.addLast(new LoggingHandler("UscAgentTcp PLUGIN1", LogLevel.TRACE)); } }); if (callHome) { cb.group(callHomeGroup); cb.channel(NioSocketChannel.class); cb.handler(new ChannelInitializer<NioSocketChannel>() { @Override public void initChannel(NioSocketChannel ch) throws Exception { if (secureService == null) { LOG.error("UscSecureService is not initialized!"); return; } ChannelPipeline p = ch.pipeline(); agentServerChannel = ch; p.addLast(new LoggingHandler("UscAgentTcp Handler5", LogLevel.TRACE)); p.addLast(secureService.getTcpClientHandler(ch)); p.addLast(new LoggingHandler("UscAgentTcp Handler4", LogLevel.TRACE)); p.addLast(new UscFrameEncoderTcp()); p.addLast(new LoggingHandler("UscAgentTcp Handler3", LogLevel.TRACE)); p.addLast(new UscFrameDecoderTcp()); p.addLast(new LoggingHandler("UscAgentTcp Handler2", LogLevel.TRACE)); p.addLast(new UscAgentTcpHandler(agent, ch)); p.addLast(new LoggingHandler("UscAgentTcp Handler1", LogLevel.TRACE)); } }); try { cb.connect(host, 1069).sync(); } catch (Exception e) { // TODO Auto-generated catch block LOG.error("USC Plugin call home port not available; call home disabled"); // e.printStackTrace(); } } } protected ConcurrentMap<Integer, SettableFuture<Boolean>> getCloseFuture() { return closeFuture; } public SettableFuture<Boolean> closeClientInternalConnection(Channel clientChannel) { try { UscSessionImpl session = clientChannel.attr(UscPlugin.SESSION).get().get(); closeFuture.remove(session.getSessionId()); closeFuture.putIfAbsent(session.getSessionId(), SettableFuture.<Boolean>create()); UscControl data = new UscControl(session.getPort(), session.getSessionId(), 1); if (agentServerChannel != null) agentServerChannel.writeAndFlush(data); LOG.trace("UscAgentTcp closeClientInternalConnection port#: " + session.getPort() + " ,session#: " + session.getSessionId()); return closeFuture.get(session.getSessionId()); } catch (Exception e) { e.printStackTrace(); return null; } } @Override public void run() { // Start the server. try { ChannelFuture f = b.bind(PORT).sync(); LOG.trace("UscAgentTcp initialized"); // Wait until the server socket is closed. f.channel().closeFuture().sync(); } catch (InterruptedException e) { // TODO Auto-generated catch block e.printStackTrace(); } } @Override public void close() throws Exception { bossGroup.shutdownGracefully(); workerGroup.shutdownGracefully(); callHomeGroup.shutdownGracefully(); } public static void main(String[] args) throws Exception { boolean callHome = false; InetAddress host = InetAddress.getLoopbackAddress(); if (args.length > 0) { try { InetAddress ip = InetAddress.getByName(args[0]); callHome = true; host = ip; } catch (UnknownHostException e) { System.err.println("Argument " + args[0] + " must be an iP address (callhome IP)."); System.exit(1); } } try (UscAgentTcp agent = new UscAgentTcp(callHome, host)) { agent.run(); } } }