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

Add starknet node version detector (#663)

Browse Source 
* Add starknet node version detector

* add juno version detecting
This commit is contained in:
Anton
2025-05-12 16:35:51 +03:00
committed by GitHub
parent 455bfdb8dd
commit 03383cf1b5
2 changed files with 83 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/kotlin/io/emeraldpay/dshackle/upstream/starknet/StarknetChainSpecific.kt
View File

@@ -14,6 +14,7 @@ import io.emeraldpay.dshackle.upstream.GenericSingleCallValidator
import io.emeraldpay.dshackle.upstream.SingleValidator
import io.emeraldpay.dshackle.upstream.Upstream
import io.emeraldpay.dshackle.upstream.UpstreamAvailability
import io.emeraldpay.dshackle.upstream.UpstreamSettingsDetector
import io.emeraldpay.dshackle.upstream.ValidateUpstreamSettingsResult
import io.emeraldpay.dshackle.upstream.generic.AbstractPollChainSpecific
import io.emeraldpay.dshackle.upstream.lowerbound.LowerBoundService
@@ -107,6 +108,13 @@ object StarknetChainSpecific : AbstractPollChainSpecific() {
override fun latestBlockRequest(): ChainRequest =
ChainRequest("starknet_getBlockWithTxHashes", ListParams("latest"))
override fun upstreamSettingsDetector(
chain: Chain,
upstream: Upstream,
): UpstreamSettingsDetector {
return StarknetUpstreamSettingsDetector(upstream, chain)
}
}
@JsonIgnoreProperties(ignoreUnknown = true)

75
src/main/kotlin/io/emeraldpay/dshackle/upstream/starknet/StarknetUpstreamSettingsDetector.kt Normal file
View File

@@ -0,0 +1,75 @@
package io.emeraldpay.dshackle.upstream.starknet
import io.emeraldpay.dshackle.Chain
import io.emeraldpay.dshackle.upstream.ChainRequest
import io.emeraldpay.dshackle.upstream.ChainResponse
import io.emeraldpay.dshackle.upstream.Upstream
import io.emeraldpay.dshackle.upstream.UpstreamSettingsDetector
import io.emeraldpay.dshackle.upstream.rpcclient.ListParams
import reactor.core.publisher.Flux
class StarknetUpstreamSettingsDetector(
private val upstream: Upstream,
private val chain: Chain,
) : UpstreamSettingsDetector(upstream) {
override fun internalDetectLabels(): Flux<Pair<String, String>> {
return Flux.merge(
detectNodeType(),
)
}
private fun detectNodeType(): Flux<Pair<String, String>?> {
return upstream
.getIngressReader()
.read(pathfinderVersionRequest())
.flatMap(ChainResponse::requireResult)
.flatMapMany { data ->
val version = parseClientVersion(data)
if (version.isEmpty()) {
throw Exception()
}
val labels =
mutableListOf("client_type" to "pathfinder", "client_version" to version)
Flux.fromIterable(labels)
}.onErrorResume {
upstream
.getIngressReader()
.read(clientVersionRequest())
.flatMap(ChainResponse::requireResult)
.flatMapMany { data ->
val version = parseClientVersion(data)
if (version.isEmpty()) {
throw Exception()
}
val labels =
mutableListOf(
"client_type" to "juno",
"client_version" to version,
)
Flux.fromIterable(labels)
}
.onErrorResume { error ->
log.warn(
"Can't detect the node type of upstream ${upstream.getId()}, reason - {}",
error.message,
)
Flux.empty()
}
}
}
override fun clientVersionRequest(): ChainRequest = ChainRequest("juno_version", ListParams())
private fun pathfinderVersionRequest() = ChainRequest("pathfinder_version", ListParams())
override fun parseClientVersion(data: ByteArray): String {
var version = String(data)
if (version.startsWith("\"") && version.endsWith("\"")) {
version = version.substring(1, version.length - 1)
}
if (version.startsWith("v")) {
version = version.substring(1, version.length)
}
return version
}
}