Add A1 controller client in policy-agent
[nonrtric.git] / policy-agent / src / main / java / org / oransc / policyagent / configuration / ApplicationConfigParser.java
index 487c3a9..530ac98 100644 (file)
 
 package org.oransc.policyagent.configuration;
 
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+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;
 
 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() {
     }
 
-    String example;
-
     public void parse(JsonObject root) throws ServiceException {
-        JsonObject config = root.getAsJsonObject(CONFIG);
-        example = getAsString(config, "exampleProperty");
+        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")) {
+            result.add(gson.fromJson(ricElem.getAsJsonObject(), ImmutableRicConfig.class));
+        }
+        return result;
     }
 
     private static JsonElement get(JsonObject obj, String memberName) throws ServiceException {
@@ -47,12 +82,73 @@ public class ApplicationConfigParser {
         return elem;
     }
 
-    private static String getAsString(JsonObject obj, String memberName) throws ServiceException {
+    private JsonArray getAsJsonArray(JsonObject obj, String memberName) throws ServiceException {
+        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 static JsonObject getAsJson(JsonObject obj, String memberName) throws ServiceException {
-        return get(obj, memberName).getAsJsonObject();
+    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);
+    }
 }