spring cloud实现websocket

最新推荐文章于 2024-06-16 09:36:16 发布
最新推荐文章于 2024-06-16 09:36:16 发布
阅读量3k 收藏 4
点赞数 3
分类专栏： spring cloud
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u013730110/article/details/97632106
版权

1、java源代码

(1)、WebSocketServer类

import java.lang.management.ManagementFactory;
import java.util.ArrayList;
import java.util.List;

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import com.portal.config.ApplicationContextRegister;
import com.portal.model.MessageVO;

@Component
@ServerEndpoint(value = "/socket/{token}/{bsCode}")
public class WebSocketServer {
	// 解决socket无法注入问题方案1
	// private static ApplicationContext applicationContext;
	// public static void setApplicationContext(ApplicationContext
	// applicationContext) {
	// WebSocketServer.applicationContext = applicationContext;
	// }
	// 解决socket无法注入问题方案2
	ApplicationContext act = ApplicationContextRegister.getApplicationContext();
	public common _common = act.getBean(common.class);

	private final static Logger logger = LoggerFactory.getLogger(WebSocketServer.class);
	public static final String NO = "NO";

	/*
	 * socket连接方法
	 */
	@OnOpen
	public void onOpen(@PathParam(value = "token") String token, @PathParam(value = "bsCode") String bsCode,
			Session session) {
		try {
			SaveUserSession(token, bsCode, session);
		} catch (Exception e) {
			logger.error("onOpen(),msg:" + e.getMessage() + ",---" + e.getStackTrace() + ",---" + e.getCause());
		}
	}

	/**
	 * 连接关闭调用的方法
	 */
	@OnClose
	public void onClose(@PathParam(value = "token") String token, @PathParam(value = "bsCode") String type,
			Session session) {
		try {
			String key = getUserKey(token, type);

			// 心跳检查是否在遍历对象
			if (!common.IsHeartbeatCheck) {
				if (!key.equals(NO)) {
					System.out.println("用户关闭,key:" + key);
					common.sl.remove(key);
					//common.curOnlineCount--;
				}
				//logger.info("移除连接,key:" + key);

				// 方法2
//				if (!key.equals(NO)) {
//					System.out.println("用户关闭,token:" + token);
//					List<Session> curUserList = new ArrayList<>();
//					if (common.sl.containsKey(key)) {
//						curUserList = (List<Session>) common.sl.get(key);
//						if (curUserList != null && curUserList.size() > 0) {
//							curUserList.remove(session);
//						}
//						common.sl.remove(key);
//						common.sl.put(key, curUserList);
//					}
//				}
			}
		} catch (Exception e) {
			logger.error("onClose(),msg:" + e.getMessage() + "," + e.getStackTrace());
		}
	}

	/**
	 * 收到客户端消息后调用的方法
	 *
	 * @param message
	 *            客户端发送过来的消息
	 */
	@OnMessage
	public void onMessage(String message, Session session) {
		MessageVO vo = new MessageVO();
		try {
			logger.info("message:" + message + ",msgType:");
			vo.setData("服务器接收消息成功");
			_common.sendMessage(vo, session);
		} catch (Exception e) {
			System.out.println("onMessage(),msg:" + e.getMessage() + "," + e.getStackTrace());
		}

	}

	// 拼接用户key
	public String getUserKey(String token, String bsCode) {
		String key = "";
		String userLoginName = _common.getUserLoginNameByToken(token);
		if (userLoginName.length() > 0) {
			key = userLoginName + "|" + bsCode; // 账号加业务类型作为key
		} else {
			key = NO;
		}
		return key;
	}

	/*
	 * desc:保存用户连接对象
	 */
	public void SaveUserSession(String token, String bsCode, Session session) {
		String key = getUserKey(token, bsCode);
		MessageVO vo = new MessageVO();
		List<Session> sArray = new ArrayList<>();
		if (!key.equals(NO)) {
			 logger.info("用户：token:" + token + ",连接成功");
			 //logger.info("用户key:" + key);
			 //若存在此连接对象，则移除再添加
			 if(common.sl.containsKey(key)) {
			 common.sl.remove(key);
			 }
			 common.sl.put(key, session);
			 //common.curOnlineCount++;
			 //logger.info("当前已连接总数："+common.curOnlineCount);
			 //logger.info("进程ID:"+ManagementFactory.getRuntimeMXBean().getName());
			 //logger.info("线程ID:"+Thread.currentThread().getId()+",name:"+Thread.currentThread().getName());
			
			 // 方案2
//			if (common.sl.containsKey(key)) {
//				sArray = (List<Session>) common.sl.get(key);
//				sArray.add(session);
//				common.sl.remove(key);
//			} else {
//				sArray.add(session);
//			}
//			common.sl.put(key, sArray);
			 
			vo.setStatus(1);
			vo.setData("连接成功,您好：" + key);
		} else {
			vo.setStatus(1);
			vo.setData("无效用户,您好：" + key);
		}
		_common.sendMessage(vo, session);
	}
}

