add client for cpumen plugin

Author: CMGS

resource/plugins/cpumem/client/node.go

@@ -0,0 +1,70 @@
+package client
+
+import (
+	"encoding/json"
+	"strconv"
+	"strings"
+
+	"github.com/projecteru2/core/resource/plugins/cpumem/types"
+)
+
+type cpuType interface {
+	string | int64
+}
+
+// NodeResourceRequest includes all possible fields passed by eru-core for editing node, it not parsed!
+type NodeResourceRequest[T cpuType] struct {
+	CPU        T        `json:"cpu"`
+	Share      int64    `json:"share"`
+	Memory     int64    `json:"memory"`
+	NUMA       []string `json:"numa-cpu"`
+	NUMAMemory []int64  `json:"numa-memory"`
+
+	NUMACPUMap    types.NUMA       `json:"-"`
+	NUMAMemoryMap types.NUMAMemory `json:"-"`
+}
+
+// NewNodeResourceRequest .
+func NewNodeResourceRequest[T cpuType](cpu T, share, memory int64, NUMACPUMap types.NUMA, NUMAMemoryMap types.NUMAMemory) (*NodeResourceRequest[T], error) {
+	r := &NodeResourceRequest[T]{
+		CPU:    cpu,
+		Share:  share,
+		Memory: memory,
+	}
+	tmpNUMA := map[int][]string{}
+	for cpuID, nodeID := range NUMACPUMap {
+		nID, err := strconv.Atoi(nodeID)
+		if err != nil {
+			return nil, err
+		}
+		if tmpNUMA[nID] == nil {
+			tmpNUMA[nID] = []string{cpuID}
+			continue
+		}
+		tmpNUMA[nID] = append(tmpNUMA[nID], cpuID)
+	}
+	r.NUMA = make([]string, len(tmpNUMA))
+	for nodeID, cpus := range tmpNUMA {
+		r.NUMA[nodeID] = strings.Join(cpus, ",")
+	}
+
+	tmpNUMAMemory := map[int]int64{}
+	for nodeID, memory := range NUMAMemoryMap {
+		nID, err := strconv.Atoi(nodeID)
+		if err != nil {
+			return nil, err
+		}
+		tmpNUMAMemory[nID] = memory
+	}
+	r.NUMAMemory = make([]int64, len(tmpNUMAMemory))
+	for nodeID, memory := range tmpNUMAMemory {
+		r.NUMAMemory[nodeID] = memory
+	}
+
+	return r, nil
+}
+
+// Encode .
+func (n NodeResourceRequest[T]) Encode() ([]byte, error) {
+	return json.Marshal(n)
+}

resource/plugins/cpumem/client/workload.go

@@ -0,0 +1,30 @@
+package client
+
+import "encoding/json"
+
+// WorkloadResourceRequest .
+type WorkloadResourceRequest struct {
+	CPUBind     bool    `json:"cpu-bind"`
+	KeepCPUBind bool    `json:"keep-cpu-bind"`
+	CPURequest  float64 `json:"cpu-request"`
+	CPULimit    float64 `json:"cpu-limit"`
+	MemRequest  int64   `json:"mem-request"`
+	MemLimit    int64   `json:"mem-limit"`
+}
+
+// NewWorkloadResourceRequest .
+func NewWorkloadResourceRequest(cpuBind, keepCPUBind bool, CPURequest, CPULimit float64, MemRequest, MemLimit int64) (*WorkloadResourceRequest, error) {
+	return &WorkloadResourceRequest{
+		CPUBind:     cpuBind,
+		KeepCPUBind: keepCPUBind,
+		CPURequest:  CPURequest,
+		CPULimit:    CPULimit,
+		MemRequest:  MemRequest,
+		MemLimit:    MemLimit,
+	}, nil
+}
+
+// Encode .
+func (w WorkloadResourceRequest) Encode() ([]byte, error) {
+	return json.Marshal(w)
+}