com.github.jleskovar.btcrpc.websocket.JsonAsyncWebSocketRpcClient.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of btc-rpc-client Show documentation

Kotlin-based JSON-RPC client for bitcoind/btcd

The newest version!

package com.github.jleskovar.btcrpc.websocket

import java.io.ByteArrayInputStream
import java.io.ByteArrayOutputStream
import java.lang.reflect.ParameterizedType
import java.lang.reflect.Type
import java.util.concurrent.CompletableFuture
import javax.net.ssl.SSLContext

class JsonAsyncWebSocketRpcClient(wsUrl: String, sslContext: SSLContext) : AbstractJsonWebSocketRpcClient(wsUrl, sslContext) {
    private val responses = mutableMapOf>>()

    override fun handleTextMessage(text: String?) {
        val id = fastExtractId(text!!)
        val returnType = responses[id]?.first
        val responseValue = super.readResponse(returnType, ByteArrayInputStream(text.toByteArray()))
        responses[id]?.second?.complete(responseValue)
        responses.remove(id)
    }

    override fun invoke(methodName: String?, argument: Any?, returnType: Type?, extraHeaders: MutableMap?): Any? {
        val outputStream = ByteArrayOutputStream()
        super.invoke(methodName, argument, outputStream)
        val output = outputStream.toString()

        val id = fastExtractId(output)

        val realReturnType = when(returnType) {
            is ParameterizedType -> returnType.actualTypeArguments[0] // Dig out "T" from a "CompletableFuture"
            else -> returnType
        }

        val completableFuture = CompletableFuture()
        responses[id] = Pair(realReturnType!!, completableFuture)
        socket.sendText(output)
        return completableFuture
    }
}