scheduling policies pluggable from commandline

schedulers/store.go

@@ -0,0 +1,30 @@
+package schedulers
+
+import "github.com/mesos/mesos-go/scheduler"
+
+// Names of different scheduling policies.
+const (
+	ff  = "first-fit"
+	bp  = "bin-packing"
+	mgm = "max-greedymins"
+	mm  = "max-min"
+)
+
+// Scheduler class factory
+var Schedulers map[string]scheduler.Scheduler = map[string]scheduler.Scheduler{
+	ff:  &FirstFit{base: base{}},
+	bp:  &BinPacking{base: base{}},
+	mgm: &MaxGreedyMins{base: base{}},
+	mm:  &MaxMin{base: base{}},
+}
+
+// build the scheduling policy with the options being applied
+func BuildSchedPolicy(s scheduler.Scheduler, opts ...schedPolicyOption) {
+	s.(ElectronScheduler).init(opts...)
+}
+
+func SchedFactory(schedPolicyName string, opts ...schedPolicyOption) scheduler.Scheduler {
+	s := Schedulers[schedPolicyName]
+	BuildSchedPolicy(s, opts...)
+	return s
+}