seaweedfs/go/topology/configuration.go

66 lines
1.3 KiB
Go
Raw Normal View History

2012-09-22 07:01:13 +00:00
package topology
import (
"encoding/xml"
2012-09-22 07:01:13 +00:00
)
type loc struct {
dcName string
rackName string
}
2012-09-22 07:01:13 +00:00
type rack struct {
Name string `xml:"name,attr"`
Ips []string `xml:"Ip"`
2012-09-22 07:01:13 +00:00
}
type dataCenter struct {
Name string `xml:"name,attr"`
Racks []rack `xml:"Rack"`
2012-09-22 07:01:13 +00:00
}
type topology struct {
DataCenters []dataCenter `xml:"DataCenter"`
}
type Configuration struct {
XMLName xml.Name `xml:"Configuration"`
Topo topology `xml:"Topology"`
ip2location map[string]loc
2012-09-22 07:01:13 +00:00
}
func NewConfiguration(b []byte) (*Configuration, error) {
c := &Configuration{}
err := xml.Unmarshal(b, c)
c.ip2location = make(map[string]loc)
for _, dc := range c.Topo.DataCenters {
for _, rack := range dc.Racks {
for _, ip := range rack.Ips {
c.ip2location[ip] = loc{dcName: dc.Name, rackName: rack.Name}
}
}
}
return c, err
2012-09-22 07:01:13 +00:00
}
func (c *Configuration) String() string {
if b, e := xml.MarshalIndent(c, " ", " "); e == nil {
return string(b)
}
return ""
2012-09-22 07:01:13 +00:00
}
func (c *Configuration) Locate(ip string, dcName string, rackName string) (dc string, rack string) {
if dcName == "" {
if c != nil && c.ip2location != nil {
if loc, ok := c.ip2location[ip]; ok {
return loc.dcName, loc.rackName
}
}
} else {
if rackName == "" {
return dcName, "DefaultRack"
} else {
return dcName, rackName
}
}
return "DefaultDataCenter", "DefaultRack"
2012-09-22 07:01:13 +00:00
}