websocket 接收消息无法获取用户id-Toy模板网

这篇具有很好参考价值的文章主要介绍了websocket 接收消息无法获取用户id。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

1.遇到问题

公司项目是基于ruoyi 框架快速搭建开发，使用多线程搜索查询，所以以用户区分任务，保证可以搜索任务和取消搜索，所以我这需要获得用户id，使用 SecurityUtils 共工工具类从请求头获取token，然后解析得到用户基本信息和权限。在开发的过程中正常使用，建立socket连接也正常，但是在后续发消息的时候，在后续将代码移动websocket 出现获取用户ID 异常的错误，其实在获取 loginUser 是空的。

/**
 * 安全服务工具类
 * 
 * @author trinity
 */
public class SecurityUtils
{
    /**
     * 用户ID
     **/
    public static Long getUserId()
    {
        try
        {
            return getLoginUser().getUserId();
        }
        catch (Exception e)
        {
            throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED);
        }
    }

2.解决思路

1.同事告诉我可以采用另外一种方式，建立连接的时候将让前端传过来，让后将userId和session存放一起，以下代码就可以实现。

@Slf4j
@Component
@ServerEndpoint(value = "/websocket/{userid}")
public class SearchSocketServer {

    @OnOpen
    public void onOpen(Session session, @PathParam("userid") String userid) {
        log.info("WebSocket connection opened for userid: {}", userid);

        // 处理其他逻辑...
    }
}

上面是基于注解开发，实现起来的简单。我也问了前端同事是否可以我传用户id，回答也是是。我这人就是最求代码高bi格，就使用下面了这种

实现 HandshakeInterceptor 接口就能在 websocket 三次握手四次挥手（至于websocket用的什么不清楚），因为socket 建立其实发消息其实就不会发请求了，也就是我在代码中获取不到用户id的问题所在。所以我只要在建立socket 请求时获取用户id 写到请求头里后面就可以无限使用了。不知道还有没有更好的方式

/**
 * @author LJL
 * @version 1.0
 * @title IdHandshakeInterceptor
 * @date 2023/8/25 9:18
 * @description TODO
 */
public class IdHandshakeInterceptor  implements HandshakeInterceptor {

    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
        // 从外部请求中获取id，并设置到WebSocket的属性中
        Long uid = SecurityUtils.getUserId();
        attributes.put("uid", uid);
        return true;
    }

    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
        // 握手完成后的处理逻辑（可选）
    }
}

实现了拦截器别忘记添加进去文章来源地址https://www.toymoban.com/news/detail-679314.html

@Configuration
@EnableWebSocket
public class SpringSocketConfig implements WebSocketConfigurer {

    @Autowired
    private SpringSocketHandle springSocketHandle;

    @Autowired
    private SpringAbstractWebSocketHandler springAbstractWebSocketHandler;

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(springSocketHandle, "/spring-ws").setAllowedOrigins("*")
                .addHandler(springAbstractWebSocketHandler, "/spring-ws1").addInterceptors(new IdHandshakeInterceptor()).setAllowedOrigins("*");
    }

到了这里，关于websocket 接收消息无法获取用户id的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！