netty学习十一:NIO客户端服务端通讯demo

原创 2017年08月20日 15:26:38

概述


本文展示了如何使用NIO来进行客户端和服务端的通讯.


服务端代码


package nio;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.UUID;

public class NioServer {
    private static Map<String,SocketChannel> clientMap = new HashMap<>();

    public static void main(String[] args) throws IOException {
        //创建一个服务端的socket channel,但是还未绑定监听地址
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        //非阻塞
        serverSocketChannel.configureBlocking(false);

        ServerSocket serverSocket = serverSocketChannel.socket();
        //通过ServerSocet为ServerSocketChannel指定监听地址
        //如果不绑定的话,调用ServerSocketChannel的accept方法会抛出NotYetBoundException
        serverSocket.bind(new InetSocketAddress(8899));

        Selector selector = Selector.open();
        //注册连接建立的事件
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        while (true) {
            //一直阻塞,直到有感兴趣的事件发生,select方法才会返回
            selector.select();
            //注册selector在上的事件可能不止一个,这些事件也可能同时发生,所以这里是SelectionKey Set集合
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            for (Iterator<SelectionKey> iterator = selectedKeys.iterator(); iterator.hasNext();) {
                SelectionKey selectionKey = (SelectionKey) iterator.next();
                final SocketChannel client;
                //isAcceptable==true,说明客户端和服务端已经建立好连接了
                if(selectionKey.isAcceptable()) {
                    ServerSocketChannel server = (ServerSocketChannel)selectionKey.channel();
                    client = server.accept();

                    client.configureBlocking(false);
                    client.register(selector,SelectionKey.OP_READ);

                    String key = "【"+UUID.randomUUID()+"】";
                    clientMap.put(key, client);
                    continue;
                }

                //isReadable==true,说明可以从SocketChannel中读取数据了
                if(selectionKey.isReadable()) {
                    client = (SocketChannel)selectionKey.channel();
                    ByteBuffer readBuffer = ByteBuffer.allocate(1024);
                    //从channel中读取数据
                    int count = client.read(readBuffer);
                    if(count > 0) {
                        readBuffer.flip();
                        Charset charset = Charset.forName("utf-8");
                        //将bytebuffer转换成字符串
                        String receiveMessage = String.valueOf(charset.decode(readBuffer).array());
                        System.out.println(client+":"+receiveMessage);

                        //找出发送消息的客户端
                        String sendKey = "";
                        for (String key : clientMap.keySet()) {
                            SocketChannel socketChannel = clientMap.get(key);
                            if (socketChannel == client) {
                                sendKey = key;
                                break;
                            }
                        }

                        for (String key : clientMap.keySet()) {
                            SocketChannel socketChannel = clientMap.get(key);
                            ByteBuffer writeBuffer = ByteBuffer.allocate(1024);
                            writeBuffer.put((sendKey + ":" +receiveMessage).getBytes());
                            writeBuffer.flip();
                            socketChannel.write(writeBuffer);
                        }
                    }
                }
            }
            //清除事件
            selectedKeys.clear();
        }
    }
}

客户端代码


