shell: optionally read filer address from master
This commit is contained in:
70
weed/server/master_server_cluster.go
Normal file
70
weed/server/master_server_cluster.go
Normal file
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user