自從 springboot 官方引入 kotlin 的支持后，一切又變得方便了很多，以前我們需要先弄出一個(gè) java 項(xiàng)目，然后加入 kotlin 的支持，而現(xiàn)在，直接 kotlin 覆蓋了全部的東西。

那么這一篇就帶大家來玩一下如何用 springboot + kotlin 來構(gòu)建 websocket 應(yīng)用。在這里我們用的 springboot 版本是 2.2.5，對(duì)應(yīng)的 kotlin 版本為 1.3.61，在此特別需要注意的是，我們應(yīng)當(dāng)使用 org.springframework.boot:spring-boot-dependencies 來整體更新依賴版本，如果單獨(dú)更新某個(gè)依賴，會(huì)發(fā)生不兼容的情況。如果需要使用 Kotlin 1.3.72 版本，則需要更新 springboot 到 2.3.1。

建好項(xiàng)目后，添加 websocket 的依賴:

dependencies {
    ... ...
    implementation 'org.springframework.boot:spring-boot-starter-websocket:2.2.5.RELEASE'
}

注意此處版本號(hào)，要與 springboot 版本號(hào)一致。

然后我們可以很輕松的在 springboot 里建立起一個(gè) WebSocket 服務(wù):

@Component
open class SocketController : WebSocketHandler {
    init {
        println("SocketController init")
    }

    override fun handleTransportError(session: WebSocketSession, t: Throwable) {
        println("handleTransportError")
    }

    override fun afterConnectionClosed(session: WebSocketSession, cs: CloseStatus) {
        println("afterConnectionClosed")
    }

    override fun handleMessage(session: WebSocketSession, message: WebSocketMessage<*>) {
        println("handleMessage: $message")
        session.sendMessage(RESPONSE_MESSAGE.toMessage())
    }

    override fun afterConnectionEstablished(session: WebSocketSession) {
        println("afterConnectionEstablished")
    }

    override fun supportsPartialMessages() = false
}

其中對(duì)于 handleMessage 方法，將發(fā)送返回?cái)?shù)據(jù)，此處有一個(gè)擴(kuò)展，即是將字符串包裝為 WebSocketMessage 對(duì)象，實(shí)現(xiàn)如下:

fun String.toMessage() = TextMessage(this)

下一步我們需要將服務(wù)暴露出來，使得前端可以訪問到它:

@Configuration
@EnableWebSocket
open class WebSocketConfig: WebSocketConfigurer {

    private fun getSocketHandler(): WebSocketHandler = SocketController()

    override fun registerWebSocketHandlers(registry: WebSocketHandlerRegistry) {
        registry.addHandler(getSocketHandler(), "/ws").setAllowedOrigins("*")
    }
}

注意加入 @EnableWebSocket 注解，同時(shí)還需要注意，由于注解的存在，所以 class 必須是open 的，不然會(huì)產(chǎn)生 注解標(biāo)注于 final class 的異常。

然后我們可以直接把項(xiàng)目跑起來，看到以下日志時(shí)，說明 WebSocket 已經(jīng)正常工作了。

INFO 48488 --- [main] c.i.boot.web.monitor.MonitorAspect       : Boot Monitor-void com.rarnu.sample.socket.WebSocketConfig.registerWebSocketHandlers(WebSocketHandlerRegistry)，in：4ms

接著來實(shí)現(xiàn)客戶端，這里分兩種實(shí)現(xiàn)方式，先來看看用 kotlin 怎么做吧:

@ClientEndpoint
class SocketClient(uri: URI,val openHandler: () -> Unit,val closeHandler: () -> Unit,val errorHandler: (t: Throwable) -> Unit,val msgHandler:(msg: String) -> Unit) {

    private var session: Session? = null

    init {
        try {
            val container = ContainerProvider.getWebSocketContainer()
            container.connectToServer(this, uri)
        } catch (th: Throwable) {
            throw RuntimeException(th)
        }
    }

    @OnOpen
    fun onOpen(session: Session) {
        this.session = session
        openHandler()
    }

    @OnClose
    fun onClose(session: Session, reason: CloseReason) {
        this.session = null
        closeHandler()
    }

    @OnMessage
    fun onMessage(message: String) {
        msgHandler(message)
    }

    @OnError
    fun onError(t: Throwable) {
        errorHandler(t)
    }

    fun sendMessage(message: String) {
        session?.basicRemote?.sendText(message)
    }
}

class SocketIO {

    private var endPoint: SocketClient? = null
    private var timerHeartbeat: Timer? = null
    private var working = false

    init {
        endPoint = SocketClient(URI("ws://127.0.0.1:9011/ws"),
                {
                    // open callback
                    startHeartHeat()
                },
                {
                    // close callback
                    working = false
                },
                {
                    // error callback
                    println("SocketIO Error: $it")
                },
                {
                    // message callback
                    println("received: $it")
                })
        }
    }

    private fun startHeartHeat() {
        working = true
        timerHeartbeat = Timer("_socket_heartbeat_")
        timerHeartbeat?.schedule(timerTask {
            if (!working) {
                this.cancel()
                timerHeartbeat?.cancel()
                timerHeartbeat = null
                return@timerTask
            }
            endPoint?.sendMessage(MSG_HEARTBEAT)
        }, 0, 10000L)
    }
}

這里啟用了一個(gè)簡單的心跳包機(jī)制，來確保服務(wù)不斷。

再來個(gè) js 的調(diào)用方法，也很簡單:

<script type="text/javascript">
    let ws = new WebSocket("ws://127.0.0.1:9011/ws");
    ws.onopen = function (event) {
        console.log("Opened.");
    }
    ws.onmessage = function (msg) {
            console.log(msg.data);
    }
    ws.onclose = function (event) {
        console.log("Closed.")
    }
    function sendText() {
        let txt = $("#txt").value;
        ws.send(txt);
    }
</script>

當(dāng)然了，springboot 集成 WebSocket 還有很多方法，比如用 @ServerEndpoint 注解來完成，又或者是集成一些第三方的框架，但是不論如何我們都能實(shí)現(xiàn)功能，那么下面就剩設(shè)計(jì)一個(gè)靠譜的數(shù)據(jù)傳輸協(xié)議了。