parent
542d11370c
commit
2f4cc28e83
@ -1,103 +0,0 @@ |
|||||||
package com.bellmann.config; |
|
||||||
|
|
||||||
import cn.hutool.core.util.StrUtil; |
|
||||||
import cn.hutool.jwt.JWTPayload; |
|
||||||
import com.bellmann.security.util.JwtUtils; |
|
||||||
import lombok.extern.slf4j.Slf4j; |
|
||||||
import org.springframework.context.annotation.Configuration; |
|
||||||
import org.springframework.http.HttpHeaders; |
|
||||||
import org.springframework.messaging.Message; |
|
||||||
import org.springframework.messaging.MessageChannel; |
|
||||||
import org.springframework.messaging.simp.config.ChannelRegistration; |
|
||||||
import org.springframework.messaging.simp.config.MessageBrokerRegistry; |
|
||||||
import org.springframework.messaging.simp.stomp.StompCommand; |
|
||||||
import org.springframework.messaging.simp.stomp.StompHeaderAccessor; |
|
||||||
import org.springframework.messaging.support.ChannelInterceptor; |
|
||||||
import org.springframework.messaging.support.MessageHeaderAccessor; |
|
||||||
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker; |
|
||||||
import org.springframework.web.socket.config.annotation.StompEndpointRegistry; |
|
||||||
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer; |
|
||||||
|
|
||||||
/** |
|
||||||
* WebSocket 配置 |
|
||||||
* |
|
||||||
* |
|
||||||
* @since 2.4.0 |
|
||||||
*/ |
|
||||||
@Configuration |
|
||||||
@EnableWebSocketMessageBroker // 启用WebSocket消息代理功能和配置STOMP协议，实现实时双向通信和消息传递
|
|
||||||
@Slf4j |
|
||||||
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { |
|
||||||
|
|
||||||
/** |
|
||||||
* 注册一个端点，客户端通过这个端点进行连接 |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public void registerStompEndpoints(StompEndpointRegistry registry) { |
|
||||||
registry |
|
||||||
.addEndpoint("/ws") // 注册了一个 /ws 的端点
|
|
||||||
.setAllowedOriginPatterns("*") // 允许跨域的 WebSocket 连接
|
|
||||||
.withSockJS(); // 启用 SockJS (浏览器不支持WebSocket，SockJS 将会提供兼容性支持)
|
|
||||||
registry.addEndpoint("/ws-app").setAllowedOriginPatterns("*"); // 注册了一个 /ws-app 的端点，支持 uni-app 的 ws 连接协议
|
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
/** |
|
||||||
* 配置消息代理 |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public void configureMessageBroker(MessageBrokerRegistry registry) { |
|
||||||
// 客户端发送消息的请求前缀
|
|
||||||
registry.setApplicationDestinationPrefixes("/app"); |
|
||||||
|
|
||||||
// 客户端订阅消息的请求前缀，topic一般用于广播推送，queue用于点对点推送
|
|
||||||
registry.enableSimpleBroker("/topic", "/queue"); |
|
||||||
|
|
||||||
// 服务端通知客户端的前缀，可以不设置，默认为user
|
|
||||||
registry.setUserDestinationPrefix("/user"); |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
/** |
|
||||||
* 配置客户端入站通道拦截器 |
|
||||||
* <p> |
|
||||||
* 添加 ChannelInterceptor 拦截器，用于在消息发送前，从请求头中获取 token 并解析出用户信息(username)，用于点对点发送消息给指定用户 |
|
||||||
* |
|
||||||
* @param registration 通道注册器 |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public void configureClientInboundChannel(ChannelRegistration registration) { |
|
||||||
registration.interceptors(new ChannelInterceptor() { |
|
||||||
@Override |
|
||||||
public Message<?> preSend(Message<?> message, MessageChannel channel) { |
|
||||||
StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class); |
|
||||||
// 如果是连接请求（CONNECT 命令），从请求头中取出 token 并设置到认证信息中
|
|
||||||
if (accessor != null && StompCommand.CONNECT.equals(accessor.getCommand())) { |
|
||||||
// 从连接头中提取授权令牌
|
|
||||||
String bearerToken = accessor.getFirstNativeHeader(HttpHeaders.AUTHORIZATION); |
|
||||||
|
|
||||||
// 验证令牌格式并提取用户信息
|
|
||||||
if (StrUtil.isNotBlank(bearerToken) && bearerToken.startsWith("Bearer ")) { |
|
||||||
try { |
|
||||||
// 移除 "Bearer " 前缀，从令牌中提取用户信息(username), 并设置到认证信息中
|
|
||||||
|
|
||||||
// 这里不应该用"name"
|
|
||||||
// String username = JwtUtils.parseToken(bearerToken).get("name").toString();
|
|
||||||
String username = JwtUtils.parseToken(bearerToken).get(JWTPayload.SUBJECT).toString(); |
|
||||||
|
|
||||||
if (StrUtil.isNotBlank(username)) { |
|
||||||
accessor.setUser(() -> username); |
|
||||||
return message; |
|
||||||
} |
|
||||||
} catch (Exception e) { |
|
||||||
log.error("Failed to process authentication token.", e); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
// 不是连接请求，直接放行
|
|
||||||
return ChannelInterceptor.super.preSend(message, channel); |
|
||||||
} |
|
||||||
}); |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
@ -1,61 +0,0 @@ |
|||||||
package com.bellmann.controller; |
|
||||||
|
|
||||||
import com.bellmann.model.dto.ChatMessage; |
|
||||||
import lombok.RequiredArgsConstructor; |
|
||||||
import lombok.extern.slf4j.Slf4j; |
|
||||||
import org.springframework.messaging.handler.annotation.DestinationVariable; |
|
||||||
import org.springframework.messaging.handler.annotation.MessageMapping; |
|
||||||
import org.springframework.messaging.handler.annotation.SendTo; |
|
||||||
import org.springframework.messaging.simp.SimpMessagingTemplate; |
|
||||||
import org.springframework.web.bind.annotation.RequestMapping; |
|
||||||
import org.springframework.web.bind.annotation.RestController; |
|
||||||
|
|
||||||
import java.security.Principal; |
|
||||||
|
|
||||||
/** |
|
||||||
* WebSocket 测试控制器 |
|
||||||
* |
|
||||||
* |
|
||||||
* @since 2.3.0 |
|
||||||
*/ |
|
||||||
@RestController |
|
||||||
@RequestMapping("/websocket") |
|
||||||
@RequiredArgsConstructor |
|
||||||
@Slf4j |
|
||||||
public class WebsocketController { |
|
||||||
|
|
||||||
private final SimpMessagingTemplate messagingTemplate; |
|
||||||
|
|
||||||
|
|
||||||
/** |
|
||||||
* 广播发送消息 |
|
||||||
* |
|
||||||
* @param message 消息内容 |
|
||||||
*/ |
|
||||||
@MessageMapping("/sendToAll") |
|
||||||
@SendTo("/topic/notice") |
|
||||||
public String sendToAll(String message) { |
|
||||||
return "服务端通知: " + message; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 点对点发送消息 |
|
||||||
* <p> |
|
||||||
* 模拟 张三 给 李四 发送消息场景 |
|
||||||
* |
|
||||||
* @param principal 当前用户 |
|
||||||
* @param username 接收消息的用户 |
|
||||||
* @param message 消息内容 |
|
||||||
*/ |
|
||||||
@MessageMapping("/sendToUser/{username}") |
|
||||||
public void sendToUser(Principal principal, @DestinationVariable String username, String message) { |
|
||||||
|
|
||||||
String sender = principal.getName(); // 发送人
|
|
||||||
String receiver = username; // 接收人
|
|
||||||
|
|
||||||
log.info("发送人:{}; 接收人:{}", sender, receiver); |
|
||||||
// 发送消息给指定用户，拼接后路径 /user/{receiver}/queue/greeting
|
|
||||||
messagingTemplate.convertAndSendToUser(receiver, "/queue/greeting", new ChatMessage(sender, message)); |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
@ -0,0 +1,14 @@ |
|||||||
|
package com.bellmann.model.form; |
||||||
|
|
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
@Data |
||||||
|
public class PwdForm { |
||||||
|
|
||||||
|
|
||||||
|
private String oldPwd; |
||||||
|
|
||||||
|
private String newPwd; |
||||||
|
|
||||||
|
private String lastPwd; |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
package com.bellmann.model.vo; |
||||||
|
|
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
import java.util.Date; |
||||||
|
|
||||||
|
@Data |
||||||
|
public class OperateTaskResultVO { |
||||||
|
|
||||||
|
private Long operTaskId; |
||||||
|
|
||||||
|
private String userName; |
||||||
|
|
||||||
|
private String errorDesc; |
||||||
|
|
||||||
|
private String resultState; |
||||||
|
|
||||||
|
private Date operTaskCreateTime; |
||||||
|
|
||||||
|
private Date operStartTime; |
||||||
|
|
||||||
|
private Date operEndTime; |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue