shell: optionally read filer address from master

1 files changed, 70 insertions, 0 deletions

diff --git a/weed/server/master_server_cluster.go b/weed/server/master_server_cluster.go
new file mode 100644
index 000000000..52e1526f9
--- /dev/null
+++ b/weed/server/master_server_cluster.go
@@ -0,0 +1,70 @@
+package weed_server
+
+import (
+	"github.com/chrislusf/seaweedfs/weed/pb"
+	"sync"
+)
+
+type NodeType int
+
+const (
+	filerNodeType NodeType = iota
+)
+
+type ClusterNode struct {
+	address pb.ServerAddress
+	version string
+}
+
+type Cluster struct {
+	filers     map[pb.ServerAddress]*ClusterNode
+	filersLock sync.RWMutex
+}
+
+func NewCluster() *Cluster {
+	return &Cluster{
+		filers: make(map[pb.ServerAddress]*ClusterNode),
+	}
+}
+
+func (cluster *Cluster) AddClusterNode(nodeType string, address pb.ServerAddress, version string) {
+	switch nodeType {
+	case "filer":
+		cluster.filersLock.Lock()
+		defer cluster.filersLock.Unlock()
+		if _, found := cluster.filers[address]; found {
+			return
+		}
+		cluster.filers[address] = &ClusterNode{
+			address: address,
+			version: version,
+		}
+	case "master":
+	}
+}
+
+func (cluster *Cluster) RemoveClusterNode(nodeType string, address pb.ServerAddress) {
+	switch nodeType {
+	case "filer":
+		cluster.filersLock.Lock()
+		defer cluster.filersLock.Unlock()
+		if _, found := cluster.filers[address]; !found {
+			return
+		}
+		delete(cluster.filers, address)
+	case "master":
+	}
+}
+
+func (cluster *Cluster) ListClusterNode(nodeType string) (nodes []*ClusterNode){
+	switch nodeType {
+	case "filer":
+		cluster.filersLock.RLock()
+		defer cluster.filersLock.RUnlock()
+		for _, node := range cluster.filers {
+			nodes = append(nodes, node)
+		}
+	case "master":
+	}
+	return
+}