java 帮我写一个接受websock请求的工具

以下是一个简单的Java工具类，用于接受WebSocket请求：

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

public class WebSocketServer {

    private int port;
    private List<WebSocketConnection> connections;

    public WebSocketServer(int port) {
        this.port = port;
        this.connections = new ArrayList<>();
    }

    public void start() {
        try {
            ServerSocket serverSocket = new ServerSocket(port);
            System.out.println("WebSocket server started on port " + port);

            while (true) {
                Socket socket = serverSocket.accept();
                WebSocketConnection connection = new WebSocketConnection(socket);
                connections.add(connection);

                Thread thread = new Thread(connection);
                thread.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        WebSocketServer server = new WebSocketServer(8080);
        server.start();
    }
}

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

public class WebSocketConnection implements Runnable {

    private Socket socket;
    private InputStream inputStream;
    private OutputStream outputStream;

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

    @Override
    public void run() {
        try {
            inputStream = socket.getInputStream();
            outputStream = socket.getOutputStream();

            // 处理WebSocket握手请求
            handleHandshake();

            // 处理WebSocket消息
            handleMessages();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    private void handleHandshake() throws IOException {
        // 读取握手请求的数据
        byte[] buffer = new byte[1024];
        int bytesRead = inputStream.read(buffer);

        // 解析握手请求
        String request = new String(buffer, 0, bytesRead);
        String response = "HTTP/1.1 101 Switching Protocols\r\n"
                + "Upgrade: websocket\r\n"
                + "Connection: Upgrade\r\n"
                + "Sec-WebSocket-Accept: " + generateWebSocketAccept(request) + "\r\n\r\n";

        // 发送握手响应
        outputStream.write(response.getBytes());
        outputStream.flush();
    }

    private void handleMessages() throws IOException {
        while (true) {
            // 读取WebSocket消息的头部
            byte[] header = new byte[2];
            inputStream.read(header, 0, 2);

            // 解析消息的长度
            int payloadLength = header[1] & 0x7F;
            if (payloadLength == 126) {
                byte[] extendedLengthBytes = new byte[2];
                inputStream.read(extendedLengthBytes, 0, 2);
                payloadLength = ((extendedLengthBytes[0] & 0xFF) << 8) | (extendedLengthBytes[1] & 0xFF);
            } else if (payloadLength == 127) {
                byte[] extendedLengthBytes = new byte[8];
                inputStream.read(extendedLengthBytes, 0, 8);
                payloadLength = ((extendedLengthBytes[0] & 0xFF) << 56) |
                        ((extendedLengthBytes[1] & 0xFF) << 48) |
                        ((extendedLengthBytes[2] & 0xFF) << 40) |
                        ((extendedLengthBytes[3] & 0xFF) << 32) |
                        ((extendedLengthBytes[4] & 0xFF) << 24) |
                        ((extendedLengthBytes[5] & 0xFF) << 16) |
                        ((extendedLengthBytes[6] & 0xFF) << 8) |
                        (extendedLengthBytes[7] & 0xFF);
            }

            // 读取消息的内容
            byte[] payload = new byte[payloadLength];
            inputStream.read(payload, 0, payloadLength);

            // 处理WebSocket消息
            String message = new String(payload);
            System.out.println("Received message: " + message);

            // 发送WebSocket消息
            String responseMessage = "Hello from server!";
            sendWebSocketMessage(responseMessage);
        }
    }

    private void sendWebSocketMessage(String message) throws IOException {
        byte[] messageBytes = message.getBytes();

        // 计算消息头部的长度
        byte[] header = new byte[2];
        header[0] = (byte) 0x81;
        if (messageBytes.length <= 125) {
            header[1] = (byte) messageBytes.length;
        } else if (messageBytes.length >= 126 && messageBytes.length <= 65535) {
            header[1] = (byte) 126;
            header[2] = (byte) ((messageBytes.length >> 8) & 0xFF);
            header[3] = (byte) (messageBytes.length & 0xFF);
        } else {
            header[1] = (byte) 127;
            header[2] = (byte) ((messageBytes.length >> 56) & 0xFF);
            header[3] = (byte) ((messageBytes.length >> 48) & 0xFF);
            header[4] = (byte) ((messageBytes.length >> 40) & 0xFF);
            header[5] = (byte) ((messageBytes.length >> 32) & 0xFF);
            header[6] = (byte) ((messageBytes.length >> 24) & 0xFF);
            header[7] = (byte) ((messageBytes.length >> 16) & 0xFF);
            header[8] = (byte) ((messageBytes.length >> 8) & 0xFF);
            header[9] = (byte) (messageBytes.length & 0xFF);
        }

        // 发送消息
        outputStream.write(header);
        outputStream.write(messageBytes);
        outputStream.flush();
    }

    private String generateWebSocketAccept(String request) {
        String key = request.split("Sec-WebSocket-Key: ")[1].split("\r\n")[0].trim();
        String magicKey = key + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
        byte[] sha1 = DigestUtils.sha1(magicKey.getBytes());
        return Base64.encodeBase64String(sha1);
    }
}

请注意，这只是一个简单的示例，用于说明如何处理WebSocket握手请求和消息。在实际应用中，您可能需要根据WebSocket协议规范进行更复杂的处理和错误处理。您还可以使用现有的WebSocket库，如Jetty或Tomcat，以简化WebSocket开发