add zone in rpc node

projecteru2 · Jan 18, 2017 · 732d8ba · 732d8ba
1 parent be055d9
commit 732d8ba
Show file tree

Hide file tree

Showing 8 changed files with 167 additions and 142 deletions.
diff --git a/cluster/calcium/meta.go b/cluster/calcium/meta.go
@@ -67,3 +67,7 @@ func (c *calcium) GetContainer(id string) (*types.Container, error) {
 func (c *calcium) GetContainers(ids []string) ([]*types.Container, error) {
 	return c.store.GetContainers(ids)
 }
+
+func (c *calcium) GetZone() string {
+	return c.config.Zone
+}
diff --git a/cluster/cluster.go b/cluster/cluster.go
@@ -24,4 +24,7 @@ type Cluster interface {
 	UpgradeContainer(ids []string, image string) (chan *types.UpgradeContainerMessage, error)
 	RemoveContainer(ids []string) (chan *types.RemoveContainerMessage, error)
 	RemoveImage(podname, nodename string, images []string) (chan *types.RemoveImageMessage, error)
+
+	// cluster attribute methods
+	GetZone() string
 }
diff --git a/devtools/core_pb2.py b/devtools/core_pb2.py
diff --git a/rpc/gen/core.pb.go b/rpc/gen/core.pb.go
diff --git a/rpc/gen/core.proto b/rpc/gen/core.proto
@@ -52,6 +52,7 @@ message Node {
     map<string, int64> cpu = 5;
     string info = 6;
     bool available = 7;
+    string zone = 8;
 }
 
 message Nodes {

diff --git a/rpc/rpc.go b/rpc/rpc.go
@@ -63,7 +63,7 @@ func (v *virbranium) AddNode(ctx context.Context, opts *pb.AddNodeOptions) (*pb.
 		return nil, err
 	}
 
-	return toRPCNode(n), nil
+	return toRPCNode(n, v.cluster.GetZone()), nil
 }
 
 // AddNode saves a node and returns it to client
@@ -84,7 +84,7 @@ func (v *virbranium) GetNode(ctx context.Context, opts *pb.GetNodeOptions) (*pb.
 		return nil, err
 	}
 
-	return toRPCNode(n), nil
+	return toRPCNode(n, v.cluster.GetZone()), nil
 }
 
 // ListPodNodes returns a list of node for pod
@@ -96,7 +96,7 @@ func (v *virbranium) ListPodNodes(ctx context.Context, opts *pb.ListNodesOptions
 
 	nodes := []*pb.Node{}
 	for _, n := range ns {
-		nodes = append(nodes, toRPCNode(n))
+		nodes = append(nodes, toRPCNode(n, v.cluster.GetZone()))
 	}
 	return &pb.Nodes{Nodes: nodes}, nil
 }
@@ -172,7 +172,7 @@ func (v *virbranium) SetNodeAvailable(ctx context.Context, opts *pb.NodeAvailabl
 	if err != nil {
 		return nil, err
 	}
-	return toRPCNode(n), nil
+	return toRPCNode(n, v.cluster.GetZone()), nil
 }
 
 // streamed returned functions

diff --git a/rpc/transform.go b/rpc/transform.go
@@ -23,7 +23,7 @@ func toRPCNetwork(n *types.Network) *pb.Network {
 	return &pb.Network{Name: n.Name, Subnets: n.Subnets}
 }
 
-func toRPCNode(n *types.Node) *pb.Node {
+func toRPCNode(n *types.Node, zone string) *pb.Node {
 	bytes := []byte("")
 	if info, err := n.Info(); err == nil {
 		bytes, _ = json.Marshal(info)
@@ -37,6 +37,7 @@ func toRPCNode(n *types.Node) *pb.Node {
 		Cpu:       toRPCCPUMap(n.CPU),
 		Info:      string(bytes),
 		Available: n.Available,
+		Zone:      zone,
 	}
 }
 

diff --git a/types/config.go b/types/config.go
@@ -9,6 +9,7 @@ type Config struct {
 	EtcdLockPrefix string   `yaml:"etcd_lock_prefix"` // etcd lock prefix, all locks will be created under this dir
 	ResourceAlloc  string   `yaml:"resource_alloc"`   // scheduler or cpu-period TODO give it a good name
 	Statsd         string   `yaml:"statsd"`           // Statsd host and port
+	Zone           string   `yaml:"zone"`             // zone for core, e.g. C1, C2
 
 	Git       GitConfig    `yaml:"git"`
 	Docker    DockerConfig `yaml:"docker"`