github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-27 13:37:30 +00:00
Code Issues Projects Releases Wiki Activity

Move scheduler to own package.

Browse Source
This commit is contained in:
Daniel Smith 2014-06-28 14:21:08 -07:00
parent 18cba57863
commit 21e63cf75a
2 changed files with 24 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pkg/registry/scheduler.go → pkg/scheduler/scheduler.go
View File

@ -24,26 +24,36 @@ import (
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels" "github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
) )
// Anything that can list minions for a scheduler.
type MinionLister interface {
List() (machines []string, err error)
}
// Make a MinionLister from a []string
type StringMinionLister []string
func (s StringMinionLister) List() ([]string, error) {
return []string(s), nil
}
// Scheduler is an interface implemented by things that know how to schedule pods onto machines. // Scheduler is an interface implemented by things that know how to schedule pods onto machines.
type Scheduler interface { type Scheduler interface {
Schedule(api.Pod) (string, error) Schedule(api.Pod, MinionLister) (string, error)
} }
// RandomScheduler choses machines uniformly at random. // RandomScheduler choses machines uniformly at random.
type RandomScheduler struct { type RandomScheduler struct {
machines MinionRegistry
random rand.Rand random rand.Rand
} }
func MakeRandomScheduler(machines MinionRegistry, random rand.Rand) Scheduler { func MakeRandomScheduler(random rand.Rand) Scheduler {
return &RandomScheduler{ return &RandomScheduler{
machines: machines,
random: random, random: random,
} }
} }
func (s *RandomScheduler) Schedule(pod api.Pod) (string, error) { func (s *RandomScheduler) Schedule(pod api.Pod, minionLister MinionLister) (string, error) {
machines, err := s.machines.List() machines, err := minionLister.List()
if err != nil { if err != nil {
return "", err return "", err
} }
@ -52,19 +62,17 @@ func (s *RandomScheduler) Schedule(pod api.Pod) (string, error) {
// RoundRobinScheduler chooses machines in order. // RoundRobinScheduler chooses machines in order.
type RoundRobinScheduler struct { type RoundRobinScheduler struct {
machines MinionRegistry
currentIndex int currentIndex int
} }
func MakeRoundRobinScheduler(machines MinionRegistry) Scheduler { func MakeRoundRobinScheduler() Scheduler {
return &RoundRobinScheduler{ return &RoundRobinScheduler{
machines: machines,
currentIndex: -1, currentIndex: -1,
} }
} }
func (s *RoundRobinScheduler) Schedule(pod api.Pod) (string, error) { func (s *RoundRobinScheduler) Schedule(pod api.Pod, minionLister MinionLister) (string, error) {
machines, err := s.machines.List() machines, err := minionLister.List()
if err != nil { if err != nil {
return "", err return "", err
} }
@ -74,14 +82,12 @@ func (s *RoundRobinScheduler) Schedule(pod api.Pod) (string, error) {
} }
type FirstFitScheduler struct { type FirstFitScheduler struct {
machines MinionRegistry
registry PodRegistry registry PodRegistry
random *rand.Rand random *rand.Rand
} }
func MakeFirstFitScheduler(machines MinionRegistry, registry PodRegistry, random *rand.Rand) Scheduler { func MakeFirstFitScheduler(registry PodRegistry, random *rand.Rand) Scheduler {
return &FirstFitScheduler{ return &FirstFitScheduler{
machines: machines,
registry: registry, registry: registry,
random: random, random: random,
} }
@ -98,8 +104,8 @@ func (s *FirstFitScheduler) containsPort(pod api.Pod, port api.Port) bool {
return false return false
} }
func (s *FirstFitScheduler) Schedule(pod api.Pod) (string, error) { func (s *FirstFitScheduler) Schedule(pod api.Pod, minionLister MinionLister) (string, error) {
machines, err := s.machines.List() machines, err := minionLister.List()
if err != nil { if err != nil {
return "", err return "", err
} }

0
pkg/registry/scheduler_test.go → pkg/scheduler/scheduler_test.go
View File