nimbus-eth2/benchmarks/rest_api_benchmark.nim

55 lines
1.6 KiB
Nim
Raw Normal View History

import
chronicles, chronicles/[topics_registry, timings],
confutils, confutils/std/net,
../beacon_chain/spec/eth2_apis/rest_beacon_client
type
Config = object
serverIpAddress {.
defaultValue: ValidIpAddress.init("127.0.0.1")
defaultValueDesc: "127.0.0.1"
desc: "IP address of the beacon node's REST server"
abbr: "a"
name: "address" }: ValidIpAddress
serverPort {.
defaultValue: 5052
desc: "Listening port of the beacon node's REST server"
abbr: "p"
name: "port" }: Port
startSlot {.
defaultValue: 0
desc: "The starting slot from which to start history traversal"
abbr: "s"
name: "start-slot" }: uint
requestsCount {.
desc: "Number of requests to send to the beacon node's REST server"
abbr: "n"
name: "count" }: uint
proc main =
let config = Config.load
let serverAddress = initTAddress(config.serverIpAddress, config.serverPort)
let client = RestClientRef.new(serverAddress)
setLogLevel(LogLevel.INFO)
template benchmark(apiNameIdent: untyped): untyped {.dirty.} =
block:
const apiName = astToStr apiNameIdent
info "Benchmarking ...", apiName
info.logTime(apiName):
for slot in config.startSlot ..< (config.startSlot + config.requestsCount):
let ident = StateIdent(kind: StateQueryKind.Slot, slot: slot.Slot)
discard waitFor client.`apiNameIdent`(ident)
benchmark(getStateRoot)
benchmark(getStateFork)
benchmark(getStateFinalityCheckpoints)
benchmark(getStateValidatorBalances)
when isMainModule:
main()