status-im
/
nim-abc
mirror of https://github.com/status-im/nim-abc.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Convert History.transactions to a set

This commit is contained in:
Mark Spanbroek 2021-07-08 14:07:07 +02:00
parent 32be2fd97d
commit 524689ac59
2 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
abc/history.nim
View File

@ -1,14 +1,17 @@
import std/sets
import ./txstore
export sets
type
History* = object
transactions*: seq[TxHash]
transactions*: HashSet[TxHash]
func past(store: TxStore, txHash: TxHash, history: var History) =
if transaction =? store[txHash]:
for (hash, _) in transaction.inputs:
if not history.transactions.contains hash:
history.transactions.add(hash)
history.transactions.incl(hash)
store.past(hash, history)
func past(store: TxStore, ackHash: AckHash, history: var History) =
@ -17,7 +20,7 @@ func past(store: TxStore, ackHash: AckHash, history: var History) =
store.past(previous, history)
for txHash in ack.transactions:
if not history.transactions.contains txHash:
history.transactions.add(txHash)
history.transactions.incl(txHash)
store.past(txHash, history)
func past*(store: TxStore, hash: TxHash|AckHash): History =

28
tests/abc/testHistory.nim
View File

@ -24,33 +24,33 @@ suite "Past transactions and acknowledgements":
ack2 = !Ack.init([tx2.hash], victor)
ack3 = !Ack.init(ack1.hash, [tx2.hash, tx3.hash], victor)
func set(transactions: varargs[TxHash]): HashSet[TxHash] =
transactions.toHashSet
test "finds all transactions that precede a transaction":
var store = TxStore.init(genesis)
store.add(tx1, tx2, tx3)
check store.past(tx1.hash).transactions == [genesis.hash]
check store.past(tx2.hash).transactions == [genesis.hash]
check store.past(tx3.hash).transactions == [
tx1.hash, genesis.hash, tx2.hash
]
check store.past(tx1.hash).transactions == set(genesis.hash)
check store.past(tx2.hash).transactions == set(genesis.hash)
check store.past(tx3.hash).transactions ==
set(genesis.hash, tx1.hash, tx2.hash)
test "past is empty when transaction cannot be found":
let store = TxStore.init(genesis)
check store.past(tx1.hash).transactions == []
check store.past(tx1.hash).transactions == set()
test "finds all transactions that precede an acknowledgement":
var store = TxStore.init(genesis)
store.add(tx1, tx2, tx3)
store.add(ack1, ack2, ack3)
check store.past(ack1.hash).transactions == [tx1.hash, genesis.hash]
check store.past(ack2.hash).transactions == [tx2.hash, genesis.hash]
check store.past(ack3.hash).transactions == [
tx1.hash, genesis.hash, tx2.hash, tx3.hash
]
check store.past(ack1.hash).transactions == set(genesis.hash, tx1.hash)
check store.past(ack2.hash).transactions == set(genesis.hash, tx2.hash)
check store.past(ack3.hash).transactions ==
set(genesis.hash, tx1.hash, tx2.hash, tx3.hash)
test "finds all transactions that precede a set of acknowledgements":
var store = TxStore.init(genesis)
store.add(tx1, tx2, tx3)
store.add(ack1, ack2, ack3)
check store.past(ack1.hash, ack2.hash).transactions == [
tx1.hash, genesis.hash, tx2.hash
]
check store.past(ack1.hash, ack2.hash).transactions ==
set(genesis.hash, tx1.hash, tx2.hash)