一个服务存在多个实例时，WebSocket通过网关会被负载均衡连接到其中任意一个实例上

而当一个实例发送消息时，连接另一个实例的客户端就会收不到消息

为了解决这个问题，该库提供了一种解决方案

只需要添加一个配置注解，就可以像单体应用一样使用WebSocket，开箱即用

也可以通过简单的自定义来支持更复杂的业务

本库同时兼容Web和Webflux，使用方式上没有任何区别

implementation 'com.github.linyuzai:concept-websocket-loadbalance-spring-boot-starter:1.1.2'

implementation 'org.springframework.boot:spring-boot-starter-websocket'//webmvc需要添加websocket依赖，webflux不需要

<dependency>
  <groupId>com.github.linyuzai</groupId>
  <artifactId>concept-websocket-loadbalance-spring-boot-starter</artifactId>
  <version>1.1.2</version>
</dependency>

<!--webmvc需要添加websocket依赖，webflux不需要-->
<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

在启动类上添加注解@EnableWebSocketLoadBalanceConcept启用功能

@EnableWebSocketLoadBalanceConcept
@EnableDiscoveryClient
@SpringBootApplication
public class AServiceApplication {

    public static void main(String[] args) {
        SpringApplication.run(AServiceApplication.class, args);
    }
}

注入WebSocketLoadBalanceConcept就可以跨实例发送消息

@RestController
@RequestMapping("/ws")
public class WsController {

    @Autowired
    private WebSocketLoadBalanceConcept concept;

    @RequestMapping("/send")
    public void send(@RequestParam String msg) {
        concept.send(msg);
    }
}

客户端的连接地址为ws://{服务的地址}/concept-websocket/{自定义路径}

其中concept-websocket为默认的固定前缀

concept:
  websocket:
    type: auto #JAVAX/SERVLET/REACTIVE，AUTO自动适配，默认AUTO
    server: #服务配置
      default-endpoint: #默认端点
        enabled: true #是否启用默认端点，默认true
        path-selector: #Path选择器
          enabled: false #是否启用Path选择器，默认false
        user-selector: #User选择器
          enabled: false #是否启用User选择器，默认false
      heartbeat: #心跳配置
        enabled: true #是否启用心跳，默认true
        period: 60000 #心跳间隔，单位ms，默认1分钟
        timeout: 210000 #超时时间，单位ms，默认3.5分钟，3次心跳间隔
    load-balance: #负载均衡（转发）配置
      protocol: ws #服务间连接协议，默认ws
      logger: true #是否启用日志，互相连接的日志打印，默认true
      monitor: #监控配置
        enabled: true #是否启用监控，默认true
        period: 30000 #轮训间隔，单位ms，默认30s
        logger: false #是否启用日志，默认false
      heartbeat: #心跳配置
        enabled: true #是否启用心跳，默认true
        period: 60000 #心跳间隔，单位ms，默认1分钟
        timeout: 210000 #超时时间，单位ms，默认3.5分钟，3次心跳间隔

通过服务间进行相互的WebSocket连接来实现消息转发

将消息转发抽象为对其他服务实例消息的订阅

通过ConnectionSubscriber来和其他服务实例建立连接

默认实现了WebSocket的双向连接

可以自定义ConnectionSubscriber使用MQ或HTTP等其他消息转发方式

需要注意，该流程在服务启动后触发，通过ApplicationRunner实现

通过注册中心获得其他服务实例的信息（默认通过注册中心获取，同时支持自定义服务列表）并向这些服务实例发起连接

连接成功后发送自身服务实例信息，其他服务实例通过收到的服务实例信息进行反向连接

存在一个定时任务定时检查（默认30s）服务实例间的连接是否完整

每个服务实例会检查自己需要连接的其他服务实例

当发现与某个服务实例不存在连接会尝试连接，或连接已经死亡（超过一定时间没有心跳时间更新）会重新连接

通过ConnectionServerProvider来获取服务实例的信息（从抽象的层面上讲并不一定局限于同服务的实例）

默认通过Spring Cloud的服务发现来获得注册中心上维护的同服务的实例信息

通过Message体现

通过MessageFactory将我们传入的任意对象封装成消息对象

默认使用ObjectMessageFactory讲对象封装成ObjectMessage

可以针对部分类型数据自定义消息工厂

通过MessageEncoder在发送消息时进行编码

通过MessageDecoder在接收消息时进行解码

通过MessageCodecAdapter统一编解码器的入口

• 普通客户端收发消息的编解码器
• 服务实例间订阅消息的编解码器
• 消息转发的编解码器

可以自定义编解码适配器用于特定的消息编解码

默认编码为json字符串

当一个消息发送时将会适配连接选择器ConnectionSelector

连接选择器将会根据消息返回需要发送该消息的连接

消息类Message的headers字段用于自定义消息头

连接类Connection的metadata字段用于自定义元数据

可以通过匹配两者来筛选连接以支持复杂的业务场景

假设前端连接的WebSocket地址为ws://localhost:8080/concept-websocket/sample

其中concept-websocket为默认的固定前缀，sample为我们自定义路径

在配置中启用路径选择器

concept:
  websocket:
    server: 
      default-endpoint: 
        path-selector: 
          enabled: true #启用Path选择器

使用PathMessage给所有的sample客户端发送消息

@RestController
@RequestMapping("/ws")
public class WsController {

    @Autowired
    private WebSocketLoadBalanceConcept concept;

    @RequestMapping("/send-path")
    public void sendPath(@RequestParam String msg) {
        concept.send(new PathMessage(msg, "sample"));
    }
}

假设前端连接的WebSocket地址为ws://localhost:8080/concept-websocket/user?userId=1

其中userId为固定参数名

在配置中启用路径选择器

concept:
  websocket:
    server: 
      default-endpoint: 
        user-selector: 
          enabled: true #启用Path选择器

使用UserMessage给指定的用户发送消息

@RestController
@RequestMapping("/ws")
public class WsController {

    @Autowired
    private WebSocketLoadBalanceConcept concept;

    @RequestMapping("/send-user")
    public void sendUser(@RequestParam String msg) {
        concept.send(new UserMessage(msg, "1"));
    }
}

通过实现MessageHandler接收客户端发送的消息

通过实现LifecycleListener监听生命周期

通过实现ErrorHandler处理异常，默认将会通过logger打印

可以自定义DefaultEndpointCustomizer来配置

Servlet环境下会回调WebSocketHandlerRegistration

Reactive环境下会回调ReactiveWebSocketServerHandlerMapping

• 支持Undertow

• 多线程发送消息的并发问题