如何在Webflux中将Spring Boot WebSocket消息转换为自定义Java对象（Bean/DTO）

现有代码问题 #

• 主动调用subscribe()会让消息流脱离Spring容器管控，异常无法被捕获处理
• handle方法需要返回Mono<Void>作为消息处理完成的信号，现有代码没有合法返回值，会触发编译错误

标准实现（JSON消息场景） #

默认你的WebSocket消息是JSON格式，使用Jackson做反序列化，实现代码如下：

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.web.reactive.socket.WebSocketHandler;
import org.springframework.web.reactive.socket.WebSocketSession;
import reactor.core.publisher.Mono;

public class CustomWebSocketHandler implements WebSocketHandler {
    // 注入Jackson序列化工具，也可通过Spring容器注入
    private final ObjectMapper objectMapper = new ObjectMapper();

    @Override
    public Mono<Void> handle(WebSocketSession session) {
        return session.receive()
                // 提取消息的文本 payload
                .map(webSocketMessage -> webSocketMessage.getPayloadAsText())
                // 反序列化为自定义MyObject对象
                .map(payload -> {
                    try {
                        return objectMapper.readValue(payload, MyObject.class);
                    } catch (JsonProcessingException e) {
                        // 自定义解析失败的异常处理逻辑
                        throw new IllegalArgumentException("消息格式不合法", e);
                    }
                })
                // 后续业务逻辑处理
                .doOnNext(myObject -> {
                    // 示例：打印对象属性
                    System.out.printf("解析得到对象：name=%s，address=%s%n", 
                            myObject.getName(), myObject.getAddress());
                })
                // 转换为Mono<Void>返回，标记全流程处理完成
                .then();
    }
}

自定义格式消息适配 #

如果你的消息是自定义文本格式（比如逗号分隔），替换反序列化逻辑即可：

.map(payload -> {
    String[] params = payload.split(",");
    MyObject myObject = new MyObject();
    myObject.setName(params[0]);
    myObject.setAddress(params[1]);
    return myObject;
})

注意：所有流操作都需要拼接在最终返回的响应式序列中，不可主动调用subscribe，由Spring WebFlux容器负责订阅执行，避免出现消息丢失、异常漏处理等问题。

内容的提问来源于stack exchange，提问作者sahim