Add A1 controller client in policy-agent

[nonrtric.git] / policy-agent / src / main / java / org / oransc / policyagent / configuration / ApplicationConfigParser.java

diff --git a/policy-agent/src/main/java/org/oransc/policyagent/configuration/ApplicationConfigParser.java b/policy-agent/src/main/java/org/oransc/policyagent/configuration/ApplicationConfigParser.java
index 400a681..530ac98 100644 (file)
--- a/policy-agent/src/main/java/org/oransc/policyagent/configuration/ApplicationConfigParser.java
+++ b/policy-agent/src/main/java/org/oransc/policyagent/configuration/ApplicationConfigParser.java
@@ -26,31 +26,46 @@ import com.google.gson.JsonArray;
 import com.google.gson.JsonElement;
 import com.google.gson.JsonObject;
 
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.util.Map.Entry;
+import java.util.Properties;
+import java.util.Set;
 import java.util.Vector;
 
+import javax.validation.constraints.NotNull;
+
 import org.oransc.policyagent.exceptions.ServiceException;
 
-class ApplicationConfigParser {
+public class ApplicationConfigParser {
 
     private static final String CONFIG = "config";
+
     private static Gson gson = new GsonBuilder() //
         .serializeNulls() //
         .create(); //
 
     private Vector<RicConfig> ricConfig;
+    private Properties dmaapConsumerConfig;
 
     public ApplicationConfigParser() {
     }
 
     public void parse(JsonObject root) throws ServiceException {
-        JsonObject config = root.getAsJsonObject(CONFIG);
-        ricConfig = parseRics(config);
+        JsonObject ricConfigJson = root.getAsJsonObject(CONFIG);
+        ricConfig = parseRics(ricConfigJson);
+        JsonObject dmaapConfigJson = root.getAsJsonObject("streams_subscribes");
+        dmaapConsumerConfig = parseDmaapConsumerConfig(dmaapConfigJson);
     }
 
     public Vector<RicConfig> getRicConfigs() {
         return this.ricConfig;
     }
 
+    public Properties getDmaapConsumerConfig() {
+        return dmaapConsumerConfig;
+    }
+
     private Vector<RicConfig> parseRics(JsonObject config) throws ServiceException {
         Vector<RicConfig> result = new Vector<RicConfig>();
         for (JsonElement ricElem : getAsJsonArray(config, "ric")) {
@@ -71,4 +86,69 @@ class ApplicationConfigParser {
         return get(obj, memberName).getAsJsonArray();
     }
 
+    private Properties parseDmaapConsumerConfig(JsonObject consumerCfg) throws ServiceException {
+        Set<Entry<String, JsonElement>> topics = consumerCfg.entrySet();
+        if (topics.size() != 1) {
+            throw new ServiceException("Invalid configuration, number of topic must be one, config: " + topics);
+        }
+        JsonObject topic = topics.iterator().next().getValue().getAsJsonObject();
+        JsonObject dmaapInfo = get(topic, "dmaap_info").getAsJsonObject();
+        String topicUrl = getAsString(dmaapInfo, "topic_url");
+
+        Properties dmaapProps = new Properties();
+        try {
+            URL url = new URL(topicUrl);
+            String passwd = "";
+            String userName = "";
+            if (url.getUserInfo() != null) {
+                String[] userInfo = url.getUserInfo().split(":");
+                userName = userInfo[0];
+                passwd = userInfo[1];
+            }
+            String urlPath = url.getPath();
+            DmaapConsumerUrlPath path = parseDmaapUrlPath(urlPath);
+
+            dmaapProps.put("port", url.getPort());
+            dmaapProps.put("server", url.getHost());
+            dmaapProps.put("topic", path.dmaapTopicName);
+            dmaapProps.put("consumerGroup", path.consumerGroup);
+            dmaapProps.put("consumerInstance", path.consumerId);
+            dmaapProps.put("fetchTimeout", 15000);
+            dmaapProps.put("fetchLimit", 1000);
+            dmaapProps.put("userName", userName);
+            dmaapProps.put("password", passwd);
+        } catch (MalformedURLException e) {
+            throw new ServiceException("Could not parse the URL", e);
+        }
+
+        return dmaapProps;
+    }
+
+    private static @NotNull String getAsString(JsonObject obj, String memberName) throws ServiceException {
+        return get(obj, memberName).getAsString();
+    }
+
+    private class DmaapConsumerUrlPath {
+        final String dmaapTopicName;
+        final String consumerGroup;
+        final String consumerId;
+
+        DmaapConsumerUrlPath(String dmaapTopicName, String consumerGroup, String consumerId) {
+            this.dmaapTopicName = dmaapTopicName;
+            this.consumerGroup = consumerGroup;
+            this.consumerId = consumerId;
+        }
+    }
+
+    private DmaapConsumerUrlPath parseDmaapUrlPath(String urlPath) throws ServiceException {
+        String[] tokens = urlPath.split("/"); // /events/A1-P/users/sdnc1
+        if (tokens.length != 5) {
+            throw new ServiceException("The path has incorrect syntax: " + urlPath);
+        }
+
+        final String dmaapTopicName = tokens[1] + "/" + tokens[2]; // /events/A1-P
+        final String consumerGroup = tokens[3]; // users
+        final String consumerId = tokens[4]; // sdnc1
+        return new DmaapConsumerUrlPath(dmaapTopicName, consumerGroup, consumerId);
+    }
 }