JacksonJsonHandler.java

package com.meilisearch.sdk.json;

import com.fasterxml.jackson.annotation.JsonInclude.Include;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.fasterxml.jackson.databind.util.StdDateFormat;
import com.meilisearch.sdk.exceptions.JsonDecodingException;
import com.meilisearch.sdk.exceptions.JsonEncodingException;
import com.meilisearch.sdk.exceptions.MeilisearchException;
import com.meilisearch.sdk.model.FilterableAttributesConfig;
import java.io.IOException;

public class JacksonJsonHandler implements JsonHandler {

    private final ObjectMapper mapper;

    /**
     * this constructor uses a default ObjectMapper with enabled 'FAIL_ON_UNKNOWN_PROPERTIES'
     * feature.
     */
    public JacksonJsonHandler() {
        this.mapper = new ObjectMapper();
        this.mapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
        this.mapper.setDateFormat(new StdDateFormat().withColonInTimeZone(true));
        this.mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        registerFilterableAttributesModule(this.mapper);
    }

    /**
     * @param mapper ObjectMapper
     */
    public JacksonJsonHandler(ObjectMapper mapper) {
        this.mapper = mapper;
        registerFilterableAttributesModule(this.mapper);
    }

    /** {@inheritDoc} */
    @Override
    public String encode(Object o) throws MeilisearchException {
        if (o != null && o.getClass() == String.class) {
            return (String) o;
        }
        try {
            this.mapper.setSerializationInclusion(Include.NON_NULL);
            return mapper.writeValueAsString(o);
        } catch (JsonProcessingException e) {
            throw new JsonEncodingException(e);
        }
    }

    /** {@inheritDoc} */
    @SuppressWarnings("unchecked")
    @Override
    public <T> T decode(Object o, Class<T> targetClass, Class<?>... parameters)
            throws MeilisearchException {
        if (o == null) {
            throw new JsonDecodingException("Response to deserialize is null");
        }
        if (targetClass == String.class) {
            return (T) o;
        }
        try {
            if (parameters == null || parameters.length == 0) {
                return mapper.readValue((String) o, targetClass);
            } else {
                return mapper.readValue(
                        (String) o,
                        mapper.getTypeFactory().constructParametricType(targetClass, parameters));
            }
        } catch (IOException e) {
            throw new JsonDecodingException(e);
        }
    }

    private void registerFilterableAttributesModule(ObjectMapper mapper) {
        SimpleModule filterableModule = new SimpleModule();
        filterableModule.addSerializer(
                FilterableAttributesConfig.class,
                new JacksonFilterableAttributesConfigSerializer());
        filterableModule.addDeserializer(
                FilterableAttributesConfig.class,
                new JacksonFilterableAttributesConfigDeserializer());
        mapper.registerModule(filterableModule);
    }
}