X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=policy-agent%2Fsrc%2Fmain%2Fjava%2Forg%2Foransc%2Fpolicyagent%2Frepository%2FPolicyTypes.java;h=9dee6f91154cd3ef7f56abdecf723cfa43b877bc;hb=637540bc28fbf337e0c4c58c051a6b4f7ceb321d;hp=f686ee4c1f81ec4772715a49a8dfac25c7144cc0;hpb=296f7c083ed15a57e59ea4546eb0624bf235d29b;p=nonrtric.git

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 f686ee4c..9dee6f91 100644
--- 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;
 
@@ -39,15 +40,27 @@ public class PolicyTypes {
         return t;
     }
 
+    public synchronized PolicyType get(String name) {
+        return types.get(name);
+    }
+
     public synchronized void put(PolicyType type) {
         types.put(type.name(), type);
     }
 
-    public boolean contains(PolicyType policyType) {
-        return types.containsKey(policyType.name());
+    public boolean contains(String policyType) {
+        return types.containsKey(policyType);
+    }
+
+    public synchronized Collection<PolicyType> getAll() {
+        return types.values();
     }
 
     public int size() {
         return types.size();
     }
+
+    public void clear() {
+        this.types.clear();
+    }
 }