WebSocket服务端

记录用到的WebSocket服务端。

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
 
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
 
import net.sf.json.JSONObject;
@ServerEndpoint("/webSocket/{username}")  
    public class WebSocket { 
        private static int onlineCount = 0; 
        private static Map<String, WebSocket> clients = new ConcurrentHashMap<String, WebSocket>(); 
        private Session session; 
        private String username; 
           
        @OnOpen 
        public void onOpen(@PathParam("username") String username, Session session) throws IOException { 
       
            this.username = username; 
            this.session = session; 
               
            addOnlineCount(); 
            clients.put(username, this);
            System.out.println(username+"已连接");
        } 
       
        @OnClose 
        public void onClose() throws IOException { 
            clients.remove(username); 
            try {
				this.session.close();
			} catch (Exception e) {
				// TODO: handle exception
				e.printStackTrace();
			}
            subOnlineCount(); 
            System.out.println("关闭");
        } 
       
        @OnMessage 
        public void onMessage(String message) throws IOException { 
       
            JSONObject jsonTo = JSONObject.fromObject(message); 
            String mes = (String) jsonTo.get("message");
             
            if (!jsonTo.get("To").equals("All")){ 
                sendMessageTo(mes, jsonTo.get("To").toString()); 
            }else{ 
                sendMessageAll("给所有人"); 
            } 
        } 
       
        @OnError 
        public void onError(Session session, Throwable error) throws IOException { 
        	onClose();
        	System.out.println("发生错误");
            error.printStackTrace(); 
        } 
       
        public static void sendMessageTo(String message, String To) throws IOException { 
            // session.getBasicRemote().sendText(message); 
            //session.getAsyncRemote().sendText(message); 
            for (WebSocket item : clients.values()) { 
                if (item.username.equals(To) ) 
                    item.session.getAsyncRemote().sendText(message); 
            } 
        } 
           
        public void sendMessageAll(String message) throws IOException { 
            for (WebSocket item : clients.values()) { 
                item.session.getAsyncRemote().sendText(message); 
            } 
        } 
       
        public static synchronized int getOnlineCount() { 
            return onlineCount; 
        } 
       
        public static synchronized void addOnlineCount() { 
            WebSocket.onlineCount++; 
        } 
       
        public static synchronized void subOnlineCount() { 
            WebSocket.onlineCount--; 
        } 
       
        public static synchronized Map<String, WebSocket> getClients() { 
            return clients; 
        } 
}