Skip to content

Commit 99b8ede

Browse files
committed
comment: split runtime and persistent types
1 parent 52dafd1 commit 99b8ede

File tree

1 file changed

+28
-24
lines changed

1 file changed

+28
-24
lines changed

mempool/mempool.go

+28-24
Original file line numberDiff line numberDiff line change
@@ -20,23 +20,27 @@ type BroadcastedTransaction struct {
2020
DeclaredClass core.Class
2121
}
2222

23-
// storageElem defines a node for both the
24-
// in-memory and persistent linked-list
25-
type storageElem struct {
23+
// runtime mempool txn
24+
type memPoolTxn struct {
25+
Txn BroadcastedTransaction
26+
Next *memPoolTxn
27+
}
28+
29+
// persistent db txn value
30+
type dbPoolTxn struct {
2631
Txn BroadcastedTransaction
27-
NextHash *felt.Felt // persistent db
28-
Next *storageElem // in-memory
32+
NextHash *felt.Felt
2933
}
3034

31-
// memTxnList represents a linked list of user transactions at runtime"
35+
// memTxnList represents a linked list of user transactions at runtime
3236
type memTxnList struct {
33-
head *storageElem
34-
tail *storageElem
37+
head *memPoolTxn
38+
tail *memPoolTxn
3539
len int
3640
mu sync.Mutex
3741
}
3842

39-
func (t *memTxnList) push(newNode *storageElem) {
43+
func (t *memTxnList) push(newNode *memPoolTxn) {
4044
t.mu.Lock()
4145
defer t.mu.Unlock()
4246
if t.tail != nil {
@@ -130,24 +134,24 @@ func (p *Pool) LoadFromDB() error {
130134
// loop through the persistent pool and push nodes to the in-memory pool
131135
currentHash := headValue
132136
for currentHash != nil {
133-
curElem, err := p.readDBElem(txn, currentHash)
137+
curDBElem, err := p.readDBElem(txn, currentHash)
134138
if err != nil {
135139
return err
136140
}
137-
newNode := &storageElem{
138-
Txn: curElem.Txn,
141+
newMemPoolTxn := &memPoolTxn{
142+
Txn: curDBElem.Txn,
139143
}
140-
if curElem.NextHash != nil {
141-
nxtElem, err := p.readDBElem(txn, curElem.NextHash)
144+
if curDBElem.NextHash != nil {
145+
nextDBTxn, err := p.readDBElem(txn, curDBElem.NextHash)
142146
if err != nil {
143147
return err
144148
}
145-
newNode.Next = &storageElem{
146-
Txn: nxtElem.Txn,
149+
newMemPoolTxn.Next = &memPoolTxn{
150+
Txn: nextDBTxn.Txn,
147151
}
148152
}
149-
p.memTxnList.push(newNode)
150-
currentHash = curElem.NextHash
153+
p.memTxnList.push(newMemPoolTxn)
154+
currentHash = curDBElem.NextHash
151155
}
152156
return nil
153157
})
@@ -163,12 +167,12 @@ func (p *Pool) writeToDB(userTxn *BroadcastedTransaction) error {
163167
}
164168
tailValue = nil
165169
}
166-
if err := p.setDBElem(dbTxn, &storageElem{Txn: *userTxn}); err != nil {
170+
if err := p.setDBElem(dbTxn, &dbPoolTxn{Txn: *userTxn}); err != nil {
167171
return err
168172
}
169173
if tailValue != nil {
170174
// Update old tail to point to the new item
171-
var oldTailElem storageElem
175+
var oldTailElem dbPoolTxn
172176
oldTailElem, err := p.readDBElem(dbTxn, tailValue)
173177
if err != nil {
174178
return err
@@ -215,7 +219,7 @@ func (p *Pool) Push(userTxn *BroadcastedTransaction) error {
215219
}
216220
}
217221

218-
newNode := &storageElem{Txn: *userTxn, Next: nil}
222+
newNode := &memPoolTxn{Txn: *userTxn, Next: nil}
219223
p.memTxnList.push(newNode)
220224

221225
select {
@@ -339,16 +343,16 @@ func (p *Pool) updateTail(txn db.Transaction, tail *felt.Felt) error {
339343
return txn.Set(db.MempoolTail.Key(), tail.Marshal())
340344
}
341345

342-
func (p *Pool) readDBElem(txn db.Transaction, itemKey *felt.Felt) (storageElem, error) {
343-
var item storageElem
346+
func (p *Pool) readDBElem(txn db.Transaction, itemKey *felt.Felt) (dbPoolTxn, error) {
347+
var item dbPoolTxn
344348
keyBytes := itemKey.Bytes()
345349
err := txn.Get(db.MempoolNode.Key(keyBytes[:]), func(b []byte) error {
346350
return encoder.Unmarshal(b, &item)
347351
})
348352
return item, err
349353
}
350354

351-
func (p *Pool) setDBElem(txn db.Transaction, item *storageElem) error {
355+
func (p *Pool) setDBElem(txn db.Transaction, item *dbPoolTxn) error {
352356
itemBytes, err := encoder.Marshal(item)
353357
if err != nil {
354358
return err

0 commit comments

Comments
 (0)