89 lines
2.6 KiB
Nim
89 lines
2.6 KiB
Nim
import std/sets
|
|
import pkg/upraises
|
|
import pkg/questionable
|
|
import pkg/chronicles
|
|
import ./storageproofs
|
|
import ./clock
|
|
|
|
export sets
|
|
export storageproofs
|
|
|
|
type
|
|
Proving* = ref object
|
|
proofs: Proofs
|
|
clock: Clock
|
|
loop: ?Future[void]
|
|
slots*: HashSet[Slot]
|
|
onProve: ?OnProve
|
|
OnProve* = proc(slot: Slot): Future[seq[byte]] {.gcsafe, upraises: [].}
|
|
|
|
func new*(_: type Proving, proofs: Proofs, clock: Clock): Proving =
|
|
Proving(proofs: proofs, clock: clock)
|
|
|
|
proc onProve*(proving: Proving): ?OnProve =
|
|
proving.onProve
|
|
|
|
proc `onProve=`*(proving: Proving, callback: OnProve) =
|
|
proving.onProve = some callback
|
|
|
|
func add*(proving: Proving, slot: Slot) =
|
|
proving.slots.incl(slot)
|
|
|
|
proc getCurrentPeriod(proving: Proving): Future[Period] {.async.} =
|
|
let periodicity = await proving.proofs.periodicity()
|
|
return periodicity.periodOf(proving.clock.now().u256)
|
|
|
|
proc waitUntilPeriod(proving: Proving, period: Period) {.async.} =
|
|
let periodicity = await proving.proofs.periodicity()
|
|
await proving.clock.waitUntil(periodicity.periodStart(period).truncate(int64))
|
|
|
|
proc removeEndedContracts(proving: Proving) {.async.} =
|
|
var ended: HashSet[Slot]
|
|
for slot in proving.slots:
|
|
let state = await proving.proofs.slotState(slot.id)
|
|
if state != SlotState.Filled:
|
|
ended.incl(slot)
|
|
proving.slots.excl(ended)
|
|
|
|
proc prove(proving: Proving, slot: Slot) {.async.} =
|
|
without onProve =? proving.onProve:
|
|
raiseAssert "onProve callback not set"
|
|
try:
|
|
let proof = await onProve(slot)
|
|
await proving.proofs.submitProof(slot.id, proof)
|
|
except CatchableError as e:
|
|
error "Submitting proof failed", msg = e.msg
|
|
|
|
proc run(proving: Proving) {.async.} =
|
|
try:
|
|
while true:
|
|
let currentPeriod = await proving.getCurrentPeriod()
|
|
await proving.removeEndedContracts()
|
|
for slot in proving.slots:
|
|
let id = slot.id
|
|
if (await proving.proofs.isProofRequired(id)) or
|
|
(await proving.proofs.willProofBeRequired(id)):
|
|
asyncSpawn proving.prove(slot)
|
|
await proving.waitUntilPeriod(currentPeriod + 1)
|
|
except CancelledError:
|
|
discard
|
|
except CatchableError as e:
|
|
error "Proving failed", msg = e.msg
|
|
|
|
proc start*(proving: Proving) {.async.} =
|
|
if proving.loop.isSome:
|
|
return
|
|
|
|
proving.loop = some proving.run()
|
|
|
|
proc stop*(proving: Proving) {.async.} =
|
|
if loop =? proving.loop:
|
|
proving.loop = Future[void].none
|
|
if not loop.finished:
|
|
await loop.cancelAndWait()
|
|
|
|
proc subscribeProofSubmission*(proving: Proving,
|
|
callback: OnProofSubmitted):
|
|
Future[Subscription] =
|
|
proving.proofs.subscribeProofSubmission(callback)
|