View Javadoc

1   /*
2    * Copyright 2012 The Netty Project
3    *
4    * The Netty Project licenses this file to you under the Apache License,
5    * version 2.0 (the "License"); you may not use this file except in compliance
6    * with the License. You may obtain a copy of the License at:
7    *
8    *   http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13   * License for the specific language governing permissions and limitations
14   * under the License.
15   */
16  package io.netty.example.objectecho;
17  
18  import io.netty.bootstrap.ServerBootstrap;
19  import io.netty.channel.ChannelInitializer;
20  import io.netty.channel.ChannelPipeline;
21  import io.netty.channel.EventLoopGroup;
22  import io.netty.channel.nio.NioEventLoopGroup;
23  import io.netty.channel.socket.SocketChannel;
24  import io.netty.channel.socket.nio.NioServerSocketChannel;
25  import io.netty.example.echo.EchoServer;
26  import io.netty.handler.codec.serialization.ClassResolvers;
27  import io.netty.handler.codec.serialization.ObjectDecoder;
28  import io.netty.handler.codec.serialization.ObjectEncoder;
29  import io.netty.handler.logging.LogLevel;
30  import io.netty.handler.logging.LoggingHandler;
31  import io.netty.handler.ssl.SslContext;
32  import io.netty.handler.ssl.SslContextBuilder;
33  import io.netty.handler.ssl.util.SelfSignedCertificate;
34  
35  /**
36   * Modification of {@link EchoServer} which utilizes Java object serialization.
37   */
38  public final class ObjectEchoServer {
39  
40      static final boolean SSL = System.getProperty("ssl") != null;
41      static final int PORT = Integer.parseInt(System.getProperty("port", "8007"));
42  
43      public static void main(String[] args) throws Exception {
44          // Configure SSL.
45          final SslContext sslCtx;
46          if (SSL) {
47              SelfSignedCertificate ssc = new SelfSignedCertificate();
48              sslCtx = SslContextBuilder.forServer(ssc.certificate(), ssc.privateKey()).build();
49          } else {
50              sslCtx = null;
51          }
52  
53          EventLoopGroup bossGroup = new NioEventLoopGroup(1);
54          EventLoopGroup workerGroup = new NioEventLoopGroup();
55          try {
56              ServerBootstrap b = new ServerBootstrap();
57              b.group(bossGroup, workerGroup)
58               .channel(NioServerSocketChannel.class)
59               .handler(new LoggingHandler(LogLevel.INFO))
60               .childHandler(new ChannelInitializer<SocketChannel>() {
61                  @Override
62                  public void initChannel(SocketChannel ch) throws Exception {
63                      ChannelPipeline p = ch.pipeline();
64                      if (sslCtx != null) {
65                          p.addLast(sslCtx.newHandler(ch.alloc()));
66                      }
67                      p.addLast(
68                              new ObjectEncoder(),
69                              new ObjectDecoder(ClassResolvers.cacheDisabled(null)),
70                              new ObjectEchoServerHandler());
71                  }
72               });
73  
74              // Bind and start to accept incoming connections.
75              b.bind(PORT).sync().channel().closeFuture().sync();
76          } finally {
77              bossGroup.shutdownGracefully();
78              workerGroup.shutdownGracefully();
79          }
80      }
81  }