97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
|
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 moveSelector struct {
|
||
|
|
index stores.SectorIndex
|
||
|
|
sector abi.SectorID
|
||
|
|
alloc storiface.SectorFileType
|
||
|
|
destPtype storiface.PathType
|
||
|
|
}
|
||
|
|
|
||
|
|
func newMoveSelector(index stores.SectorIndex, sector abi.SectorID, alloc storiface.SectorFileType, destPtype storiface.PathType) *moveSelector {
|
||
|
|
return &moveSelector{
|
||
|
|
index: index,
|
||
|
|
sector: sector,
|
||
|
|
alloc: alloc,
|
||
|
|
destPtype: destPtype,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *moveSelector) Ok(ctx context.Context, task sealtasks.TaskType, spt abi.RegisteredSealProof, whnd *WorkerHandle) (bool, bool, error) {
|
||
|
|
tasks, err := whnd.TaskTypes(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return false, false, xerrors.Errorf("getting supported worker task types: %w", err)
|
||
|
|
}
|
||
|
|
if _, supported := tasks[task]; !supported {
|
||
|
|
return false, false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
paths, err := whnd.workerRpc.Paths(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return false, false, xerrors.Errorf("getting worker paths: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
workerPaths := map[storiface.ID]int{}
|
||
|
|
for _, path := range paths {
|
||
|
|
workerPaths[path.ID] = 0
|
||
|
|
}
|
||
|
|
|
||
|
|
ssize, err := spt.SectorSize()
|
||
|
|
if err != nil {
|
||
|
|
return false, false, xerrors.Errorf("getting sector size: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// note: allowFetch is always false here, because we want to find workers with
|
||
|
|
// the sector available locally
|
||
|
|
preferred, err := s.index.StorageFindSector(ctx, s.sector, s.alloc, ssize, false)
|
||
|
|
if err != nil {
|
||
|
|
return false, false, xerrors.Errorf("finding preferred storage: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, info := range preferred {
|
||
|
|
if _, ok := workerPaths[info.ID]; ok {
|
||
|
|
workerPaths[info.ID]++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
best, err := s.index.StorageBestAlloc(ctx, s.alloc, ssize, s.destPtype)
|
||
|
|
if err != nil {
|
||
|
|
return false, false, xerrors.Errorf("finding best dest storage: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var ok bool
|
||
|
|
|
||
|
|
for _, info := range best {
|
||
|
|
if n, has := workerPaths[info.ID]; has {
|
||
|
|
ok = true
|
||
|
|
|
||
|
|
// if the worker has a local path with the sector already in it
|
||
|
|
// prefer that worker; This usually meant that the move operation is
|
||
|
|
// either a no-op because the sector is already in the correct path,
|
||
|
|
// or the move a local move.
|
||
|
|
if n > 0 {
|
||
|
|
return true, true, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return ok, false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *moveSelector) Cmp(ctx context.Context, task sealtasks.TaskType, a, b *WorkerHandle) (bool, error) {
|
||
|
|
return a.Utilization() < b.Utilization(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ WorkerSelector = &moveSelector{}
|