java NIO学习记录

缺乏、安全感 2023-07-17 05:57 4阅读 0赞

server端，最原始的写法

package Test01;
    
    import javax.sound.midi.Soundbank;
    import java.io.IOException;
    import java.net.InetSocketAddress;
    import java.nio.ByteBuffer;
    import java.nio.channels.ServerSocketChannel;
    import java.nio.channels.SocketChannel;
    import java.util.ArrayList;
    import java.util.List;
    
    public class NioTest {
        private List<SocketChannel> list = new ArrayList<SocketChannel>();
        ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
        public void test(){
            try {
                ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
                serverSocketChannel.configureBlocking(false);
                serverSocketChannel.bind(new InetSocketAddress(1000));
                while(true){
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    if(socketChannel == null){
                        try {
                            System.out.println("wait connecting");
                            Thread.sleep(1500);
    
                        } catch (InterruptedException e) {
                            e.printStackTrace();
                        }
                        for(SocketChannel tmp:list){
                            //每次都主动检查时候有客户端发送数据过来
                            int length = tmp.read(byteBuffer);
                            byteBuffer.flip();
                            System.out.println(byteBuffer.toString());
                            if(length>0){
                                System.out.println(byteBuffer.toString());
                            }
                        }
                    }else{
                        //有人连接上来
                        System.out.println("hava -------------------------");
                        socketChannel.configureBlocking(false);
                        list.add(socketChannel);
                        for(SocketChannel tmp:list){
                            //每次都主动检查时候有客户端发送数据过来
                            int length = tmp.read(byteBuffer);
                            byteBuffer.flip();
                            System.out.println(byteBuffer.toString());
                            if(length>0){
                                System.out.println(byteBuffer.toString());
                            }
                        }
                    }
    
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
    
        }
    }

client层，最原始的写法

package Test01;
    
    import java.io.IOException;
    import java.io.OutputStream;
    import java.net.InetSocketAddress;
    import java.net.Socket;
    import java.net.SocketAddress;
    import java.util.Scanner;
    
    public class ClientSocket {
        public static void main(String[] args) {
            ClientSocket clientSocket = new ClientSocket();
            clientSocket.test();
        }
        public void test(){
            Socket socket = new Socket();
            try {
                socket.connect(new InetSocketAddress("127.0.0.1",1000));
                OutputStream out = socket.getOutputStream();
                out.write(123);
                out.flush();
                Scanner scanner = new Scanner(System.in);
                scanner.hasNext();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

\-------------------------------------------------------------------------------------------------------------------------------------------------------------------------

使用Selector

package Test01.Selector;
    
    import com.sun.jndi.cosnaming.IiopUrl;
    
    import java.io.IOException;
    import java.net.InetSocketAddress;
    import java.net.InterfaceAddress;
    import java.net.ServerSocket;
    import java.nio.Buffer;
    import java.nio.ByteBuffer;
    import java.nio.channels.*;
    import java.util.Date;
    import java.util.Iterator;
    
    public class NioSelectorTest {
        public static void main(String[] args) {
            NioSelectorTest nioSelectorTest = new NioSelectorTest();
            nioSelectorTest.test();
        }
        public void test(){
            try {
                ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
                ServerSocketChannel channel = ServerSocketChannel.open();
                channel.configureBlocking(false);
                channel.bind(new InetSocketAddress(1000));
                channel.accept();
                Selector selector = Selector.open();
                channel.register(selector, SelectionKey.OP_ACCEPT);
                while(selector.select()>0){
                    Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
                    while(iterator.hasNext()){
                        SelectionKey key = iterator.next();
                        if(key.isConnectable()){
                            System.out.println("key.isConnectable()");
                        }else if(key.isAcceptable()){
                            SocketChannel socketChannel = channel.accept();
                            socketChannel.configureBlocking(false);
                            //对这个通道进行注册
                            socketChannel.register(selector,SelectionKey.OP_READ);
                            System.out.println("key.isAcceptable()");
                        }else if(key.isReadable()){
                            System.out.println("key.isReadable()");
                            SocketChannel socketChannel = (SocketChannel) key.channel();
                            byteBuffer.clear();
                            socketChannel.read(byteBuffer);
                            byteBuffer.flip();
                            System.out.println(new Date().toString());
                            System.out.println(new String(byteBuffer.array(),0,byteBuffer.limit()));
                        }else if(key.isWritable()){
                            System.out.println("key.isWritable()");
                        }
                        System.out.println("*");
                    }
                    iterator.remove();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
    
        }
    }

对应的测试客户端

package Test01;
    
    import java.io.IOException;
    import java.io.OutputStream;
    import java.net.InetSocketAddress;
    import java.net.Socket;
    import java.util.Scanner;
    
    public class ClientSocket2 {
        public static void main(String[] args) {
            ClientSocket2 clientSocket = new ClientSocket2();
            clientSocket.test();
        }
        public void test(){
            Socket socket = new Socket();
            try {
                socket.connect(new InetSocketAddress("127.0.0.1",1000));
                OutputStream out = socket.getOutputStream();
    
                while(true){
                    Scanner scanner = new Scanner(System.in);
                    out.write(scanner.next().getBytes());
                    out.flush();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }