1.在websocketService里面定义一个静态ApplicationContext;
@ServerEndpoint("/imserver/{userId}")
@Component
public class WebSocketServer {
private static ApplicationContext applicationContext;
public static void setApplicationContext(ApplicationContext context){
applicationContext = context;
}
//下面这些都是外部引用的service
private ExpService expService;
private ExpRecordService expRecordService;
private ExpRegistrationService expRegistrationService;
...
}
2.在Springboot的启动器Application的main方法里面,去接收该context并赋值给上文的applicationContext
@SpringBootApplication
public class EvaluateApplication extends SpringBootServletInitializer {
public static void main(String[] args) {
ConfigurableApplicationContext context = SpringApplication.run(EvaluateApplication.class, args);
WebSocketServer.setApplicationContext(context);
}
}
3.在sendMessage方法里面给这些service赋值
/**
* 实现服务器主动推送
*/
public void sendMessage(String message) throws IOException {
expRegistrationService = (ExpRegistrationService)applicationContext.getBean(ExpRegistrationService.class);
expRecordService = (ExpRecordService)applicationContext.getBean(ExpRecordService.class);
expService = (ExpService)applicationContext.getBean(ExpService.class);
this.session.getBasicRemote().sendText(message);
}