optimizely · zashraf1985 · Aug 2, 2022 · Jul 28, 2022 · Jul 28, 2022 · Jul 29, 2022
diff --git a/core-api/src/main/java/com/optimizely/ab/internal/JsonParserProvider.java b/core-api/src/main/java/com/optimizely/ab/internal/JsonParserProvider.java
@@ -0,0 +1,74 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.internal;
+
+import com.optimizely.ab.config.parser.MissingJsonParserException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public enum JsonParserProvider {
+    GSON_CONFIG_PARSER("com.google.gson.Gson"),
+    JACKSON_CONFIG_PARSER("com.fasterxml.jackson.databind.ObjectMapper" ),
+    JSON_CONFIG_PARSER("org.json.JSONObject"),
+    JSON_SIMPLE_CONFIG_PARSER("org.json.simple.JSONObject");
+
+    private static final Logger logger = LoggerFactory.getLogger(JsonParserProvider.class);
+
+    private final String className;
+
+    JsonParserProvider(String className) {
+        this.className = className;
+    }
+
+    private boolean isAvailable() {
+        try {
+            Class.forName(className);
+            return true;
+        } catch (ClassNotFoundException e) {
+            return false;
+        }
+    }
+
+    public static JsonParserProvider getDefaultParser() {
+        String defaultParserName = PropertyUtils.get("default_parser");
+
+        if (defaultParserName != null) {
+            try {
+                JsonParserProvider parser = JsonParserProvider.valueOf(defaultParserName);
+                if (parser.isAvailable()) {
+                    logger.debug("using json parser: {}, based on override config", parser.className);
+                    return parser;
+                }
+
+                logger.warn("configured parser {} is not available in the classpath", defaultParserName);
+            } catch (IllegalArgumentException e) {
+                logger.warn("configured parser {} is not a valid value", defaultParserName);
+            }
+        }
+
+        for (JsonParserProvider parser: JsonParserProvider.values()) {
+            if (!parser.isAvailable()) {
+                continue;
+            }
+
+            logger.info("using json parser: {}", parser.className);
+            return parser;
+        }
+
+        throw new MissingJsonParserException("unable to locate a JSON parser. "
+            + "Please see <link> for more information");
+    }
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/ODPApiManager.java b/core-api/src/main/java/com/optimizely/ab/odp/ODPApiManager.java
@@ -0,0 +1,22 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp;
+
+import java.util.List;
+
+public interface ODPApiManager {
+    String fetchQualifiedSegments(String apiKey, String apiEndpoint, String userKey, String userValue, List<String> segmentsToCheck);
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/parser/ResponseJsonParser.java b/core-api/src/main/java/com/optimizely/ab/odp/parser/ResponseJsonParser.java
@@ -0,0 +1,22 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp.parser;
+
+import java.util.List;
+
+public interface ResponseJsonParser {
+    public List<String> parseQualifiedSegments(String responseJson);
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/parser/ResponseJsonParserFactory.java b/core-api/src/main/java/com/optimizely/ab/odp/parser/ResponseJsonParserFactory.java
@@ -0,0 +1,49 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp.parser;
+
+import com.optimizely.ab.internal.JsonParserProvider;
+import com.optimizely.ab.odp.parser.impl.GsonParser;
+import com.optimizely.ab.odp.parser.impl.JacksonParser;
+import com.optimizely.ab.odp.parser.impl.JsonParser;
+import com.optimizely.ab.odp.parser.impl.JsonSimpleParser;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class ResponseJsonParserFactory {
+    private static final Logger logger = LoggerFactory.getLogger(ResponseJsonParserFactory.class);
+
+    public static ResponseJsonParser getParser() {
+        JsonParserProvider parserProvider = JsonParserProvider.getDefaultParser();
+        ResponseJsonParser jsonParser = null;
+        switch (parserProvider) {
+            case GSON_CONFIG_PARSER:
+                jsonParser = new GsonParser();
+                break;
+            case JACKSON_CONFIG_PARSER:
+                jsonParser = new JacksonParser();
+                break;
+            case JSON_CONFIG_PARSER:
+                jsonParser = new JsonParser();
+                break;
+            case JSON_SIMPLE_CONFIG_PARSER:
+                jsonParser = new JsonSimpleParser();
+                break;
+        }
+        logger.info("Using " + parserProvider.toString() + " parser");
+        return jsonParser;
+    }
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/GsonParser.java b/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/GsonParser.java
@@ -0,0 +1,62 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp.parser.impl;
+
+import com.google.gson.*;
+import com.google.gson.JsonParser;
+import com.optimizely.ab.odp.parser.ResponseJsonParser;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class GsonParser implements ResponseJsonParser {
+    private static final Logger logger = LoggerFactory.getLogger(GsonParser.class);
+
+    @Override
+    public List<String> parseQualifiedSegments(String responseJson) {
+        List<String> parsedSegments = new ArrayList<>();
+        try {
+            JsonObject root = JsonParser.parseString(responseJson).getAsJsonObject();
+
+            if (root.has("errors")) {
+                JsonArray errors =  root.getAsJsonArray("errors");
+                StringBuilder logMessage = new StringBuilder();
+                for (int i = 0; i < errors.size(); i++) {
+                    if (i > 0) {
+                        logMessage.append(", ");
+                    }
+                    logMessage.append(errors.get(i).getAsJsonObject().get("message").getAsString());
+                }
+                logger.error(logMessage.toString());
+                return null;
+            }
+
+            JsonArray edges = root.getAsJsonObject("data").getAsJsonObject("customer").getAsJsonObject("audiences").getAsJsonArray("edges");
+            for (int i = 0; i < edges.size(); i++) {
+                JsonObject node = edges.get(i).getAsJsonObject().getAsJsonObject("node");
+                if (node.has("state") && node.get("state").getAsString().equals("qualified")) {
+                    parsedSegments.add(node.get("name").getAsString());
+                }
+            }
+            return parsedSegments;
+        } catch (JsonSyntaxException e) {
+            logger.error("Error parsing qualified segments from response", e);
+            return null;
+        }
+    }
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/JacksonParser.java b/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/JacksonParser.java
@@ -0,0 +1,66 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp.parser.impl;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.optimizely.ab.odp.parser.ResponseJsonParser;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.ArrayList;
+
+import java.util.List;
+
+public class JacksonParser implements ResponseJsonParser {
+    private static final Logger logger = LoggerFactory.getLogger(JacksonParser.class);
+
+    @Override
+    public List<String> parseQualifiedSegments(String responseJson) {
+        ObjectMapper objectMapper = new ObjectMapper();
+        List<String> parsedSegments = new ArrayList<>();
+        JsonNode root;
+        try {
+            root = objectMapper.readTree(responseJson);
+
+            if (root.has("errors")) {
+                JsonNode errors = root.path("errors");
+                StringBuilder logMessage = new StringBuilder();
+                for (int i = 0; i < errors.size(); i++) {
+                    if (i > 0) {
+                        logMessage.append(", ");
+                    }
+                    logMessage.append(errors.get(i).path("message"));
+                }
+                logger.error(logMessage.toString());
+                return null;
+            }
+
+            JsonNode edges = root.path("data").path("customer").path("audiences").path("edges");
+            for (JsonNode edgeNode : edges) {
+                String state = edgeNode.path("node").path("state").asText();
+                if (state.equals("qualified")) {
+                    parsedSegments.add(edgeNode.path("node").path("name").asText());
+                }
+            }
+            return parsedSegments;
+        } catch (JsonProcessingException e) {
+            logger.error("Error parsing qualified segments from response", e);
+            return null;
+        }
+    }
+}
diff --git a/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/JsonParser.java b/core-api/src/main/java/com/optimizely/ab/odp/parser/impl/JsonParser.java
@@ -0,0 +1,63 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+package com.optimizely.ab.odp.parser.impl;
+
+import com.optimizely.ab.odp.parser.ResponseJsonParser;
+import org.json.JSONArray;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class JsonParser implements ResponseJsonParser {
+    private static final Logger logger = LoggerFactory.getLogger(JsonParser.class);
+
+    @Override
+    public List<String> parseQualifiedSegments(String responseJson) {
+        List<String> parsedSegments = new ArrayList<>();
+        try {
+            JSONObject root = new JSONObject(responseJson);
+
+            if (root.has("errors")) {
+                JSONArray errors =  root.getJSONArray("errors");
+                StringBuilder logMessage = new StringBuilder();
+                for (int i = 0; i < errors.length(); i++) {
+                    if (i > 0) {
+                        logMessage.append(", ");
+                    }
+                    logMessage.append(errors.getJSONObject(i).getString("message"));
+                }
+                logger.error(logMessage.toString());
+                return null;
+            }
+
+            JSONArray edges = root.getJSONObject("data").getJSONObject("customer").getJSONObject("audiences").getJSONArray("edges");
+            for (int i = 0; i < edges.length(); i++) {
+                JSONObject node = edges.getJSONObject(i).getJSONObject("node");
+                if (node.has("state") && node.getString("state").equals("qualified")) {
+                    parsedSegments.add(node.getString("name"));
+                }
+            }
+            return parsedSegments;
+        } catch (JSONException e) {
+            logger.error("Error parsing qualified segments from response", e);
+            return  null;
+        }
+    }
+}