seaweedfs/weed/server/master_server_cluster.go

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
}