SocketChannel_NIO_编程_聊天Demo

Server端：

package Server;

import java.io.IOException;

import java.net.InetSocketAddress;

import java.net.Socket;

import java.nio.ByteBuffer;

import java.nio.channels.*;

import java.nio.charset.Charset;

import java.util.Iterator;

import java.util.Scanner;

import java.util.concurrent.TimeUnit;

public class SeverChannel {

    public static void main(String[] args) {

        // TODO Auto-generated method stub

        ByteBuffer bb=ByteBuffer.allocate(1024);

        Start(bb);

    }

    public static void Start( ByteBuffer bb) {

  

        ServerSocketChannel ssc = null;

        Selector s = null;

        boolean tag=false;

        try {

            ssc = ServerSocketChannel.open();

            s = Selector.open();

            ssc.socket().bind(new InetSocketAddress(10086));

            ssc.configureBlocking(false);

            ssc.register(s, SelectionKey.OP_ACCEPT);

            while (true) {

                if (s.select() == 0) {

                    System.out.println("Next Loop");

                    continue;

                }

                Iterator<SelectionKey> keys = s.selectedKeys().iterator();

                while (keys.hasNext()) {

                    SelectionKey key = keys.next();

                    if (key.isAcceptable()) {

                        System.out.println(key + "______________Accept");

                        HandleAccept(key,  bb);

                    }

                    if (key.isConnectable())

                        ;

                    //System.out.println(key + "______________Connect");

                    if (key.isReadable()) {

                        //System.out.println(key + "______________Read");

                        HandleRead(key);

                    }

                 if (key.isWritable()) {

                        //System.out.println(key + "______________Write");

                        //HandleWrite(key);

                     if(!tag) {

                         inputThread(key);

                         tag=true;

                     }

                   }

                    keys.remove();

                }

                try {

                    Thread.sleep(3000);

                } catch (InterruptedException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

            }

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } finally {

            try {

                ssc.close();

                s.close();

            } catch (IOException e) {

                // TODO Auto-generated catch block

                e.printStackTrace();

            }

        }

    }

    public static void HandleAccept(SelectionKey key, ByteBuffer bb) {

        ServerSocketChannel ssc = (ServerSocketChannel) key.channel();

        SocketChannel sc = null;

        try {

            sc = ssc.accept();

            sc.configureBlocking(false);

            sc.register(key.selector(), SelectionKey.OP_READ|SelectionKey.OP_WRITE,bb);

         

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    public static void HandleRead(SelectionKey key) {

        SocketChannel sc = null;

        sc = (SocketChannel) key.channel();

        ByteBuffer bb=(ByteBuffer) key.attachment();

        try {

                sc.read(bb);

                bb.flip();

                System.out.println("小姐姐说："+Charset.forName("UTF-8").newDecoder().decode(bb));

                bb.clear();

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

       

    }

    public static void inputThread(SelectionKey key) {

        SocketChannel sc=(SocketChannel) key.channel();

        ByteBuffer bb=(ByteBuffer) key.attachment();

        new Thread(new Runnable() {

           @Override

         

           public void run() {

               // TODO Auto-generated method stub

               try {  

                System.out.println("Connect with "+sc.getRemoteAddress()+" Sucessed");      

               Scanner scan = new Scanner(System.in);

               while(true){

               String str = scan.nextLine(); 

              

                   bb.put(str.getBytes("UTF-8"));

                   bb.flip();

                   sc.write(bb);

                   bb.clear();

               }

               } catch (IOException e) {

                   // TODO Auto-generated catch block

                   e.printStackTrace();

               }

               }

           }

                ).start();

   }

//    public static void HandleWrite(SelectionKey key) {

//        SocketChannel sc = null;

//        sc = (SocketChannel) key.channel();

//        ByteBuffer bb = (ByteBuffer) key.attachment();

//        Scanner scan = new Scanner(System.in);

//        String str = scan.nextLine();

//        //while(!str.equals("\n")) {

//        bb.clear();

//        bb.put(str.getBytes());

//        bb.flip();

//        try {

//            sc.write(bb);

//        } catch (IOException e) {

//            // TODO Auto-generated catch block

//            e.printStackTrace();

//        }

//

//    }
}

Client端：

package Client;

import java.io.IOException;

import java.io.UnsupportedEncodingException;

import java.net.InetAddress;

import java.net.InetSocketAddress;

import java.nio.ByteBuffer;

import java.nio.channels.*;

import java.nio.charset.Charset;

import java.util.Iterator;

import java.util.Scanner;

import java.util.concurrent.TimeUnit;

import javax.sound.midi.VoiceStatus;

public class ClientChannel {

    public static void main(String[] args) {

        //client();

        ByteBuffer bb = ByteBuffer.allocate(1024);

        Start(bb);

    }

    public static void Start(ByteBuffer bb) {

        SocketChannel sc = null;

        Selector s = null;

        try {

            sc = SocketChannel.open();

            s = Selector.open();

            sc.configureBlocking(false);

            sc.connect(new InetSocketAddress("localhost", 10086));

            sc.register(s, SelectionKey.OP_CONNECT, bb);

            boolean tag = false;

            while (true) {

                if (s.select() == 0) {

                    System.out.println("Next Loop");

                    continue;

                }

                Iterator<SelectionKey> keys =
4000
s.selectedKeys().iterator();

                SelectionKey key = null;

                while (keys.hasNext()) {

                    key = keys.next();

                    if (key.isConnectable()) {

                        System.out.println("Connectable");

                        sc.finishConnect();

                        key.interestOps(key.OP_WRITE | key.OP_READ);

                    }

                    if (key.isReadable()) {

                        // System.out.println("Read");

                        HandleRead(key);

                    }

                    if (key.isWritable()) {

                        //                        System.out.println("Write");

                        //                        System.out.println(key.interestOps());

                        // HandleWrite(key);

                        if (!tag) {

                            inputThread(key);

                            tag = true;

                        }

                    }

                    keys.remove();

                }

            }

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } finally {

            if (sc != null)

                try {

                    sc.close();

                } catch (IOException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

        }

    }

    //    public static void HandleWrite(SelectionKey key) {

    //        SocketChannel sc = null;

    //

    //        sc = (SocketChannel) key.channel();

    //        ByteBuffer bb = ByteBuffer.allocate(1024);

    //        Scanner scan = new Scanner(System.in);

    //        System.out.println("Please Input Some Info:");

    //        String str = scan.nextLine();

    //        //while(!str.equals("\n")) {

    //        bb.clear();

    //        bb.put(str.getBytes());

    //        bb.flip();

    //        try {

    //            sc.write(bb);

    //        } catch (IOException e) {

    //            // TODO Auto-generated catch block

    //            e.printStackTrace();

    //        }

    //

    //    }

    public static void HandleRead(SelectionKey key) {

        SocketChannel sc = null;

        sc = (SocketChannel) key.channel();

        ByteBuffer bb = (ByteBuffer) key.attachment();

        try {

            sc.read(bb);

            bb.flip();

            while (bb.hasRemaining())

                System.out.println(Charset.forName("UTF-8").newDecoder().decode(bb));

            bb.clear();

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    public static void inputThread(SelectionKey key) {

        SocketChannel sc = (SocketChannel) key.channel();

        ByteBuffer bb = (ByteBuffer) key.attachment();

        new Thread(new Runnable() {

            @Override

            public void run() {

                // TODO Auto-generated method stub

                try {

                    System.out.println("Connect with " + sc.getRemoteAddress() + " Sucessed");

                    Scanner scan = new Scanner(System.in);

                    while (true) {

                        String str = scan.nextLine();

                        bb.put(str.getBytes("UTF-8"));

                        bb.flip();

                        sc.write(bb);

                        bb.clear();

                    }

                } catch (IOException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

            }

        }).start();

    }

}