fr.vsct.tock.nlp.entity.ValueResolverRepository.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2017 VSCT
 *
 * 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 fr.vsct.tock.nlp.entity

import fr.vsct.tock.nlp.entity.date.DateEntityValue
import fr.vsct.tock.nlp.entity.date.DateIntervalEntityValue
import fr.vsct.tock.nlp.entity.temperature.TemperatureValue
import java.util.ResourceBundle
import java.util.concurrent.ConcurrentHashMap
import java.util.logging.Level
import java.util.logging.Logger
import kotlin.reflect.KClass

/**
 * Repository used to register custom [Value] serializer/deserializer.
 */
object ValueResolverRepository {

    private val idClassMap = ConcurrentHashMap>()
    private val classIdMap = ConcurrentHashMap, String>()

    init {
        listOf(
            DateEntityValue::class,
            DateIntervalEntityValue::class,
            NumberValue::class,
            OrdinalValue::class,
            DistanceValue::class,
            TemperatureValue::class,
            VolumeValue::class,
            AmountOfMoneyValue::class,
            EmailValue::class,
            UrlValue::class,
            PhoneNumberValue::class,
            CustomValueWrapper::class,
            UnknownValue::class,
            StringValue::class,
            DurationValue::class
        )
            .forEach { registerType(it) }
    }

    /**
     * Register a new [Value] type.
     *
     * @mapper the jackson mapper
     * @kClass the class to register
     */
    fun  registerType(kClass: KClass) {
        kClass.simpleName!!
            .replace("Value", "")
            .decapitalize()
            .apply {
                idClassMap.put(this, kClass)
                classIdMap.put(kClass, this)
            }
    }

    internal fun getType(id: String): KClass {
        return idClassMap[id] ?: try {
            @Suppress("UNCHECKED_CAST")
            (Class.forName(id) as Class).kotlin
        } catch (e: ClassNotFoundException) {
            val className = ValueResolverRepository::class.qualifiedName
            Logger.getLogger(className).logrb(
                Level.FINE,
                className,
                className,
                null as ResourceBundle?,
                "${e.message}: Please call ValueResolverRepository#registerType during app initialization for value of type $id",
                e
            )

            UnknownValue::class
        }
    }

    internal fun  getId(kClass: KClass): String {
        return classIdMap[kClass] ?: kClass.qualifiedName!!
    }
}