spring boot +websocket 生成 文件上传代码

在Spring Boot中集成WebSocket并实现文件上传功能，通常需要以下几个步骤：

1. **添加依赖**：
首先，在`pom.xml`文件中添加WebSocket和文件上传相关的依赖，如`spring-websocket`和`spring-boot-starter-web`。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-fileupload</artifactId>
</dependency>

2. **配置WebSocket支持**：
在`application.properties`或`application.yml`中启用WebSocket支持，并设置端口（如果自定义的话）。

server.port=8080
spring.websocket.server.port=8443

3. **创建WebSocket服务**：
创建一个`WebSocketConfig`类来配置WebSocket连接及其处理器。例如：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
}

4. **处理文件上传**：
使用Spring MVC的`MultipartFile`接收上传的文件。你可以创建一个Controller来处理WebSocket消息和文件上传请求：

@Controller
public class FileUploadController {

    @PostMapping("/upload")
    public String handleFileUpload(@RequestParam("file") MultipartFile file) {
        // 检查文件是否有效，保存到服务器等操作
        if (file.isEmpty()) {
            return "error";
        }
        saveUploadedFile(file);
        return "success"; // 返回给前端的消息
    }

    // 实现文件保存逻辑
    private void saveUploadedFile(MultipartFile file) {
        try {
            Files.write(file.getOriginalFilename(), file.getBytes());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
    
    @MessageMapping("/upload")
    @SendTo("/topic/file-uploads")
    public String handleWebSocketFileUpload(@RequestBody MultipartFile file) {
        // 对于WebSocket消息，可以异步处理并将结果发送回客户端
        saveUploadedFile(file);
        return "file uploaded";
    }
}

5. **客户端连接和文件上传**：
使用JavaScript库（如`sockjs-client`和`stomp.js`）与服务器建立WebSocket连接，并通过HTTP POST发送文件数据。

var stompClient = Stomp.over('ws://localhost:8080/ws');
stompClient.connect({}, function(frame) {
    stompClient.send('/app/upload', {}, JSON.stringify({file: file}));
});