（2）、common类

import java.text.MessageFormat;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.websocket.Session;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

import com.alibaba.fastjson.JSON;
import com.google.inject.spi.Message;
import com.portal.model.MessageVO;


@RestController
public class common {
	private final static Logger logger = LoggerFactory.getLogger(common.class);

	//public static Map<String,Session>  sessionlist=new HashMap<>();
	
	//保存建立连接用户的连接对象
	public static Map<String,Object>  sl=new HashMap<>();
	
	//当前连接数
	//public static int curOnlineCount;
	
	//判断是否在执行心跳检查服务
	public static boolean IsHeartbeatCheck=false;
	@Autowired
	public RestTemplate _RestTemplate;
	
	
	private String getLoginNameAddress="***";
	/*
	  * 回复客户端消息
	  * */
	 public void sendMessage(MessageVO vo,Session session)  {
		try {
			 String jsonObjStr=JSON.toJSONString(vo);
			 session.getBasicRemote().sendText(jsonObjStr);
		} catch (Exception e) {
			logger.error("回复客户端异常："+e.getMessage()+","+e.getStackTrace());
		}
	        
	 }

	 
	 /*
	  * desc: http get请求方法
	  * params:url 请求地址
	  * **/
	 public String doGet(String resquestUrl) {
		String result="";
		try {
			  result =_RestTemplate.getForObject(resquestUrl,String.class);
			  //logger.info("请求结果:"+result);
		} catch (Exception e) {
			logger.error("doGet请求异常："+e.getMessage()+","+e.getStackTrace());
		}
		return result;	
	 }
	 /*
	  * desc:获取用户token获取用户登录名
	  * params:token
	  * */
	 public  String getUserLoginNameByToken(String token) {
		 String result="";
		 result = doGet(MessageFormat.format(getLoginNameAddress, token)).replaceAll("\"","") ;
		 return result;
	 }
	 
	 
}

以下是配置类文件
（3）、CorsConfigurer（设置全局跨域）

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.CorsRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

@Configuration
public class CorsConfigurer extends WebMvcConfigurerAdapter{
	
	public void addCorsMappings(CorsRegistry registry) {
        registry.addMapping("/**")
                .allowedOrigins("*") 
                .allowedMethods("GET", "HEAD", "POST","PUT", "DELETE", "OPTIONS")
                .allowCredentials(false).maxAge(3600);
    }

}

2、前端实现代码

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8" />
    <title></title>
</head>
<body>
    <input id="text" type="text" />
    <button onclick="send()">Send</button>
    <button onclick="closeWebSocket()">Close</button>
    <br />
    <div id="message">
    </div>
    <script src="https://code.jquery.com/jquery-3.1.1.min.js"></script>
    <script>

        var token = "220627AF-52DD-4D6C-99E8-C6CD7B50E2DD";
        if (getQueryString("token") != null && getQueryString("token").length > 0) {
            token = getQueryString("token");
        }
        var bsCode = "TASK1";
        if (getQueryString("bsCode") != null && getQueryString("bsCode").length > 0) {
            bsCode = getQueryString("bsCode");
        }
        var websocket = null;
        $(function () {
            connectWebSocket();
        })
        //建立WebSocket连接
        function connectWebSocket() {
            console.log("开始...");
            //建立webSocket连接
            websocket = new WebSocket("ws://localhost:3005/socket/" + token + "/" + bsCode);
            //打开webSokcet连接时，回调该函数
            websocket.onopen = function () {
                console.log("用户:" + token);
                console.log("onpen");
            }
            //关闭webSocket连接时，回调该函数
            websocket.onclose = function () {
                //关闭连接
                console.log("onclose");
            }
            //接收信息
            websocket.onmessage = function (msg) {
                //console.log(msg.data);
                var obj = JSON.parse(msg.data);
                console.log(obj);
                console.log("消息正文:");
                //console.log(JSON.parse(obj.data));
                $("#message").append("<div><span style='color:red;'>推送消息正文：</span><br>" + msg.data + "<div>");
            }
        }
        //发送消息
        function send() {
            var postValue = {};
            postValue.id = token;
            postValue.message = $("#text").val();
            postValue.msgType = "测试发送其它字段添加字段";
            websocket.send(JSON.stringify(postValue));
        }
        //关闭连接
        function closeWebSocket() {
            if (websocket != null) {
                websocket.close();
            }
        }
        function getQueryString(name) {
            var reg = new RegExp("(^|&)" + name + "=([^&]*)(&|$)");
            var r = window.location.search.substr(1).match(reg);
            if (r != null) return unescape(r[2]); return null;
        }
        //window.open('http://www.imooc.com','_blank')//')
    </script>
