com.hubspot.rosetta.internal.StoredAsJsonSerializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of RosettaCore Show documentation
The newest version!
package com.hubspot.rosetta.internal;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.BeanProperty;
import com.fasterxml.jackson.databind.JavaType;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.jsonFormatVisitors.JsonFormatVisitorWrapper;
import com.fasterxml.jackson.databind.ser.ContextualSerializer;
import com.fasterxml.jackson.databind.ser.std.NonTypedScalarSerializerBase;
import com.fasterxml.jackson.databind.ser.std.StringSerializer;
import java.io.IOException;
import java.lang.reflect.Type;

public class StoredAsJsonSerializer
  extends NonTypedScalarSerializerBase
  implements ContextualSerializer {

  private static final StringSerializer DELEGATE = new StringSerializer();

  public StoredAsJsonSerializer(Class t) {
    super(t);
  }

  @Override
  public void serialize(T value, JsonGenerator jgen, SerializerProvider provider)
    throws IOException {
    ObjectMapper mapper = (ObjectMapper) jgen.getCodec();
    JsonNode tree = mapper.valueToTree(value);
    if (tree.isNull()) {
      provider.defaultSerializeNull(jgen);
    } else {
      jgen.writeString(mapper.writeValueAsString(tree));
    }
  }

  @Override
  public JsonNode getSchema(SerializerProvider provider, Type typeHint) {
    return DELEGATE.getSchema(provider, typeHint);
  }

  @Override
  public void acceptJsonFormatVisitor(
    JsonFormatVisitorWrapper visitor,
    JavaType typeHint
  ) {
    // try/catch is for 2.1.x compatibility
    try {
      DELEGATE.acceptJsonFormatVisitor(visitor, typeHint);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  @Override
  public JsonSerializer createContextual(
    SerializerProvider prov,
    BeanProperty property
  ) throws JsonMappingException {
    if (property == null) {
      return this;
    } else {
      return new ContextualStoredAsJsonSerializer(handledType(), property) {
        @Override
        public void serialize(T value, JsonGenerator gen, SerializerProvider provider)
          throws IOException {
          serializeAsString(value, gen, provider);
        }
      };
    }
  }
}