60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package modules
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
|
|
"go.uber.org/fx"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/lib/backupds"
|
|
"github.com/filecoin-project/lotus/node/modules/dtypes"
|
|
"github.com/filecoin-project/lotus/node/modules/helpers"
|
|
"github.com/filecoin-project/lotus/node/repo"
|
|
)
|
|
|
|
func LockedRepo(lr repo.LockedRepo) func(lc fx.Lifecycle) repo.LockedRepo {
|
|
return func(lc fx.Lifecycle) repo.LockedRepo {
|
|
lc.Append(fx.Hook{
|
|
OnStop: func(_ context.Context) error {
|
|
return lr.Close()
|
|
},
|
|
})
|
|
|
|
return lr
|
|
}
|
|
}
|
|
|
|
func KeyStore(lr repo.LockedRepo) (types.KeyStore, error) {
|
|
return lr.KeyStore()
|
|
}
|
|
|
|
func Datastore(disableLog bool) func(lc fx.Lifecycle, mctx helpers.MetricsCtx, r repo.LockedRepo) (dtypes.MetadataDS, error) {
|
|
return func(lc fx.Lifecycle, mctx helpers.MetricsCtx, r repo.LockedRepo) (dtypes.MetadataDS, error) {
|
|
ctx := helpers.LifecycleCtx(mctx, lc)
|
|
mds, err := r.Datastore(ctx, "/metadata")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var logdir string
|
|
if !disableLog {
|
|
logdir = filepath.Join(r.Path(), "kvlog/metadata")
|
|
}
|
|
|
|
bds, err := backupds.Wrap(mds, logdir)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("opening backupds: %w", err)
|
|
}
|
|
|
|
lc.Append(fx.Hook{
|
|
OnStop: func(_ context.Context) error {
|
|
return bds.CloseLog()
|
|
},
|
|
})
|
|
|
|
return bds, nil
|
|
}
|
|
}
|