2012-08-29 07:58:03 +00:00
|
|
|
package topology
|
|
|
|
|
|
|
|
import (
|
2012-09-02 21:33:48 +00:00
|
|
|
_ "fmt"
|
|
|
|
"pkg/storage"
|
2012-08-29 07:58:03 +00:00
|
|
|
)
|
|
|
|
|
2012-09-08 23:25:44 +00:00
|
|
|
type DataNode struct {
|
2012-09-02 21:33:48 +00:00
|
|
|
NodeImpl
|
2012-09-19 08:45:30 +00:00
|
|
|
volumes map[storage.VolumeId]storage.VolumeInfo
|
2012-09-12 08:07:23 +00:00
|
|
|
Ip string
|
|
|
|
Port int
|
|
|
|
PublicUrl string
|
2012-09-17 08:48:09 +00:00
|
|
|
LastSeen int64 // unix time in seconds
|
2012-09-18 21:05:12 +00:00
|
|
|
Dead bool
|
2012-08-29 07:58:03 +00:00
|
|
|
}
|
2012-09-02 21:33:48 +00:00
|
|
|
|
2012-09-08 23:25:44 +00:00
|
|
|
func NewDataNode(id string) *DataNode {
|
|
|
|
s := &DataNode{}
|
2012-09-02 21:33:48 +00:00
|
|
|
s.id = NodeId(id)
|
2012-09-08 23:25:44 +00:00
|
|
|
s.nodeType = "DataNode"
|
2012-09-19 08:45:30 +00:00
|
|
|
s.volumes = make(map[storage.VolumeId]storage.VolumeInfo)
|
|
|
|
s.NodeImpl.value = s
|
2012-09-02 21:33:48 +00:00
|
|
|
return s
|
2012-08-31 08:35:11 +00:00
|
|
|
}
|
2012-09-19 08:45:30 +00:00
|
|
|
func (dn *DataNode) AddOrUpdateVolume(v storage.VolumeInfo) {
|
|
|
|
if _, ok := dn.volumes[v.Id]; !ok {
|
2012-09-17 00:31:15 +00:00
|
|
|
dn.volumes[v.Id] = v
|
|
|
|
dn.UpAdjustActiveVolumeCountDelta(1)
|
|
|
|
dn.UpAdjustMaxVolumeId(v.Id)
|
2012-09-18 21:05:12 +00:00
|
|
|
} else {
|
|
|
|
dn.volumes[v.Id] = v
|
2012-09-17 00:31:15 +00:00
|
|
|
}
|
2012-09-10 07:18:07 +00:00
|
|
|
}
|
|
|
|
func (dn *DataNode) GetTopology() *Topology {
|
2012-09-17 00:31:15 +00:00
|
|
|
p := dn.parent
|
|
|
|
for p.Parent() != nil {
|
|
|
|
p = p.Parent()
|
|
|
|
}
|
|
|
|
t := p.(*Topology)
|
|
|
|
return t
|
2012-08-29 07:58:03 +00:00
|
|
|
}
|
2012-09-14 08:17:13 +00:00
|
|
|
func (dn *DataNode) MatchLocation(ip string, port int) bool {
|
2012-09-17 00:31:15 +00:00
|
|
|
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()
|
2012-09-20 09:11:08 +00:00
|
|
|
ret["Max"] = dn.GetMaxVolumeCount()
|
|
|
|
ret["Free"] = dn.FreeSpace()
|
2012-09-17 00:31:15 +00:00
|
|
|
ret["PublicUrl"] = dn.PublicUrl
|
|
|
|
return ret
|
2012-09-14 08:17:13 +00:00
|
|
|
}
|