feat: add capacity for kubelet

IPADSIntern-MiniK8s · May 29, 2023 · 175f106 · 175f106
1 parent 6a38b6a
commit 175f106
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/cmd/kubelet.go b/cmd/kubelet.go
@@ -31,6 +31,8 @@ var KubeletConfig = kubelet.Config{
 	IP:            "192.168.1.14",
 	Labels:        map[string]string{},
 	ListenAddr:    "localhost:10250",
+	CPU:	"2",
+	Memory:	"3Gi",
 }
 
 func initConfig() {

diff --git a/config/kubelet-config.yaml b/config/kubelet-config.yaml
@@ -8,4 +8,6 @@ FlannelSubnet : 10.2.9.1/24
 IP            : 192.168.1.14
 Labels        :
   - resource : gpu
-ListenAddr    : 192.168.1.14:10250
+ListenAddr    : 192.168.1.14:10250
+CPU:    "2"
+Memory: "3Gi"
diff --git a/pkg/kubelet/kubelet.go b/pkg/kubelet/kubelet.go
@@ -20,6 +20,8 @@ type Kubelet struct {
 	IP            string            //host ip
 	Labels        map[string]string // for nodeSelector
 	ListenAddr    string            //MetricsServer listen for auto-scaling
+	CPU		string
+	Memory		string
 	Server        *metricsserver.MetricsServer
 }
 
@@ -30,6 +32,8 @@ func NewKubelet(config Config) *Kubelet {
 		IP:            config.IP,
 		Labels:        config.Labels,
 		ListenAddr:    config.ListenAddr,
+		CPU:		config.CPU,
+		Memory:		config.Memory,
 		Server:        metricsserver.NewMetricsServer(),
 	}
 }
@@ -48,6 +52,10 @@ func (kl *Kubelet) register() {
 			PodCIDR:       kl.FlannelSubnet,
 		},
 		Status: apiobject.NodeStatus{
+			Capability:map[string]string{
+				"cpu":kl.CPU,
+				"memory":kl.Memory,
+			}
 			Addresses: []apiobject.Address{
 				{
 					Type:    "InternalIP",