Netty寫一個(gè) echo 客戶端

2018-08-02 14:20 更新

在上節(jié)的內(nèi)容中我們完成了echo服務(wù)器的編寫,接下來就讓我們一起來學(xué)習(xí)Netty中如何寫一個(gè)echo的客戶端,這樣才能讓連接客戶端,并完成信息的傳送。

客戶端的工作內(nèi)容:

  • 連接服務(wù)器
  • 發(fā)送信息
  • 發(fā)送的每個(gè)信息,等待和接收從服務(wù)器返回的同樣的信息
  • 關(guān)閉連接

用 ChannelHandler 實(shí)現(xiàn)客戶端邏輯

跟寫服務(wù)器一樣,我們提供 ChannelInboundHandler 來處理數(shù)據(jù)。下面例子,我們用 SimpleChannelInboundHandler 來處理所有的任務(wù),需要覆蓋三個(gè)方法:

  • channelActive() - 服務(wù)器的連接被建立后調(diào)用
  • channelRead0() - 數(shù)據(jù)后從服務(wù)器接收到調(diào)用
  • exceptionCaught() - 捕獲一個(gè)異常時(shí)調(diào)用

Listing 2.4 ChannelHandler for the client

@Sharable                                //1
public class EchoClientHandler extends
        SimpleChannelInboundHandler<ByteBuf> {

    @Override
    public void channelActive(ChannelHandlerContext ctx) {
        ctx.writeAndFlush(Unpooled.copiedBuffer("Netty rocks!", //2
        CharsetUtil.UTF_8));
    }

    @Override
    public void channelRead0(ChannelHandlerContext ctx,
        ByteBuf in) {
        System.out.println("Client received: " + in.toString(CharsetUtil.UTF_8));    //3
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx,
        Throwable cause) {                    //4
        cause.printStackTrace();
        ctx.close();
    }
}

1.@Sharable標(biāo)記這個(gè)類的實(shí)例可以在 channel 里共享

2.當(dāng)被通知該 channel 是活動(dòng)的時(shí)候就發(fā)送信息

3.記錄接收到的消息

4.記錄日志錯(cuò)誤并關(guān)閉 channel

建立連接后該 channelActive() 方法被調(diào)用一次。邏輯很簡(jiǎn)單:一旦建立了連接,字節(jié)序列被發(fā)送到服務(wù)器。該消息的內(nèi)容并不重要;在這里,我們使用了 Netty 編碼字符串 “Netty rocks!” 通過覆蓋這種方法,我們確保東西被盡快寫入到服務(wù)器。

接下來,我們覆蓋方法 channelRead0()。這種方法會(huì)在接收到數(shù)據(jù)時(shí)被調(diào)用。注意,由服務(wù)器所發(fā)送的消息可以以塊的形式被接收。即,當(dāng)服務(wù)器發(fā)送 5 個(gè)字節(jié)是不是保證所有的 5 個(gè)字節(jié)會(huì)立刻收到 - 即使是只有 5 個(gè)字節(jié),channelRead0() 方法可被調(diào)用兩次,第一次用一個(gè)ByteBuf(Netty的字節(jié)容器)裝載3個(gè)字節(jié)和第二次一個(gè) ByteBuf 裝載 2 個(gè)字節(jié)。唯一要保證的是,該字節(jié)將按照它們發(fā)送的順序分別被接收。 (注意,這是真實(shí)的,只有面向流的協(xié)議如TCP)。

第三個(gè)方法重寫是 exceptionCaught()。正如在 EchoServerHandler (清單2.2),所述的記錄 Throwable 并且關(guān)閉通道,在這種情況下終止 連接到服務(wù)器。

SimpleChannelInboundHandler vs. ChannelInboundHandler

何時(shí)用這兩個(gè)要看具體業(yè)務(wù)的需要。在客戶端,當(dāng) channelRead0() 完成,我們已經(jīng)拿到的入站的信息。當(dāng)方法返回時(shí),SimpleChannelInboundHandler 會(huì)小心的釋放對(duì) ByteBuf(保存信息) 的引用。而在 EchoServerHandler,我們需要將入站的信息返回給發(fā)送者,由于 write() 是異步的,在 channelRead() 返回時(shí),可能還沒有完成。所以,我們使用 ChannelInboundHandlerAdapter,無需釋放信息。最后在 channelReadComplete() 我們調(diào)用 ctxWriteAndFlush() 來釋放信息。詳見第5、6章

引導(dǎo)客戶端

客戶端引導(dǎo)需要 host 、port 兩個(gè)參數(shù)連接服務(wù)器。

Listing 2.5 Main class for the client

public class EchoClient {

    private final String host;
    private final int port;

    public EchoClient(String host, int port) {
        this.host = host;
        this.port = port;
    }

    public void start() throws Exception {
        EventLoopGroup group = new NioEventLoopGroup();
        try {
            Bootstrap b = new Bootstrap();                //1
            b.group(group)                                //2
             .channel(NioSocketChannel.class)            //3
             .remoteAddress(new InetSocketAddress(host, port))    //4
             .handler(new ChannelInitializer<SocketChannel>() {    //5
                 @Override
                 public void initChannel(SocketChannel ch) 
                     throws Exception {
                     ch.pipeline().addLast(
                             new EchoClientHandler());
                 }
             });

            ChannelFuture f = b.connect().sync();        //6

            f.channel().closeFuture().sync();            //7
        } finally {
            group.shutdownGracefully().sync();            //8
        }
    }

    public static void main(String[] args) throws Exception {
        if (args.length != 2) {
            System.err.println(
                    "Usage: " + EchoClient.class.getSimpleName() +
                    " <host> <port>");
            return;
        }

        final String host = args[0];
        final int port = Integer.parseInt(args[1]);

        new EchoClient(host, port).start();
    }
}

1.創(chuàng)建 Bootstrap

2.指定 EventLoopGroup 來處理客戶端事件。由于我們使用 NIO 傳輸,所以用到了 NioEventLoopGroup 的實(shí)現(xiàn)

3.使用的 channel 類型是一個(gè)用于 NIO 傳輸

4.設(shè)置服務(wù)器的 InetSocketAddress

5.當(dāng)建立一個(gè)連接和一個(gè)新的通道時(shí),創(chuàng)建添加到 EchoClientHandler 實(shí)例 到 channel pipeline

6.連接到遠(yuǎn)程;等待連接完成

7.阻塞直到 Channel 關(guān)閉

8.調(diào)用 shutdownGracefully() 來關(guān)閉線程池和釋放所有資源

與以前一樣,在這里使用了 NIO 傳輸。請(qǐng)注意,您可以在 客戶端和服務(wù)器 使用不同的傳輸 ,例如 NIO 在服務(wù)器端和 OIO 客戶端。在第四章中,我們將研究一些具體的因素和情況,這將導(dǎo)致 您可以選擇一種傳輸,而不是另一種。

本節(jié)要點(diǎn)回顧:

  • 創(chuàng)建一個(gè) Bootstrap 來初始化客戶端
  • 一個(gè) NioEventLoopGroup 實(shí)例被分配給處理該事件的處理,這包括創(chuàng)建新的連接和處理入站和出站數(shù)據(jù)
  • 創(chuàng)建一個(gè) InetSocketAddress 以連接到服務(wù)器
  • 連接好服務(wù)器之時(shí),將安裝一個(gè) EchoClientHandler 在 pipeline 
  • 之后 Bootstrap.connect()被調(diào)用連接到遠(yuǎn)程的 - 本例就是 echo(回聲)服務(wù)器。



以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)