Blame view

src/main/java/com/huaheng/pc/stompwebsocket/StompHandshakeHandler.java 780 Bytes
1
package com.huaheng.pc.stompwebsocket;
周峰 authored
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

import com.huaheng.common.utils.security.ShiroUtils;
import com.huaheng.pc.system.user.domain.User;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.DefaultHandshakeHandler;

import java.security.Principal;
import java.util.Map;

public class StompHandshakeHandler extends DefaultHandshakeHandler {
    @Override
    protected Principal determineUser(ServerHttpRequest request,
                                      WebSocketHandler wsHandler,
                                      Map<String, Object> attributes) {
        User user = ShiroUtils.getUser();
        return new StompPrincipal(user.getLoginName());
    }
}