com.mindee.parsing.generated.GeneratedV1PredictionDeserializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mindee-api-java Show documentation

Java Library to call Mindee's Off-The-Shelf and Custom APIs

The newest version!

package com.mindee.parsing.generated;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.mindee.product.generated.GeneratedV1Document;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * JSON deserializer for generated documents v1.x.
 */
public class GeneratedV1PredictionDeserializer extends StdDeserializer {

  public GeneratedV1PredictionDeserializer(Class vc) {
    super(vc);
  }

  private static final ObjectMapper mapper = new ObjectMapper();

  public GeneratedV1PredictionDeserializer() {
    this(null);
  }

  @Override
  public GeneratedV1Document deserialize(
      JsonParser jsonParser,
      DeserializationContext deserializationContext
  ) throws IOException {
    ObjectNode rootNode = jsonParser.getCodec().readTree(jsonParser);
    Map features = new HashMap<>();

    for (Iterator> subNode = rootNode.fields(); subNode.hasNext(); ) {
      Map.Entry featureNode = subNode.next();
      String featureName = featureNode.getKey();

      GeneratedFeature feature;

      if (featureNode.getValue().isArray()) {
        feature = new GeneratedFeature(true);
        for (JsonNode item : featureNode.getValue()) {
          GeneratedObject value = mapper
              .readerFor(new TypeReference() {})
              .readValue(item);
          feature.add(value);
        }
        features.put(featureName, feature);
      } else {
        feature = new GeneratedFeature(false);
        GeneratedObject value = mapper
            .readerFor(new TypeReference() {})
            .readValue(featureNode.getValue());
        feature.add(value);
      }
      features.put(featureName, feature);
    }
    return new GeneratedV1Document(features);
  }
}