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 org.jboss.netty.example.echo;
    17
    18  import java.net.InetSocketAddress;
    19  import java.util.concurrent.Executors;
    20
    21  import org.jboss.netty.bootstrap.ClientBootstrap;
    22  import org.jboss.netty.channel.ChannelFuture;
    23  import org.jboss.netty.channel.ChannelPipeline;
    24  import org.jboss.netty.channel.ChannelPipelineFactory;
    25  import org.jboss.netty.channel.Channels;
    26  import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;
    27
    28  /**
    29   * Sends one message when a connection is open and echoes back any received
    30   * data to the server.  Simply put, the echo client initiates the ping-pong
    31   * traffic between the echo client and server by sending the first message to
    32   * the server.
    33   */
    34  public class EchoClient {
    35
    36      private final String host;
    37      private final int port;
    38      private final int firstMessageSize;
    39
    40      public EchoClient(String host, int port, int firstMessageSize) {
    41          this.host = host;
    42          this.port = port;
    43          this.firstMessageSize = firstMessageSize;
    44      }
    45
    46      public void run() {
    47          // Configure the client.
    48          ClientBootstrap bootstrap = new ClientBootstrap(
    49                  new NioClientSocketChannelFactory(
    50                          Executors.newCachedThreadPool(),
    51                          Executors.newCachedThreadPool()));
    52
    53          // Set up the pipeline factory.
    54          bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
    55              public ChannelPipeline getPipeline() throws Exception {
    56                  return Channels.pipeline(
    57                          new EchoClientHandler(firstMessageSize));
    58              }
    59          });
    60
    61          // Start the connection attempt.
    62          ChannelFuture future = bootstrap.connect(new InetSocketAddress(host, port));
    63
    64          // Wait until the connection is closed or the connection attempt fails.
    65          future.getChannel().getCloseFuture().awaitUninterruptibly();
    66
    67          // Shut down thread pools to exit.
    68          bootstrap.releaseExternalResources();
    69      }
    70
    71      public static void main(String[] args) throws Exception {
    72          // Print usage if no argument is specified.
    73          if (args.length < 2 || args.length > 3) {
    74              System.err.println(
    75                      "Usage: " + EchoClient.class.getSimpleName() +
    76                      " <host> <port> [<first message size>]");
    77              return;
    78          }
    79
    80          // Parse options.
    81          final String host = args[0];
    82          final int port = Integer.parseInt(args[1]);
    83          final int firstMessageSize;
    84          if (args.length == 3) {
    85              firstMessageSize = Integer.parseInt(args[2]);
    86          } else {
    87              firstMessageSize = 256;
    88          }
    89
    90          new EchoClient(host, port, firstMessageSize).run();
    91      }
    92  }