004NIO服务端

package nio;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import java.util.Set;

public class ChatSer {
    Selector selector;
    ServerSocketChannel serChannel;


    public ChatSer() throws IOException {
        serChannel = ServerSocketChannel.open();
        serChannel.socket().bind(new InetSocketAddress(6767));
        serChannel.configureBlocking(false);
        selector = Selector.open();
        serChannel.register(selector, SelectionKey.OP_ACCEPT);

        listen();


    }

    private void listen() throws IOException {
        while (true) {
            int num = selector.select();
            if (num > 0) {
                Set<SelectionKey> selectionKeys = selector.selectedKeys();
                Iterator<SelectionKey> iterator = selectionKeys.iterator();
                while (iterator.hasNext()) {
                    SelectionKey next = iterator.next();
                    if(next.isAcceptable()){
                        SocketChannel channel = serChannel.accept();
                        channel.configureBlocking(false);
                        channel.register(selector, SelectionKey.OP_READ);
                        System.out.println(channel.getRemoteAddress()+":来了");
                    }


                    if (next.isReadable()) {
                        read(next);
                    }
                    iterator.remove();


                }

            }
        }
    }

    private void read(SelectionKey next)   {
        SocketChannel channel=null;
        try {
            channel = (SocketChannel) next.channel();
//        ByteBuffer buf = (ByteBuffer) next.attachment();
            ByteBuffer buf = ByteBuffer.allocate(1234);
            int count = channel.read(buf);//按照实际的
            if (count>0) {
                String msg = new String(buf.array(), 0, count);
                System.out.println("client:"+msg);
                copy(msg,channel);
            }
        } catch (IOException e) {
            try {
                System.out.println(channel.getRemoteAddress()+"走了。。。。。。。。。");
                next.cancel();
                channel.close();
            } catch (IOException e1) {
                e1.printStackTrace();
            }
            e.printStackTrace();
        }
    }

    private void copy(String msg, SocketChannel next) throws IOException {
        System.out.println("begin copy to others");
        Set<SelectionKey> keys = selector.keys();
        System.out.println("keys.size() = " + keys.size());
        for (SelectionKey key : keys) {
            SelectableChannel channel = key.channel();
            if (channel instanceof SocketChannel&&channel!=next) {
                System.out.println("----------+++++++++");
                SocketChannel other = (SocketChannel) channel;
                ByteBuffer buf = ByteBuffer.wrap(msg.getBytes());
                other.write(buf);


            }

        }
    }

    public static void main(String[] args) throws IOException {
        ChatSer chatSer = new ChatSer();
        chatSer.listen();

    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值