You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
3.0 KiB
149 lines
3.0 KiB
package topology
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/chrislusf/seaweedfs/go/storage"
|
|
)
|
|
|
|
type VolumeLocationList struct {
|
|
list []*DataNode
|
|
}
|
|
|
|
func NewVolumeLocationList() *VolumeLocationList {
|
|
return &VolumeLocationList{}
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) String() string {
|
|
return fmt.Sprintf("%v", dnll.list)
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) Head() *DataNode {
|
|
//mark first node as master volume
|
|
return dnll.list[0]
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) Length() int {
|
|
return len(dnll.list)
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) CalcReplicaPlacement() (rp *storage.ReplicaPlacement) {
|
|
//TODO CalcReplicaPlacement
|
|
return nil
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) Set(loc *DataNode) {
|
|
for i := 0; i < len(dnll.list); i++ {
|
|
if loc.Ip == dnll.list[i].Ip && loc.Port == dnll.list[i].Port {
|
|
dnll.list[i] = loc
|
|
return
|
|
}
|
|
}
|
|
dnll.list = append(dnll.list, loc)
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) Remove(loc *DataNode) bool {
|
|
for i, dnl := range dnll.list {
|
|
if loc.Ip == dnl.Ip && loc.Port == dnl.Port {
|
|
dnll.list = append(dnll.list[:i], dnll.list[i+1:]...)
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (dnll *VolumeLocationList) Refresh(freshThreshHold int64) {
|
|
var changed bool
|
|
for _, dnl := range dnll.list {
|
|
if dnl.LastSeen < freshThreshHold {
|
|
changed = true
|
|
break
|
|
}
|
|
}
|
|
if changed {
|
|
var l []*DataNode
|
|
for _, dnl := range dnll.list {
|
|
if dnl.LastSeen >= freshThreshHold {
|
|
l = append(l, dnl)
|
|
}
|
|
}
|
|
dnll.list = l
|
|
}
|
|
}
|
|
|
|
// return all data centers, first is main data center
|
|
func (dnll *VolumeLocationList) DataCenters() []*DataCenter {
|
|
m := make(map[*DataCenter]int)
|
|
maxCount := 0
|
|
var mainDC *DataCenter
|
|
for _, dn := range dnll.list {
|
|
var dc *DataCenter
|
|
if dc = dn.GetDataCenter(); dc == nil {
|
|
continue
|
|
}
|
|
m[dc] = m[dc] + 1
|
|
if m[dc] > maxCount {
|
|
mainDC = dc
|
|
maxCount = m[dc]
|
|
}
|
|
}
|
|
dataCenters := make([]*DataCenter, 0, len(m))
|
|
if mainDC != nil {
|
|
dataCenters = append(dataCenters, mainDC)
|
|
}
|
|
for dc := range m {
|
|
if dc != mainDC {
|
|
dataCenters = append(dataCenters, dc)
|
|
}
|
|
}
|
|
return dataCenters
|
|
}
|
|
|
|
// return all racks if data center set nil
|
|
func (dnll *VolumeLocationList) Racks(dc *DataCenter) []*Rack {
|
|
m := make(map[*Rack]int)
|
|
maxCount := 0
|
|
var mainRack *Rack
|
|
for _, dn := range dnll.list {
|
|
if dc != nil && dn.GetDataCenter() != dc {
|
|
continue
|
|
}
|
|
var rack *Rack
|
|
if rack = dn.GetRack(); rack == nil {
|
|
continue
|
|
}
|
|
m[rack] = m[rack] + 1
|
|
if m[rack] > maxCount {
|
|
mainRack = rack
|
|
maxCount = m[rack]
|
|
}
|
|
}
|
|
racks := make([]*Rack, 0, len(m))
|
|
if mainRack != nil {
|
|
racks = append(racks, mainRack)
|
|
}
|
|
for rack := range m {
|
|
racks = append(racks, rack)
|
|
}
|
|
return racks
|
|
}
|
|
|
|
|
|
func (dnll *VolumeLocationList) Servers(rack *Rack) []*DataNode {
|
|
servers := make([]*DataNode)
|
|
for _, dn := range dnll.list {
|
|
if rack != nil && dn.GetRack() != rack {
|
|
continue
|
|
}
|
|
var rack *Rack
|
|
if rack = dn.GetRack(); rack == nil {
|
|
continue
|
|
}
|
|
servers = append(servers, dn)
|
|
}
|
|
return servers
|
|
}
|
|
|
|
//func (dnll *VolumeLocationList)ContainDataNode(nodeType, id string)bool {
|
|
//
|
|
//}
|