nimbus-eth1/fluffy/network/state/state_network.nim

68 lines
2.2 KiB
Nim
Raw Normal View History

# Nimbus
# Copyright (c) 2021 Status Research & Development GmbH
# Licensed and distributed under either of
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
# at your option. This file may not be copied, modified, or distributed except according to those terms.
import
std/[options, sugar],
stew/results, chronos,
eth/p2p/discoveryv5/[protocol, enr],
../../content_db,
../wire/[portal_protocol, portal_stream],
./state_content,
./state_distance
const
stateProtocolId* = [byte 0x50, 0x0A]
type StateNetwork* = ref object
portalProtocol*: PortalProtocol
contentDB*: ContentDB
proc toContentIdHandler(contentKey: ByteList): Option[ContentId] =
toContentId(contentKey)
proc getContent*(n: StateNetwork, key: ContentKey):
2021-09-13 13:56:44 +00:00
Future[Option[seq[byte]]] {.async.} =
let
keyEncoded = encode(key)
contentId = toContentId(key)
contentInRange = n.portalProtocol.inRange(contentId)
# When the content id is in the radius range, try to look it up in the db.
if contentInRange:
let contentFromDB = n.contentDB.get(contentId)
if contentFromDB.isSome():
return contentFromDB
let content = await n.portalProtocol.contentLookup(keyEncoded, contentId)
# When content is found on the network and is in the radius range, store it.
if content.isSome() and contentInRange:
n.contentDB.put(contentId, content.get().asSeq())
# TODO: for now returning bytes, ultimately it would be nice to return proper
# domain types.
return content.map(x => x.asSeq())
proc new*(
T: type StateNetwork,
baseProtocol: protocol.Protocol,
contentDB: ContentDB,
portalStream: PortalStream,
dataRadius = UInt256.high(),
bootstrapRecords: openArray[Record] = []): T =
let portalProtocol = PortalProtocol.new(
baseProtocol, stateProtocolId, contentDB, toContentIdHandler,
portalStream, dataRadius, bootstrapRecords, stateDistanceCalculator)
2021-09-13 13:56:44 +00:00
return StateNetwork(portalProtocol: portalProtocol, contentDB: contentDB)
proc start*(n: StateNetwork) =
n.portalProtocol.start()
proc stop*(n: StateNetwork) =
n.portalProtocol.stop()