com.jaxio.celerio.configuration.database.h2.H2Extension Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 JAXIO http://www.jaxio.com
 *
 * 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.jaxio.celerio.configuration.database.h2;

import com.jaxio.celerio.configuration.database.Column;
import com.jaxio.celerio.configuration.database.Metadata;
import com.jaxio.celerio.configuration.database.Table;
import com.jaxio.celerio.configuration.database.support.Extension;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.datasource.SingleConnectionDataSource;
import org.springframework.stereotype.Service;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;

@Service
public class H2Extension implements Extension {

    private static final String H2_DEFAULT_SCHEMA_NAME = "PUBLIC";

    private static final String H2_DATABASE = "H2";

    public class Constraint {
        private String tableName;
        private String columnName;
        private String constraint;

        public String getTableName() {
            return tableName;
        }

        public void setTableName(String tableName) {
            this.tableName = tableName;
        }

        public String getColumnName() {
            return columnName;
        }

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

        public String getConstraint() {
            return constraint;
        }

        public void setConstraint(String constraint) {
            this.constraint = constraint;
        }

    }

    @Override
    public boolean applyable(Metadata metadata) {
        return H2_DATABASE.equalsIgnoreCase(metadata.getDatabaseInfo().getDatabaseProductName());
    }

    @Override
    public void apply(Connection connection, Metadata metadata) {
        JdbcTemplate jdbcTemplate = getJdbcTemplate(connection);
        String schemaName = getSchemaName(metadata);

        List constraints = jdbcTemplate.query(
                "SELECT table_name, column_name, check_constraint FROM information_schema.columns WHERE table_schema = ? AND LENGTH(check_constraint) > 0",
                new Object[]{schemaName}, new RowMapper() {
                    @Override
                    public Constraint mapRow(ResultSet rs, int rowNum) throws SQLException {
                        Constraint constraint = new Constraint();
                        constraint.setTableName(rs.getString("table_name"));
                        constraint.setColumnName(rs.getString("column_name"));
                        constraint.setConstraint(rs.getString("check_constraint"));
                        return constraint;
                    }

                });
        for (Constraint constraint : constraints) {
            addEnumValue(metadata, constraint.getTableName(), constraint.getColumnName(), constraint.getConstraint());
        }
    }

    private void addEnumValue(Metadata metadata, String tableName, String columnName, String constraint) {
        Table table = metadata.getTableByName(tableName);
        if (table == null) {
            // TODO: log
            return;
        }
        Column column = table.getColumnByName(columnName);
        if (column == null) {
            // TODO: log
            return;
        }
        if (!column.getType().isString()) {
            // TODO : ordinal not supported yet
            return;
        }
        H2EnumValuesExtractor extractor = new H2EnumValuesExtractor();
        List values = extractor.extract(constraint);
        for (String value : values) {
            column.addEnumValue(value);
        }
    }

    private JdbcTemplate getJdbcTemplate(Connection connection) {
        DataSource dataSource = new SingleConnectionDataSource(connection, true);
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
        return jdbcTemplate;
    }

    private String getSchemaName(Metadata metadata) {
        String schemaName = metadata.getJdbcConnectivity().getSchemaName();
        if (schemaName == null) {
            schemaName = H2_DEFAULT_SCHEMA_NAME;
        }
        return schemaName;
    }
}