Сервер WebSocket в Kotlin
Сервер наследуют от WebSocketServer и передают
InetSocketAddress с портом 0, чтобы
ОС выдала свободный номер. Метод start
запускает приём в фоне; в onStart удобно
сигнализировать, что порт уже известен.
Текст от клиента приходит в onMessage вторым
аргументом. Ответ на то же соединение шлют через
send у объекта WebSocket:
import java.net.InetSocketAddress
import java.net.URI
import java.util.concurrent.CountDownLatch
import org.java_websocket.WebSocket
import org.java_websocket.client.WebSocketClient
import org.java_websocket.handshake.ClientHandshake
import org.java_websocket.handshake.ServerHandshake
import org.java_websocket.server.WebSocketServer
fun main() {
val ready = CountDownLatch(1)
val done = CountDownLatch(1)
val server = object : WebSocketServer(
InetSocketAddress(0),
) {
override fun onStart() {
ready.countDown()
}
override fun onOpen(
conn: WebSocket,
handshake: ClientHandshake,
) {}
override fun onClose(
conn: WebSocket,
code: Int,
reason: String,
remote: Boolean,
) {}
override fun onMessage(
conn: WebSocket,
message: String,
) {
conn.send(message)
}
override fun onError(
conn: WebSocket,
ex: Exception,
) {}
}
server.start()
ready.await()
val port = server.port
val uri = URI("ws://127.0.0.1:$port")
val client = object : WebSocketClient(uri) {
override fun onOpen(
handshakedata: ServerHandshake,
) {}
override fun onMessage(message: String) {
done.countDown()
}
override fun onClose(
code: Int,
reason: String,
remote: Boolean,
) {}
override fun onError(ex: Exception) {}
}
client.connectBlocking()
client.send("ping")
done.await()
client.close()
server.stop()
}
Тот же приём подходит для другой фразы: в
onMessage достаточно передать строку
обратно без изменений. Клиент в примере лишь
дожидается кадра, чтобы программа не завершилась
раньше ответа:
import java.net.InetSocketAddress
import java.net.URI
import java.util.concurrent.CountDownLatch
import org.java_websocket.WebSocket
import org.java_websocket.client.WebSocketClient
import org.java_websocket.handshake.ClientHandshake
import org.java_websocket.handshake.ServerHandshake
import org.java_websocket.server.WebSocketServer
fun main() {
val ready = CountDownLatch(1)
val done = CountDownLatch(1)
val server = object : WebSocketServer(
InetSocketAddress(0),
) {
override fun onStart() {
ready.countDown()
}
override fun onOpen(
conn: WebSocket,
handshake: ClientHandshake,
) {}
override fun onClose(
conn: WebSocket,
code: Int,
reason: String,
remote: Boolean,
) {}
override fun onMessage(
conn: WebSocket,
message: String,
) {
conn.send(message)
}
override fun onError(
conn: WebSocket,
ex: Exception,
) {}
}
server.start()
ready.await()
val port = server.port
val uri = URI("ws://127.0.0.1:$port")
val client = object : WebSocketClient(uri) {
override fun onOpen(
handshakedata: ServerHandshake,
) {}
override fun onMessage(message: String) {
done.countDown()
}
override fun onClose(
code: Int,
reason: String,
remote: Boolean,
) {}
override fun onError(ex: Exception) {}
}
client.connectBlocking()
client.send("order 7")
done.await()
client.close()
server.stop()
}
На входящую строку "check" сервер должен
вернуть ту же строку клиенту на localhost.
На запрос "version" ответ сервера в
кадре должен совпадать с текстом запроса.