49 lines
1.3 KiB
Go
49 lines
1.3 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"
|
|
)
|
|
|
|
type taskSelector struct {
|
|
best []stores.StorageInfo //nolint: unused, structcheck
|
|
}
|
|
|
|
func newTaskSelector() *taskSelector {
|
|
return &taskSelector{}
|
|
}
|
|
|
|
func (s *taskSelector) Ok(ctx context.Context, task sealtasks.TaskType, spt abi.RegisteredSealProof, whnd *workerHandle) (bool, error) {
|
|
tasks, err := whnd.workerRpc.TaskTypes(ctx)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("getting supported worker task types: %w", err)
|
|
}
|
|
_, supported := tasks[task]
|
|
|
|
return supported, nil
|
|
}
|
|
|
|
func (s *taskSelector) Cmp(ctx context.Context, _ sealtasks.TaskType, a, b *workerHandle) (bool, error) {
|
|
atasks, err := a.workerRpc.TaskTypes(ctx)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("getting supported worker task types: %w", err)
|
|
}
|
|
btasks, err := b.workerRpc.TaskTypes(ctx)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("getting supported worker task types: %w", err)
|
|
}
|
|
if len(atasks) != len(btasks) {
|
|
return len(atasks) < len(btasks), nil // prefer workers which can do less
|
|
}
|
|
|
|
return a.utilization() < b.utilization(), nil
|
|
}
|
|
|
|
var _ WorkerSelector = &taskSelector{}
|