ai.vespa.metricsproxy.metric.model.json.JacksonUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics-proxy Show documentation
There is a newer version: 8.441.21
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.metricsproxy.metric.model.json;

import com.yahoo.json.Jackson;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.Version;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.module.SimpleModule;

import java.io.IOException;
import java.text.DecimalFormat;
import java.text.DecimalFormatSymbols;
import java.text.NumberFormat;
import java.util.Locale;

/**
 * @author smorgrav
 * @author gjoranv
 */
public class JacksonUtil {

    private static final ThreadLocal withinLongRangeFormat = ThreadLocal.withInitial(JacksonUtil::createWithinLongRangeFormat);
    private static final ThreadLocal outsideLongRangeFormat = ThreadLocal.withInitial(JacksonUtil::createOutsideLongRangeFormat);
    private static final ObjectMapper objectMapper = createObjectMapper();
    private static DecimalFormat createWithinLongRangeFormat() {
        DecimalFormat df = new DecimalFormat("#.####", new DecimalFormatSymbols(Locale.ENGLISH));
        df.setMaximumFractionDigits(13);
        return df;
    }
    private static DecimalFormat createOutsideLongRangeFormat() {
        DecimalFormat df = new DecimalFormat("#.0###", new DecimalFormatSymbols(Locale.ENGLISH));
        df.setMaximumFractionDigits(13);
        return df;
    }
    /**
     * Returns an object mapper with a custom floating point serializer to avoid scientific notation
     */
    private static ObjectMapper createObjectMapper() {
        ObjectMapper mapper = Jackson.createMapper();
        SimpleModule module = new SimpleModule("DoubleSerializer",
                                               new Version(1, 0, 0, "", null, null));
        module.addSerializer(Double.class, new DoubleSerializer());
        mapper.registerModule(module);
        return mapper;
    }
    public static ObjectMapper objectMapper() {
        return objectMapper;
    }
    /**
     * Returns an object mapper with a custom floating point serializer to avoid scientific notation
     */
    public static void writeDouble(JsonGenerator jgen, Double value) throws IOException {
        jgen.writeNumber(format(value));
    }

    public static String format(Double value) {
        return format(value, withinLongRangeFormat.get(), outsideLongRangeFormat.get());
    }

    private static String format(Double value, NumberFormat withinLongRange, NumberFormat outsideLongRange) {
        if ((value <= Long.MAX_VALUE) && (value >= Long.MIN_VALUE)) {
            return withinLongRange.format(value);
        } else {
            return outsideLongRange.format(value);
        }
    }

    public static class DoubleSerializer extends JsonSerializer {
        private final DecimalFormat withinLongRangeFormat = createWithinLongRangeFormat();
        private final DecimalFormat outsideLongRangeFormat = createOutsideLongRangeFormat();
        @Override
        public void serialize(Double value, JsonGenerator jgen, SerializerProvider provider) throws IOException {
            jgen.writeNumber(format(value, withinLongRangeFormat, outsideLongRangeFormat));
        }
    }

}