store node init resource

projecteru2 · Nov 16, 2018 · 275e6f2 · 275e6f2
1 parent 1039dc8
commit 275e6f2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 15 deletions.
diff --git a/store/etcdv3/node.go b/store/etcdv3/node.go
@@ -83,13 +83,15 @@ func (m *Mercury) AddNode(ctx context.Context, name, endpoint, podname, ca, cert
 	}
 
 	node := &types.Node{
-		Name:      name,
-		Endpoint:  endpoint,
-		Podname:   podname,
-		CPU:       cpumap,
-		MemCap:    memcap,
-		Available: true,
-		Labels:    labels,
+		Name:       name,
+		Endpoint:   endpoint,
+		Podname:    podname,
+		CPU:        cpumap,
+		MemCap:     memcap,
+		InitCPU:    cpumap,
+		InitMemCap: memcap,
+		Available:  true,
+		Labels:     labels,
 	}
 
 	bytes, err := json.Marshal(node)

diff --git a/types/node.go b/types/node.go
@@ -52,14 +52,16 @@ func (c CPUMap) Sub(q CPUMap) {
 
 // Node store node info
 type Node struct {
-	Name      string            `json:"name"`
-	Endpoint  string            `json:"endpoint"`
-	Podname   string            `json:"podname"`
-	Available bool              `json:"available"`
-	CPU       CPUMap            `json:"cpu"`
-	MemCap    int64             `json:"memcap"`
-	Labels    map[string]string `json:"labels"`
-	Engine    *engineapi.Client `json:"-"`
+	Name       string            `json:"name"`
+	Endpoint   string            `json:"endpoint"`
+	Podname    string            `json:"podname"`
+	Available  bool              `json:"available"`
+	CPU        CPUMap            `json:"cpu"`
+	MemCap     int64             `json:"memcap"`
+	Labels     map[string]string `json:"labels"`
+	InitCPU    CPUMap            `json:"init_cpu"`
+	InitMemCap int64             `json:"init_memcap"`
+	Engine     *engineapi.Client `json:"-"`
 }
 
 // Info show node info