Java Socket如何处理并发连接

在Java中，使用Socket处理并发连接通常有两种方法：多线程和NIO（非阻塞I/O）。下面分别介绍这两种方法的实现。

1. 多线程方法：

在这种方法中，为每个客户端连接创建一个新的线程。当客户端连接到服务器时，服务器将创建一个新的线程来处理该连接。这种方法简单易实现，但在大量并发连接的情况下可能会导致性能问题。

以下是一个简单的多线程服务器示例：

import java.io.*;
import java.net.*;

public class MultiThreadedServer {
    public static void main(String[] args) throws IOException {
        ServerSocket serverSocket = new ServerSocket(8888);
        System.out.println("Server started on port 8888");

        while (true) {
            Socket socket = serverSocket.accept();
            System.out.println("New client connected");

            ClientHandler clientHandler = new ClientHandler(socket);
            new Thread(clientHandler).start();
        }
    }
}

class ClientHandler implements Runnable {
    private Socket socket;

    public ClientHandler(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {
        try {
            InputStream input = socket.getInputStream();
            BufferedReader reader = new BufferedReader(new InputStreamReader(input));

            OutputStream output = socket.getOutputStream();
            PrintWriter writer = new PrintWriter(output, true);

            String message;
            while ((message = reader.readLine()) != null) {
                System.out.println("Received message from client: " + message);
                writer.println("Server received: " + message);
            }

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

2. NIO（非阻塞I/O）方法：

NIO提供了更高效的I/O操作，可以在单个线程中处理多个并发连接。这种方法使用Selector来监听多个通道（Channel），当某个通道上有事件发生时（例如数据可读或可写），Selector会通知相应的处理器进行处理。这种方法在大量并发连接的情况下性能更好。

以下是一个简单的NIO服务器示例：

import java.io.*;
import java.net.*;
import java.nio.*;
import java.nio.channels.*;
import java.util.*;

public class NIOServer {
    public static void main(String[] args) throws IOException {
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.socket().bind(new InetSocketAddress(8888));
        serverSocketChannel.configureBlocking(false);

        Selector selector = Selector.open();
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        System.out.println("Server started on port 8888");

        while (true) {
            selector.select();
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            Iterator<SelectionKey> iterator = selectedKeys.iterator();

            while (iterator.hasNext()) {
                SelectionKey key = iterator.next();

                if (key.isAcceptable()) {
                    handleAccept(key, selector);
                } else if (key.isReadable()) {
                    handleRead(key);
                }

                iterator.remove();
            }
        }
    }

    private static void handleAccept(SelectionKey key, Selector selector) throws IOException {
        ServerSocketChannel serverSocketChannel = (ServerSocketChannel) key.channel();
        SocketChannel socketChannel = serverSocketChannel.accept();
        socketChannel.configureBlocking(false);
        socketChannel.register(selector, SelectionKey.OP_READ);
        System.out.println("New client connected");
    }

    private static void handleRead(SelectionKey key) throws IOException {
        SocketChannel socketChannel = (SocketChannel) key.channel();
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        int bytesRead = socketChannel.read(buffer);

        if (bytesRead > 0) {
            buffer.flip();
            byte[] data = new byte[buffer.remaining()];
            buffer.get(data);
            String message = new String(data).trim();
            System.out.println("Received message from client: " + message);

            ByteBuffer responseBuffer = ByteBuffer.wrap(("Server received: " + message).getBytes());
            socketChannel.write(responseBuffer);
        } else if (bytesRead < 0) {
            socketChannel.close();
        }
    }
}

这两种方法都可以处理并发连接，具体选择哪种方法取决于你的需求和性能要求。对于大量并发连接的情况，NIO方法通常更合适。