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["MaxVolumeCount"] = dn.GetMaxVolumeCount()
	ret["FreeVolumeCount"] = dn.FreeSpace()
	ret["PublicUrl"] = dn.PublicUrl
	return ret
}