added getters for QuotaTree and TreeController

Author: atantawi

pkg/quotaplugins/quota-forest/quota-manager/quota/core/quotatree.go

@@ -209,6 +209,15 @@ func (qt *QuotaTree) GetQuotaSize() int {
 	return len(qt.resourceNames)
 }
 
+// GetNodes : get a map of all quota nodes in the tree
+func (qt *QuotaTree) GetNodes() map[string]*QuotaNode {
+	nodeMap := make(map[string]*QuotaNode)
+	for _, n := range qt.Tree.GetNodeListBFS() {
+		nodeMap[n.GetID()] = (*QuotaNode)(unsafe.Pointer(n))
+	}
+	return nodeMap
+}
+
 // StringSimply :
 func (qt *QuotaTree) StringSimply() string {
 	var b bytes.Buffer

pkg/quotaplugins/quota-forest/quota-manager/quota/core/treecontroller.go

@@ -138,6 +138,11 @@ func (controller *Controller) DeAllocate(consumerID string) bool {
 	return false
 }
 
+// GetConsumers : get a map of consumers in controller
+func (controller *Controller) GetConsumers() map[string]*Consumer {
+	return controller.consumers
+}
+
 // GetPreemptedConsumers : get a list of the preempted consumer IDs
 func (controller *Controller) GetPreemptedConsumers() []string {
 	return controller.preemptedConsumers