項目中需要向頁面推送消息..決定采用websocket

部分代碼copy網(wǎng)上的

config: websocket的配置

@Configuration
@EnableWebSocket
open class WebSocketConfig : WebSocketConfigurer {
    override fun registerWebSocketHandlers(p0: WebSocketHandlerRegistry?) {
        p0?.addHandler(MyWebSocketHander(),"/test")?.addInterceptors(MyHandshakeInterceptor())
    }
    @Bean open fun serverEndpointExporter() : ServerEndpointExporter {
        return ServerEndpointExporter()
    }
}

socketHandler類

  @ServerEndpoint("/test")
  @Component
  class MyWebSocket {
      companion object {
          val webSocketSet = CopyOnWriteArraySet<MyWebSocket>()
          var onlineCount = 0
      }
    var session: Session? = null
    @OnOpen
    fun onOpen(session: Session) {
        this.session = session
        webSocketSet.add(this)
        addOnlineCount()
        println("有新客戶加入!當前在線人數(shù) $onlineCount")
        sendMsg("你好!客戶端")
        try {
            sendMsg("test")
        } catch(e: Exception) {
            println("${e.message}")
        }
    }
    @OnClose
    fun onClose() {
        webSocketSet.remove(this)
        subOnlineCount()
        println("客戶退出!當前在線人數(shù) $onlineCount")
    }
    @OnMessage
    fun onMsg(msg: String, session: Session) {
        println("客戶端傳來的消息:$msg")
        for (item in webSocketSet) {
            item.sendMsg("msg:$msg")
        }
    }
    fun subOnlineCount() {
        onlineCount--
    }
    fun addOnlineCount(): Int {
        return onlineCount++
    }
    fun sendMsg(msg: String) {
        this.session?.basicRemote?.sendText(msg)
    }
}

最后就是控制類

@PostMapping("xxxx")
@MessageMapping("/test")
    fun test(){
      val webSocketSet = MyWebSocket.webSocketSet
            for (item in webSocketSet) {
                   item.sendMsg("有訂單導入!!")
                }
      }

到這里就可以實現(xiàn)服務器向web推送消息的方式了.