com.eclecticlogic.pedal.dialect.postgresql.CopyAttribute Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pedal-dialect Show documentation

Pedal-dialect is a collection of dialect (database) and provider (e.g. Hibernate) specific implementations for use with JPA. It enables non-standard SQL data types (arrays, sets, bit-sets) and dialect specific features such as support for the Postgresql Copy command.

There is a newer version: 1.5.6

Show newest version

/*
 * Copyright (c) 2017 Eclectic Logic LLC
 *
 * 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.eclecticlogic.pedal.dialect.postgresql;

import javax.persistence.Column;
import javax.persistence.Convert;
import javax.persistence.Converter;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.atomic.AtomicLong;

/**
 * Created by kabram on 4/6/17.
 */
public class CopyAttribute {

    private final List methods = new ArrayList<>();
    private AtomicLong variableCounter = new AtomicLong();
    private ThreadLocal currentVariable = new ThreadLocal<>();
    private String columnName; // Name of db column.


    public List getMethods() {
        return methods;
    }


    public Method getEntityMethod() {
        return methods.get(0);
    }


    public String getColumnName() {
        return columnName;
    }


    public void setColumnName(String columnName) {
        this.columnName = columnName;
    }


    public String getNewVariable() {
        currentVariable.set("v" + variableCounter.getAndIncrement());
        return getVariable();
    }


    public String getVariable() {
        return currentVariable.get();
    }


    public boolean isCopyConverter() {
        return getEntityMethod().isAnnotationPresent(CopyConverter.class);
    }


    public Class getCopyConverterClass() {
        return getEntityMethod().getAnnotation(CopyConverter.class).value();
    }


    public boolean isCopyAsBitString() {
        return getEntityMethod().isAnnotationPresent(CopyAsBitString.class);
    }


    public int getColumnLength() {
        return getEntityMethod().getAnnotation(Column.class).length();
    }


    public boolean isJpaConverter() {
        return getEntityMethod().isAnnotationPresent(Convert.class);
    }


    public Class getJpaConverterClass() {
        return getEntityMethod().getAnnotation(Convert.class).converter();
    }


    public boolean isCollection() {
        return Collection.class.isAssignableFrom(getEntityMethod().getReturnType());
    }


    public boolean isCopyEmptyAsNull() {
        return getEntityMethod().isAnnotationPresent(CopyEmptyAsNull.class);
    }


    public boolean isJoinColumn() {
        return getEntityMethod().isAnnotationPresent(JoinColumn.class);
    }


    public Method getJoinColumnIdMethod() {
        return Arrays.stream(getEntityMethod().getReturnType().getMethods()) //
                .filter(it -> it.isAnnotationPresent(Id.class)).findFirst().get();
    }
}