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

solution: transfer actual status and availability through grpc

Browse Source
This commit is contained in:
Igor Artamonov
2019-07-28 18:04:54 -04:00
parent b996edabef
commit 90e96ceb70
4 changed files with 54 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/main/kotlin/io/emeraldpay/dshackle/rpc/Describe.kt
View File

@@ -4,40 +4,38 @@ import io.emeraldpay.api.proto.BlockchainOuterClass
import io.emeraldpay.api.proto.Common
import io.emeraldpay.dshackle.upstream.UpstreamAvailability
import io.emeraldpay.dshackle.upstream.ConfiguredUpstreams
import io.emeraldpay.dshackle.upstream.Upstream
import io.emeraldpay.dshackle.upstream.Upstreams
import io.emeraldpay.grpc.Chain
import io.grpc.stub.StreamObserver
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Service
@Service
class Describe(
@Autowired private val upstreams: Upstreams
@Autowired private val upstreams: Upstreams,
@Autowired private val subscribeStatus: SubscribeStatus
) {
fun describe(request: BlockchainOuterClass.DescribeRequest, responseObserver: StreamObserver<BlockchainOuterClass.DescribeResponse>) {
val resp = BlockchainOuterClass.DescribeResponse.newBuilder()
upstreams.getAvailable().forEach { chain ->
upstreams.getUpstream(chain)?.let { chainUpstreams ->
val quorum = chainUpstreams.getAll().map { u ->
if (u.getStatus() == UpstreamAvailability.OK) {
u.getOptions().quorum
} else {
0
chainUpstreams.getAll().let { ups ->
if (ups.isNotEmpty()) {
val status = subscribeStatus.chainStatus(chain, ups)
resp.addChains(
BlockchainOuterClass.DescribeChain.newBuilder()
.setChain(Common.ChainRef.forNumber(chain.id))
.setStatus(status)
.build()
)
}
}.sum()
val available = chainUpstreams.getAll().any { u ->
u.getStatus() == UpstreamAvailability.OK
}
resp.addChains(
BlockchainOuterClass.DescribeChain.newBuilder()
.setChain(Common.ChainRef.forNumber(chain.id))
.setQuorum(quorum)
.setAvailable(available)
.build()
)
}
}
responseObserver.onNext(resp.build())
responseObserver.onCompleted()
}
}

32
src/main/kotlin/io/emeraldpay/dshackle/rpc/SubscribeStatus.kt
View File

@@ -2,8 +2,10 @@ package io.emeraldpay.dshackle.rpc
import io.emeraldpay.api.proto.BlockchainOuterClass
import io.emeraldpay.api.proto.Common
import io.emeraldpay.dshackle.upstream.Upstream
import io.emeraldpay.dshackle.upstream.UpstreamAvailability
import io.emeraldpay.dshackle.upstream.Upstreams
import io.emeraldpay.grpc.Chain
import io.grpc.StatusRuntimeException
import io.grpc.stub.StreamObserver
import org.springframework.beans.factory.annotation.Autowired
@@ -18,18 +20,11 @@ class SubscribeStatus(
fun subscribeStatus(request: BlockchainOuterClass.StatusRequest, responseObserver: StreamObserver<BlockchainOuterClass.ChainStatus>) {
upstreams.getAvailable().forEach { chain ->
var d: Disposable? = null
d = upstreams.getUpstream(chain)?.observeStatus()?.subscribe { availability ->
val chainStatus = BlockchainOuterClass.ChainStatus.newBuilder()
.setChain(Common.ChainRef.forNumber(chain.id))
.setAvailable(availability == UpstreamAvailability.OK)
.setQuorum(0)
if (availability == UpstreamAvailability.OK) {
upstreams.getUpstream(chain)?.getOptions()?.let { opts ->
chainStatus.setQuorum(opts.quorum)
}
}
val chainUpstream = upstreams.getUpstream(chain)
d = chainUpstream?.observeStatus()?.subscribe { availability ->
val status = chainStatus(chain, chainUpstream.getAll())
try {
responseObserver.onNext(chainStatus.build())
responseObserver.onNext(status)
} catch (e: StatusRuntimeException) {
// gRPC channel was closed
d?.dispose()
@@ -38,4 +33,19 @@ class SubscribeStatus(
}
}
fun chainStatus(chain: Chain, ups: List<Upstream>): BlockchainOuterClass.ChainStatus {
val available = ups.map { u ->
u.getStatus()
}.min()!!
val quorum = ups.filter {
it.getStatus() > UpstreamAvailability.UNAVAILABLE
}.count()
val status = BlockchainOuterClass.ChainStatus.newBuilder()
.setAvailability(BlockchainOuterClass.AvailabilityEnum.forNumber(available.grpcId))
.setChain(Common.ChainRef.forNumber(chain.id))
.setQuorum(quorum)
.build()
return status
}
}

10
src/main/kotlin/io/emeraldpay/dshackle/upstream/GrpcUpstream.kt
View File

@@ -86,18 +86,14 @@ open class GrpcUpstream(
}
fun init(conf: BlockchainOuterClass.DescribeChain) {
val available = conf.available
val quorum = conf.quorum
setStatus(
if (available && quorum > 0) UpstreamAvailability.OK else UpstreamAvailability.UNAVAILABLE
)
conf.status?.let { status -> onStatus(status) }
}
fun onStatus(value: BlockchainOuterClass.ChainStatus) {
val available = value.available
val available = value.availability
val quorum = value.quorum
setStatus(
if (available && quorum > 0) UpstreamAvailability.OK else UpstreamAvailability.UNAVAILABLE
if (available != null) UpstreamAvailability.fromGrpc(available.number) else UpstreamAvailability.UNAVAILABLE
)
}

22
src/main/kotlin/io/emeraldpay/dshackle/upstream/UpstreamAvailability.kt
View File

@@ -1,11 +1,21 @@
package io.emeraldpay.dshackle.upstream
enum class UpstreamAvailability {
enum class UpstreamAvailability(val grpcId: Int) {
OK,
IMMATURE,
SYNCING,
LAGGING,
UNAVAILABLE
OK(1),
IMMATURE(2),
SYNCING(3),
LAGGING(4),
UNAVAILABLE(5);
companion object {
fun fromGrpc(id: Int?): UpstreamAvailability {
if (id == null) {
return UNAVAILABLE
}
return UpstreamAvailability.values().find {
it.grpcId == id
} ?: UNAVAILABLE
}
}
}