seaweedfs/weed-fs/src/pkg/topology/rack.go

package topology

import (
	"strconv"
	"time"
)

type Rack struct {
	NodeImpl
	ipRange *IpRange
}

func NewRack(id string) *Rack {
	r := &Rack{}
	r.id = NodeId(id)
	r.nodeType = "Rack"
	r.children = make(map[NodeId]Node)
	return r
}

func (r *Rack) MatchLocationRange(ip string) bool {
	if r.ipRange == nil {
		return true
	}
	return r.ipRange.Match(ip)
}

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("DataNode" + 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["Free"] = rack.FreeSpace()
	var dns []interface{}
	for _, c := range rack.Children() {
		dn := c.(*DataNode)
		dns = append(dns, dn.ToMap())
	}
	m["DataNodes"] = dns
	return m
}