aboutsummaryrefslogtreecommitdiff
path: root/weed/topology/topology.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/topology/topology.go')
-rw-r--r--weed/topology/topology.go16
1 files changed, 16 insertions, 0 deletions
diff --git a/weed/topology/topology.go b/weed/topology/topology.go
index 0a4cb4050..01822cbf2 100644
--- a/weed/topology/topology.go
+++ b/weed/topology/topology.go
@@ -285,6 +285,22 @@ func (t *Topology) UnRegisterVolumeLayout(v storage.VolumeInfo, dn *DataNode) {
}
}
+func (t *Topology) DataCenterExists(dcName string) bool {
+ return dcName == "" || t.GetOrCreateDataCenter(dcName) != nil
+}
+
+func (t *Topology) GetDataCenter(dcName string) (dc *DataCenter) {
+ t.RLock()
+ defer t.RUnlock()
+ for _, c := range t.children {
+ dc = c.(*DataCenter)
+ if string(dc.Id()) == dcName {
+ return dc
+ }
+ }
+ return dc
+}
+
func (t *Topology) GetOrCreateDataCenter(dcName string) *DataCenter {
t.Lock()
defer t.Unlock()