com.palantir.conjure.java.serialization.LenientLongModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of conjure-java-jackson-serialization Show documentation
Palantir open source project
There is a newer version: 8.16.0
Show newest version
/*
 * (c) Copyright 2019 Palantir Technologies Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.palantir.conjure.java.serialization;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.cfg.CoercionAction;
import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
import com.fasterxml.jackson.databind.exc.InvalidFormatException;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.fasterxml.jackson.datatype.jdk8.OptionalLongDeserializer;
import com.palantir.logsafe.exceptions.SafeIoException;
import java.io.IOException;
import java.util.OptionalLong;

/**
 * Provides support for the {@link Long} deserialization from JSON string and numeric values regardless of
 *
 * MapperFeature.ALLOW_COERCION_OF_SCALARS
 *
 * configuration.
 */
final class LenientLongModule extends SimpleModule {

    LenientLongModule() {
        super("lenient long");
        // Register to both Long.TYPE and Long.class
        this.addDeserializer(long.class, new LongAsStringDeserializer())
                .addDeserializer(Long.class, new LongAsStringDeserializer())
                .addDeserializer(OptionalLong.class, new OptionalLongAsStringDeserializer());
    }

    private static final class LongAsStringDeserializer extends StdDeserializer {

        private LongAsStringDeserializer() {
            super(Long.TYPE);
        }

        @Override
        public Long deserialize(JsonParser jsonParser, DeserializationContext _ctxt) throws IOException {
            switch (jsonParser.currentToken()) {
                case VALUE_NUMBER_INT:
                    return jsonParser.getLongValue();
                case VALUE_STRING:
                    return parseLong(jsonParser);
                case VALUE_NULL:
                    return null;
                default:
                    throw new SafeIoException("Expected a long value");
            }
        }

        @Override
        public boolean isCachable() {
            return true;
        }

        private static Long parseLong(JsonParser jsonParser) throws IOException {
            String value = jsonParser.getValueAsString();
            try {
                return Long.valueOf(value);
            } catch (NumberFormatException e) {
                InvalidFormatException failure =
                        new InvalidFormatException(jsonParser, "not a valid long value", value, long.class);
                failure.initCause(e);
                throw failure;
            }
        }
    }

    private static final class OptionalLongAsStringDeserializer extends OptionalLongDeserializer {

        private OptionalLongAsStringDeserializer() {}

        @Override
        protected CoercionAction _checkFromStringCoercion(DeserializationContext _ctxt, String _value) {
            return CoercionAction.TryConvert;
        }
    }
}