cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

use the transaction Luke!

Browse Source
This commit is contained in:
vyzo 2023-03-12 13:35:50 +02:00
parent d9ca214309
commit 0d274df977
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
chain/index/msgindex.go
View File

@ -292,42 +292,42 @@ func (x *msgIndex) processHeadChanges(ctx context.Context) error {
x.pend = nil x.pend = nil
x.mx.Unlock() x.mx.Unlock()
txn, err := x.db.Begin() tx, err := x.db.Begin()
if err != nil { if err != nil {
return xerrors.Errorf("error creating transaction: %w", err) return xerrors.Errorf("error creating transaction: %w", err)
} }
for _, hc := range pend { for _, hc := range pend {
for _, ts := range hc.rev { for _, ts := range hc.rev {
if err := x.doRevert(ctx, ts); err != nil { if err := x.doRevert(ctx, tx, ts); err != nil {
txn.Rollback() tx.Rollback()
return xerrors.Errorf("error reverting %s: %w", ts, err) return xerrors.Errorf("error reverting %s: %w", ts, err)
} }
} }
for _, ts := range hc.app { for _, ts := range hc.app {
if err := x.doApply(ctx, ts); err != nil { if err := x.doApply(ctx, tx, ts); err != nil {
txn.Rollback() tx.Rollback()
return xerrors.Errorf("error applying %s: %w", ts, err) return xerrors.Errorf("error applying %s: %w", ts, err)
} }
} }
} }
return txn.Commit() return tx.Commit()
} }
func (x *msgIndex) doRevert(ctx context.Context, ts *types.TipSet) error { func (x *msgIndex) doRevert(ctx context.Context, tx *sql.Tx, ts *types.TipSet) error {
tskey, err := ts.Key().Cid() tskey, err := ts.Key().Cid()
if err != nil { if err != nil {
return xerrors.Errorf("error computing tipset cid: %w", err) return xerrors.Errorf("error computing tipset cid: %w", err)
} }
key := tskey.String() key := tskey.String()
_, err = x.deleteTipSetStmt.Exec(key) _, err = tx.Stmt(x.deleteTipSetStmt).Exec(key)
return err return err
} }
func (x *msgIndex) doApply(ctx context.Context, ts *types.TipSet) error { func (x *msgIndex) doApply(ctx context.Context, tx *sql.Tx, ts *types.TipSet) error {
tscid, err := ts.Key().Cid() tscid, err := ts.Key().Cid()
if err != nil { if err != nil {
return xerrors.Errorf("error computing tipset cid: %w", err) return xerrors.Errorf("error computing tipset cid: %w", err)
@ -343,7 +343,7 @@ func (x *msgIndex) doApply(ctx context.Context, ts *types.TipSet) error {
return nil return nil
} }
if _, err := x.insertMsgStmt.Exec(key, tskey, xepoch, xindex); err != nil { if _, err := tx.Stmt(x.insertMsgStmt).Exec(key, tskey, xepoch, xindex); err != nil {
return err return err
} }
seen[key] = struct{}{} seen[key] = struct{}{}