mirror of
https://github.com/codex-storage/nim-codex.git
synced 2025-02-21 09:08:11 +00:00
88caa011b6
Documentation in the REST OpenAPI spec were showing incorrect information about Purchasing endpoints. Update the open api spec to reflect information about purchases, not requests.
98 lines
2.5 KiB
Nim
98 lines
2.5 KiB
Nim
## Nim-Codex
|
|
## Copyright (c) 2022 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import std/sequtils
|
|
import std/tables
|
|
import std/algorithm
|
|
|
|
import pkg/upraises
|
|
|
|
push: {.upraises: [].}
|
|
|
|
import pkg/chronos
|
|
import pkg/chronicles
|
|
import pkg/libp2p
|
|
|
|
import ../protobuf/blockexc
|
|
|
|
import ./peercontext
|
|
export peercontext
|
|
|
|
logScope:
|
|
topics = "codex peerctxstore"
|
|
|
|
type
|
|
PeerCtxStore* = ref object of RootObj
|
|
peers*: OrderedTable[PeerId, BlockExcPeerCtx]
|
|
|
|
iterator items*(self: PeerCtxStore): BlockExcPeerCtx =
|
|
for p in self.peers.values:
|
|
yield p
|
|
|
|
proc contains*(a: openArray[BlockExcPeerCtx], b: PeerId): bool =
|
|
## Convenience method to check for peer precense
|
|
##
|
|
|
|
a.anyIt( it.id == b )
|
|
|
|
func contains*(self: PeerCtxStore, peerId: PeerId): bool =
|
|
peerId in self.peers
|
|
|
|
func add*(self: PeerCtxStore, peer: BlockExcPeerCtx) =
|
|
trace "Adding peer to peer context store", peer = peer.id
|
|
self.peers[peer.id] = peer
|
|
|
|
func remove*(self: PeerCtxStore, peerId: PeerId) =
|
|
trace "Removing peer from peer context store", peer = peerId
|
|
self.peers.del(peerId)
|
|
|
|
func get*(self: PeerCtxStore, peerId: PeerId): BlockExcPeerCtx =
|
|
trace "Retrieving peer from peer context store", peer = peerId
|
|
self.peers.getOrDefault(peerId, nil)
|
|
|
|
func len*(self: PeerCtxStore): int =
|
|
self.peers.len
|
|
|
|
func peersHave*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
toSeq(self.peers.values).filterIt( it.peerHave.anyIt( it == cid ) )
|
|
|
|
func peersWant*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
toSeq(self.peers.values).filterIt( it.peerWants.anyIt( it.cid == cid ) )
|
|
|
|
func selectCheapest*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
var
|
|
peers = self.peersHave(cid)
|
|
|
|
trace "Selecting cheapest peers", peers = peers.len
|
|
func cmp(a, b: BlockExcPeerCtx): int =
|
|
var
|
|
priceA = 0.u256
|
|
priceB = 0.u256
|
|
|
|
a.blocks.withValue(cid, precense):
|
|
priceA = precense[].price
|
|
|
|
b.blocks.withValue(cid, precense):
|
|
priceB = precense[].price
|
|
|
|
if priceA == priceB:
|
|
0
|
|
elif priceA > priceB:
|
|
1
|
|
else:
|
|
-1
|
|
|
|
peers.sort(cmp)
|
|
trace "Selected cheapest peers", peers = peers.len
|
|
return peers
|
|
|
|
proc new*(T: type PeerCtxStore): PeerCtxStore =
|
|
T(
|
|
peers: initOrderedTable[PeerId, BlockExcPeerCtx]())
|