com.factual.driver.JsonUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of factual-java-driver Show documentation
Factual's officially supported Java driver
The newest version!
package com.factual.driver;

import java.io.IOException;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.codehaus.jackson.JsonGenerationException;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

public class JsonUtil {

  /**
   * Takes a JSONArray of records, where each record is a dictionary, and
   * returns the translated List of Maps.
   */
  public static List> data(JSONArray arr) throws JSONException {
    List> data = Lists.newArrayList();
    for(int i=0; i> data(JSONObject jo) throws JSONException {
    Map> data = Maps.newHashMap();
    Iterator iter = jo.keys();
    while (iter.hasNext()) {
      String key = iter.next().toString();
      data.put(key, row(jo.getJSONObject(key)));
    }
    return data;
  }

  private static Map row(JSONObject jo) throws JSONException {
    return toMap(jo);
  }

  public static Map toMap(JSONObject jo) throws JSONException {
    Map row = Maps.newHashMap();
    Iterator iter = jo.keys();
    while(iter.hasNext()) {
      String key = iter.next().toString();
      Object value = jo.get(key);
      row.put(key, value);
    }
    return row;
  }

  /**
   * Serialize a Java object to json.
   * Use Maps and Lists to specify object and array structures respectively.
   * 
   * For example, the following value as input:
   *    * {@code
   * new HashMap() {{
   * put("$and", new Map[] {
   * 	new HashMap() {{
   * 		put("name", new HashMap() {{
   *  			put("$bw", "McDonald's");
   *  		}});
   *  		put("category", new HashMap() {{
   *  			put("$bw", "Food & Beverage");
   *  		}});
   * 	}}});
   * }};
   * }
   * 
   * Will be serialized to json as:
   * {"$and":[{"name":{"$bw":"McDonald's"},"category":{"$bw":"Food & Beverage"}}
   * 
   * @param obj the object to be serialized as json
   * @return the json string representing the passed in object
   */
  public static String toJsonStr(Object obj) {
    try {
      return new ObjectMapper().writeValueAsString(obj);
    } catch (JsonGenerationException e) {
      throw new RuntimeException(e);
    } catch (JsonMappingException e) {
      throw new RuntimeException(e);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

}