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

Update stores list when deleting

Browse Source 
Signed-off-by: Jakub Sztandera <kubuxu@protocol.ai>
This commit is contained in:
Jakub Sztandera 2020-07-17 22:44:16 +02:00
parent f2ebeaa7e4
commit 0f5e56289e
No known key found for this signature in database
GPG Key ID: 9A9AF56F8B3879BA
1 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
node/repo/importmgr/multistore.go
View File

@ -66,6 +66,24 @@ func (mds *MultiStore) Next() int {
return mds.next return mds.next
} }
func (mds *MultiStore) updateStores() error {
stores := make([]int, 0, len(mds.open))
for k := range mds.open {
stores = append(stores, k)
}
sort.Ints(stores)
listBytes, err := json.Marshal(stores)
if err != nil {
return xerrors.Errorf("could not marshal list: %w", err)
}
err = mds.ds.Put(dsListKey, listBytes)
if err != nil {
return xerrors.Errorf("could not save stores list: %w", err)
}
return nil
}
func (mds *MultiStore) Get(i int) (*Store, error) { func (mds *MultiStore) Get(i int) (*Store, error) {
mds.lk.Lock() mds.lk.Lock()
defer mds.lk.Unlock() defer mds.lk.Unlock()
@ -85,22 +103,12 @@ func (mds *MultiStore) Get(i int) (*Store, error) {
return nil, xerrors.Errorf("could not open new store: %w", err) return nil, xerrors.Errorf("could not open new store: %w", err)
} }
stores := make([]int, 0, len(mds.open)) err = mds.updateStores()
for k := range mds.open {
stores = append(stores, k)
}
sort.Ints(stores)
listBytes, err := json.Marshal(stores)
if err != nil { if err != nil {
return nil, xerrors.Errorf("could not marshal list: %w", err) return nil, xerrors.Errorf("updating stores: %w", err)
}
err = mds.ds.Put(dsListKey, listBytes)
if err != nil {
return nil, xerrors.Errorf("could not save stores list: %w", err)
} }
return mds.open[i], err return mds.open[i], nil
} }
func (mds *MultiStore) List() []int { func (mds *MultiStore) List() []int {
@ -130,6 +138,11 @@ func (mds *MultiStore) Delete(i int) error {
return xerrors.Errorf("closing store: %w", err) return xerrors.Errorf("closing store: %w", err)
} }
err = mds.updateStores()
if err != nil {
return xerrors.Errorf("updating stores: %w", err)
}
qres, err := store.ds.Query(query.Query{KeysOnly: true}) qres, err := store.ds.Query(query.Query{KeysOnly: true})
if err != nil { if err != nil {
return xerrors.Errorf("query error: %w", err) return xerrors.Errorf("query error: %w", err)