mirror of
https://github.com/logos-storage/logos-storage-network-crawler.git
synced 2026-01-02 21:43:11 +00:00
67 lines
1.8 KiB
Nim
67 lines
1.8 KiB
Nim
import pkg/chronicles
|
|
import pkg/chronos
|
|
import pkg/questionable/results
|
|
|
|
import ./nodestore
|
|
import ../services/dht
|
|
import ../component
|
|
import ../state
|
|
import ../types
|
|
import ../utils/asyncdataevent
|
|
|
|
logScope:
|
|
topics = "timetracker"
|
|
|
|
type TimeTracker* = ref object of Component
|
|
state: State
|
|
nodestore: NodeStore
|
|
dht: Dht
|
|
|
|
proc checkForExpiredNodes(t: TimeTracker): Future[?!void] {.async: (raises: []).} =
|
|
trace "Checking for expired nodes..."
|
|
let expiry =
|
|
(Moment.now().epochSeconds - (t.state.config.revisitDelayMins * 60)).uint64
|
|
|
|
var expired = newSeq[Nid]()
|
|
proc checkNode(item: NodeEntry): Future[?!void] {.async: (raises: []), gcsafe.} =
|
|
if item.lastVisit < expiry:
|
|
expired.add(item.id)
|
|
return success()
|
|
|
|
?await t.nodestore.iterateAll(checkNode)
|
|
?await t.state.events.nodesExpired.fire(expired)
|
|
return success()
|
|
|
|
proc raiseRoutingTableNodes(t: TimeTracker): Future[?!void] {.async: (raises: []).} =
|
|
trace "Raising routing table nodes..."
|
|
let nids = t.dht.getRoutingTableNodeIds()
|
|
if err =? (await t.state.events.nodesFound.fire(nids)).errorOption:
|
|
return failure(err)
|
|
return success()
|
|
|
|
proc step(t: TimeTracker): Future[?!void] {.async: (raises: []).} =
|
|
?await t.checkForExpiredNodes()
|
|
?await t.raiseRoutingTableNodes()
|
|
return success()
|
|
|
|
method start*(t: TimeTracker): Future[?!void] {.async.} =
|
|
info "Starting..."
|
|
|
|
proc onStep(): Future[?!void] {.async: (raises: []), gcsafe.} =
|
|
await t.step()
|
|
|
|
var delay = t.state.config.revisitDelayMins div 100
|
|
if delay < 1:
|
|
delay = 1
|
|
|
|
await t.state.whileRunning(onStep, delay.minutes)
|
|
return success()
|
|
|
|
method stop*(t: TimeTracker): Future[?!void] {.async.} =
|
|
return success()
|
|
|
|
proc new*(
|
|
T: type TimeTracker, state: State, nodestore: NodeStore, dht: Dht
|
|
): TimeTracker =
|
|
TimeTracker(state: state, nodestore: nodestore, dht: dht)
|