com.mybatis.jpa.definition.AnnotationDefinitionRegistry Maven / Gradle / Ivy

Go to download
package com.mybatis.jpa.definition;

import com.mybatis.jpa.annotation.InsertDefinition;
import com.mybatis.jpa.annotation.UpdateDefinition;
import com.mybatis.jpa.definition.adaptor.AnnotationAdaptable;
import com.mybatis.jpa.definition.adaptor.InsertDefinitionAdaptor;
import com.mybatis.jpa.definition.adaptor.UpdateDefinitionAdaptor;

import java.lang.annotation.Annotation;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * @author svili
 **/
public class AnnotationDefinitionRegistry {

    private final Map, AnnotationAdaptable> ANNOTATION_ADAPTORS = new HashMap<>();

    public AnnotationDefinitionRegistry() {
        registerAdaptor(InsertDefinition.class, new InsertDefinitionAdaptor());
        registerAdaptor(UpdateDefinition.class, new UpdateDefinitionAdaptor());
    }

    public AnnotationAdaptable resolveAdaptor(Class annotationType) {
        if (annotationType == null) {
            return null;
        }
        return ANNOTATION_ADAPTORS.get(annotationType);
    }

    public void registerAdaptor(Class annotationType, AnnotationAdaptable adaptor) {
        if (annotationType == null) {
            throw new RuntimeException("The parameter annotationType cannot be null");
        }
        if (adaptor == null) {
            throw new RuntimeException("The parameter register cannot be null");
        }
        ANNOTATION_ADAPTORS.put(annotationType, adaptor);
    }

    public Map, AnnotationAdaptable> getAnnotationAdaptors() {
        return Collections.unmodifiableMap(ANNOTATION_ADAPTORS);
    }
}