ru.curs.celesta.dbutils.adaptors.column.ColumnDefinerFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of celesta-sql Show documentation
The newest version!
package ru.curs.celesta.dbutils.adaptors.column;

import ru.curs.celesta.DBType;
import ru.curs.celesta.score.BinaryColumn;
import ru.curs.celesta.score.BooleanColumn;
import ru.curs.celesta.score.Column;
import ru.curs.celesta.score.DateTimeColumn;
import ru.curs.celesta.score.DecimalColumn;
import ru.curs.celesta.score.FloatingColumn;
import ru.curs.celesta.score.IntegerColumn;
import ru.curs.celesta.score.StringColumn;
import ru.curs.celesta.score.ZonedDateTimeColumn;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;
import java.util.function.Supplier;

/**
 * Column definer factory.

 * 

 * Usage pattern: {@code ColumnDefinerFactory.getColumnDefiner(...);}
 */
public final class ColumnDefinerFactory {

    private static final Map>, ColumnDefiner>>
                         COLUMN_DEFINERS = new HashMap<>();
    private static final Map>, ColumnDefiner>>
                         FACTORY_METHODS = new HashMap<>();
    private static final Map>, Supplier>
                         H2_METHODS = new HashMap<>();
    private static final Map>, Supplier>
                         POSTGRES_METHODS = new HashMap<>();
    private static final Map>, Supplier>
                         MS_SQL_METHODS = new HashMap<>();
    private static final Map>, Supplier>
                         ORA_METHODS = new HashMap<>();
    private static final Map>, Supplier>
                        FIREBIRD_METHODS = new HashMap<>();

    static {
        FACTORY_METHODS.put(
                DBType.H2,
                cls -> H2_METHODS.get(cls).get()
        );
        FACTORY_METHODS.put(
                DBType.POSTGRESQL,
                cls -> POSTGRES_METHODS.get(cls).get()
        );
        FACTORY_METHODS.put(
                DBType.MSSQL,
                cls -> MS_SQL_METHODS.get(cls).get()
        );
        FACTORY_METHODS.put(
                DBType.ORACLE,
                cls -> ORA_METHODS.get(cls).get()
        );
        FACTORY_METHODS.put(
            DBType.FIREBIRD,
            cls -> FIREBIRD_METHODS.get(cls).get()
        );

        H2_METHODS.put(IntegerColumn.class, H2IntegerColumnDefiner::new);
        H2_METHODS.put(FloatingColumn.class, H2FloatingColumnDefiner::new);
        H2_METHODS.put(DecimalColumn.class, H2DecimalColumnDefiner::new);
        H2_METHODS.put(BooleanColumn.class, H2BooleanColumnDefiner::new);
        H2_METHODS.put(StringColumn.class, H2StringColumnDefiner::new);
        H2_METHODS.put(BinaryColumn.class, H2BinaryColumnDefiner::new);
        H2_METHODS.put(DateTimeColumn.class, H2DateTimeColumnDefiner::new);
        H2_METHODS.put(ZonedDateTimeColumn.class, H2ZonedDateTimeColumnDefiner::new);

        POSTGRES_METHODS.put(IntegerColumn.class, PostgresIntegerColumnDefiner::new);
        POSTGRES_METHODS.put(FloatingColumn.class, PostgresFloatingColumnDefiner::new);
        POSTGRES_METHODS.put(DecimalColumn.class, PostgresDecimalColumnDefiner::new);
        POSTGRES_METHODS.put(BooleanColumn.class, PostgresBooleanColumnDefiner::new);
        POSTGRES_METHODS.put(StringColumn.class, PostgresStringColumnDefiner::new);
        POSTGRES_METHODS.put(BinaryColumn.class, PostgresBinaryColumnDefiner::new);
        POSTGRES_METHODS.put(DateTimeColumn.class, PostgresDateTimeColumnDefiner::new);
        POSTGRES_METHODS.put(ZonedDateTimeColumn.class, PostgresZonedDateTimeColumnDefiner::new);

        MS_SQL_METHODS.put(IntegerColumn.class, MsSqlIntegerColumnDefiner::new);
        MS_SQL_METHODS.put(FloatingColumn.class, MsSqlFloatingColumnDefiner::new);
        MS_SQL_METHODS.put(DecimalColumn.class, MsSqlDecimalColumnDefiner::new);
        MS_SQL_METHODS.put(BooleanColumn.class, MsSqlBooleanColumnDefiner::new);
        MS_SQL_METHODS.put(StringColumn.class, MsSqlStringColumnDefiner::new);
        MS_SQL_METHODS.put(BinaryColumn.class, MsSqlBinaryColumnDefiner::new);
        MS_SQL_METHODS.put(DateTimeColumn.class, MsSqlDateTimeColumnDefiner::new);
        MS_SQL_METHODS.put(ZonedDateTimeColumn.class, MsSqlZonedDateTimeColumnDefiner::new);

        ORA_METHODS.put(IntegerColumn.class, OraIntegerColumnDefiner::new);
        ORA_METHODS.put(FloatingColumn.class, OraFloatingColumnDefiner::new);
        ORA_METHODS.put(DecimalColumn.class, OraDecimalColumnDefiner::new);
        ORA_METHODS.put(BooleanColumn.class, OraBooleanColumnDefiner::new);
        ORA_METHODS.put(StringColumn.class, OraStringColumnDefiner::new);
        ORA_METHODS.put(BinaryColumn.class, OraBinaryColumnDefiner::new);
        ORA_METHODS.put(DateTimeColumn.class, OraDateTimeColumnDefiner::new);
        ORA_METHODS.put(ZonedDateTimeColumn.class, OraZonedDateTimeColumnDefiner::new);


        FIREBIRD_METHODS.put(IntegerColumn.class, FireBirdIntegerColumnDefiner::new);
        FIREBIRD_METHODS.put(FloatingColumn.class, FireBirdFloatingColumnDefiner::new);
        FIREBIRD_METHODS.put(DecimalColumn.class, FireBirdDecimalColumnDefiner::new);
        FIREBIRD_METHODS.put(BooleanColumn.class, FireBirdBooleanColumnDefiner::new);
        FIREBIRD_METHODS.put(StringColumn.class, FireBirdStringColumnDefiner::new);
        FIREBIRD_METHODS.put(BinaryColumn.class, FireBirdBinaryColumnDefiner::new);
        FIREBIRD_METHODS.put(DateTimeColumn.class, FireBirdDateTimeColumnDefiner::new);
        FIREBIRD_METHODS.put(ZonedDateTimeColumn.class, FireBirdZonedDateTimeColumnDefiner::new);
    }

    private ColumnDefinerFactory() {
        throw new AssertionError();
    }

    /**
     * Returns a column definer depending on the DB type and column type.
     *
     * @param dbType  DB type.
     * @param cls  column class
     */
    public static ColumnDefiner getColumnDefiner(DBType dbType, Class> cls) {
        Map>, ColumnDefiner> definers = COLUMN_DEFINERS.computeIfAbsent(
                dbType,
                dt -> new HashMap<>()
        );

        return definers.computeIfAbsent(
                cls,
                c -> FACTORY_METHODS.get(dbType).apply(c)
        );
    }

}