652 lines
15 KiB
Go
652 lines
15 KiB
Go
package stores
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"math/bits"
|
|
"math/rand"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
"time"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/extern/sector-storage/fsutil"
|
|
"github.com/filecoin-project/lotus/extern/sector-storage/storiface"
|
|
)
|
|
|
|
type StoragePath struct {
|
|
ID ID
|
|
Weight uint64
|
|
|
|
LocalPath string
|
|
|
|
CanSeal bool
|
|
CanStore bool
|
|
}
|
|
|
|
// LocalStorageMeta [path]/sectorstore.json
|
|
type LocalStorageMeta struct {
|
|
ID ID
|
|
|
|
// A high weight means data is more likely to be stored in this path
|
|
Weight uint64 // 0 = readonly
|
|
|
|
// Intermediate data for the sealing process will be stored here
|
|
CanSeal bool
|
|
|
|
// Finalized sectors that will be proved over time will be stored here
|
|
CanStore bool
|
|
}
|
|
|
|
// StorageConfig .lotusstorage/storage.json
|
|
type StorageConfig struct {
|
|
StoragePaths []LocalPath
|
|
}
|
|
|
|
type LocalPath struct {
|
|
Path string
|
|
}
|
|
|
|
type LocalStorage interface {
|
|
GetStorage() (StorageConfig, error)
|
|
SetStorage(func(*StorageConfig)) error
|
|
|
|
Stat(path string) (fsutil.FsStat, error)
|
|
|
|
// returns real disk usage for a file/directory
|
|
// os.ErrNotExit when file doesn't exist
|
|
DiskUsage(path string) (int64, error)
|
|
}
|
|
|
|
const MetaFile = "sectorstore.json"
|
|
|
|
type Local struct {
|
|
localStorage LocalStorage
|
|
index SectorIndex
|
|
urls []string
|
|
|
|
paths map[ID]*path
|
|
|
|
localLk sync.RWMutex
|
|
}
|
|
|
|
type path struct {
|
|
local string // absolute local path
|
|
|
|
reserved int64
|
|
reservations map[abi.SectorID]storiface.SectorFileType
|
|
}
|
|
|
|
func (p *path) stat(ls LocalStorage) (fsutil.FsStat, error) {
|
|
stat, err := ls.Stat(p.local)
|
|
if err != nil {
|
|
return fsutil.FsStat{}, xerrors.Errorf("stat %s: %w", p.local, err)
|
|
}
|
|
|
|
stat.Reserved = p.reserved
|
|
|
|
for id, ft := range p.reservations {
|
|
for _, fileType := range storiface.PathTypes {
|
|
if fileType&ft == 0 {
|
|
continue
|
|
}
|
|
|
|
sp := p.sectorPath(id, fileType)
|
|
|
|
used, err := ls.DiskUsage(sp)
|
|
if err == os.ErrNotExist {
|
|
p, ferr := tempFetchDest(sp, false)
|
|
if ferr != nil {
|
|
return fsutil.FsStat{}, ferr
|
|
}
|
|
|
|
used, err = ls.DiskUsage(p)
|
|
}
|
|
if err != nil {
|
|
log.Errorf("getting disk usage of '%s': %+v", p.sectorPath(id, fileType), err)
|
|
continue
|
|
}
|
|
|
|
stat.Reserved -= used
|
|
}
|
|
}
|
|
|
|
if stat.Reserved < 0 {
|
|
log.Warnf("negative reserved storage: p.reserved=%d, reserved: %d", p.reserved, stat.Reserved)
|
|
stat.Reserved = 0
|
|
}
|
|
|
|
stat.Available -= stat.Reserved
|
|
if stat.Available < 0 {
|
|
stat.Available = 0
|
|
}
|
|
|
|
return stat, err
|
|
}
|
|
|
|
func (p *path) sectorPath(sid abi.SectorID, fileType storiface.SectorFileType) string {
|
|
return filepath.Join(p.local, fileType.String(), storiface.SectorName(sid))
|
|
}
|
|
|
|
func NewLocal(ctx context.Context, ls LocalStorage, index SectorIndex, urls []string) (*Local, error) {
|
|
l := &Local{
|
|
localStorage: ls,
|
|
index: index,
|
|
urls: urls,
|
|
|
|
paths: map[ID]*path{},
|
|
}
|
|
return l, l.open(ctx)
|
|
}
|
|
|
|
func (st *Local) OpenPath(ctx context.Context, p string) error {
|
|
st.localLk.Lock()
|
|
defer st.localLk.Unlock()
|
|
|
|
mb, err := ioutil.ReadFile(filepath.Join(p, MetaFile))
|
|
if err != nil {
|
|
return xerrors.Errorf("reading storage metadata for %s: %w", p, err)
|
|
}
|
|
|
|
var meta LocalStorageMeta
|
|
if err := json.Unmarshal(mb, &meta); err != nil {
|
|
return xerrors.Errorf("unmarshalling storage metadata for %s: %w", p, err)
|
|
}
|
|
|
|
// TODO: Check existing / dedupe
|
|
|
|
out := &path{
|
|
local: p,
|
|
|
|
reserved: 0,
|
|
reservations: map[abi.SectorID]storiface.SectorFileType{},
|
|
}
|
|
|
|
fst, err := out.stat(st.localStorage)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = st.index.StorageAttach(ctx, StorageInfo{
|
|
ID: meta.ID,
|
|
URLs: st.urls,
|
|
Weight: meta.Weight,
|
|
CanSeal: meta.CanSeal,
|
|
CanStore: meta.CanStore,
|
|
}, fst)
|
|
if err != nil {
|
|
return xerrors.Errorf("declaring storage in index: %w", err)
|
|
}
|
|
|
|
if err := st.declareSectors(ctx, p, meta.ID, meta.CanStore); err != nil {
|
|
return err
|
|
}
|
|
|
|
st.paths[meta.ID] = out
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) open(ctx context.Context) error {
|
|
cfg, err := st.localStorage.GetStorage()
|
|
if err != nil {
|
|
return xerrors.Errorf("getting local storage config: %w", err)
|
|
}
|
|
|
|
for _, path := range cfg.StoragePaths {
|
|
err := st.OpenPath(ctx, path.Path)
|
|
if err != nil {
|
|
return xerrors.Errorf("opening path %s: %w", path.Path, err)
|
|
}
|
|
}
|
|
|
|
go st.reportHealth(ctx)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) Redeclare(ctx context.Context) error {
|
|
st.localLk.Lock()
|
|
defer st.localLk.Unlock()
|
|
|
|
for id, p := range st.paths {
|
|
mb, err := ioutil.ReadFile(filepath.Join(p.local, MetaFile))
|
|
if err != nil {
|
|
return xerrors.Errorf("reading storage metadata for %s: %w", p.local, err)
|
|
}
|
|
|
|
var meta LocalStorageMeta
|
|
if err := json.Unmarshal(mb, &meta); err != nil {
|
|
return xerrors.Errorf("unmarshalling storage metadata for %s: %w", p.local, err)
|
|
}
|
|
|
|
fst, err := p.stat(st.localStorage)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if id != meta.ID {
|
|
log.Errorf("storage path ID changed: %s; %s -> %s", p.local, id, meta.ID)
|
|
continue
|
|
}
|
|
|
|
err = st.index.StorageAttach(ctx, StorageInfo{
|
|
ID: id,
|
|
URLs: st.urls,
|
|
Weight: meta.Weight,
|
|
CanSeal: meta.CanSeal,
|
|
CanStore: meta.CanStore,
|
|
}, fst)
|
|
if err != nil {
|
|
return xerrors.Errorf("redeclaring storage in index: %w", err)
|
|
}
|
|
|
|
if err := st.declareSectors(ctx, p.local, meta.ID, meta.CanStore); err != nil {
|
|
return xerrors.Errorf("redeclaring sectors: %w", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) declareSectors(ctx context.Context, p string, id ID, primary bool) error {
|
|
for _, t := range storiface.PathTypes {
|
|
ents, err := ioutil.ReadDir(filepath.Join(p, t.String()))
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
if err := os.MkdirAll(filepath.Join(p, t.String()), 0755); err != nil { // nolint
|
|
return xerrors.Errorf("openPath mkdir '%s': %w", filepath.Join(p, t.String()), err)
|
|
}
|
|
|
|
continue
|
|
}
|
|
return xerrors.Errorf("listing %s: %w", filepath.Join(p, t.String()), err)
|
|
}
|
|
|
|
for _, ent := range ents {
|
|
if ent.Name() == FetchTempSubdir {
|
|
continue
|
|
}
|
|
|
|
sid, err := storiface.ParseSectorID(ent.Name())
|
|
if err != nil {
|
|
return xerrors.Errorf("parse sector id %s: %w", ent.Name(), err)
|
|
}
|
|
|
|
if err := st.index.StorageDeclareSector(ctx, id, sid, t, primary); err != nil {
|
|
return xerrors.Errorf("declare sector %d(t:%d) -> %s: %w", sid, t, id, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) reportHealth(ctx context.Context) {
|
|
// randomize interval by ~10%
|
|
interval := (HeartbeatInterval*100_000 + time.Duration(rand.Int63n(10_000))) / 100_000
|
|
|
|
for {
|
|
select {
|
|
case <-time.After(interval):
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
|
|
st.reportStorage(ctx)
|
|
}
|
|
}
|
|
|
|
func (st *Local) reportStorage(ctx context.Context) {
|
|
st.localLk.RLock()
|
|
|
|
toReport := map[ID]HealthReport{}
|
|
for id, p := range st.paths {
|
|
stat, err := p.stat(st.localStorage)
|
|
|
|
toReport[id] = HealthReport{
|
|
Stat: stat,
|
|
Err: err,
|
|
}
|
|
}
|
|
|
|
st.localLk.RUnlock()
|
|
|
|
for id, report := range toReport {
|
|
if err := st.index.StorageReportHealth(ctx, id, report); err != nil {
|
|
log.Warnf("error reporting storage health for %s (%+v): %+v", id, report, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (st *Local) Reserve(ctx context.Context, sid abi.SectorID, ssize abi.SectorSize, ft storiface.SectorFileType, storageIDs storiface.SectorPaths, overheadTab map[storiface.SectorFileType]int) (func(), error) {
|
|
st.localLk.Lock()
|
|
|
|
done := func() {}
|
|
deferredDone := func() { done() }
|
|
defer func() {
|
|
st.localLk.Unlock()
|
|
deferredDone()
|
|
}()
|
|
|
|
for _, fileType := range storiface.PathTypes {
|
|
if fileType&ft == 0 {
|
|
continue
|
|
}
|
|
|
|
id := ID(storiface.PathByType(storageIDs, fileType))
|
|
|
|
p, ok := st.paths[id]
|
|
if !ok {
|
|
return nil, errPathNotFound
|
|
}
|
|
|
|
stat, err := p.stat(st.localStorage)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("getting local storage stat: %w", err)
|
|
}
|
|
|
|
overhead := int64(overheadTab[fileType]) * int64(ssize) / storiface.FSOverheadDen
|
|
|
|
if stat.Available < overhead {
|
|
return nil, xerrors.Errorf("can't reserve %d bytes in '%s' (id:%s), only %d available", overhead, p.local, id, stat.Available)
|
|
}
|
|
|
|
p.reserved += overhead
|
|
|
|
prevDone := done
|
|
done = func() {
|
|
prevDone()
|
|
|
|
st.localLk.Lock()
|
|
defer st.localLk.Unlock()
|
|
|
|
p.reserved -= overhead
|
|
}
|
|
}
|
|
|
|
deferredDone = func() {}
|
|
return done, nil
|
|
}
|
|
|
|
func (st *Local) AcquireSector(ctx context.Context, sid abi.SectorID, ssize abi.SectorSize, existing storiface.SectorFileType, allocate storiface.SectorFileType, pathType storiface.PathType, op storiface.AcquireMode) (storiface.SectorPaths, storiface.SectorPaths, error) {
|
|
if existing|allocate != existing^allocate {
|
|
return storiface.SectorPaths{}, storiface.SectorPaths{}, xerrors.New("can't both find and allocate a sector")
|
|
}
|
|
|
|
st.localLk.RLock()
|
|
defer st.localLk.RUnlock()
|
|
|
|
var out storiface.SectorPaths
|
|
var storageIDs storiface.SectorPaths
|
|
|
|
for _, fileType := range storiface.PathTypes {
|
|
if fileType&existing == 0 {
|
|
continue
|
|
}
|
|
|
|
si, err := st.index.StorageFindSector(ctx, sid, fileType, ssize, false)
|
|
if err != nil {
|
|
log.Warnf("finding existing sector %d(t:%d) failed: %+v", sid, fileType, err)
|
|
continue
|
|
}
|
|
|
|
for _, info := range si {
|
|
p, ok := st.paths[info.ID]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if p.local == "" { // TODO: can that even be the case?
|
|
continue
|
|
}
|
|
|
|
spath := p.sectorPath(sid, fileType)
|
|
storiface.SetPathByType(&out, fileType, spath)
|
|
storiface.SetPathByType(&storageIDs, fileType, string(info.ID))
|
|
|
|
existing ^= fileType
|
|
break
|
|
}
|
|
}
|
|
|
|
for _, fileType := range storiface.PathTypes {
|
|
if fileType&allocate == 0 {
|
|
continue
|
|
}
|
|
|
|
sis, err := st.index.StorageBestAlloc(ctx, fileType, ssize, pathType)
|
|
if err != nil {
|
|
return storiface.SectorPaths{}, storiface.SectorPaths{}, xerrors.Errorf("finding best storage for allocating : %w", err)
|
|
}
|
|
|
|
var best string
|
|
var bestID ID
|
|
|
|
for _, si := range sis {
|
|
p, ok := st.paths[si.ID]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if p.local == "" { // TODO: can that even be the case?
|
|
continue
|
|
}
|
|
|
|
if (pathType == storiface.PathSealing) && !si.CanSeal {
|
|
continue
|
|
}
|
|
|
|
if (pathType == storiface.PathStorage) && !si.CanStore {
|
|
continue
|
|
}
|
|
|
|
// TODO: Check free space
|
|
|
|
best = p.sectorPath(sid, fileType)
|
|
bestID = si.ID
|
|
break
|
|
}
|
|
|
|
if best == "" {
|
|
return storiface.SectorPaths{}, storiface.SectorPaths{}, xerrors.Errorf("couldn't find a suitable path for a sector")
|
|
}
|
|
|
|
storiface.SetPathByType(&out, fileType, best)
|
|
storiface.SetPathByType(&storageIDs, fileType, string(bestID))
|
|
allocate ^= fileType
|
|
}
|
|
|
|
return out, storageIDs, nil
|
|
}
|
|
|
|
func (st *Local) Local(ctx context.Context) ([]StoragePath, error) {
|
|
st.localLk.RLock()
|
|
defer st.localLk.RUnlock()
|
|
|
|
var out []StoragePath
|
|
for id, p := range st.paths {
|
|
if p.local == "" {
|
|
continue
|
|
}
|
|
|
|
si, err := st.index.StorageInfo(ctx, id)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("get storage info for %s: %w", id, err)
|
|
}
|
|
|
|
out = append(out, StoragePath{
|
|
ID: id,
|
|
Weight: si.Weight,
|
|
LocalPath: p.local,
|
|
CanSeal: si.CanSeal,
|
|
CanStore: si.CanStore,
|
|
})
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
func (st *Local) Remove(ctx context.Context, sid abi.SectorID, typ storiface.SectorFileType, force bool) error {
|
|
if bits.OnesCount(uint(typ)) != 1 {
|
|
return xerrors.New("delete expects one file type")
|
|
}
|
|
|
|
si, err := st.index.StorageFindSector(ctx, sid, typ, 0, false)
|
|
if err != nil {
|
|
return xerrors.Errorf("finding existing sector %d(t:%d) failed: %w", sid, typ, err)
|
|
}
|
|
|
|
if len(si) == 0 && !force {
|
|
return xerrors.Errorf("can't delete sector %v(%d), not found", sid, typ)
|
|
}
|
|
|
|
for _, info := range si {
|
|
if err := st.removeSector(ctx, sid, typ, info.ID); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) RemoveCopies(ctx context.Context, sid abi.SectorID, typ storiface.SectorFileType) error {
|
|
if bits.OnesCount(uint(typ)) != 1 {
|
|
return xerrors.New("delete expects one file type")
|
|
}
|
|
|
|
si, err := st.index.StorageFindSector(ctx, sid, typ, 0, false)
|
|
if err != nil {
|
|
return xerrors.Errorf("finding existing sector %d(t:%d) failed: %w", sid, typ, err)
|
|
}
|
|
|
|
var hasPrimary bool
|
|
for _, info := range si {
|
|
if info.Primary {
|
|
hasPrimary = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !hasPrimary {
|
|
log.Warnf("RemoveCopies: no primary copies of sector %v (%s), not removing anything", sid, typ)
|
|
return nil
|
|
}
|
|
|
|
for _, info := range si {
|
|
if info.Primary {
|
|
continue
|
|
}
|
|
|
|
if err := st.removeSector(ctx, sid, typ, info.ID); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) removeSector(ctx context.Context, sid abi.SectorID, typ storiface.SectorFileType, storage ID) error {
|
|
p, ok := st.paths[storage]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
if p.local == "" { // TODO: can that even be the case?
|
|
return nil
|
|
}
|
|
|
|
if err := st.index.StorageDropSector(ctx, storage, sid, typ); err != nil {
|
|
return xerrors.Errorf("dropping sector from index: %w", err)
|
|
}
|
|
|
|
spath := p.sectorPath(sid, typ)
|
|
log.Infof("remove %s", spath)
|
|
|
|
if err := os.RemoveAll(spath); err != nil {
|
|
log.Errorf("removing sector (%v) from %s: %+v", sid, spath, err)
|
|
}
|
|
|
|
st.reportStorage(ctx) // report freed space
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *Local) MoveStorage(ctx context.Context, s abi.SectorID, ssize abi.SectorSize, types storiface.SectorFileType) error {
|
|
dest, destIds, err := st.AcquireSector(ctx, s, ssize, storiface.FTNone, types, storiface.PathStorage, storiface.AcquireMove)
|
|
if err != nil {
|
|
return xerrors.Errorf("acquire dest storage: %w", err)
|
|
}
|
|
|
|
src, srcIds, err := st.AcquireSector(ctx, s, ssize, types, storiface.FTNone, storiface.PathStorage, storiface.AcquireMove)
|
|
if err != nil {
|
|
return xerrors.Errorf("acquire src storage: %w", err)
|
|
}
|
|
|
|
for _, fileType := range storiface.PathTypes {
|
|
if fileType&types == 0 {
|
|
continue
|
|
}
|
|
|
|
sst, err := st.index.StorageInfo(ctx, ID(storiface.PathByType(srcIds, fileType)))
|
|
if err != nil {
|
|
return xerrors.Errorf("failed to get source storage info: %w", err)
|
|
}
|
|
|
|
dst, err := st.index.StorageInfo(ctx, ID(storiface.PathByType(destIds, fileType)))
|
|
if err != nil {
|
|
return xerrors.Errorf("failed to get source storage info: %w", err)
|
|
}
|
|
|
|
if sst.ID == dst.ID {
|
|
log.Debugf("not moving %v(%d); src and dest are the same", s, fileType)
|
|
continue
|
|
}
|
|
|
|
if sst.CanStore {
|
|
log.Debugf("not moving %v(%d); source supports storage", s, fileType)
|
|
continue
|
|
}
|
|
|
|
log.Debugf("moving %v(%d) to storage: %s(se:%t; st:%t) -> %s(se:%t; st:%t)", s, fileType, sst.ID, sst.CanSeal, sst.CanStore, dst.ID, dst.CanSeal, dst.CanStore)
|
|
|
|
if err := st.index.StorageDropSector(ctx, ID(storiface.PathByType(srcIds, fileType)), s, fileType); err != nil {
|
|
return xerrors.Errorf("dropping source sector from index: %w", err)
|
|
}
|
|
|
|
if err := move(storiface.PathByType(src, fileType), storiface.PathByType(dest, fileType)); err != nil {
|
|
// TODO: attempt some recovery (check if src is still there, re-declare)
|
|
return xerrors.Errorf("moving sector %v(%d): %w", s, fileType, err)
|
|
}
|
|
|
|
if err := st.index.StorageDeclareSector(ctx, ID(storiface.PathByType(destIds, fileType)), s, fileType, true); err != nil {
|
|
return xerrors.Errorf("declare sector %d(t:%d) -> %s: %w", s, fileType, ID(storiface.PathByType(destIds, fileType)), err)
|
|
}
|
|
}
|
|
|
|
st.reportStorage(ctx) // report space use changes
|
|
|
|
return nil
|
|
}
|
|
|
|
var errPathNotFound = xerrors.Errorf("fsstat: path not found")
|
|
|
|
func (st *Local) FsStat(ctx context.Context, id ID) (fsutil.FsStat, error) {
|
|
st.localLk.RLock()
|
|
defer st.localLk.RUnlock()
|
|
|
|
p, ok := st.paths[id]
|
|
if !ok {
|
|
return fsutil.FsStat{}, errPathNotFound
|
|
}
|
|
|
|
return p.stat(st.localStorage)
|
|
}
|
|
|
|
var _ Store = &Local{}
|