50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package sealer
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/storage/sealer/sealtasks"
|
|
"github.com/filecoin-project/lotus/storage/sealer/storiface"
|
|
)
|
|
|
|
type taskSelector struct {
|
|
best []storiface.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, bool, error) {
|
|
tasks, err := whnd.TaskTypes(ctx)
|
|
if err != nil {
|
|
return false, false, xerrors.Errorf("getting supported worker task types: %w", err)
|
|
}
|
|
_, supported := tasks[task]
|
|
|
|
return supported, false, nil
|
|
}
|
|
|
|
func (s *taskSelector) Cmp(ctx context.Context, _ sealtasks.TaskType, a, b *WorkerHandle) (bool, error) {
|
|
atasks, err := a.TaskTypes(ctx)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("getting supported worker task types: %w", err)
|
|
}
|
|
|
|
btasks, err := b.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{}
|