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.discard;
    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   * Keeps sending random data to the specified address.
    30   */
    31  public class DiscardClient {
    32
    33      private final String host;
    34      private final int port;
    35      private final int firstMessageSize;
    36
    37      public DiscardClient(String host, int port, int firstMessageSize) {
    38          this.host = host;
    39          this.port = port;
    40          this.firstMessageSize = firstMessageSize;
    41      }
    42
    43      public void run() {
    44          // Configure the client.
    45          ClientBootstrap bootstrap = new ClientBootstrap(
    46                  new NioClientSocketChannelFactory(
    47                          Executors.newCachedThreadPool(),
    48                          Executors.newCachedThreadPool()));
    49
    50          // Set up the pipeline factory.
    51          bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
    52              public ChannelPipeline getPipeline() throws Exception {
    53                  return Channels.pipeline(
    54                          new DiscardClientHandler(firstMessageSize));
    55              }
    56          });
    57
    58          // Start the connection attempt.
    59          ChannelFuture future = bootstrap.connect(new InetSocketAddress(host, port));
    60
    61          // Wait until the connection is closed or the connection attempt fails.
    62          future.getChannel().getCloseFuture().awaitUninterruptibly();
    63
    64          // Shut down thread pools to exit.
    65          bootstrap.releaseExternalResources();
    66      }
    67
    68      public static void main(String[] args) throws Exception {
    69          // Print usage if no argument is specified.
    70          if (args.length < 2 || args.length > 3) {
    71              System.err.println(
    72                      "Usage: " + DiscardClient.class.getSimpleName() +
    73                      " <host> <port> [<first message size>]");
    74              return;
    75          }
    76
    77          // Parse options.
    78          final String host = args[0];
    79          final int port = Integer.parseInt(args[1]);
    80          final int firstMessageSize;
    81          if (args.length == 3) {
    82              firstMessageSize = Integer.parseInt(args[2]);
    83          } else {
    84              firstMessageSize = 256;
    85          }
    86
    87          new DiscardClient(host, port, firstMessageSize).run();
    88      }
    89  }