package nio;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.time.LocalDateTime;
import java.util.Set;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class NioClient {
    public static void main(String[] args) throws IOException {
        SocketChannel socketChannel = SocketChannel.open();
        //非阻塞
        socketChannel.configureBlocking(false);

        //客户端同样需要selector去监听channel
        Selector selector = Selector.open();
        socketChannel.register(selector, SelectionKey.OP_CONNECT);
        socketChannel.connect(new InetSocketAddress("127.0.0.1", 8899));

        while(true) {
            selector.select();
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            for (SelectionKey selectionKey : selectedKeys) {
                //isConnectable=true,说明已经跟服务端建立连接了
                if(selectionKey.isConnectable()) {
                    SocketChannel client = (SocketChannel)selectionKey.channel();
                    if (client.isConnectionPending()) {
                        client.finishConnect();
                        ByteBuffer writeBuffer = ByteBuffer.allocate(1024);
                        writeBuffer.put((LocalDateTime.now() + "连接成功").getBytes());
                        writeBuffer.flip();
                        client.write(writeBuffer);

                        //异步给服务端写信息,避免阻塞主线程
                        ExecutorService executorService = Executors.newSingleThreadExecutor(Executors.defaultThreadFactory());
                        executorService.submit(new Runnable() {
                            @Override
                            public void run() {
                                while(true) {
                                    writeBuffer.clear();
                                    InputStreamReader inputStreamReader = new InputStreamReader(System.in);
                                    BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
                                    try {
                                        String sendMessage = bufferedReader.readLine();
                                        writeBuffer.put(sendMessage.getBytes());
                                        writeBuffer.flip();
                                        client.write(writeBuffer);
                                    } catch (IOException e) {
                                        e.printStackTrace();
                                    }
                                }
                            }
                        });
                    }

                    //一旦客户端和服务端建立连接了,立刻注册读事件
                    client.register(selector, SelectionKey.OP_READ);
                    continue;
                }

                if(selectionKey.isReadable()) {
                    SocketChannel client = (SocketChannel)selectionKey.channel();
                    ByteBuffer readBuffer = ByteBuffer.allocate(1024);
                    int count = client.read(readBuffer);
                    if (count > 0) {
                        String receiveMessage = new String(readBuffer.array(),0,count);
                        System.out.println(receiveMessage);
                    }
                }
            }

            //清除事件
            selectedKeys.clear();
        }
    }
}

csdn code 路径


这个项目的源代码放置在csdn code上,欢迎访问。
netty_study

版权声明:本文为博主原创文章,未经博主允许不得转载。 举报

相关文章推荐

深入理解Tomcat系列之一:系统架构

前言Tomcat是Apache基金组织下的开源项目,性质是一个Web服务器。下面这种情况很普遍:在eclipse床架一个web项目并部署到Tomcat中,启动tomcat,在浏览器中输入一个类似htt...

Java并发编程系列之三:重排序与顺序一致性

前言在我们编写程序并运行的时候,编译器给我们一个错觉:程序编译的顺序与编写的顺序是一致的。但是实际上,为了提高性能,编译器和处理器常常会对指令进行重排序。重排序主要分为两类:编译器优化的重排序、指令级...

我是如何成为一名python大咖的?

人生苦短,都说必须python,那么我分享下我是如何从小白成为Python资深开发者的吧。2014年我大学刚毕业..

java指令重排序的问题

转载:http://my.oschina.net/004/blog/222069?fromerr=ER2mp62C 指令重排序是个比较复杂、觉得有些不可思议的问题,同样是先以例子开头(建议大家跑下例...

一个Csh小技巧--记录访问的所有的系统路径(pushd,dirs)

摘自我的博客:http://my.unix-center.net/~xiaoshe 前注:此方按后面的脚本只支持Csh.不支持bash,如果是bash的话,只能脚本实现了。但是这个方法所有的shel...

网络协议系列之二:HTTP(2)

这篇文章继续对HTTP协议进行说明。一个http事务包括客户向HTTP服务器发送请求以及服务器向客户端发送响应两个过程。根据事务的特点,这两个过程具有原子性,任何一个过程失败都会进行回滚(恢复事务之前...

不同证件的照片大小汇总

我自己照了一张相片,要怎么用photoshop制作一寸的证件照?(请会的朋友给我说一下这个步骤,还有一寸的照片规格是多少?我记不清是不是2.5乘以3.6,还是多少?谢谢了!我要用证件照去报名考试,有点...

网络协议系列之一:HTTP(1)

HTTP协议是web开发必须掌握的协议之一,文章的内容并没有按照教科书式对协议本身进行说明。这里特意将一些web开发遇到的常见问题进行了总结。也方便以后自己开发进行迅速查阅。1、Http的报文结构 ...

用jQuery全选、取消checkbox方法

$(function() { $("#checkedAll").click(function() { if ($(this).attr("checked") == true) {  $("input[...

解决Eclipse 导入工程包 总是出现红叉的问题

Eclipse导入包类型的工程(is library 勾选的工程;常见的工程是App类型的,library类型的工程的特点是:工程bin目录下的生成的是jar包,而不是Apk)总是出现红叉,不能正确引...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)