package topology import ( _ "fmt" "pkg/storage" ) type DataNode struct { NodeImpl volumes map[storage.VolumeId]storage.VolumeInfo Ip string Port int PublicUrl string LastSeen int64 // unix time in seconds Dead bool } func NewDataNode(id string) *DataNode { s := &DataNode{} s.id = NodeId(id) s.nodeType = "DataNode" s.volumes = make(map[storage.VolumeId]storage.VolumeInfo) s.NodeImpl.value = s return s } func (dn *DataNode) AddOrUpdateVolume(v storage.VolumeInfo) { if _, ok := dn.volumes[v.Id]; !ok { dn.volumes[v.Id] = v dn.UpAdjustActiveVolumeCountDelta(1) dn.UpAdjustMaxVolumeId(v.Id) } else { dn.volumes[v.Id] = v } } func (dn *DataNode) GetTopology() *Topology { p := dn.parent for p.Parent() != nil { p = p.Parent() } t := p.(*Topology) return t } func (dn *DataNode) MatchLocation(ip string, port int) bool { return dn.Ip == ip && dn.Port == port } func (dn *DataNode) ToMap() interface{} { ret := make(map[string]interface{}) ret["Ip"] = dn.Ip ret["Port"] = dn.Port ret["Volumes"] = dn.GetActiveVolumeCount() ret["Max"] = dn.GetMaxVolumeCount() ret["Free"] = dn.FreeSpace() ret["PublicUrl"] = dn.PublicUrl return ret }