Java使用WebSocket

2019独角兽企业重金招聘Python工程师标准>>>

maven引入jar包或者使用tomcat8以上中的websocket-api.jar

<dependency>
	<groupId>javax</groupId>
	<artifactId>javaee-api</artifactId>
	<version>7.0</version>
    <scope>provided</scope>
</dependency>

WebSocket.java

package com.web.common;

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.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * WebSocket连接 sessionKey是url中的参数
 */
@ServerEndpoint("/websocket/{sessionKey}")
public class WebSocket {

    private static final Logger log = LoggerFactory.getLogger(WebSocket.class);

    //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;

    //concurrent包的线程安全，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
    private static Map<String, WebSocket> webSockets = new ConcurrentHashMap<>();

    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;

    /**
     * 连接建立成功调用的方法
     * @param session 可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
     * @param sessionKey url地址参数
     */
    @OnOpen
    public void onOpen(Session session, @PathParam("sessionKey") String sessionKey){

        if (!webSockets.containsKey(sessionKey)){
            this.session = session;
            webSockets.put(sessionKey, this);
            addOnlineCount();
            log.info("当前websocket连接数：" + onlineCount);
        }
    }

    /**
     * 连接关闭调用的方法
     * @param sessionKey url地址参数
     */
    @OnClose
    public void onClose(@PathParam("sessionKey") String sessionKey){

        if (webSockets.containsKey(sessionKey)){
            webSockets.remove(sessionKey);
            subOnlineCount();
            log.info("当前websocket连接数：" + onlineCount);
        }
    }

    /**
     * 收到客户端消息后调用的方法
     * @param message 客户端发送过来的消息
     * @param session 可选的参数
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        log.info("来自客户端的消息:" + message);
    }

    /**
     * 发生错误时调用
     * @param session 可选的参数
     * @param error 错误消息
     */
    @OnError
    public void onError(Session session, Throwable error){
        log.error("websocket发生错误：" + error);
    }

    /**
     * 该方法没有用注解，是根据自己需要添加的方法。在自己的业务中调用，发送消息给前端。
     * @param sessionKey
     * @param message 返回的结果
     * @throws IOException
     */
    public static void sendMessage(String sessionKey,String message) throws IOException {

        WebSocket webSocket = webSockets.get(sessionKey);

        if (null != webSocket){

            log.info("websocket发送消息：" + message);

            //同步发送 发送第二条时，必须等第一条发送完成
            webSocket.session.getBasicRemote().sendText(message);

            //异步发送
            //webSocket.session.getAsyncRemote().sendText(message);
        }
    }

    public static synchronized int getOnlineCount() {
        return onlineCount;
    }

    public static synchronized void addOnlineCount() {
        onlineCount++;
    }

    public static synchronized void subOnlineCount() {
        onlineCount--;
    }

}

前端js

    var webSocket = null;

    $(function () {

        //判断当前浏览器是否支持WebSocket  
        if ('WebSocket' in window) { 
            webSocket = new _WebSocket("${WEBSOCKET_URL}/websocket/${(receipt.id)!''}");
        } else {
            alert('当前浏览器不支持webSocket');
        }

       //连接成功建立的回调方法  
        websocket.onopen = function() {  
            console.info("WebSocket连接成功");  
        };

        //连接出错
        webSocket.onerror = function (ev) {
            console.info('连接错误', ev);
        };

        //接收到消息的回调方法
        webSocket.onmessage = function (ev) {
            console.info(ev);
        };

        //连接关闭的回调方法
        webSocket.onclose = function () {
            console.info('WebSocket连接关闭');
        };
       
    });

    //关闭WebSocket连接
    function closeWebSocket() {
        webSocket.close();
    }

 

转载于:https://my.oschina.net/u/3445245/blog/3003208