Modify method calls in A1 client

[nonrtric.git] / policy-agent / src / main / java / org / oransc / policyagent / repository / PolicyTypes.java

diff --git a/policy-agent/src/main/java/org/oransc/policyagent/repository/PolicyTypes.java b/policy-agent/src/main/java/org/oransc/policyagent/repository/PolicyTypes.java
index 6cf8ff4..63450cb 100644 (file)
--- a/policy-agent/src/main/java/org/oransc/policyagent/repository/PolicyTypes.java
+++ b/policy-agent/src/main/java/org/oransc/policyagent/repository/PolicyTypes.java
@@ -20,6 +20,7 @@
 
 package org.oransc.policyagent.repository;
 
+import java.util.Collection;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -43,4 +44,19 @@ public class PolicyTypes {
         types.put(type.name(), type);
     }
 
+    public boolean contains(PolicyType policyType) {
+        return types.containsKey(policyType.name());
+    }
+
+    public synchronized Collection<PolicyType> getAll() {
+        return types.values();
+    }
+
+    public int size() {
+        return types.size();
+    }
+
+    public void clear() {
+        this.types.clear();
+    }
 }