tech.ytsaurus.client.rows.JavaPersistenceApi Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ytsaurus-client Show documentation
YTsaurus client
The newest version!
package tech.ytsaurus.client.rows;

import java.lang.annotation.Annotation;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import javax.annotation.Nullable;

import static tech.ytsaurus.core.utils.ClassUtils.anyMatchWithAnnotation;
import static tech.ytsaurus.core.utils.ClassUtils.getValueOfAnnotationProperty;

class JavaPersistenceApi {
    private static final List PACKAGES = List.of(
            "javax.persistence", "jakarta.persistence"
    );
    private static final String ENTITY = "Entity";
    private static final String ENTITY_NAME = "name";
    private static final String TRANSIENT = "Transient";
    private static final String COLUMN = "Column";
    private static final String COLUMN_NAME = "name";
    private static final String COLUMN_NULLABLE = "nullable";
    private static final String COLUMN_PRECISION = "precision";
    private static final String COLUMN_SCALE = "scale";
    private static final String COLUMN_DEFINITION = "columnDefinition";
    private static final String EMBEDDABLE = "Embeddable";
    private static final String EMBEDDED = "Embedded";
    private static final Set ENTITY_ANNOTATIONS = getAnnotationsFor(ENTITY);
    private static final Set TRANSIENT_ANNOTATIONS = getAnnotationsFor(TRANSIENT);
    private static final Set COLUMN_ANNOTATIONS = getAnnotationsFor(COLUMN);
    private static final Set EMBEDDABLE_ANNOTATIONS = getAnnotationsFor(EMBEDDABLE);
    private static final Set EMBEDDED_ANNOTATIONS = getAnnotationsFor(EMBEDDED);

    private JavaPersistenceApi() {
    }

    static Set transientAnnotations() {
        return TRANSIENT_ANNOTATIONS;
    }

    static Set entityAnnotations() {
        return ENTITY_ANNOTATIONS;
    }

    static Set columnAnnotations() {
        return COLUMN_ANNOTATIONS;
    }

    static Set embeddableAnnotations() {
        return EMBEDDABLE_ANNOTATIONS;
    }

    static Set embeddedAnnotations() {
        return EMBEDDED_ANNOTATIONS;
    }
    static boolean isColumnAnnotationPresent(@Nullable Annotation annotation) {
        return annotation != null &&
                anyMatchWithAnnotation(annotation, JavaPersistenceApi.columnAnnotations());
    }

    static String getEntityName(Annotation entityAnnotation) {
        return getValueOfAnnotationProperty(entityAnnotation, ENTITY_NAME);
    }

    static String getColumnName(Annotation columnAnnotation) {
        return getValueOfAnnotationProperty(columnAnnotation, COLUMN_NAME);
    }

    static boolean isColumnNullable(Annotation columnAnnotation) {
        return getValueOfAnnotationProperty(columnAnnotation, COLUMN_NULLABLE);
    }

    static int getColumnPrecision(Annotation columnAnnotation) {
        return getValueOfAnnotationProperty(columnAnnotation, COLUMN_PRECISION);
    }

    static int getColumnScale(Annotation columnAnnotation) {
        return getValueOfAnnotationProperty(columnAnnotation, COLUMN_SCALE);
    }

    static String getColumnDefinition(Annotation columnAnnotation) {
        return getValueOfAnnotationProperty(columnAnnotation, COLUMN_DEFINITION);
    }

    private static Set getAnnotationsFor(String annotationName) {
        return PACKAGES.stream()
                .map(pack -> pack + "." + annotationName)
                .collect(Collectors.toUnmodifiableSet());
    }
}