mirror of
https://github.com/status-im/nim-dagger.git
synced 2025-03-01 14:50:41 +00:00
87590f43ce
* fix(statemachine): do not raise from state.run * fix rebase * fix exception handling in SaleProvingSimulated.prove - re-raise CancelledError - don't return State on CatchableError - expect the Proofs_InvalidProof custom error instead of checking a string * asyncSpawn salesagent.onCancelled This was swallowing a KeyError in one of the tests (fixed in the previous commit) * remove error handling states in asyncstatemachine * revert unneeded changes * formatting * PR feedback, logging updates
67 lines
2.1 KiB
Nim
67 lines
2.1 KiB
Nim
import pkg/questionable
|
|
import pkg/metrics
|
|
|
|
import ../../logutils
|
|
import ../../market
|
|
import ../../utils/exceptions
|
|
import ../salesagent
|
|
import ../statemachine
|
|
import ./cancelled
|
|
import ./failed
|
|
import ./ignored
|
|
import ./downloading
|
|
import ./errored
|
|
|
|
type SaleSlotReserving* = ref object of SaleState
|
|
|
|
logScope:
|
|
topics = "marketplace sales reserving"
|
|
|
|
method `$`*(state: SaleSlotReserving): string =
|
|
"SaleSlotReserving"
|
|
|
|
method onCancelled*(state: SaleSlotReserving, request: StorageRequest): ?State =
|
|
return some State(SaleCancelled())
|
|
|
|
method onFailed*(state: SaleSlotReserving, request: StorageRequest): ?State =
|
|
return some State(SaleFailed())
|
|
|
|
method run*(
|
|
state: SaleSlotReserving, machine: Machine
|
|
): Future[?State] {.async: (raises: []).} =
|
|
let agent = SalesAgent(machine)
|
|
let data = agent.data
|
|
let context = agent.context
|
|
let market = context.market
|
|
|
|
logScope:
|
|
requestId = data.requestId
|
|
slotIndex = data.slotIndex
|
|
|
|
try:
|
|
let canReserve = await market.canReserveSlot(data.requestId, data.slotIndex)
|
|
if canReserve:
|
|
try:
|
|
trace "Reserving slot"
|
|
await market.reserveSlot(data.requestId, data.slotIndex)
|
|
except MarketError as e:
|
|
if e.msg.contains "SlotReservations_ReservationNotAllowed":
|
|
debug "Slot cannot be reserved, ignoring", error = e.msg
|
|
return some State(SaleIgnored(reprocessSlot: false, returnBytes: true))
|
|
else:
|
|
return some State(SaleErrored(error: e))
|
|
# other CatchableErrors are handled "automatically" by the SaleState
|
|
|
|
trace "Slot successfully reserved"
|
|
return some State(SaleDownloading())
|
|
else:
|
|
# do not re-add this slot to the queue, and return bytes from Reservation to
|
|
# the Availability
|
|
debug "Slot cannot be reserved, ignoring"
|
|
return some State(SaleIgnored(reprocessSlot: false, returnBytes: true))
|
|
except CancelledError as e:
|
|
trace "SaleSlotReserving.run was cancelled", error = e.msgDetail
|
|
except CatchableError as e:
|
|
error "Error during SaleSlotReserving.run", error = e.msgDetail
|
|
return some State(SaleErrored(error: e))
|