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

springboot基礎(chǔ)環(huán)境,請參考springboot文檔

maven依賴

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

代碼準(zhǔn)備

準(zhǔn)備常量類

    //webSocket相關(guān)配置
    //鏈接地址
    public static String WEBSOCKETPATHPERFIX = "/ws-push";
    public static String WEBSOCKETPATH = "/endpointWisely";
    //消息代理路徑
    public static String WEBSOCKETBROADCASTPATH = "/topic";
    //前端發(fā)送給服務(wù)端請求地址
    public static final String FORETOSERVERPATH = "/welcome";
    //服務(wù)端生產(chǎn)地址,客戶端訂閱此地址以接收服務(wù)端生產(chǎn)的消息
    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;
    }
}

后臺發(fā)送消息實體

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協(xié)議來傳輸基于代理（MessageBroker）的消息，這時候控制器（controller）
// 開始支持@MessageMapping,就像是使用@requestMapping一樣。
@EnableWebSocketMessageBroker
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {

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

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

service

@Service
public class WebSocketService {

    @Autowired
    private SimpMessagingTemplate template;

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

    /**
     * 發(fā)送給指定用戶
     * @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功能類似，用于設(shè)置URL映射地址，瀏覽器向服務(wù)器發(fā)起請求，需要通過該地址。
    @SendTo(Constant.PRODUCERPATH)//如果服務(wù)器接受到了消息，就會對訂閱了@SendTo括號中的地址傳送消息。
    public WiselyResponse say(WiselyMessage message) throws Exception {
        List users = Lists.newArrayList();
        users.add("d892bf12bf7d11e793b69c5c8e6f60fb");//此處寫死只是為了方便測試,此值需要對應(yīng)頁面中訂閱個人消息的userId
        webSocketService.send2Users(users, new WiselyResponse("admin hello"));

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

頁面

    
    





    

        連接
        斷開連接
    
    

        輸入你的名字
        發(fā)送
        

        

    

測試

點擊連接控制臺輸出

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

此時在文本框內(nèi)輸入任意值
結(jié)果如圖所示則代表成功

控制臺中顯示依次為,發(fā)送信息,目標(biāo)長度內(nèi)容

此時服務(wù)端控制器接收到請求

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

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