mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-11-27 12:49:41 +08:00
27c74a7e66
change replication_type to ReplicaPlacement, hopefully cleaner code works for 9 possible ReplicaPlacement xyz x : number of copies on other data centers y : number of copies on other racks z : number of copies on current rack x y z each can be 0,1,2 Minor: weed server "-mdir" default to "-dir" if empty
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package topology
|
|
|
|
import (
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type Rack struct {
|
|
NodeImpl
|
|
}
|
|
|
|
func NewRack(id string) *Rack {
|
|
r := &Rack{}
|
|
r.id = NodeId(id)
|
|
r.nodeType = "Rack"
|
|
r.children = make(map[NodeId]Node)
|
|
r.NodeImpl.value = r
|
|
return r
|
|
}
|
|
|
|
func (r *Rack) FindDataNode(ip string, port int) *DataNode {
|
|
for _, c := range r.Children() {
|
|
dn := c.(*DataNode)
|
|
if dn.MatchLocation(ip, port) {
|
|
return dn
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
func (r *Rack) GetOrCreateDataNode(ip string, port int, publicUrl string, maxVolumeCount int) *DataNode {
|
|
for _, c := range r.Children() {
|
|
dn := c.(*DataNode)
|
|
if dn.MatchLocation(ip, port) {
|
|
dn.LastSeen = time.Now().Unix()
|
|
if dn.Dead {
|
|
dn.Dead = false
|
|
r.GetTopology().chanRecoveredDataNodes <- dn
|
|
dn.UpAdjustMaxVolumeCountDelta(maxVolumeCount - dn.maxVolumeCount)
|
|
}
|
|
return dn
|
|
}
|
|
}
|
|
dn := NewDataNode(ip + ":" + strconv.Itoa(port))
|
|
dn.Ip = ip
|
|
dn.Port = port
|
|
dn.PublicUrl = publicUrl
|
|
dn.maxVolumeCount = maxVolumeCount
|
|
dn.LastSeen = time.Now().Unix()
|
|
r.LinkChildNode(dn)
|
|
return dn
|
|
}
|
|
|
|
func (rack *Rack) ToMap() interface{} {
|
|
m := make(map[string]interface{})
|
|
m["Id"] = rack.Id()
|
|
m["Max"] = rack.GetMaxVolumeCount()
|
|
m["Free"] = rack.FreeSpace()
|
|
var dns []interface{}
|
|
for _, c := range rack.Children() {
|
|
dn := c.(*DataNode)
|
|
dns = append(dns, dn.ToMap())
|
|
}
|
|
m["DataNodes"] = dns
|
|
return m
|
|
}
|