1 files changed, 41 insertions, 0 deletions
diff --git a/weed/topology/data_center.go b/weed/topology/data_center.go
new file mode 100644
index 000000000..a3b2b7d13
--- /dev/null
+++ b/weed/topology/data_center.go
@@ -0,0 +1,41 @@
+package topology
+
+import ()
+
+type DataCenter struct {
+	NodeImpl
+}
+
+func NewDataCenter(id string) *DataCenter {
+	dc := &DataCenter{}
+	dc.id = NodeId(id)
+	dc.nodeType = "DataCenter"
+	dc.children = make(map[NodeId]Node)
+	dc.NodeImpl.value = dc
+	return dc
+}
+
+func (dc *DataCenter) GetOrCreateRack(rackName string) *Rack {
+	for _, c := range dc.Children() {
+		rack := c.(*Rack)
+		if string(rack.Id()) == rackName {
+			return rack
+		}
+	}
+	rack := NewRack(rackName)
+	dc.LinkChildNode(rack)
+	return rack
+}
+
+func (dc *DataCenter) ToMap() interface{} {
+	m := make(map[string]interface{})
+	m["Max"] = dc.GetMaxVolumeCount()
+	m["Free"] = dc.FreeSpace()
+	var racks []interface{}
+	for _, c := range dc.Children() {
+		rack := c.(*Rack)
+		racks = append(racks, rack.ToMap())
+	}
+	m["Racks"] = racks
+	return m
+}