</body>
</html>

以下为个人功能添加代码，与websocket无关

(4) taskController(推送消息方法，非核心方法，个人使用)


import java.util.ArrayList;
import java.util.List;

import javax.websocket.Session;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.portal.model.MessageVO;
import com.portal.model.ReceiveDTO;

@RestController
@RequestMapping("socket")
public class taskController {
	private final static Logger logger = LoggerFactory.getLogger(taskController.class);
	
	@Autowired
	public common _common;
	@PostMapping(value = "/pushNotification")
	public MessageVO pushNotification(@RequestBody ReceiveDTO receiveDto ) {
		String key="";
		String userLoginName="";
		String bsCode="";
		String jsonData="";
		Session ses=null;
		MessageVO vo=new MessageVO();
		MessageVO res=new MessageVO();
		//List<Session> sArray=new ArrayList<>();
		try {
			if(receiveDto!=null) {
				res.setData("消息接收成功");
				res.setStatus(1);
				jsonData=receiveDto.getData();
				List<String> userList=receiveDto.getUserList();
				List<String> bsCodeList=receiveDto.getBsCodeList();
				if(userList!=null&&userList.size()>0&&bsCodeList!=null&bsCodeList.size()>0) {
					for (int i = 0,len=userList.size(); i < len; i++) {
						userLoginName=userList.get(i);
						for (int j = 0,len2=bsCodeList.size(); j <len2; j++) {
							bsCode=bsCodeList.get(j);
							key=userLoginName+"|"+bsCode;
							ses=(Session)common.sl.get(key);
							vo.setStatus(1);
							vo.setData(jsonData);
							if(ses!=null&&ses.isOpen()) {
								_common.sendMessage(vo, ses);
								//logger.info("用户key:"+key+",jsonData:"+jsonData+",消息推送成功");
								logger.info("用户key:"+key+",消息推送成功");
							}else {
								logger.info("消息推送失败，用户key:"+key+",已关闭连接");
								res.setData("消息推送失败，用户未连接");
								res.setStatus(-1);
							}
							//方案2
//							sArray=(List<Session>)common.sl.get(key);
//							if(sArray!=null&&sArray.size()>0) {
//								vo.setStatus(1);
//								vo.setData(jsonData);
//								sendUserMsg(sArray,vo,key,jsonData);
//							}
							
						}
					}
				}else {
					res.setData("userList或bsCodeList参数为空");
					res.setStatus(0);
				}
			}else {
				res.setData("参数为空");
				res.setStatus(0);
			}
		} catch (Exception e) {
			logger.error("pushNotification()，错误信息："+e.getMessage()+",StackTrace:"+e.getStackTrace());
		}
		return res;
	} 
	
	
	//推送消息
	public  void sendUserMsgByObj(Session ses,MessageVO vo,String key,String jsonData) {
		if(ses!=null&&ses.isOpen()) {
			_common.sendMessage(vo, ses);
			logger.info("用户key:"+key+",jsonData:"+jsonData+",消息推送成功");
		}else {
			logger.info("用户key:"+key+",已关闭连接");
		}
	}
	//推送消息
	public  void sendUserMsgByArray(List<Session> sArray,MessageVO vo,String key,String jsonData) {
		Session ses=null;
		for (int i = 0,len=sArray.size(); i < len; i++) {
			ses=sArray.get(i);
			if(ses.isOpen()) {
				_common.sendMessage(vo, ses);
				logger.info("用户key:"+key+",jsonData:"+jsonData+",消息推送成功");
			}else {
				logger.info("用户key:"+key+",已关闭连接");
			}
			
		}
	}

（4）、ReceiveDTO（消息推送接收信息，个人使用）

import java.util.List;

public class ReceiveDTO {
	public List<String> userList;
	public List<String> bsCodeList;
	public String data;
	
