Switch from gson to Jackson for JSON->object conversion

rockset · Dec 20, 2018 · e6a9a25 · e6a9a25
1 parent 7ba3267
commit e6a9a25
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/rockset/RocksetClientWrapper.java b/src/main/java/rockset/RocksetClientWrapper.java
@@ -1,6 +1,7 @@
 package rockset;
 
-import com.google.gson.Gson;
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.rockset.client.RocksetClient;
 import com.rockset.client.model.AddDocumentsRequest;
 import com.rockset.client.model.AddDocumentsResponse;
@@ -10,7 +11,6 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-
 public class RocksetClientWrapper {
   private static Logger log = LoggerFactory.getLogger(RocksetClientWrapper.class);
   private RocksetClient client;
@@ -24,10 +24,10 @@ public RocksetClientWrapper(String apiKey, String apiServer) {
   public void addDoc(String collection, String json) {
     LinkedList<Object> list = new LinkedList<>();
     AddDocumentsResponse res = null;
+    ObjectMapper mapper = new ObjectMapper();
 
     try {
-      Map map = new Gson().fromJson(json, Map.class);
-      list.add(map);
+      list.add(mapper.readValue(json, new TypeReference<Map<String, Object>>(){}));
       AddDocumentsRequest documentsRequest = new AddDocumentsRequest().data(list);
       res = client.addDocuments(collection, documentsRequest);
     } catch (Exception e) {