package sectorstorage import ( "context" "golang.org/x/xerrors" "github.com/filecoin-project/go-state-types/abi" "github.com/filecoin-project/lotus/extern/sector-storage/sealtasks" "github.com/filecoin-project/lotus/extern/sector-storage/stores" "github.com/filecoin-project/lotus/extern/sector-storage/storiface" ) type allocSelector struct { index stores.SectorIndex alloc storiface.SectorFileType ptype storiface.PathType } func newAllocSelector(index stores.SectorIndex, alloc storiface.SectorFileType, ptype storiface.PathType) *allocSelector { return &allocSelector{ index: index, alloc: alloc, ptype: ptype, } } func (s *allocSelector) Ok(ctx context.Context, task sealtasks.TaskType, spt abi.RegisteredSealProof, whnd *WorkerHandle) (bool, error) { tasks, err := whnd.TaskTypes(ctx) if err != nil { return false, xerrors.Errorf("getting supported worker task types: %w", err) } if _, supported := tasks[task]; !supported { return false, nil } paths, err := whnd.workerRpc.Paths(ctx) if err != nil { return false, xerrors.Errorf("getting worker paths: %w", err) } have := map[storiface.ID]struct{}{} for _, path := range paths { have[path.ID] = struct{}{} } ssize, err := spt.SectorSize() if err != nil { return false, xerrors.Errorf("getting sector size: %w", err) } best, err := s.index.StorageBestAlloc(ctx, s.alloc, ssize, s.ptype) if err != nil { return false, xerrors.Errorf("finding best alloc storage: %w", err) } for _, info := range best { if _, ok := have[info.ID]; ok { return true, nil } } return false, nil } func (s *allocSelector) Cmp(ctx context.Context, task sealtasks.TaskType, a, b *WorkerHandle) (bool, error) { return a.Utilization() < b.Utilization(), nil } var _ WorkerSelector = &allocSelector{}