首页 > 其他分享> > websocket即时通讯

websocket即时通讯

2021-07-21 21:03:27 作者：互联网

jar

<dependency>
    <groupId>javax.websocket</groupId>
    <artifactId>javax.websocket-api</artifactId>
    <version>1.1</version>
</dependency>

java

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.*;

@ServerEndpoint(value="/shopWebsocket/{shopId}")
public class ShopWebSocket {
    private static final String SESSION_KEY_PRE = "websocketId";
    public static Map<String, Session> sessions = new HashMap<>();
    private static String shopId;

    private Logger log = LoggerFactory.getLogger(ShopWebSocket.class);

    //连接时执行
    @OnOpen
    public void onOpen(@PathParam("shopId") String shopId, Session session) throws IOException {
        this.shopId = shopId;
        log.info("session id: " + session.getId());
        log.info("新连接：{}", shopId);
        sessions.put(shopId, session);
    }

    //关闭时执行
    @OnClose
    public void onClose(){
        log.info("连接：{} 关闭", this.shopId);
        sessions.remove(shopId);
    }

    //收到消息时执行
    @OnMessage
    public void onMessage(String message, Session session) throws IOException {
        log.info("收到用户{}的消息{}", this.shopId, message);
        if (message.equals("ping")) {
            session.getBasicRemote().sendText("heartCheck"); //回复用户
            return;
        }
        session.getBasicRemote().sendText("收到 " + this.shopId + " 的消息: " + message); //回复用户
    }

    //连接错误时执行
    @OnError
    public void one rror(Session session, Throwable error){
        log.info("用户id为：{}的连接发送错误", this.shopId);
        error.printStackTrace();
    }

    /**
     * 发送消息
     * @author zhuxiang
     * @date 3:52 下午 2021/7/19
     * @param: shopId
     * @param: msg
     * @return void
     */
    public static boolean sendMsg(String shopId, String msg)
        throws IOException {
        Session session = sessions.get(shopId);
        if (session == null) {
            return false;
        }
        session.getBasicRemote().sendText(msg);  // 发送消息
        return true;
    }

    /**
     * 获取药店ID
     * @author zhuxiang
     * @date 10:54 上午 2021/7/20
     * @param:
     * @return java.util.List<java.lang.String>
     */
    public synchronized static List<String> shopIdList() {
        List<String> list = new ArrayList<>();
        for (String shopId : ShopWebSocket.sessions.keySet()) {
            list.add(shopId);
        }
        return list;
    }

}

js

<!DOCTYPE html>
<html>
<head>

</head>
<body>
<script src="http://cdn.jsdelivr.net/sockjs/1/sockjs.min.js"></script>

<script type="text/javascript">
	var ws = null;
	function openWebSocket(){
	    //判断当前浏览器是否支持WebSocket
	    if ('WebSocket' in window) {
	        ws = new WebSocket("ws://192.168.0.122:8080/taodoctor/shopWebsocket/user000");
	    } else {
	        ws = new SockJS("http://localhost:8080/taodoctor/sockjs/myWebSocket/info?type=mall");
	    }
	    ws.onopen = function () {

	    };　　//这个事件是接受后端传过来的数据
	    ws.onmessage = function (event) {
	        //根据业务逻辑解析数据
	        console.log(event);
	    };
	    ws.onclose = function (event) {

	    };
	}
	openWebSocket();
</script>
</body>
</html>

标签：websocket,String,shopId,即时通讯,session,ws,import,public
来源： https://www.cnblogs.com/zhuxiang1633/p/15041563.html