@ServerEndpoint生成一个可以通过websocket传输图片，视频文档的代码

以下是一个可以通过 WebSocket 传输图片，视频和文档的示例代码，其中 `@ServerEndpoint` 注解用于创建 WebSocket 服务器端点：

import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.Base64;
import javax.websocket.OnMessage;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint("/file-transfer")
public class FileTransferEndpoint {

    @OnMessage
    public void onMessage(Session session, ByteBuffer buffer) throws IOException {
        byte[] data = buffer.array();
        String type = session.getUserProperties().getOrDefault("type", "").toString();
        if (type.equals("image")) {
            String base64 = Base64.getEncoder().encodeToString(data);
            session.getBasicRemote().sendText("data:image/png;base64," + base64);
        } else if (type.equals("video")) {
            session.getBasicRemote().sendBinary(buffer);
        } else {
            session.getBasicRemote().sendBinary(buffer);
        }
    }
}

上述代码中，`onMessage` 方法用于处理客户端发送的消息，其中 `ByteBuffer` 对象可以用于传输二进制数据。通过 `Session` 对象的 `getUserProperties` 方法可以获取客户端发送的文件类型（如图片、视频或文档），并根据不同类型的文件进行处理。如果收到的是图片，将二进制数据转换为 Base64 编码的字符串，并添加图片的数据类型后发送；如果收到的是视频或文档，直接将二进制数据发送回客户端。

在客户端发送文件时，可以使用 JavaScript 的 `FileReader` 对象将文件读取为二进制数据，并通过 WebSocket 发送到服务器端：

var file = document.getElementById("file").files[0];
var reader = new FileReader();
reader.onload = function(event) {
    var data = event.target.result;
    var buffer = new Uint8Array(data);
    socket.send(buffer);
};
reader.readAsArrayBuffer(file);

上述代码中，`file` 对象表示要发送的文件，`FileReader` 对象用于将文件读取为二进制数据，`Uint8Array` 对象用于将二进制数据转换为 `ArrayBuffer` 对象，最后调用 WebSocket 的 `send` 方法发送数据到服务器端。