node scheduler

Author: sp-yduck

cloud/interfaces.go

@@ -46,7 +46,7 @@ type ClusterSettter interface {
 // MachineGetter is an interface which can get machine information.
 type MachineGetter interface {
 	Client
-	GetScheduler() *scheduler.Scheduler
+	GetScheduler(client *proxmox.Service) *scheduler.Scheduler
 	Name() string
 	Namespace() string
 	// Zone() string

cloud/scheduler/framework/cycle_state.go

@@ -0,0 +1,4 @@
+package framework
+
+type CycleState struct {
+}

cloud/scheduler/framework/interface.go

@@ -0,0 +1,22 @@
+package framework
+
+import (
+	"context"
+
+	"github.com/sp-yduck/proxmox-go/api"
+)
+
+type Plugin interface {
+	// return plugin name
+	Name() string
+}
+
+type NodeFilterPlugin interface {
+	Plugin
+	Filter(ctx context.Context, state *CycleState, config api.VirtualMachineCreateOptions, nodeInfo *NodeInfo) *Status
+}
+
+type NodeScorePlugin interface {
+	Plugin
+	Score(ctx context.Context, state *CycleState, config api.VirtualMachineCreateOptions, nodeInfo *NodeInfo) (int64, *Status)
+}

cloud/scheduler/framework/types.go

@@ -0,0 +1,87 @@
+package framework
+
+import (
+	"context"
+
+	"github.com/sp-yduck/proxmox-go/api"
+	"github.com/sp-yduck/proxmox-go/proxmox"
+)
+
+type Status struct {
+	code         int
+	reasons      []string
+	err          error
+	failedPlugin string
+}
+
+func NewStatus() *Status {
+	return &Status{code: 0}
+}
+
+func (s *Status) Code() int {
+	return s.code
+}
+
+func (s *Status) SetCode(code int) {
+	s.code = code
+}
+
+func (s *Status) Reasons() []string {
+	if s.err != nil {
+		return append([]string{s.err.Error()}, s.reasons...)
+	}
+	return s.reasons
+}
+
+func (s *Status) FailedPlugin() string {
+	return s.failedPlugin
+}
+
+func (s *Status) SetFailedPlugin(name string) {
+	s.failedPlugin = name
+}
+
+func (s *Status) IsSuccess() bool {
+	return s.code == 0
+}
+
+func (s *Status) Error() error {
+	return s.err
+}
+
+// NodeInfo is node level aggregated information
+type NodeInfo struct {
+	node *api.Node
+
+	// qemus assigned to the node
+	qemus []*api.VirtualMachine
+}
+
+func GetNodeInfoList(ctx context.Context, client *proxmox.Service) ([]*NodeInfo, error) {
+	nodes, err := client.Nodes(ctx)
+	if err != nil {
+		return nil, err
+	}
+	nodeInfos := []*NodeInfo{}
+	for _, node := range nodes {
+		qemus, err := client.RESTClient().GetVirtualMachines(ctx, node.Node)
+		if err != nil {
+			return nil, err
+		}
+		nodeInfos = append(nodeInfos, &NodeInfo{node: node, qemus: qemus})
+	}
+	return nodeInfos, nil
+}
+
+func (n NodeInfo) Node() *api.Node {
+	return n.node
+}
+
+// NodeScoreList declares a list of nodes and their scores.
+type NodeScoreList []NodeScore
+
+// NodeScore is a struct with node name and score.
+type NodeScore struct {
+	Name  string
+	Score int64
+}

cloud/scheduler/plugins/node/names/names.go

@@ -0,0 +1,12 @@
+package names
+
+// node plugins
+const (
+	// filter plugins
+
+	NodeName = "NodeName"
+
+	// score plugins
+
+	Random = "Random"
+)

cloud/scheduler/plugins/node/nodename/node_name.go

@@ -0,0 +1,37 @@
+package nodename
+
+import (
+	"context"
+
+	"github.com/sp-yduck/proxmox-go/api"
+
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/framework"
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/plugins/node/names"
+)
+
+type NodeName struct{}
+
+var _ framework.NodeFilterPlugin = &NodeName{}
+
+const (
+	Name      = names.NodeName
+	ErrReason = "node didn't match the requested node name"
+)
+
+func (pl *NodeName) Name() string {
+	return Name
+}
+
+func (pl *NodeName) Filter(ctx context.Context, _ *framework.CycleState, config api.VirtualMachineCreateOptions, nodeInfo *framework.NodeInfo) *framework.Status {
+	if !Fits(config, nodeInfo) {
+		status := framework.NewStatus()
+		status.SetCode(1)
+		return status
+	}
+	return &framework.Status{}
+}
+
+// return true if config.Node is empty or match with node name
+func Fits(config api.VirtualMachineCreateOptions, nodeInfo *framework.NodeInfo) bool {
+	return config.Node == "" || config.Node == nodeInfo.Node().Node
+}

cloud/scheduler/plugins/node/random/random.go

@@ -0,0 +1,32 @@
+package random
+
+import (
+	"context"
+	"math/rand"
+	"time"
+
+	"github.com/sp-yduck/proxmox-go/api"
+
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/framework"
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/plugins/node/names"
+)
+
+type Random struct{}
+
+var _ framework.NodeScorePlugin = &Random{}
+
+const (
+	Name = names.Random
+)
+
+func (pl *Random) Name() string {
+	return Name
+}
+
+// return random score: 0 <= n < 100
+func (pl *Random) Score(ctx context.Context, state *framework.CycleState, config api.VirtualMachineCreateOptions, nodeInfo *framework.NodeInfo) (int64, *framework.Status) {
+	src := rand.NewSource(time.Now().Unix())
+	r := rand.New(src)
+	score := r.Int63n(100)
+	return score, nil
+}

cloud/scheduler/plugins/registry.go

@@ -0,0 +1,15 @@
+package plugins
+
+import (
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/framework"
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/plugins/node/nodename"
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/scheduler/plugins/node/random"
+)
+
+func NewNodeFilterPlugins() []framework.NodeFilterPlugin {
+	return []framework.NodeFilterPlugin{&nodename.NodeName{}}
+}
+
+func NewNodeScorePlugins() []framework.NodeScorePlugin {
+	return []framework.NodeScorePlugin{&random.Random{}}
+}