Java NIO 同步非阻塞Socket DEMO

最新推荐文章于 2025-02-14 13:34:50 发布

杨安康

最新推荐文章于 2025-02-14 13:34:50 发布

阅读量702

点赞数

分类专栏： Java

Java 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一个基于Java NIO技术的简单聊天室实现方案。该方案包括Server端和Client端两部分，Server端使用ServerSocketChannel监听客户端连接，并通过Selector进行事件管理；Client端则通过SocketChannel与Server建立连接并发送消息。文章详细展示了如何利用NIO的非阻塞特性完成高效的消息收发。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这个例子上做的修改：http://blog.youkuaiyun.com/eclipser1987/article/details/7329362

Server.java

import java.io.IOException;
import java.net.InetSocketAddress;
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.channels.spi.SelectorProvider;
import java.util.Iterator;

/**
 * Created by yangankang on 16/2/20.
 */
public class Server implements Runnable {

    private int port = 8888;
    private ServerSocketChannel socketChannel;
    private SocketChannel clientChannel;
    private Selector selector;
    private ByteBuffer byteBuffer = ByteBuffer.allocate(512);

    public Server() {
        init();
    }

    public void init() {
        try {
            this.selector = SelectorProvider.provider().openSelector();
            this.socketChannel = ServerSocketChannel.open();
            this.socketChannel.configureBlocking(false);
            this.socketChannel.socket().bind(new InetSocketAddress("localhost", port));
            this.socketChannel.register(this.selector, SelectionKey.OP_ACCEPT);

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String args[]) {
        Server server = new Server();
        Thread thread = new Thread(server);
        thread.start();
    }

    @Override
    public void run() {
        while (true) {
            try {
                System.err.println("Server Start.");
                this.selector.select();
                Iterator iterator = this.selector.selectedKeys().iterator();
                while (iterator.hasNext()) {
                    SelectionKey key = (SelectionKey) iterator.next();
                    iterator.remove();
                    if (!key.isValid()) {
                        continue;
                    }
                    if (key.isAcceptable()) {
                        this.accept(key);
                    } else if (key.isReadable()) {
                        this.read(key);
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public void accept(SelectionKey key) {
        ServerSocketChannel socketChannel = (ServerSocketChannel) key.channel();
        try {
            clientChannel = socketChannel.accept();
            clientChannel.configureBlocking(false);
            clientChannel.register(this.selector, SelectionKey.OP_READ);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void read(SelectionKey key) {
        this.byteBuffer.clear();
        SocketChannel socketChannel = (SocketChannel) key.channel();
        try {
            int count = socketChannel.read(this.byteBuffer);
            if (count == -1) {
                key.channel().close();
                key.channel();
                return;
            }
            String string = new String(this.byteBuffer.array()).trim();
            System.out.println("from client message:" + string);
            socketChannel.write(ByteBuffer.wrap(("you say:" + string).getBytes()));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

Client.java

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;

/**
 * Created by yangankang on 16/2/21.
 */
public class Client {

    private ByteBuffer buffer = ByteBuffer.allocate(512);
    private String url = "localhost";
    private int port = 8888;

    public void send() {
        InetSocketAddress address = new InetSocketAddress(url, port);

        SocketChannel socketChannel = null;
        while (true) {
            byte[] b = new byte[512];
            try {
                System.err.print("input your message:");
                System.in.read(b);
                socketChannel = SocketChannel.open();
                socketChannel.connect(address);
                buffer.clear();
                buffer.put(b);
                buffer.flip();
                socketChannel.write(buffer);
                buffer.clear();
                socketChannel.configureBlocking(true);
                ByteBuffer readBuffer = ByteBuffer.allocate(512);
                socketChannel.read(readBuffer);
                System.out.println("from client message:" + (new String(readBuffer.array()).trim()));
                socketChannel.close();
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                if (socketChannel != null)
                    try {
                        socketChannel.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
            }
        }
    }

    public static void main(String args[]) {
        Client client = new Client();
        client.send();
    }
}