使用WebSocket接收前端传输的大文件

当使用Java编写WebSocket传输大文件的后端代码时，你可以使用Java WebSocket API来实现。以下是一个简单的示例代码，演示了如何在Java中使用WebSocket接收前端传输的大文件：

废话不多说，直接上代码

java
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint("/fileupload")
public class FileUploadEndpoint {

    private static final String FILE_PATH = "/path/to/save/file";
    private String fileName;
    private FileOutputStream fileOutputStream;

    @OnOpen
    public void onOpen(Session session) {
        System.out.println("Client connected");
    }

    @OnMessage
    public void onMessage(Session session, ByteBuffer byteBuffer, boolean last, String fileName) {
        try {
            if (this.fileName == null) {
                this.fileName = fileName;
                fileOutputStream = new FileOutputStream(FILE_PATH + this.fileName);
            }

            byte[] bytes = new byte[byteBuffer.remaining()];
            byteBuffer.get(bytes);
            fileOutputStream.write(bytes);

            if (last) {
                fileOutputStream.close();
                System.out.println("File received and saved: " + FILE_PATH + this.fileName);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在上面的代码中，我们使用@ServerEndpoint注解将一个POJO类声明为WebSocket端点，并定义了onOpen和onMessage方法来处理WebSocket连接建立和接收文件块的操作。在onMessage方法中，我们接收ByteBuffer对象表示文件块数据，并根据last参数判断当前文件块是否为最后一个文件块，并将文件块保存到服务器端文件系统中。