JsonProduct.java

  1. package gov.usgs.earthquake.product.io;

  3. import gov.usgs.earthquake.product.AbstractContent;
  4. import gov.usgs.earthquake.product.Content;
  5. import gov.usgs.earthquake.product.Product;
  6. import gov.usgs.earthquake.product.ProductId;
  7. import gov.usgs.earthquake.product.URLContent;
  8. import gov.usgs.util.CryptoUtils.Version;
  9. import gov.usgs.util.protocolhandlers.data.Handler;
  10. import gov.usgs.util.StreamUtils;
  11. import gov.usgs.util.XmlUtils;

  13. import java.math.BigDecimal;
  14. import java.net.URI;
  15. import java.net.URL;
  16. import java.util.ArrayList;
  17. import java.util.Base64;
  18. import java.util.Date;
  19. import java.util.HashMap;
  20. import java.util.List;
  21. import java.util.Map;
  22. import javax.json.Json;
  23. import javax.json.JsonArray;
  24. import javax.json.JsonArrayBuilder;
  25. import javax.json.JsonObject;
  26. import javax.json.JsonObjectBuilder;
  27. import javax.json.JsonValue;

  29. public class JsonProduct {

  31.   static {
  32.     // make sure data protocol handler registered
  33.     Handler.register();
  34.   }

  36.   /**
  37.    * Convert product object to json.
  38.    *
  39.    * @param product a product
  40.    * @return a json object
  41.    * @throws Exception if error occurs
  42.    */
  43.   public JsonObject getJsonObject(final Product product) throws Exception {
  44.     JsonObjectBuilder json = Json.createObjectBuilder();

  46.     json.add("contents", getContentsJson(product.getContents()));
  47.     JsonObjectBuilder geometry = getGeometryJson(product);
  48.     if (geometry == null) {
  49.       json.addNull("geometry");
  50.     } else {
  51.       json.add("geometry", geometry);
  52.     }
  53.     final ProductId id = product.getId();
  54.     json.add("id", getIdJson(id));
  55.     json.add("links", getLinksJson(product.getLinks()));
  56.     json.add("properties", getPropertiesJson(product.getProperties()));
  57.     if (product.getSignature() == null) {
  58.       json.addNull("signature");
  59.     } else {
  60.       json.add("signature", product.getSignature());
  61.     }
  62.     json.add("signatureVersion", product.getSignatureVersion().toString());
  63.     json.add("status", product.getStatus());
  64.     json.add("type", "Feature");
  65.     return json.build();
  66.   }

  68.   /**
  69.    * Convert json object to product.
  70.    * @param json a json object
  71.    * @return a product
  72.    * @throws Exception if error occurs
  73.    */
  74.   public Product getProduct(final JsonObject json) throws Exception {
  75.     Product product = new Product(getId(json.getJsonObject("id")));
  76.     product.setContents(getContents(json.getJsonArray("contents")));
  77.     product.setLinks(getLinks(json.getJsonArray("links")));
  78.     product.setProperties(getProperties(json.getJsonObject("properties")));
  79.     product.setStatus(json.getString("status"));
  80.     try {
  81.       product.setSignature(json.getString("signature"));
  82.     } catch (Exception e) {
  83.       product.setSignature(null);
  84.     }
  85.     product.setSignatureVersion(Version.fromString(json.getString("signatureVersion")));
  86.     product.setTrackerURL(new URL("data:,"));
  87.     return product;
  88.   }

  90.   /**
  91.    * Convert contents map to json.
  92.    *
  93.    * @param contents contents map
  94.    * @return JSOnArrayBuilder
  95.    * @throws Exception if error occurs
  96.    */
  97.   public JsonArrayBuilder getContentsJson(final Map<String, Content> contents) throws Exception {
  98.     final JsonArrayBuilder builder = Json.createArrayBuilder();
  99.     for (final String path : contents.keySet()) {
  100.       final Content content = contents.get(path);
  101.       final JsonObjectBuilder jsonContent =
  102.           Json.createObjectBuilder()
  103.               .add("length", content.getLength())
  104.               .add("modified", XmlUtils.formatDate(content.getLastModified()))
  105.               .add("path", path)
  106.               .add("sha256", content.getSha256())
  107.               .add("type", content.getContentType());
  108.       if (content instanceof URLContent) {
  109.         jsonContent.add("url", ((URLContent) content).getURL().toString());
  110.       } else if ("".equals(path)) {
  111.         jsonContent.add(
  112.             "url",
  113.             "data:"
  114.                 + content.getContentType()
  115.                 + ";base64,"
  116.                 + Base64.getEncoder()
  117.                     .encodeToString(StreamUtils.readStream(content.getInputStream())));
  118.       } else {
  119.         // no url, will throw parse error
  120.         // this is used to get upload urls, and returned object includes urls...
  121.         jsonContent.addNull("url");
  122.       }
  123.       builder.add(jsonContent);
  124.     }
  125.     return builder;
  126.   }

  128.   /**
  129.    * Convert contents json to map.
  130.    *
  131.    * @param json JsonArray
  132.    * @return Contents map
  133.    * @throws Exception if error occurs
  134.    */
  135.   public Map<String, Content> getContents(final JsonArray json) throws Exception {
  136.     Map<String, Content> contents = new HashMap<String, Content>();
  137.     for (JsonValue value : json) {
  138.       JsonObject object = value.asJsonObject();
  139.       Long length = object.getJsonNumber("length").longValue();
  140.       Date modified = XmlUtils.getDate(object.getString("modified"));
  141.       String path = object.getString("path");
  142.       String sha256 = object.getString("sha256");
  143.       String type = object.getString("type");
  144.       String url = object.getString("url");

  146.       AbstractContent content = new URLContent(new URL(url));
  147.       content.setContentType(type);
  148.       content.setLastModified(modified);
  149.       content.setLength(length);
  150.       content.setSha256(sha256);
  151.       contents.put(path, content);
  152.     }
  153.     return contents;
  154.   }

  156.   /**
  157.    * Create json geometry from product properties.
  158.    *
  159.    * @param product a product
  160.    * @return JSON geometry via JsonObjectBuilder
  161.    * @throws Exception if error occurs
  162.    */
  163.   public JsonObjectBuilder getGeometryJson(final Product product) throws Exception {
  164.     final BigDecimal latitude = product.getLatitude();
  165.     final BigDecimal longitude = product.getLongitude();
  166.     final BigDecimal depth = product.getDepth();
  167.     if (latitude != null || longitude != null || depth != null) {
  168.       final JsonArrayBuilder coordinates = Json.createArrayBuilder();
  169.       if (latitude != null) {
  170.         coordinates.add(latitude);
  171.       } else {
  172.         coordinates.addNull();
  173.       }
  174.       if (longitude != null) {
  175.         coordinates.add(longitude);
  176.       } else {
  177.         coordinates.addNull();
  178.       }
  179.       if (depth != null) {
  180.         coordinates.add(depth);
  181.       } else {
  182.         coordinates.addNull();
  183.       }
  184.       return Json.createObjectBuilder()
  185.           .add("type", "Point")
  186.           .add("coordinates", coordinates);
  187.     }
  188.     return null;
  189.   }

  191.   /**
  192.    * Convert json id to ProductId object.
  193.    *
  194.    * @param json A JsonObject ID
  195.    * @return a productId
  196.    * @throws Exception if error occurs
  197.    */
  198.   public ProductId getId(final JsonObject json) throws Exception {
  199.     final String code = json.getString("code");
  200.     final String source = json.getString("source");
  201.     final String type = json.getString("type");
  202.     final Date updateTime = XmlUtils.getDate(json.getString("updateTime"));
  203.     return new ProductId(source, type, code, updateTime);
  204.   }

  206.   /**
  207.    * Convert ProductId to json object.
  208.    * @param id A ProductId
  209.    * @return JsonObjectBuilder
  210.    * @throws Exception if error occurs
  211.    */
  212.   public JsonObjectBuilder getIdJson(final ProductId id) throws Exception {
  213.     return Json.createObjectBuilder()
  214.         .add("code", id.getCode())
  215.         .add("source", id.getSource())
  216.         .add("type", id.getType())
  217.         .add("updateTime", XmlUtils.formatDate(id.getUpdateTime()));
  218.   }

  220.   /**
  221.    * Convert json links to map.
  222.    * @param json a Jsonarray
  223.    * @return a Map of links
  224.    * @throws Exception if error occurs
  225.    */
  226.   public Map<String, List<URI>> getLinks(final JsonArray json) throws Exception {
  227.     final Map<String, List<URI>> links = new HashMap<String, List<URI>>();
  228.     for (final JsonValue value : json) {
  229.       final JsonObject link = value.asJsonObject();
  230.       final String relation = link.getString("relation");
  231.       final URI uri = new URI(link.getString("uri"));
  232.       List<URI> relationLinks = links.get(relation);
  233.       if (relationLinks == null) {
  234.         relationLinks = new ArrayList<URI>();
  235.         links.put(relation, relationLinks);
  236.       }
  237.       relationLinks.add(uri);
  238.     }
  239.     return links;
  240.   }

  242.   /**
  243.    * Convert links map to json.
  244.    *
  245.    * @param links map
  246.    * @return JsonArray of JsonArrayBuilder
  247.    * @throws Exception if error occurs
  248.    */
  249.   public JsonArrayBuilder getLinksJson(final Map<String, List<URI>> links) throws Exception {
  250.     final JsonArrayBuilder builder = Json.createArrayBuilder();
  251.     for (final String relation : links.keySet()) {
  252.       final List<URI> relationLinks = links.get(relation);
  253.       for (final URI uri : relationLinks) {
  254.         builder.add(
  255.             Json.createObjectBuilder().add("relation", relation).add("uri", uri.toString()));
  256.       }
  257.     }
  258.     return builder;
  259.   }

  261.   /**
  262.    * Convert properties json to map.
  263.    * @param json JsonObject properties
  264.    * @return A map
  265.    * @throws Exception if error occurs
  266.    */
  267.   public Map<String, String> getProperties(final JsonObject json) throws Exception {
  268.     final Map<String, String> properties = new HashMap<String, String>();
  269.     for (final String name : json.keySet()) {
  270.       properties.put(name, json.getString(name));
  271.     }
  272.     return properties;
  273.   }

  275.   /**
  276.    * Convert properties map to json.
  277.    *
  278.    * @param properties Map of properties
  279.    * @return JsonObjectBuilder
  280.    * @throws Exception if error occurs
  281.    */
  282.   public JsonObjectBuilder getPropertiesJson(final Map<String, String> properties)
  283.       throws Exception {
  284.     final JsonObjectBuilder builder = Json.createObjectBuilder();
  285.     for (final String name : properties.keySet()) {
  286.       builder.add(name, properties.get(name));
  287.     }
  288.     return builder;
  289.   }
  290. }
Created with JaCoCo 0.8.8.202204050719