	public List<String> getUserList() {
		return userList;
	}
	public void setUserList(List<String> userList) {
		this.userList = userList;
	}
	public List<String> getBsCodeList() {
		return bsCodeList;
	}
	public void setBsCodeList(List<String> bsCodeList) {
		this.bsCodeList = bsCodeList;
	}
	public String getData() {
		return data;
	}
	public void setData(String data) {
		this.data = data;
	}
	
}

3、特殊说明

spring cloud中使用@ServerEndpoint注解后，是无法使用@Autowired注解注入其它类的，可使用以下解决方法2注入其它公共类；
在这里插入图片描述
以上方法如有疑问，可在下方回复评论

大海中一粒沙子
关注
  • 3
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
websocket替代方案_分布式WebSocket集群解决方案
weixin_39738115的博客
12-23 2030
问题起因最近做项目时遇到了需要多用户之间通信的问题，涉及到了WebSocket握手请求，以及集群中WebSocket Session共享的问题。期间我经过了几天的研究，总结出了几个实现分布式WebSocket集群的办法，从zuul到spring cloud gateway的不同尝试，总结出了这篇文章，希望能帮助到某些人，并且能一起分享这方面的想法与研究。以下是我的场景描述资源：4台服务器。其中只有...
spring websocket实现
浅粒的博客
05-18 424
注：需spring4.0及以上版本。步骤：1、导入jar包（maven方式）&lt;!-- Spring websocket --&gt; &lt;dependency&gt; &lt;groupId&gt;org.springframework&lt;/groupId&gt; &lt;artifactId&gt;spring-websocket&lt;/artifactId&gt...
SpringCloud 网关配置websocket
feiz3020的专栏
06-07 1055
this.websocket = new WebSocket(process.env.VUE_APP_WS_URL + '/orderWebsocket') // 创建WebSocket连接。this.websocket.onclose = this.onWebSocketClose // WebSocket连接关闭时的处理函数。VUE_APP_WS_URL = 'wss://域名.com/websocket/ws/'this.websocket.close() // 关闭WebSocket连接。
推荐开源项目：Spring Cloud Netflix Zuul WebSocket 支持库
最新发布
gitblog_00036的博客
06-16 294
推荐开源项目：Spring Cloud Netflix Zuul WebSocket 支持库 项目地址:https://gitcode.com/mthizo247/spring-cloud-netflix-zuul-websocket 在这个日益互联网化的时代，WebSockets已经成为实时通信的重要工具。然而，Spring Cloud Netflix Zuul——一个强大的边缘服务和API网关...
详解Spring Cloud微服务架构下的WebSocket解决方案
08-26
主要介绍了详解Spring Cloud微服务架构下的WebSocket解决方案，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
Spring Cloud整合Websocket（SockJs Stomp方式）
船长灬普朗克的博客
10-20 7097
目录一、服务端配置1、引入依赖包2、编写Config类进行websocket的配置3、编写拦截器WebSocketInterceptor4、编写消息的接收和发送的Controller二、客户端配置1、引入SockJS以及stompjs2、编写公用的websocket.js供其他地方调用3、业务代码中引用websocket进行消息发送和接收三、Spring cloud gateway网关配置1、修改gateway网关的配置文件2、在网关中添加全局过滤器四、前端代理的配置五、问题记录 一、服务端配置 整个服务端
Spring Cloud 微服务架构下的 WebSocket 解决方案
m0_67393413的博客
08-24 4354
WebSocket在现代浏览器中的应用已经算是比较普遍了，在某些业务场景下，要求必须能够在服务器端推送消息至客户端。在没有WebSocket的年代，我们使用过dwr，在那个时候dwr真实一个非常棒的方案。但是在WebSocket兴起之后，我们更愿意使用标准实现来解决问题、首先交代一下，本篇文章不讲解WebSocket的配置，主要讲的是针对在微服务架构集群模式下解决方案的选择。微服务架构大家应该都不陌生了，在微服务架构下，服务是分布式的，而且为了保证业务的可用性，每个服务都是以集群的形式存在。
springCloud(若依微服务版)集成WebSocket实现消息推送
weixin_43931108的博客
11-15 2013
websocket
微服务springcloud环境下基于Netty搭建websocket集群实现服务器消息推送----netty是yyds
weixin_45131680的博客
06-25 4810
本文主要讲述了网络编程相关的IO模型以及NIO框架netty、如何搭建websocket。
Spring环境下WebSocket简单实现
™red scarf的博客
08-12 208
1.WebSocket简介 WebSocket是HTML5提供的一种全双工通信的协议，通常是浏览器（或其他客户端）与Web服务器之间的通信。这使得它适合于高度交互的Web应用程序，如及时通讯聊天等。 WebSocket协议是基于TCP的一种新的网络协议。它实现了浏览器与服务器全双工(full-duplex)通信——可以通俗的解释为服务器主动发送信息给客户端。 WebSocket首次在H...
springcloud下通过gateway转发websocket
07-14
本文将深入探讨如何在Spring Cloud Gateway中配置和使用WebSocket，以实现基于Spring Cloud的微服务架构下的WebSocket通信。 首先，理解WebSocket的基础概念至关重要。WebSocket协议是HTTP/1.1协议的一个补充，它在...
springcloud实现gate网关转发功能和整合websocket源码
04-25
在本文中，我们将深入探讨如何使用Spring Cloud Gateway实现网关转发功能，并整合WebSocket源码，以便在微服务架构中提供高效、灵活的数据通信。首先，让我们先了解一下Spring Cloud Gateway及其重要性。 Spring ...
websocket-cluster:这是一个针对WebSocket集群服务器的Spring Cloud项目。
03-29
实战Spring Cloud的WebSocket体现此项目是一个WebSocket实施的实践，基于Spring Cloud。原理我们利用一致性哈希算法，构造一个哈希环，网关监听WebSocket服务实例的上下线消息，根据实例的变化动态地更新哈希环。将...
springcloud getway 示例 包含 网关 http websocket 两种转发包含网关JWT鉴权包含Clie
08-05
UserApplication 模拟WebSocket服务 PtGateClientTests 模拟WebSocket请求服务 http可以通过网页模拟 （开启鉴权之后需要postman在header中添加token才能通过鉴权） 导入之后 maven跑完即可完整运行
springboot、cloud整合websocket
weixin_61240867的博客
06-13 907
话不多说，直接上代码：1.首先创建websocket的配置类2.创建websocket核心服务这样其实就已经可以连接了，但在实际运用中，会出现一种情况，如图：状态码明明显示的200，但却连接之后却立即断掉了，此时很有可能是被某个过滤器拦截掉，我查看了自己项目中的过滤器，发现都放行了，却在下一步直接断开；在过滤器与websocket服务之间只有一个，那就是用户认证。所以我先将websocket的认证给关掉。如图：我用的是spring-security的配置连接成功。
WebSocket：基于 Spring Cloud 配置注解实现 WebSocket 集群方案
欢迎来到 ABin-阿斌的博客：写一生代码，创一世佳话，筑一览芳华。
06-25 1489
WebSocket大家应该是再熟悉不过了，如果是单体应用确实不会有什么问题，但是当我们的项目使用微服务架构时，就可能会存在问题比如服务A有两个实例A1和A2，前端的WebSocket客户端C通过网关的负载均衡连到了A1，这个时候当A2触发消息发送的逻辑，需要将某个消息发送给所有的客户端时，C就接受不到消息这个时候我们很快就能想到一种最简单的解决方案，就是把A2的消息转发给A1A1再把消息发送给C，这样C就能收到A2发送的消息了基于这个思路，我实现了一个库，一个配置注解搞定一切。
SpringCloud Gateway 集成 WebSocket服务
热门推荐
weixin_40261814的博客
01-27 1万+
作为一名程序员，工作也7、8年了，这是我第一次写文章，实在是惭愧。之前因工作需要一直都是百度、google，而没有想过静下心来自己研究原理、源码，导致现在还是一只菜鸟。。。 好了。言归正传，本次实践的话是基于nacos作为配置、注册中心的，各位大佬可以去了解一下nacos的官网，虽然最近这个组件爆出了漏洞，但不可否认其功能还真的是很强大的，附上官网地址：https://nacos.io/zh-cn/ 项目使用jdk1.8进行开发。使用maven做jar包管理，分了很多模块子项目，包括基础服务的commo
SpringCloud微服务整合websocket
qq_40996015的博客
10-10 2071
可以通过网关路径连接websocket。
SpringCloud整合WebSocket实现用户监控
folyh
03-05 1979
文章目录前言一、建项目1. 在父项目ams-cloud下建立maven子项目ams-websocket2.pom文件添加常用依赖，另外添加redis依赖等，我这里直接引用common模块3.添加bootstrap.yml文件4.创建application二、添加config类、拦截器类、处理器类等1.添加config类2.添加拦截器类3.添加处理器类三、添加controller1.controller如下：2.运行结果总结 前言 一、建项目 1. 在父项目ams-cloud下建立maven..
springcloud gateway websocket
04-30
在传统的架构中，要实现Websocket，需要单独搭建WebSocket服务器，并且与其他微服务的交互存在一定的难度，而SpringCloud Gateway可以通过使用统一的API网关服务，将WebSocket接入统一管理，降低了开发难度并提高了...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

大海中一粒沙子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值