Springboot项目整合WebSocket源码分析
背景
在一个Springboot项目中，写了一个WebSocket服务端代码。具体代码网上一大堆，这里不再展示。同时，我在Websocket服务端的类里面，定义了一个Boolean类型的成员变量。当客户端websocket传来的参数是666时，将该成员变量改为true。客户端传来其他值时，将该成员变量改为false。
在调试中发现，每当客户端新创建一个连接，调用服务端@OnOpen修饰的方法时，服务端的类都是新的一个对象，并不是加入Spring中管理的那个对象。所以每次调用@OnOpen方法时，Boolean成员变量的默认值都为Null，不管之前是否设置为false，新建客户端连到服务端后，都是新的对象。
这时，我发现了服务端的类里，计算在线人数时，用的static修饰的成员变量。这也再次证明，每次新建连接，Websocket服务端的类都新建个对象，并不是用Spring管理的那个单例对象。所以计算在线人数时，才会用static来统计不同对象的连接次数，即在线人数。
扫描@ServerEndpoint注解源码分析
虽然通过表象，已经看到了其原因。但是Spring中既然已经有了WebSocket服务端的一个类，而每次新建客户端还要重新生成一个类，这种操作很难理解，所以下面找到新建对象的这个源码，来落实这个猜测。
首先，SpringBoot整合了WebSocket，我们到springboot-autoconfiguration包里的spring.factories文件里，找到SpringBoot自动配置了关于WebSocket的哪些类，找到如下：
那这三个类，到底用了哪个类呢，分别点进去，打个断点，看项目启动过程中，执行了哪个断点，就用了哪个类。最后发现，是进了如下类的断点:
即在我的项目中,SpringBoot使用了WebSocketServletAutoConfiguration这个类来自动装配WebSocket。
下面对这个类进行分析:
先看这个类上的注解:
重点是@ConditionalOnClass注解，它表示项目的classpath里如果有参数里配置的类，则就加载注解修饰的类。我们看该注解的参数ServerContainer.class这个类:
public interface ServerContainer extends WebSocketContainer {
public abstract void addEndpoint(Class<?> clazz) throws DeploymentException;
public abstract void addEndpoint(ServerEndpointConfig sec) throws DeploymentException;
}
可见，其是WebSocketContainer的一个子类，因为项目里引入了WebSocket的jar包，所以项目里一定有WebSocketContainer这个类，所以SpringBoot项目就加载了WebSocketServletAutoConfiguration自动配置类。
下面看WebSocketServletAutoConfiguration里的注入代码，即刚才进入断点的代码:
可见，其往Spring容器中加入了TomcatWebSocketServletWebServerCustomizer类。下面分析这个类:
public class TomcatWebSocketServletWebServerCustomizer
implements WebServerFactoryCustomizer<TomcatServletWebServerFactory>, Ordered {
@Override
public void customize(TomcatServletWebServerFactory factory) {
factory.addContextCustomizers((context) -> context.addServletContainerInitializer(new WsSci(), null));
}
@Override
public int getOrder() {
return 0;
}
}
看它源码，一脸懵逼，不知道在干啥，那就看它类继承结构:
看WebServerFactoryCustomizer源码:
/*
* Copyright 2012-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.boot.web.server;
import org.springframework.beans.factory.config.BeanPostProcessor;
/**
* Strategy interface for customizing {@link WebServerFactory web server factories}. Any
* beans of this type will get a callback with the server factory before the server itself
* is started, so you can set the port, address, error pages etc.
*
* Beware: calls to this interface are usually made from a
* {@link WebServerFactoryCustomizerBeanPostProcessor} which is a
* {@link BeanPostProcessor} (so called very early in the ApplicationContext lifecycle).
* It might be safer to lookup dependencies lazily in the enclosing BeanFactory rather
* than injecting them with {@code @Autowired}.
*
* @param the configurable web server factory
* @author Phillip Webb
* @author Dave Syer
* @author Brian Clozel
* @since 2.0.0
* @see WebServerFactoryCustomizerBeanPostProcessor
*/
@FunctionalInterface
public interface WebServerFactoryCustomizer<T extends WebServerFactory> {
/**
* Customize the specified {@link WebServerFactory}.
* @param factory the web server factory to customize
*/
void customize(T factory);
}
重点是其注释:
自定义WebServerFactory类的策略接口。这个类的所有beans在服务启动前，都会执行一个回调函数，所以可以设置端口、地址和错误页等。
注意:对该接口的调用通常由WebServerFactoryCustomizerBeanProcessor进行（所以在SpringContext周期的早期进行），所以使用懒加载创建属性比用@Autowired更安全
从注释中可知，WebServerFactoryCustomizerBeanProcessor会调用这个接口，而WebServerFactoryCustomizerBeanProcessor是一个BeanProcessor。在类初始化完成后，就会执行BeanProcessor的方法。
下面回到TomcatWebSocketServletWebServerCustomizer类的这个代码:
可知，这行代码是由Beanprocessor触发执行的。这样，就执行了这行代码。上图中的这行代码，又new WsSci对象，下面看WsSci对象源码:
在类上有一个@HandlesTypes注解，该注解的作用为获取到所有用@ServerEndpoint修饰的类，并且把这些类赋给onStartup方法的clazzes参数上。至此，WebSocket的服务类上的@ServerEndpoint如何扫描到的源码已经找到。
@OnOpen方法每次调用都会创建新对象源码分析
找@OnOpen的源码，是利用了一点技巧。在WsServerContainer类里，直接搜索关键字open，找到了如下代码:
这里的意思就是将@OnOpen对应的方法，放入到了sec对象的userProperties属性中。这是个映射操作。那么一定会有找这个映射的代码，所以断点打到getUserProperties方法那里，去调试程序，如下图:
等到代码执行到这里后，然后一句一句往下debug，来找到执行@OnOpen方法的代码，最终看到确实是new出了一个新对象。
具体过程不再展示。
总结
下面复盘一下SpringBoot是如何自动装配WebSocket的:
首先，项目依赖了WebSocket的相关jar包，SpringBoot自动注入了一个类WebSocketServletAutoConfiguration。
这个自动装配类，又会往Spring容器加入一个TomcatWebSocketServletWebServerCustomizer类。TomcatWebSocketServletWebServerCustomizer类会被BeanProcessor执行。所以，这个类的方法里，创建了WebSocket的WsSci对象，在这个对象里，搜集了@ServerEndpoint注解，解析了WebSocket的服务类。
所以，本质还是通过BeanProcessor，来解析的@ServerEndpoint注解，不过，是利用了现有的一个BeanProcessor，自定义了一个BeanProcessor操作的类来进行了注解的解析。