WebSocket无法注入属性

踩坑一：

原因：

是因为Spring对象的创建都是以单例模式创建的，在启动时只创建一次WebSocket。而WebSocketServer在每个连接请求到来时，都会new一个对象。所以当你启动项目时，你想要注入的对象已经注入进去，但是当用户连接是，新创建的websocket对象没有你要注入的对象，所以会报NullPointerException

解决：

通过static关键字让webSocketService属于WebSocketServer类

private static WebSocketService webSocketService; //通过static关键字让webSocketService属于WebSocketServer类

@Autowired//注入到WebSocketServer类的webSocketService属性里
public void setKefuService(WebSocketService webSocketService){
    WebSocketServer.webSocketService= webSocketService;
}

踩坑二：

使用@ServerEndpoint声明的websocket服务器中自动注入

1. 错误方法，这样无法从容器中获取

@ServerEndpoint(value = "/chat/{username}")
@Service
public class WebSocketServer {

    @Resource // @Autowired
    private RabbitTemplate rabbitTemplate;  //null
}

2. 解决：使用上下文获取

@ServerEndpoint(value = "/chat/{username}")
@Service
public class WebSocketServer {

    /*
     * 提供一个spring context上下文(解决方案)
     */
    private static ApplicationContext context;

    public static void setApplicationContext(ApplicationContext applicationContext) {
        WebSocketServer.context = applicationContext;
    }
}

3. 在启动类中传入上下文

@SpringBootApplication
public class TalkApplication {
    public static void main(String[] args) {
        //解决springboot和websocket之间使用@autowired注入为空问题
        ConfigurableApplicationContext applicationContext =
                SpringApplication.run(TalkApplication.class, args);
        //这里将Spring Application注入到websocket类中定义的Application中。
        WebSocketServer.setApplicationContext(applicationContext);
    }
}

4. 在使用的地方通过上下文去获取服务context.getBean();

public void sendSimpleQueue(String message) {
    String queueName = "talk";
    // 在使用的地方通过上下文去获取服务context.getBean();
    RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
    
    rabbitTemplate.convertAndSend(queueName, message);
}