StakeSquid/dshackle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

problem: Proxy fails to serialize JSON

Browse Source 
reason: expects full JSON response from upstream, which is replaces with result field only
This commit is contained in:
Igor Artamonov
2020-06-29 22:44:44 -04:00
parent 04958784ec
commit c2636cd81e
5 changed files with 200 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/kotlin/io/emeraldpay/dshackle/Global.kt
View File

@@ -19,6 +19,7 @@ import com.fasterxml.jackson.core.Version
import com.fasterxml.jackson.databind.DeserializationFeature
import com.fasterxml.jackson.databind.ObjectMapper
import com.fasterxml.jackson.databind.module.SimpleModule
import io.emeraldpay.dshackle.upstream.rpcclient.JsonRpcResponse
import java.text.SimpleDateFormat
import java.util.*
@@ -31,6 +32,7 @@ class Global {
private fun createObjectMapper(): ObjectMapper {
val module = SimpleModule("EmeraldDshackle", Version(1, 0, 0, null, null, null))
module.addSerializer(JsonRpcResponse::class.java, JsonRpcResponse.ResponseJsonSerializer())
val objectMapper = ObjectMapper()
objectMapper.registerModule(module)

31
src/main/kotlin/io/emeraldpay/dshackle/proxy/WriteRpcJson.kt
View File

@@ -19,15 +19,11 @@ package io.emeraldpay.dshackle.proxy
import com.fasterxml.jackson.databind.ObjectMapper
import io.emeraldpay.api.proto.BlockchainOuterClass
import io.emeraldpay.dshackle.Global
import io.infinitape.etherjar.rpc.RpcResponseError
import io.infinitape.etherjar.rpc.json.ResponseJson
import io.emeraldpay.dshackle.upstream.rpcclient.JsonRpcResponse
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Service
import reactor.core.publisher.Flux
import reactor.core.publisher.Mono
import java.lang.StringBuilder
import java.time.Duration
import java.util.function.Function
/**
@@ -48,29 +44,32 @@ open class WriteRpcJson() {
open fun toJsons(call: ProxyCall): Function<Flux<BlockchainOuterClass.NativeCallReplyItem>, Flux<String>> {
return Function { flux ->
flux.flatMap { response ->
val json = ResponseJson<Any, Any>()
if (!call.ids.containsKey(response.id)) {
log.warn("ID wasn't requested: ${response.id}")
return@flatMap Flux.empty<String>()
}
json.id = call.ids[response.id]
if (response.succeed) {
val payload = objectMapper.readValue(response.payload.toByteArray(), ResponseJson::class.java)
if (payload.error != null) {
json.error = payload.error
} else {
json.result = payload.result
}
val json = toJson(call, response)
if (json == null) {
Flux.empty<String>()
} else {
json.error = RpcResponseError(-32002, response.errorMessage)
Flux.just(json)
}
Flux.just(objectMapper.writeValueAsString(json))
}.onErrorContinue { t, u ->
log.warn("Failed to convert to JSON", t)
}
}
}
open fun toJson(call: ProxyCall, response: BlockchainOuterClass.NativeCallReplyItem): String? {
val id = call.ids[response.id] ?: return null;
val json = if (response.succeed) {
JsonRpcResponse.ok(response.payload.toByteArray(), JsonRpcResponse.Id.from(id))
} else {
JsonRpcResponse.error(-32002, response.errorMessage, JsonRpcResponse.Id.from(id))
}
return objectMapper.writeValueAsString(json)
}
/**
* Format response as JSON Array, for Batch requests
*/

77
src/main/kotlin/io/emeraldpay/dshackle/upstream/rpcclient/JsonRpcResponse.kt
View File

@@ -23,9 +23,12 @@ import reactor.core.publisher.Mono
class JsonRpcResponse(
private val result: ByteArray?,
val error: ResponseError?
val error: ResponseError?,
val id: Id
) {
constructor(result: ByteArray?, error: ResponseError?) : this(result, error, IntId(0))
companion object {
private val NULL_VALUE = "null".toByteArray()
@@ -34,6 +37,11 @@ class JsonRpcResponse(
return JsonRpcResponse(value, null)
}
@JvmStatic
fun ok(value: ByteArray, id: Id): JsonRpcResponse {
return JsonRpcResponse(value, null, id);
}
@JvmStatic
fun ok(value: String): JsonRpcResponse {
return JsonRpcResponse(value.toByteArray(), null)
@@ -43,6 +51,11 @@ class JsonRpcResponse(
fun error(code: Int, msg: String): JsonRpcResponse {
return JsonRpcResponse(null, ResponseError(code, msg))
}
@JvmStatic
fun error(code: Int, msg: String, id: Id): JsonRpcResponse {
return JsonRpcResponse(null, ResponseError(code, msg), id)
}
}
fun hasResult(): Boolean {
@@ -114,11 +127,67 @@ class JsonRpcResponse(
}
}
/**
* JSON RPC wrapper. Makes sure that the id is either Int or String
*/
interface Id {
fun asInt(): Int
fun asString(): String
fun isInt(): Boolean
companion object {
fun from(id: Any): Id {
if (id is Int) {
return IntId(id)
}
if (id is Number) {
return IntId(id.toInt())
}
if (id is String) {
return StringId(id)
}
throw IllegalArgumentException("Id must be Int or String")
}
}
}
class IntId(val id: Int) : Id {
override fun asInt(): Int {
return id
}
override fun asString(): String {
throw IllegalStateException("Not string")
}
override fun isInt(): Boolean {
return true
}
}
class StringId(val id: String) : Id {
override fun asInt(): Int {
throw IllegalStateException("Not int")
}
override fun asString(): String {
return id
}
override fun isInt(): Boolean {
return false
}
}
class ResponseJsonSerializer : JsonSerializer<JsonRpcResponse>() {
override fun serialize(value: JsonRpcResponse, gen: JsonGenerator, serializers: SerializerProvider) {
gen.writeStartObject()
gen.writeStringField("jsonrpc", "2.0")
gen.writeNumberField("id", 0)
if (value.id.isInt()) {
gen.writeNumberField("id", value.id.asInt())
} else {
gen.writeStringField("id", value.id.asString())
}
if (value.error != null) {
gen.writeObjectFieldStart("error")
gen.writeNumberField("code", value.error.code)
@@ -128,7 +197,9 @@ class JsonRpcResponse(
if (value.result == null) {
throw IllegalStateException("No result set")
}
gen.writeRawUTF8String(value.result, 0, value.result.size)
gen.writeFieldName("result")
gen.writeRaw(":")
gen.writeRaw(String(value.result))
}
gen.writeEndObject()
}