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

fix lotus-shed

Browse Source
This commit is contained in:
vyzo 2021-07-11 18:00:54 +03:00
parent aec126879e
commit 4715b1f436
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
blockstore/badger/blockstore.go
View File

@ -950,3 +950,9 @@ func (b *Blockstore) StorageKey(dst []byte, cid cid.Cid) []byte {
} }
return dst[:reqsize] return dst[:reqsize]
} }
// this method is added for lotus-shed needs
// WARNING: THIS IS COMPLETELY UNSAFE; DONT USE THIS IN PRODUCTION CODE
func (b *Blockstore) DB() *badger.DB {
return b.db
}

6
cmd/lotus-shed/pruning.go
View File

@ -161,7 +161,7 @@ var stateTreePruneCmd = &cli.Command{
if cctx.Bool("only-ds-gc") { if cctx.Bool("only-ds-gc") {
fmt.Println("running datastore gc....") fmt.Println("running datastore gc....")
for i := 0; i < cctx.Int("gc-count"); i++ { for i := 0; i < cctx.Int("gc-count"); i++ {
if err := badgbs.DB.RunValueLogGC(DiscardRatio); err != nil { if err := badgbs.DB().RunValueLogGC(DiscardRatio); err != nil {
return xerrors.Errorf("datastore GC failed: %w", err) return xerrors.Errorf("datastore GC failed: %w", err)
} }
} }
@ -208,7 +208,7 @@ var stateTreePruneCmd = &cli.Command{
return nil return nil
} }
b := badgbs.DB.NewWriteBatch() b := badgbs.DB().NewWriteBatch()
defer b.Cancel() defer b.Cancel()
markForRemoval := func(c cid.Cid) error { markForRemoval := func(c cid.Cid) error {
@ -249,7 +249,7 @@ var stateTreePruneCmd = &cli.Command{
fmt.Println("running datastore gc....") fmt.Println("running datastore gc....")
for i := 0; i < cctx.Int("gc-count"); i++ { for i := 0; i < cctx.Int("gc-count"); i++ {
if err := badgbs.DB.RunValueLogGC(DiscardRatio); err != nil { if err := badgbs.DB().RunValueLogGC(DiscardRatio); err != nil {
return xerrors.Errorf("datastore GC failed: %w", err) return xerrors.Errorf("datastore GC failed: %w", err)
} }
} }