All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.hubspot.algebra.ResultSerializer Maven / Gradle / Ivy

    package com.hubspot.algebra;

import static com.hubspot.algebra.ResultModule.CASE_FIELD_NAME;
import static com.hubspot.algebra.ResultModule.ERROR_FIELD_NAME;
import static com.hubspot.algebra.ResultModule.OK_FIELD_NAME;

import com.fasterxml.jackson.annotation.JsonAnyGetter;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JavaType;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;
import com.google.common.collect.Multimap;
import com.google.common.collect.Table;
import com.hubspot.algebra.ResultModule.Case;
import java.io.IOException;
import java.util.Map;

public class ResultSerializer extends StdSerializer> {

  ResultSerializer(JavaType type) {
    super(type);
  }

  @Override
  public void serialize(
    Result value,
    JsonGenerator gen,
    SerializerProvider provider
  ) throws IOException {
    gen.writeStartObject();

    if (value.isErr()) {
      Object err = value.unwrapErrOrElseThrow();
      serializeValue(ERROR_FIELD_NAME, err, gen, provider);
      gen.writeStringField(CASE_FIELD_NAME, Case.ERR.name());
    } else {
      Object ok = value.unwrapOrElseThrow();
      serializeValue(OK_FIELD_NAME, ok, gen, provider);
      gen.writeStringField(CASE_FIELD_NAME, Case.OK.name());
    }

    gen.writeEndObject();
  }

  private static void serializeValue(
    String fieldName,
    Object value,
    JsonGenerator gen,
    SerializerProvider provider
  ) throws IOException {
    Object flattenedValue = flattenValue(value);
    JsonSerializer serializer = provider
      .findTypedValueSerializer(flattenedValue.getClass(), true, null)
      .unwrappingSerializer(null);
    if (!serializer.isUnwrappingSerializer()) {
      gen.writeFieldName(fieldName);
    }
    serializer.serialize(flattenedValue, gen, provider);
  }

  private static Object flattenValue(Object value) {
    if (value instanceof Map) {
      return new MapFlattener((Map) value);
    } else if (value instanceof Multimap) {
      return new MapFlattener(((Multimap) value).asMap());
    } else if (value instanceof Table) {
      return new MapFlattener(((Table) value).rowMap());
    } else {
      return value;
    }
  }

  private static class MapFlattener {

    private final Map map;

    private MapFlattener(Map map) {
      this.map = map;
    }

    @JsonAnyGetter
    public Map getMap() {
      return map;
    }
  }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy