前言
在原项目中,对于WebSocket的长连接,聊天系统并没有开放接口出来给第三方的系统调用,只有我们系统内部的人员才知道,确切的说系统内部也没有实际的查询接口,那么我们今天就来实现这个功能。
在Netty下的Websocket长连接中,以API形式获取在线用户数,与在线用户列表,并针对某个用户已API调用的形式进行数据发送,而不需要所谓的前端页面去创建websocket连接。
实践流程
存放Channel的容器
首先,我们需要一个类似ChannelGroup的连接池来存放我们的连接实例,这里我直接在原来本地模拟的一个LikeRedisTemplate中新建了一个ConcurrentHashMap,用于存放对应的用户名——连接实例的键值对。
方便后期API调用时可以通过这个LikeRedisTemplate中的这个Map进行获取、删除及相关信息。
/**存放链接池实例*/
private Map<Object,Object> ChannelRedisMap = new ConcurrentHashMap<>();
/**
* 存储对应的用户名与Netty链接实例
* @param name 登录用户名
* @param channel Netty链接实例
*/
public void saveChannel(Object name,Object channel){
ChannelRedisMap.put(name,channel);
}
/**
* 获取存储池中的链接实例
* @param name 登录用户名
* @return {@link io.netty.channel.Channel 链接实例}
*/
public Object getChannel(Object name){
return ChannelRedisMap.get(name);
}
/**
* 删除存储池实例
* @param name 登录用户名
*/
public void deleteChannel(Object name){
ChannelRedisMap.remove(name);
}
/**
* 获取储存池链接数
* @return 在线数
*/
public Integer getSize(){
return ChannelRedisMap.size();
}
/**
* 返回在线用户列表信息
* @return 用户名列表
*/
public Object getOnline() {
List<Object> result = new ArrayList<>();
for (Object key:ChannelRedisMap.keySet()){
result.add(key);
}
return result;
}
Handler中执行存储操作
有了容器,我们就需要在对应的位置进行连接实例的键值对存储,我目前选择了在聊天消息传输过程中进行存储,暂时还没有抽象出来。
并在连接断开时也要做相关的处理。
//用户登录判断
if (redisTemplate.check(incoming.id(),rName)){
//临时存储聊天数据
cacheTemplate.save(rName,rMsg);
//存储随机链接ID与对应登录用户名
redisTemplate.save(incoming.id(),rName);
//存储登录用户名与链接实例,方便API调用链接实例
redisTemplate.saveChannel(rName,incoming);
}else{
incoming.writeAndFlush(new TextWebSocketFrame("存在二次登陆,系统已为你自动断开本次链接"));
channels.remove(ctx.channel());
ctx.close();
return;
}
@Override
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
//删除存储池对应实例
String name = (String) redisTemplate.getName(ctx.channel().id());
redisTemplate.deleteChannel(name);
//删除默认存储对应关系
redisTemplate.delete(ctx.channel().id());
channels.remove(ctx.channel());
}
发送方法
我直接在SendUtil中写一个系统发送的方法,输出也是转为TextWebSocketFrame
/**
* 想指定链接发送数据
* @param msg 消息
* @param channel 指定链接
* @return {@link String}
*/
public static String sendTest(String msg,Channel channel) {
try {
channel.writeAndFlush(new TextWebSocketFrame( "[系统API]" + msg));
return "success";
}catch (Exception e){
e.printStackTrace();
return "error";
}
}
定义API
这个就简单一些了,定义一个统一返回的Bean,还有API的返回工具类,然后写对应的API接口方法。
@RestController
@RequestMapping("/back")
public class NCBackController {
@Autowired
private LikeRedisTemplate redisTemplate;
/**
* 获取在线用户数
* @return {@link ResultVo}
*/
@GetMapping("/size")
public ResultVo getSize(){
return ResultVOUtil.success(redisTemplate.getSize());
}
/**
* 获取在线用户列表
* @return {@link ResultVo}
*/
@GetMapping("/online")
public ResultVo getOnline(){
return ResultVOUtil.success(redisTemplate.getOnline());
}
/**
* API调用向在线用户发送消息
* @param name 用户名
* @param msg 消息
* @return {@link ResultVo}
*/
@PostMapping("/send")
public ResultVo send(@RequestParam String name,@RequestParam String msg){
Channel channel = (Channel) redisTemplate.getChannel(name);
if (channel == null){
return ResultVOUtil.error(555,"当前用户连接已断开");
}
String result = SendUtil.sendTest(msg,channel);
return ResultVOUtil.success(result);
}
}
效果
我在项目中添加Swagger方便查看与简单测试API,引入对应pom,在启动类加一个注解即可。
启动项目后登陆界面,发送了一个基本消息。
Swagger这边的页面打开后,测试几个API,都是成功的。
好了,结尾还是成功的,不过作为一个好产品是不能仅仅这样的,后续我们继续完善。
本项目是本人近期GitHub的核心发展项目,有兴趣的朋友可以去了解下
GitHub
项目名:InChat
项目地址:https://github.com/UncleCatMy...
项目介绍:基于Netty4与SpringBoot,聊天室WebSocket(文字图片)、Iot物联网-MQTT协议、TCP/IP协议单片机通信,异步存储聊天数据
如果本文对你有所帮助,欢迎关注个人技术公众号