mirror of
https://github.com/codex-storage/nim-codex.git
synced 2025-02-23 18:18:20 +00:00
0d6b3f862b
- add requestId and slotIndex to Reservation (hopefully these will prove to be useful when we persist Reservations until request are completed, to add back bytes to Availability) - add querying of all reservations, with accompanying tests - change from find to findAvailabilities - move onCleanUp from SalesContext to SalesAgent as it was getting overwritten for each slot processed - remove sales agent AFTER deleting reservation, as this was causing some SIGSEGVs - retrofit testsales and testslotqueue to match updated Reservations module API
84 lines
2.5 KiB
Nim
84 lines
2.5 KiB
Nim
import pkg/chronicles
|
|
import pkg/questionable
|
|
import pkg/questionable/results
|
|
import ../../market
|
|
import ../salesagent
|
|
import ../statemachine
|
|
import ./errorhandling
|
|
import ./cancelled
|
|
import ./failed
|
|
import ./filled
|
|
import ./ignored
|
|
import ./downloading
|
|
import ./errored
|
|
|
|
type
|
|
SalePreparing* = ref object of ErrorHandlingState
|
|
|
|
logScope:
|
|
topics = "marketplace sales preparing"
|
|
|
|
method `$`*(state: SalePreparing): string = "SalePreparing"
|
|
|
|
method onCancelled*(state: SalePreparing, request: StorageRequest): ?State =
|
|
return some State(SaleCancelled())
|
|
|
|
method onFailed*(state: SalePreparing, request: StorageRequest): ?State =
|
|
return some State(SaleFailed())
|
|
|
|
method onSlotFilled*(state: SalePreparing, requestId: RequestId,
|
|
slotIndex: UInt256): ?State =
|
|
return some State(SaleFilled())
|
|
|
|
method run*(state: SalePreparing, machine: Machine): Future[?State] {.async.} =
|
|
let agent = SalesAgent(machine)
|
|
let data = agent.data
|
|
let context = agent.context
|
|
let market = context.market
|
|
let reservations = context.reservations
|
|
|
|
await agent.retrieveRequest()
|
|
await agent.subscribe()
|
|
|
|
without request =? data.request:
|
|
raiseAssert "no sale request"
|
|
|
|
let slotId = slotId(data.requestId, data.slotIndex)
|
|
let state = await market.slotState(slotId)
|
|
if state != SlotState.Free:
|
|
return some State(SaleIgnored())
|
|
|
|
# TODO: Once implemented, check to ensure the host is allowed to fill the slot,
|
|
# due to the [sliding window mechanism](https://github.com/codex-storage/codex-research/blob/master/design/marketplace.md#dispersal)
|
|
|
|
logScope:
|
|
slotIndex = data.slotIndex
|
|
slotSize = request.ask.slotSize
|
|
duration = request.ask.duration
|
|
pricePerSlot = request.ask.pricePerSlot
|
|
|
|
# availability was checked for this slot when it entered the queue, however
|
|
# check to the ensure that there is still availability as they may have
|
|
# changed since being added (other slots may have been processed in that time)
|
|
without availability =? await reservations.findAvailability(
|
|
request.ask.slotSize,
|
|
request.ask.duration,
|
|
request.ask.pricePerSlot,
|
|
request.ask.collateral):
|
|
info "no availability found for request, ignoring"
|
|
|
|
return some State(SaleIgnored())
|
|
|
|
info "availability found for request, creating reservation"
|
|
|
|
without reservation =? await reservations.createReservation(
|
|
availability.id,
|
|
request.ask.slotSize,
|
|
request.id,
|
|
data.slotIndex
|
|
), error:
|
|
return some State(SaleErrored(error: error))
|
|
|
|
data.reservation = some reservation
|
|
return some State(SaleDownloading())
|