摘要：注冊一個的節點并指定使用協議。通過訂閱服務器的目標是接收一對一的推送消息其中由服務端傳遞過來用于表示唯一的用戶通過此值將消息精確推送給一個用戶通過向地址為的服務器地址發起請求，與里的地址對應。

springboot基礎環境,請參考springboot文檔

maven依賴

        
            org.springframework.boot
            spring-boot-starter-websocket
        

代碼準備

準備常量類

    //webSocket相關配置
    //鏈接地址
    public static String WEBSOCKETPATHPERFIX = "/ws-push";
    public static String WEBSOCKETPATH = "/endpointWisely";
    //消息代理路徑
    public static String WEBSOCKETBROADCASTPATH = "/topic";
    //前端發送給服務端請求地址
    public static final String FORETOSERVERPATH = "/welcome";
    //服務端生產地址,客戶端訂閱此地址以接收服務端生產的消息
    public static final String PRODUCERPATH = "/topic/getResponse";
    //點對點消息推送地址前綴
    public static final String P2PPUSHBASEPATH = "/user";
    //點對點消息推送地址后綴,最后的地址為/user/用戶識別碼/msg
    public static final String P2PPUSHPATH = "/msg";

接收前端消息實體

public class WiselyMessage {
    private String name;

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

后臺發送消息實體

private String responseMessage;

    public WiselyResponse(String responseMessage){
        this.responseMessage = responseMessage;
    }

    public String getResponseMessage() {
        return responseMessage;
    }

    public void setResponseMessage(String responseMessage) {
        this.responseMessage = responseMessage;
    }

配置websocket

@Configuration
// @EnableWebSocketMessageBroker注解用于開啟使用STOMP協議來傳輸基于代理（MessageBroker）的消息，這時候控制器（controller）
// 開始支持@MessageMapping,就像是使用@requestMapping一樣。
@EnableWebSocketMessageBroker
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {

    @Override
    public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
        //注冊一個Stomp的節點（endpoint）,并指定使用SockJS協議。
        stompEndpointRegistry.addEndpoint(Constant.WEBSOCKETPATH).withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        //服務端發送消息給客戶端的域,多個用逗號隔開
        registry.enableSimpleBroker(Constant.WEBSOCKETBROADCASTPATH, Constant.P2PPUSHBASEPATH);
        //定義一對一推送的時候前綴
        registry.setUserDestinationPrefix(Constant.P2PPUSHBASEPATH);
        //定義websoket前綴
        registry.setApplicationDestinationPrefixes(Constant.WEBSOCKETPATHPERFIX);
    }
}

service

@Service
public class WebSocketService {

    @Autowired
    private SimpMessagingTemplate template;

    /**
     * 廣播
     * 發給所有在線用戶
     *
     * @param msg
     */
    public void sendMsg(WiselyResponse msg) {
        template.convertAndSend(Constant.PRODUCERPATH, msg);
    }

    /**
     * 發送給指定用戶
     * @param users
     * @param msg
     */
    public void send2Users(List users, WiselyResponse msg) {
        users.forEach(userName -> {
            template.convertAndSendToUser(userName, Constant.P2PPUSHPATH, msg);
        });
    }
}

控制器

@Controller
public class WsController {

    @Resource
    WebSocketService webSocketService;

    @MessageMapping(Constant.FORETOSERVERPATH)//@MessageMapping和@RequestMapping功能類似，用于設置URL映射地址，瀏覽器向服務器發起請求，需要通過該地址。
    @SendTo(Constant.PRODUCERPATH)//如果服務器接受到了消息，就會對訂閱了@SendTo括號中的地址傳送消息。
    public WiselyResponse say(WiselyMessage message) throws Exception {
        List users = Lists.newArrayList();
        users.add("d892bf12bf7d11e793b69c5c8e6f60fb");//此處寫死只是為了方便測試,此值需要對應頁面中訂閱個人消息的userId
        webSocketService.send2Users(users, new WiselyResponse("admin hello"));

        return new WiselyResponse("Welcome, " + message.getName() + "!");
    }
}

頁面

    
    





    

        連接
        斷開連接
    
    

        輸入你的名字
        發送
        

        

    

測試

點擊連接控制臺輸出

表示連接成功并且訂閱了兩個地址

此時在文本框內輸入任意值
結果如圖所示則代表成功

控制臺中顯示依次為,發送信息,目標長度內容

此時服務端控制器接收到請求

同時給指定用戶發送了消息,所以控制臺接收到消息

同時因為控制器有注解@SendTo所以會向@SendTo的地址廣播消息,客戶端訂閱了廣播地址所有控制臺顯示接收了消息