springboot整合websocket

服務端：

1.添加依賴

<!--websocket依賴-->
<dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
2.
采用springboot內置容器啟動項目的，則需要配置一個Bean,如果用外部容器，則不需要配置

package com.mybatis.demo.demo.config;

import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * @author shafei
 * @version 1.0
 * @date 8:24 2019/8/24
 * @fun websocket啟動配置
 */
@Component
public class WebSocketConfig {
    /**
     * ServerEndpointExporter 作用
     *
     * 這個Bean會自動注冊使用@ServerEndpoint注解聲明的websocket endpoint
     *
     * @return
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

}

　　3.核心代碼，監聽客戶端，收發消息

package com.mybatis.demo.demo;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.websocket.server.ServerEndpoint;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author shafei
 * @version 1.0
 * @date 8:26 2019/8/24
 * @fun
 */
@Slf4j
@Component
@ServerEndpoint("/websocket/{name}")
public class WebSocket {

    /**
     *  與某個客戶端的連接對話，需要通過它來給客戶端發送消息
     */
    private Session session;

    /**
     * 標識當前連接客戶端的用戶名
     */
    private String name;

    /**
     *  用於存所有的連接服務的客戶端，這個對象存儲是安全的
     */
    private static ConcurrentHashMap<String,WebSocket> webSocketSet = new ConcurrentHashMap<>();


    @OnOpen
    public void OnOpen(Session session, @PathParam(value = "name") String name){
        this.session = session;
        this.name = name;
        // name是用來表示唯一客戶端，如果需要指定發送，需要指定發送通過name來區分
        webSocketSet.put(name,this);
        log.info("[WebSocket] 連接成功，當前連接客戶端數為：={}",webSocketSet.size());
    }


    @OnClose
    public void OnClose(){
        webSocketSet.remove(this.name);
        log.info("[WebSocket] 退出成功，當前連接客戶端數為：={}",webSocketSet.size());
    }

    @OnMessage
    public void OnMessage(String message){
        log.info("[WebSocket] 收到消息：{}",message);
        //判斷是否需要指定發送，具體規則自定義
        if(message.indexOf("TOUSER") == 0){
            String name = message.substring(message.indexOf("TOUSER")+6,message.indexOf(";"));
            AppointSending(name,message.substring(message.indexOf(";")+1,message.length()));
        }else{
            GroupSending(message);
        }

    }

    /**
     * 群發
     * @param message
     */
    public void GroupSending(String message){
        for (String name : webSocketSet.keySet()){
            try {
                webSocketSet.get(name).session.getBasicRemote().sendText(message);
            }catch (Exception e){
                e.printStackTrace();
            }
        }
    }

    /**
     * 指定發送
     * @param name
     * @param message
     */
    public void AppointSending(String name,String message){
        try {
            webSocketSet.get(name).session.getBasicRemote().sendText(message);
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}

　　備注：

@Slf4j這個注解坐標：

<dependency>
   <groupId>org.projectlombok</groupId>
   <artifactId>lombok</artifactId>
</dependency>

4.控制層測試發送消息，當然這一步需要將客戶端連接到我們的這個服務端，客戶端稍后下面會寫

@Controller
@RequestMapping("/test")
public class TestController {
   
    @Autowired
    private WebSocket webSocket;

    
    @GetMapping("/sendMessage")
    @ResponseBody
    public void sendMessage(){
//        AppointSending("")
        webSocket.AppointSending("test","shafei");
    }

}

　　

客戶端：

1.客戶端依賴：

<!--websocket作為客戶端-->
<dependency>
    <groupId>org.java-websocket</groupId>
    <artifactId>Java-WebSocket</artifactId>
    <version>1.3.5</version>
</dependency>
2.連接服務端和接收消息的配置類

package com.mesboot.doc.config.websocket;

import lombok.extern.slf4j.Slf4j;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ServerHandshake;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import java.net.URI;

/**
 * @author shafei
 * @version 1.0
 * @date 8:31 2019/8/24
 * @fun
 */
@Slf4j
@Component
public class WebSocketConfig {
    @Bean
    public WebSocketClient webSocketClient() {
        try {
            WebSocketClient webSocketClient = new WebSocketClient(new URI("ws://localhost:8089/mybatisDemo/websocket/test"),new Draft_6455()) {
                @Override
                public void onOpen(ServerHandshake handshakedata) {
                    log.info("[websocket] 連接成功");
                }

                @Override
                public void onMessage(String message) {
                    log.info("[websocket] 收到消息={}",message);

                }

                @Override
                public void onClose(int code, String reason, boolean remote) {
                    log.info("[websocket] 退出連接");
                }

                @Override
                public void onError(Exception ex) {
                    log.info("[websocket] 連接錯誤={}",ex.getMessage());
                }
            };
            webSocketClient.connect();
            return webSocketClient;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}

　　

以上的服務端和客戶端是可以寫在兩個程序里面，互不干擾；

 

參考代碼：https://blog.csdn.net/weixin_38111957/article/details/86352677