JAVA Socket编程学习6--NIOTCP两个线程监听两个端口

本文代码来自：http://bbs.csdn.net/topics/390748964

客户端代码：

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.util.Date;
import java.util.Set;
 
public class NIOClient implements Runnable{
    private ByteBuffer sBuffer = ByteBuffer.allocate(1024);  
    private ByteBuffer rBuffer = ByteBuffer.allocate(1024);  
    private InetSocketAddress SERVER;  
    private Selector selector;  
    private SocketChannel client;  
    private String receiveText;  
    private String sendText;  
    private int count=0;  
       
    public NIOClient(int port){  
        SERVER = new InetSocketAddress("localhost", port);  
        init(); 
    }
    public void init(){  
        try {  
            SocketChannel socketChannel = SocketChannel.open();  
            socketChannel.configureBlocking(false);  
            selector = Selector.open();  
            socketChannel.register(selector, SelectionKey.OP_CONNECT);  
            socketChannel.connect(SERVER);  
             
        } catch (Exception e) {  
            e.printStackTrace();  
        }  
    }  
       
    private void handle(SelectionKey selectionKey) throws IOException{  
        if (selectionKey.isConnectable()) {  
            client = (SocketChannel) selectionKey.channel();  
            if (client.isConnectionPending()) {  
                client.finishConnect();  
                System.out.println("connect success !");  
                sBuffer.clear();  
                sBuffer.put((new Date().toLocaleString()+" connected!").getBytes());  
                sBuffer.flip();  
                client.write(sBuffer);
                new Thread(){  
                    @Override  
                    public void run() {  
                        while(true){  
                            try {  
                                sBuffer.clear();  
                                InputStreamReader input = new InputStreamReader(System.in);  
                                BufferedReader br = new BufferedReader(input);  
                                sendText = br.readLine();  
                                sBuffer.put(sendText.getBytes());  
                                sBuffer.flip();  
                                client.write(sBuffer);  
                            } catch (IOException e) {  
                                e.printStackTrace();  
                                break;  
                            }  
                      }  
                    };  
                }.start();  
            }  
            client.register(selector, SelectionKey.OP_READ);  
        } else if (selectionKey.isReadable()) {  
            client = (SocketChannel) selectionKey.channel();  
            rBuffer.clear();  
            count=client.read(rBuffer);  
            if(count>0){  
                receiveText = new String( rBuffer.array(),0,count);  
                System.out.println(receiveText);  
                client = (SocketChannel) selectionKey.channel();  
                client.register(selector, SelectionKey.OP_READ);  
            }  
        }   
    }
    @Override
    public void run() {
        // TODO Auto-generated method stub
        try {
        while (true) {  
            selector.select();
            Set<SelectionKey> keySet = selector.selectedKeys();  
            for(final SelectionKey key : keySet){  
                handle(key);
            };  
            keySet.clear();  
        }  
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } 
    }
    
    public static void main(String args[]){
    	NIOClient client1 = new NIOClient(7788);  
        new Thread(client1).start();
    }
}

服务端代码：

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
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.Map;
import java.util.Set;
 
public class NIOSServer implements Runnable{
    private int port = 8888;
    private Charset cs = Charset.forName("gbk"); 
    private ByteBuffer sBuffer = ByteBuffer.allocate(1024);  
    private ByteBuffer rBuffer = ByteBuffer.allocate(1024);  
    private Map<String, SocketChannel> clientsMap = new HashMap<String, SocketChannel>();  
    private Selector selector;  
       
    public NIOSServer(int port){  
        this.port = port;  
        try {  
            init();  
        } catch (Exception e) {  
            e.printStackTrace();  
        }  
    }  
    private void init() throws IOException{  
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();  
        serverSocketChannel.configureBlocking(false);  
        ServerSocket serverSocket = serverSocketChannel.socket();  
        serverSocket.bind(new InetSocketAddress(port));  
        selector = Selector.open();
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);  
        System.out.println("server start on port:"+port);  
    }  
 
    private void listen(){   
    } 
     
    @Override
    public void run() {
        // TODO Auto-generated method stub
        while (true) {  
            try {  
                selector.select();
                Set<SelectionKey> selectionKeys = selector.selectedKeys();  
                for(SelectionKey key : selectionKeys){  
                    handle(key);  
                }  
                selectionKeys.clear();
            } catch (Exception e) {  
                e.printStackTrace();  
                break;  
            }  
               
        }  
    } 
    private void handle(SelectionKey selectionKey) throws IOException, InterruptedException {  
        ServerSocketChannel server = null;  
        SocketChannel client = null;  
        String receiveText=null;  
        int count=0;  
        if (selectionKey.isAcceptable()) {  
            server = (ServerSocketChannel) selectionKey.channel();  
            client = server.accept();
            //判断client 是否为空
            if(client != null){
                client.configureBlocking(false);  
                client.register(selector, SelectionKey.OP_READ);
            }
//            Thread.sleep(10*1000);
        } else if (selectionKey.isReadable()) {  
            client = (SocketChannel) selectionKey.channel();  
            rBuffer.clear();  
            count = client.read(rBuffer);  
            if (count > 0) {  
                rBuffer.flip();  
                receiveText = String.valueOf(cs.decode(rBuffer).array());  
                System.out.println(client.toString()+":"+receiveText);  
                dispatch(client, receiveText);  
                client = (SocketChannel) selectionKey.channel();  
                client.register(selector, SelectionKey.OP_READ);  
            }  
        }   
    }  
       
    private void dispatch(SocketChannel client,String info) throws IOException{  
        Socket s = client.socket();  
        String name = "["+s.getInetAddress().toString().substring(1)+":"+Integer.toHexString(client.hashCode())+"]";  
        if(!clientsMap.isEmpty()){  
            for(Map.Entry<String, SocketChannel> entry : clientsMap.entrySet()){  
                SocketChannel temp = entry.getValue();  
                if(!client.equals(temp)){  
                    sBuffer.clear();  
                    sBuffer.put((name+":"+info).getBytes());  
                    sBuffer.flip();    
                    temp.write(sBuffer);  
                }  
            }  
        }  
        clientsMap.put(name, client);  
    }  
     
    public static void main(String[] args) throws InterruptedException {
        //创建NIOSServer
        NIOSServer server1 = new NIOSServer(7778);  
        NIOSServer server2 = new NIOSServer(7777);  
        new Thread(server1).start();
        new Thread(server2).start();
//        Thread.sleep(10*1000);
        //创建NIOClient 
        NIOClient client1 = new NIOClient(7778);  
        new Thread(client1).start();
        NIOClient client2 = new NIOClient(7777);  
        new Thread(client2).start();
        NIOClient client3 = new NIOClient(7778);  
        new Thread(client3).start();
        NIOClient client4 = new NIOClient(7777);  
        new Thread(client4).start();
    }
}

运行NIOSServer，输出：
server start on port:7778
server start on port:7777
connect success !
connect success !
connect success !
connect success !
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7778 remote=/127.0.0.1:51700]:2017-11-7 15:46:06 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7777 remote=/127.0.0.1:51709]:2017-11-7 15:46:06 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7778 remote=/127.0.0.1:51706]:2017-11-7 15:46:06 connected!
java.nio.channels.SocketChannel[connected local=/127.0.0.1:7777 remote=/127.0.0.1:51703]:2017-11-7 15:46:06 connected!
[127.0.0.1:228079cf]:2017-11-7 15:46:06 connected!
[127.0.0.1:7c721de]:2017-11-7 15:46:06 connected!