mirror of https://github.com/docker/docs.git
38 lines
931 B
Go
38 lines
931 B
Go
package strategy
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"github.com/docker/swarm/cluster"
|
|
"github.com/docker/swarm/scheduler/node"
|
|
)
|
|
|
|
// SpreadPlacementStrategy places a container on the node with the fewest running containers.
|
|
type SpreadPlacementStrategy struct {
|
|
}
|
|
|
|
// Initialize a SpreadPlacementStrategy.
|
|
func (p *SpreadPlacementStrategy) Initialize() error {
|
|
return nil
|
|
}
|
|
|
|
// Name returns the name of the strategy.
|
|
func (p *SpreadPlacementStrategy) Name() string {
|
|
return "spread"
|
|
}
|
|
|
|
// RankAndSort sorts nodes based on the spread strategy applied to the container config.
|
|
func (p *SpreadPlacementStrategy) RankAndSort(config *cluster.ContainerConfig, nodes []*node.Node) ([]*node.Node, error) {
|
|
weightedNodes, err := weighNodes(config, nodes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sort.Sort(weightedNodes)
|
|
output := make([]*node.Node, len(weightedNodes))
|
|
for i, n := range weightedNodes {
|
|
output[i] = n.Node
|
|
}
|
|
return output, nil
|
|
}
|