netty-ws-starter

SpringBoot starter for websocket cluster by netty

Stars
2
Committers
1

netty websocket starter

  • 支持功能:

    • SDK随应用集成,不单独部署,并且无需考虑类似于Cookie/Session场景下的 session 共享问题
    • 广播消息
    • 心跳检测
    • HTTP握手请求时支持自定义认证
    • 自定义编解码方式,JSON Protobuf等
  • 不支持功能:

    • 消息持久化,历史消息搜索
    • 消息的已读&未读 回执
    • 发送和接收消息的 Once 语义
    • 发送离线消息
    • 断线重连时的节点亲和
    • 节点故障转移时,保证消息不丢失

Quick Start

Demo 项目

添加依赖

<!--添加源-->
<repository>
    <id>github</id>
    <url>https://maven.pkg.github.com</url>
    <snapshots>
        <enabled>true</enabled>
    </snapshots>
</repository>

<dependency>
    <groupId>com.github.kuangcp</groupId>
    <artifactId>netty-ws-spring-boot-starter</artifactId>
    <version>1.0.5-RELEASE</version>
</dependency>

基础配置

netty-ws:
  # websocket 服务 端口 
  port: 5455
  # ws url 根路径 默认为 ws://host:port/ws
  prefix: /ws
  # HTTP握手请求时最大请求体字符串长度
  max-content-length: 4096
  # 单条消息最大字符串长度,需要按业务设置合理值,防范DOS攻击
  max-frame-size: 65536
  # 60s 未读取到客户端发来的消息 认为是触发一次idle事件
  reader-idle-sec: 60
  # 当idle事件累计到2次后,关闭当前用户连接
  reader-idle-threshold: 2
  # 是否开启 token 认证机制
  connect-auth: true

自定义连接处理类


@Component
@ChannelHandler.Sharable
public class DemoHandler extends AbstractBizHandler {

    public DemoHandler(CacheDao cacheDao, UserDao userDao, WsServerConfig config) {
        super(cacheDao, userDao, config);
        this.schedulerPollQueueMsg(Executors.newScheduledThreadPool(1));
    }

    @Override
    public void connectSuccess(Long userId) {
        log.info("connected {}", userId);
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        String id = WsSocketUtil.id(ctx);
        Long userId = channelUserMap.get(id);
        log.info("disconnect {}", userId);
        super.channelInactive(ctx);
    }

    @Override
    protected void handSharkHttpRequest(ChannelHandlerContext ctx, FullHttpRequest request) {
        super.handSharkHttpRequest(ctx, request);
    }
}

实现存储扩展 com.github.kuangcp.websocket.store.CacheDao 实现认证扩展 com.github.kuangcp.websocket.store.UserDao

  • 然后启动项目, 验证websocket通信。 可使用 Js 测试客户端 测试,连接地址 ws://127.0.0.1:5455/ws?uid={uid}&token={token}
    • 注意认证的实现方式为HTTP握手时,解析 url参数或者Header中的token值。
    • 如果未开启认证(connect-auth) token可不传

设计思路

  1. Netty作为通信基础,每个用户连接时通过前置的Nginx等LB层负载均衡到WS集群中的任意节点。
  2. 用户和主机ip关系绑定到Redis map结构中 (userid -> host)
  3. 每个主机ip绑定一个Redis的list队列,存放了需要推送的消息(由其他节点写入),解决应用层向用户推送消息时,用户连接随机分散的问题。

注意:Redis可替换成任意中心存储, 已由 CacheDao 抽象, 例如关系型数据库 MySQL,PG;非关系型数据库:MongoDB 等等。

Dev

mvn clean source:jar deploy -U -DskipTests=true

Related Projects