LingandRX
8455307731
- 优化邮件验证逻辑，增加对 verifyCode 是否为空的判断 - 添加 log4j2 配置文件，实现日志记录功能 - 新增 WebSocket 相关配置和处理类 - 优化 TokenInterceptor 类，使用 Lombok 注解 - 新增 TraceIdAspect 方面类，用于添加 traceId 到日志 - 更新 RedisServiceImpl 类，简化 exists 方法的实现 - 新增 SchedulerConfig 配置类，用于配置定时任务 - 更新 SecurityConfig 配置类，添加跨域请求配置
42 lines
1.5 KiB
Java
42 lines
1.5 KiB
Java
package com.example.copykamanotes.config;
|
|
|
|
import com.example.copykamanotes.utils.JwtUtil;
|
|
import com.example.copykamanotes.websocket.MessageWebSocketHandler;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.web.socket.config.annotation.EnableWebSocket;
|
|
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
|
|
import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;
|
|
|
|
@Configuration
|
|
@EnableWebSocket
|
|
public class WebSocketConfig {
|
|
|
|
private final JwtUtil jwtUtil;
|
|
private final ObjectMapper objectMapper;
|
|
|
|
public WebSocketConfig(JwtUtil jwtUtil, ObjectMapper objectMapper) {
|
|
this.jwtUtil = jwtUtil;
|
|
this.objectMapper = objectMapper;
|
|
}
|
|
|
|
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
|
|
registry.addHandler(messageWebSocketHandler(), "ws/message")
|
|
.setAllowedOrigins("*");
|
|
}
|
|
|
|
@Bean
|
|
public MessageWebSocketHandler messageWebSocketHandler() {
|
|
return new MessageWebSocketHandler(jwtUtil, objectMapper);
|
|
}
|
|
|
|
@Bean
|
|
public ServletServerContainerFactoryBean createWebSocketContainer() {
|
|
ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();
|
|
container.setMaxTextMessageBufferSize(8192);
|
|
container.setMaxBinaryMessageBufferSize(8192);
|
|
return container;
|
|
}
|
|
}
|