SSM框架WebSocket配置

1、StartFilter.java

package cn.xydata.pharmacy.websocket;

import java.io.IOException;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;

import org.java_websocket.WebSocketImpl;

import cn.xydata.pharmacy.websocket.WSServer;;




public class StartFilter implements Filter {

    public void destroy() {

    }

    public void doFilter(ServletRequest arg0, ServletResponse arg1,
            FilterChain arg2) throws IOException, ServletException {

    }

    public void init(FilterConfig arg0) throws ServletException {
        this.startWebsocketInstantMsg();
    }

    /**
     * 啟動即時聊天服務
     */
    public void startWebsocketInstantMsg() {
        WebSocketImpl.DEBUG = false;
        WebServer s = new WebServer(6004);
        s.start();
    }
}

2、WebServer.java

package cn.xydata.pharmacy.websocket;

import java.net.InetSocketAddress;

import org.java_websocket.WebSocket;
import org.java_websocket.WebSocketImpl;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

public class WebServer extends WebSocketServer {
    public WebServer(int port) {
        super(new InetSocketAddress(port));
    }

    public WebServer(InetSocketAddress address) {
        super(address);
    }

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        // ws連接的時候觸發的代碼，onOpen中我們不做任何操作
        System.out.println("onOpen");
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        //斷開連接時候觸發代碼
        userLeave(conn);
        System.out.println(reason);
        System.out.println("onClose");
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
        System.out.println(message);
        System.out.println("onMessage");
        if(null != message){
            String userName = message.replaceFirst("online", message);//用戶名
            userJoin(conn,userName);//用戶加入
        }

    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        //錯誤時候觸發的代碼
        System.out.println("on error");
        ex.printStackTrace();
    }
    /**
     * 去除掉失效的websocket鏈接
     * @param conn
     */
    private void userLeave(WebSocket conn){
        WsPool.removeUser(conn);
    }
    /**
     * 將websocket加入用戶池
     * @param conn
     * @param userName
     */
    private void userJoin(WebSocket conn,String userName){
        WsPool.addUser(userName, conn);
    }

//    public void start() {
//        if( selectorthread != null )
//            throw new IllegalStateException( getClass().getName() + " can only be started once." );
//        new Thread( this ).start();;
//    }
    
//    public static void main(String args[]){
//        WebSocketImpl.DEBUG = false;
//        int port = 8887; // 端口
//        WsServer s = new WsServer(port);
//        s.start();
//    }

    @Override
    public void onStart() {
        System.out.println("onstart");
    }
    
}

 

3、WsPool.java

package cn.xydata.pharmacy.websocket;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.java_websocket.WebSocket;

public class WsPool {
    private static final Map<WebSocket, String> wsUserMap = new HashMap<WebSocket, String>();

    /**
     * 通過websocket連接獲取其對應的用戶
     * 
     * @param conn
     * @return
     */
    public static String getUserByWs(WebSocket conn) {
        return wsUserMap.get(conn);
    }

    /**
     * 根據userName獲取WebSocket,這是一個list,此處取第一個
     * 因為有可能多個websocket對應一個userName（但一般是只有一個，因為在close方法中，我們將失效的websocket連接去除了）
     * 
     * @param user
     */
    public static WebSocket getWsByUser(String userName) {
        Set<WebSocket> keySet = wsUserMap.keySet();
        synchronized (keySet) {
            for (WebSocket conn : keySet) {
                String cuser = wsUserMap.get(conn);
                if (cuser.equals(userName)) {
                    return conn;
                }
            }
        }
        return null;
    }

    /**
     * 向連接池中添加連接
     * 
     * @param inbound
     */
    public static void addUser(String userName, WebSocket conn) {
        wsUserMap.put(conn, userName); // 添加連接
    }

    /**
     * 獲取所有連接池中的用戶，因為set是不允許重復的，所以可以得到無重復的user數組
     * 
     * @return
     */
    public static Collection<String> getOnlineUser() {
        List<String> setUsers = new ArrayList<String>();
        Collection<String> setUser = wsUserMap.values();
        for (String u : setUser) {
            setUsers.add(u);
        }
        return setUsers;
    }

    /**
     * 移除連接池中的連接
     * 
     * @param inbound
     */
    public static boolean removeUser(WebSocket conn) {
        if (wsUserMap.containsKey(conn)) {
            wsUserMap.remove(conn); // 移除連接
            return true;
        } else {
            return false;
        }
    }

    /**
     * 向特定的用戶發送數據
     * 
     * @param user
     * @param message
     */
    public static void sendMessageToUser(WebSocket conn, String message) {
        if (null != conn && null != wsUserMap.get(conn)) {
            conn.send(message);
        }
    }

    /**
     * 向所有的用戶發送消息
     * 
     * @param message
     */
    public static void sendMessageToAll(String message) {
        Set<WebSocket> keySet = wsUserMap.keySet();
        synchronized (keySet) {
            for (WebSocket conn : keySet) {
                String user = wsUserMap.get(conn);
                if (user != null) {
                    conn.send(message);
                }
            }
        }
    }

}

4、WSServer.java

package cn.xydata.pharmacy.websocket;

import java.io.UnsupportedEncodingException;
import java.util.Collection;

import org.java_websocket.WebSocket;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;


//@ServerEndpoint注解中的內容是用戶客戶端websocket的連接url,例如ws://127.0.0.1:80/websocket/ws,
//結構形式為“ws：//ip:端口/項目名/指定的url”
@Controller
@RequestMapping(value="WSServer")
public class WSServer {
    
//    @SystemControllerLog(description="添加用戶信息")
//    @RequestMapping(value="addParticipant",method = RequestMethod.POST)
//    @ResponseBody()
//    public Object addParticipant(HttpServletRequest request,
//            HttpServletResponse response) {
//         Map<String,Object> result = new HashMap<String,Object>();
//         
//        result.put("success", "500");
//        return result;
//    }
    
    @ResponseBody
    @RequestMapping("sendWs")
    public String sendWs(String message) throws Exception {
        message = new String(message.getBytes("ISO-8859-1"),"UTF-8");    
        WsPool.sendMessageToAll(message);
        return message;
    }
    
    @ResponseBody
    @RequestMapping("sendWsOne")
    public String sendWsOne() {
        Collection<String> conns = WsPool.getOnlineUser();
        System.out.println(conns + "sendWsOne");
        return "123";
    }
    
}

另需兩個jar包：

javax.websocket-api-1.1.jar

Java-WebSocket-1.3.7.jar

點擊下載

在web.xml中配置過濾器如下：

<filter>
   <filter-name>startFilter</filter-name>
   <filter-class>StartFilter.java路徑</filter-class>
</filter>

 怎么引用？一段代碼即可：

WsPool.sendMessageToAll(p.getRealName()+","+p.getPosition()+","+mp.getSeatNo());

客戶端怎么弄？

　　　　　　　var ws = new WebSocket("ws://180.76.236.11:6005");　　//握手協議，服務器ip地址以及服務器提供給WebSocket進程端口，與StartFilter.java中startWebsocketInstantMsg方法內的端口保持一致
            
            ws.onopen = function(){
                ws.send("online-hello");
                console.log("連接...")
            }
            
            //處理服務器發送來的數據
            ws.onmessage = function(e){
                var text = e.data;
                var arr = text.split(',');
                sign(arr[0],arr[1],arr[2]); //sign函數的作用是將從服務器發送來的參會人員信息顯示在歡迎屏
            }
            
            ws.onclose = function(){
                console.log("連接關閉");
            }
            
            ws.onerror = function(){
                console.log('連接失敗');
            }