WebSocket简单样例

WebSocket简单样例

1.目的：通过WebSocekt实现Java服务端将agent状态实时推送到html页面

2. 测试环境:win7 64位 旗舰版，jdk8，Tomcat8，Mycelispe2014

3.服务端代码： 

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;

/**
 * @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务端
 * 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 */
@ServerEndpoint("/websocket/{agentcontact}")
public class TestWebSocket {
	//concurrent包的线程安全Map,用来存放每个客户端对应的WebSocket对象,其中Key为坐席标识
	private static Map agentclientmap = new ConcurrentHashMap();  
    private String agentcontact; //用户标识
	private Session session; //与某个客户端的连接会话，需要通过它来给客户端发送数据
		
         /**
		 * 连接建立成功调用的方法
		 * @param session可选的参数 session为与某个客户端的连接会话，需要通过它来给客户端发送数据
		 */
		@OnOpen
		public void onOpen(@PathParam("agentcontact") String agentcontact, Session session){
			this.agentcontact = agentcontact;  
	         this.session = session; 
              // ConcurrentHashMap key值不重复，保证每个用户标识对应一个websocket类，避免刷新，登录等操作出现一个用户标识对应多个客户端
		     agentclientmap.put(agentcontact,this);
		}

		/**
		 * 连接关闭调用的方法
		 */
		@OnClose
		public void onClose(){
              //关闭ws连接，从map中移除对应的key
			agentclientmap.remove(agentcontact);
		}

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

		/**
		 * 发生错误时调用
		 * @param session
		 * @param error
		 */
		@OnError
		public void onError(Session session, Throwable error){
			System.out.println("发生错误");
			error.printStackTrace();
		}

		/**
		 * server向client推送消息方法
		 * @param message
		 * @throws IOException
		 */
	    public void pushMessage(String agentcontact,String message) throws IOException {  
	        //全部推送
/*for (TestWebSocket item : agentclientmap.values()) {  
	                item.session.getAsyncRemote().sendText(message);  
	        }*/
         //点对点推送
	    if(agentclientmap.containsKey(agentcontact)){
        	TestWebSocket ws_push = agentclientmap.get(agentcontact);
         // getAsyncRemote() 非阻塞发送方法，可以避免某客户端推送延迟导致整个服务器推送卡顿
        	ws_push.session.getAsyncRemote().sendText(message);
      	} 
 } 
}

4.客户端代码：

   
  
  
    Welcome

    发送消息
    

    关闭WebSocket连接