improve scheduling framework so one controller can run multiple scheduler in parallel

Author: sp-yduck

cloud/scheduler/framework/cycle_state.go

@@ -1,4 +1,68 @@
 package framework
 
+import (
+	"github.com/sp-yduck/proxmox-go/api"
+	"github.com/sp-yduck/proxmox-go/proxmox"
+)
+
 type CycleState struct {
+	completed bool
+	err       error
+	result    SchedulerResult
+}
+
+type SchedulerResult struct {
+	vmid     int
+	node     string
+	instance *proxmox.VirtualMachine
+}
+
+func NewCycleState() CycleState {
+	return CycleState{completed: false, err: nil}
+}
+
+func (c *CycleState) SetComplete() {
+	c.completed = true
+}
+
+func (c *CycleState) IsCompleted() bool {
+	return c.completed
+}
+
+func (c *CycleState) SetError(err error) {
+	c.err = err
+}
+
+func (c *CycleState) Error() error {
+	return c.err
+}
+
+func (c *CycleState) QEMU() *api.VirtualMachine {
+	return c.result.instance.VM
+}
+
+func (c *CycleState) UpdateState(completed bool, err error, result *SchedulerResult) {
+	c.completed = completed
+	c.err = err
+	c.result = *result
+}
+
+func NewSchedulerResult(vmid int, node string, instance *proxmox.VirtualMachine) SchedulerResult {
+	return SchedulerResult{vmid: vmid, node: node, instance: instance}
+}
+
+func (c *CycleState) Result() SchedulerResult {
+	return c.result
+}
+
+func (r *SchedulerResult) Node() string {
+	return r.node
+}
+
+func (r *SchedulerResult) VMID() int {
+	return r.vmid
+}
+
+func (r *SchedulerResult) Instance() *proxmox.VirtualMachine {
+	return r.instance
 }

cloud/scheduler/plugins/registry.go

@@ -1,8 +1,6 @@
 package plugins
 
 import (
-	"fmt"
-
 	"github.com/sp-yduck/proxmox-go/proxmox"
 
 	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/framework"
@@ -26,11 +24,6 @@ func NewNodeScorePlugins() []framework.NodeScorePlugin {
 	return []framework.NodeScorePlugin{&random.Random{}}
 }
 
-func NewVMIDPlugin(client *proxmox.Service, name string) (framework.VMIDPlugin, error) {
-	plugins := VMIDPlugins(client)
-	plugin, ok := plugins[name]
-	if !ok {
-		return nil, fmt.Errorf("vmid plugin %s not found", name)
-	}
-	return plugin, nil
+func NewVMIDPlugins() []framework.VMIDPlugin {
+	return []framework.VMIDPlugin{&nextid.NextID{}}
 }

cloud/scheduler/queue/queue.go

@@ -0,0 +1,55 @@
+package queue
+
+import (
+	"context"
+	"sync"
+
+	"github.com/sp-yduck/proxmox-go/api"
+)
+
+type SchedulingQueue struct {
+	activeQ []*qemuSpec
+	lock    *sync.Cond
+}
+
+func New() *SchedulingQueue {
+	return &SchedulingQueue{
+		activeQ: []*qemuSpec{},
+		lock:    sync.NewCond(&sync.Mutex{}),
+	}
+}
+
+// qemu create option with context
+type qemuSpec struct {
+	ctx    context.Context
+	config *api.VirtualMachineCreateOptions
+}
+
+// add new qemuSpec to queue
+func (s *SchedulingQueue) Add(ctx context.Context, config *api.VirtualMachineCreateOptions) {
+	s.lock.L.Lock()
+	defer s.lock.L.Unlock()
+	s.activeQ = append(s.activeQ, &qemuSpec{ctx: ctx, config: config})
+	s.lock.Signal()
+}
+
+// return next qemuSpec
+func (s *SchedulingQueue) NextQEMU() *qemuSpec {
+	// wait
+	s.lock.L.Lock()
+	for len(s.activeQ) == 0 {
+		s.lock.Wait()
+	}
+	spec := s.activeQ[0]
+	s.activeQ = s.activeQ[1:]
+	s.lock.L.Unlock()
+	return spec
+}
+
+func (s *qemuSpec) Config() *api.VirtualMachineCreateOptions {
+	return s.config
+}
+
+func (s *qemuSpec) Context() context.Context {
+	return s.ctx
+}