packagecom.jeecg.webSocket;importnet.sf.json.JSONObject;import javax.websocket.*;importjavax.websocket.server.PathParam;importjavax.websocket.server.ServerEndpoint;importjava.io.IOException;importjava.util.Map;importjava.util.concurrent.ConcurrentHashMap;
@ServerEndpoint("/webSocket/{username}")public classWebSocket {private static int onlineCount = 0;private static Map clients = new ConcurrentHashMap();privateSession session;privateString username;
@OnOpenpublic void onOpen(@PathParam("username") String username, Session session) throwsIOException {this.username =username;this.session =session;
addOnlineCount();
clients.put(username,this);
System.out.println("已连接");
}
@OnClosepublic void onClose() throwsIOException {
clients.remove(username);
subOnlineCount();
}
@OnMessagepublic void onMessage(String message) throwsIOException {
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("给所有人");
}
}
@OnErrorpublic voidonError(Session session, Throwable error) {
error.printStackTrace();
}public void sendMessageTo(String message, String To) throwsIOException {for(WebSocket item : clients.values()) {if(item.username.equals(To) )
item.session.getAsyncRemote().sendText(message);
}
}public void sendMessageAll(String message) throwsIOException {for(WebSocket item : clients.values()) {
item.session.getAsyncRemote().sendText(message);
}
}public static synchronized intgetOnlineCount() {returnonlineCount;
}public static synchronized voidaddOnlineCount() {
WebSocket.onlineCount++;
}public static synchronized voidsubOnlineCount() {
WebSocket.onlineCount--;
}public static synchronized MapgetClients() {returnclients;
}
}

@JsonFormat(pattern = “yyyy-MM-dd HH:mm:ss”)在时间上面加入该注释，直接在application.yml配置是无法识别的

最近做自动化需要接收websocket消息来进行判断，做个总结：
使用websoket-client库，其git主页：https://github.com/websocket-client/websocket-client
pip install websocket_client

代码在下面，注意：cookie要发送http登录请求去获取然后传给这边的接口
from websocket import create_connection
import json
import time

url = "ws://127.0.0.1:8080/test"
header = ['Cookie: shiro.session=ffdsftedd-d234-2233;Path=/;HttpOnly, remem……']
ws = create_connection(url, header=header)
ws.send('login')  #发送握手消息
end_time = time.time() + 10
tmp_list = []
while time.time() < end_time:  #接收10s数据
	mes = ws.recv()
	print(mes)
	if json.loads(mes)["key"] == 'target': #拿自己想要的数据
		tmp_list.append(json.loads(mes))  


遇到的问题：


连接报错，如下图：WinError 10042 在getsockopt 或 setsockopt调用中指定一个未知的，无效的或不受支持的选项或层次

这个是因为socket设置了一个默认的配置，而这个配置在windows某些版本不支持（我用的是win7不支持，其它不清楚）



解决办法：

找到site-packages\websocket_socket.py文件，搜索TCP_KEEPCNT关键字，注释相关代码，如下图




如何发送cookie

开始我只是把cookie值填进去，用requests登录接口获取cookie发现送的不止是cookie值，还有其它，所以最好是发送登录接口获取cookie然后再赋给websocket。

websocket应用场景：
1、即时通信。比如QQ，微信这些。
2、站内消息推送，比如打开qq时会提示xx用户已上线之类在app或者网站内部的消息。
3、实时数据获取。比如地理位置，用户是否在线等需要实时获取的消息。
4、解决http长轮询消耗服务器资源问题。
需要使用的框架：
1、gin框架：gin-gonic/gin
2、websocket框架：gorilla/websocket
3、api文档：
https://godoc.org/github.com/gin-gonic/gin
https://godoc.org/github.com/gorilla/websocket#Conn.ReadJSON
示例代码：
package main

import (
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"net/http"
)
//设置websocket
//CheckOrigin防止跨站点的请求伪造
var upGrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

//websocket实现
func ping(c *gin.Context) {
	//升级get请求为webSocket协议
	ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		return
	}
	defer ws.Close() //返回前关闭
	for {
		//读取ws中的数据
		mt, message, err := ws.ReadMessage()
		if err != nil {
			break
		}
		//写入ws数据
		err = ws.WriteMessage(mt, message)
		if err != nil {
			break
		}
	}
}

func main() {
	r := gin.Default()
	r.GET("/ping", ping)
	r.Run(":12345")
}
测试：
websocket请求地址：ws://localhost:12345/ping
实际应用中的分析：
上面的例子只是实现了最基本的websocket功能，客户端发送消息，服务端接收到以后就按原样返回（json数据会作BASE64加密）。但在实际的项目中，需要处理的逻辑远比这个复杂。而且数据在传输过程中没有加密，因此在实际应用中，使用json加密后传输。客户端或服务端接收到的只是一串字符串，内部解密后再进行相应的业务逻辑处理。如果业务逻辑复杂，可以配合channel,redis，mysql等一起使用。
参考文章：
gin 框架中结合 gorilla 实现 webSocket - 链滴​ld246.com