com.factual.driver.ReadResponse 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.util.Collection;
import java.util.List;
import java.util.Map;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import com.google.common.base.Function;
import com.google.common.collect.Collections2;
import com.google.common.collect.Lists;



/**
 * Represents the response from running a fetch request against Factual, such as
 * a geolocation based query for specific places entities.
 *
 * @author aaron
 */
public class ReadResponse extends Response implements Tabular {
  private List> data = Lists.newArrayList();


  /**
   * Constructor, parses from a JSON response String.
   */
  public ReadResponse(InternalResponse resp) {
    super(resp);
    try{
      JSONObject rootJsonObj = new JSONObject(resp.getContent());
      Response.withMeta(this, rootJsonObj);
      JSONObject respJson = rootJsonObj.getJSONObject(Constants.RESPONSE);
      Object dataObj = respJson.get(Constants.QUERY_DATA);
      if (dataObj instanceof JSONArray) {
        data = JsonUtil.data(respJson.getJSONArray(Constants.QUERY_DATA));
      } else if (dataObj instanceof JSONObject) {
        data.add(JsonUtil.toMap((JSONObject) dataObj));
      }
    } catch (JSONException e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * @return The full JSON response from Factual
   */
  @Override
  public String getJson() {
    return resp.getContent();
  }

  /**
   * @return the first data record or null if no data was returned.
   */
  public Map first() {
    if(data.isEmpty()) {
      return null;
    } else {
      return data.get(0);
    }
  }

  /**
   * An ordered collection of the main data returned by Factual. Represented as
   * Maps, where each Map is a record in the results.
   * 
   * @return the main data returned by Factual.
   */
  @Override
  public List> getData() {
    return data;
  }

  /**
   * @return the size of the result set
   */
  public int size() {
    return data.size();
  }

  /**
   * @return a Collection of all String values found in this Response's data
   *         rows as the field attribute.
   */
  public Collection mapStrings(final String field) {
    return Collections2.transform(data, new Function, String>() {
      @Override
      public String apply(Map row) {
        Object val = row.get(field);
        return val != null ? val.toString() : null;
      }});
  }

  //TODO: Bradley: "mapToStrings, mapToDoubles, etc."

}