com.github.jleskovar.btcrpc.websocket.AbstractJsonWebSocketRpcClient.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 com.googlecode.jsonrpc4j.IJsonRpcClient
import com.googlecode.jsonrpc4j.JsonRpcClient
import com.neovisionaries.ws.client.WebSocket
import com.neovisionaries.ws.client.WebSocketAdapter
import com.neovisionaries.ws.client.WebSocketFactory
import java.lang.reflect.Type
import javax.net.ssl.SSLContext

/**
 * Created by james on 21/12/17.
 */

abstract class AbstractJsonWebSocketRpcClient(wsUrl: String, sslContext: SSLContext) : JsonRpcClient(), IJsonRpcClient {

    private val webSocketFactory = WebSocketFactory()

    protected val socket: WebSocket

    init {
        webSocketFactory.sslContext = sslContext
        socket = webSocketFactory.createSocket(wsUrl)
    }

    abstract override fun invoke(methodName: String?, argument: Any?, returnType: Type?, extraHeaders: MutableMap?): Any?

    protected abstract fun handleTextMessage(text: String?)

    fun connect() {
        socket.addListener(object: WebSocketAdapter() {
            override fun onTextMessage(websocket: WebSocket?, text: String?) {
                handleTextMessage(text)
            }
        })

        socket.connect()
    }

    fun disconnect() {
        socket.disconnect()
    }

    protected fun fastExtractId(jsonRpc: String) = jsonRpc.substringAfterLast("\"id\":\"").substringBefore("\"")

    override fun invoke(methodName: String?, argument: Any?) {
        invoke(methodName, argument, null as Type?)
    }

    override fun invoke(methodName: String?, argument: Any?, returnType: Type?): Any? {
        return invoke(methodName, argument, returnType, mutableMapOf())
    }

    override fun  invoke(methodName: String?, argument: Any?, clazz: Class?): T {
        return invoke(methodName, argument, clazz, mutableMapOf())
    }

    @Suppress("UNCHECKED_CAST")
    override fun  invoke(methodName: String?, argument: Any?, clazz: Class?, extraHeaders: MutableMap?): T {
        return invoke(methodName, argument, clazz as Type, extraHeaders) as T
    }
}