catch nil case

This commit is contained in:
Cassandra Heart 2025-02-06 15:07:22 -06:00
parent 6deaee1439
commit d61b78567b
No known key found for this signature in database
GPG Key ID: 6352152859385958
2 changed files with 18 additions and 5 deletions

View File

@ -236,6 +236,7 @@ func NewTokenExecutionEngine(
keyManager: keyManager,
clockStore: clockStore,
coinStore: coinStore,
hypergraphStore: hypergraphStore,
keyStore: keyStore,
pubSub: pubSub,
inclusionProver: inclusionProver,
@ -452,6 +453,10 @@ func (e *TokenExecutionEngine) rebuildHypergraph() {
for iter.First(); iter.Valid(); iter.Next() {
key := make([]byte, len(iter.Key()[2:]))
copy(key, iter.Key()[2:])
e.logger.Debug(
"encrypting coin",
zap.String("address", hex.EncodeToString(key)),
)
data := e.mpcithVerEnc.Encrypt(iter.Value(), config.GetGenesis().Beacon)
compressed := []hypergraph.Encrypted{}
for _, d := range data {
@ -476,6 +481,10 @@ func (e *TokenExecutionEngine) rebuildHypergraph() {
for iter.First(); iter.Valid(); iter.Next() {
key := make([]byte, len(iter.Key()[2:]))
copy(key, iter.Key()[2:])
e.logger.Debug(
"encrypting pre-coin proof",
zap.String("address", hex.EncodeToString(key)),
)
data := e.mpcithVerEnc.Encrypt(iter.Value(), config.GetGenesis().Beacon)
compressed := []hypergraph.Encrypted{}
for _, d := range data {
@ -492,14 +501,14 @@ func (e *TokenExecutionEngine) rebuildHypergraph() {
}
}
iter.Close()
e.logger.Info("saving rebuilt state tree")
e.logger.Info("saving rebuilt hypergraph")
txn, err := e.clockStore.NewTransaction(false)
if err != nil {
panic(err)
}
e.logger.Info("committing state tree")
e.logger.Info("committing hypergraph")
roots := e.hypergraph.Commit()
@ -1179,7 +1188,7 @@ func (e *TokenExecutionEngine) ProcessFrame(
}
}
e.logger.Info("committing state tree")
e.logger.Info("committing hypergraph")
roots := hg.Commit()

View File

@ -373,7 +373,11 @@ func (set *IdSet) Add(atom Atom) error {
}
func (set *IdSet) GetSize() *big.Int {
return set.tree.GetSize()
size := set.tree.GetSize()
if size == nil {
size = big.NewInt(0)
}
return size
}
func (set *IdSet) Delete(atom Atom) bool {
@ -407,7 +411,7 @@ type Hypergraph struct {
func NewHypergraph() *Hypergraph {
return &Hypergraph{
size: new(big.Int),
size: big.NewInt(0),
vertexAdds: make(map[ShardKey]*IdSet),
vertexRemoves: make(map[ShardKey]*IdSet),
hyperedgeAdds: make(map[ShardKey]*IdSet),