| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 org.apache.metamodel.jdbc; |
| |
| import java.sql.Connection; |
| import java.sql.SQLException; |
| import java.sql.Statement; |
| |
| import org.apache.metamodel.create.AbstractTableCreationBuilder; |
| import org.apache.metamodel.jdbc.dialects.IQueryRewriter; |
| import org.apache.metamodel.schema.Column; |
| import org.apache.metamodel.schema.ColumnType; |
| import org.apache.metamodel.schema.Schema; |
| import org.apache.metamodel.schema.Table; |
| import org.apache.metamodel.util.FileHelper; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| /** |
| * {@link CreateTableBuilder} implementation that issues a SQL CREATE TABLE |
| * statement. |
| * |
| * @author Kasper Sørensen |
| */ |
| final class JdbcCreateTableBuilder extends AbstractTableCreationBuilder<JdbcUpdateCallback> { |
| |
| private static final Logger logger = LoggerFactory.getLogger(JdbcCreateTableBuilder.class); |
| |
| public JdbcCreateTableBuilder(JdbcUpdateCallback updateCallback, Schema schema, String name) { |
| super(updateCallback, schema, name); |
| if (!(schema instanceof JdbcSchema)) { |
| throw new IllegalArgumentException("Not a valid JDBC schema: " + schema); |
| } |
| } |
| |
| @Override |
| public Table execute() { |
| final String sql = createSqlStatement(); |
| logger.info("Create table statement created: {}", sql); |
| |
| Connection connection = getUpdateCallback().getConnection(); |
| Statement st = null; |
| try { |
| st = connection.createStatement(); |
| int rowsAffected = st.executeUpdate(sql); |
| logger.debug("Create table statement executed, {} rows affected", rowsAffected); |
| } catch (SQLException e) { |
| throw JdbcUtils.wrapException(e, "execute create table statement: " + sql); |
| } finally { |
| FileHelper.safeClose(st); |
| } |
| |
| JdbcSchema schema = (JdbcSchema) getSchema(); |
| schema.refreshTables(); |
| return schema.getTableByName(getTable().getName()); |
| } |
| |
| protected String createSqlStatement() { |
| return createSqlStatement(getTable()); |
| } |
| |
| private String createSqlStatement(Table table) { |
| final IQueryRewriter queryRewriter = getUpdateCallback().getDataContext().getQueryRewriter(); |
| final StringBuilder sb = new StringBuilder(); |
| sb.append("CREATE TABLE "); |
| final Schema schema = getSchema(); |
| if (schema != null && schema.getName() != null) { |
| sb.append(schema.getQualifiedLabel()); |
| sb.append("."); |
| } |
| sb.append(getUpdateCallback().quoteIfNescesary(table.getName())); |
| sb.append(" ("); |
| final Column[] columns = table.getColumns(); |
| for (int i = 0; i < columns.length; i++) { |
| final Column column = columns[i]; |
| if (i != 0) { |
| sb.append(", "); |
| } |
| sb.append(getUpdateCallback().quoteIfNescesary(column.getName())); |
| sb.append(' '); |
| final String nativeType = column.getNativeType(); |
| if (nativeType == null) { |
| ColumnType columnType = column.getType(); |
| if (columnType == null) { |
| columnType = ColumnType.VARCHAR; |
| } |
| |
| final String columnTypeString = queryRewriter.rewriteColumnType(columnType); |
| |
| sb.append(columnTypeString); |
| } else { |
| sb.append(nativeType); |
| } |
| final Integer columnSize = column.getColumnSize(); |
| if (columnSize != null) { |
| sb.append('('); |
| sb.append(columnSize.intValue()); |
| sb.append(')'); |
| } |
| if (column.isNullable() != null && !column.isNullable().booleanValue()) { |
| sb.append(" NOT NULL"); |
| } |
| if (column.isPrimaryKey()) { |
| sb.append(" PRIMARY KEY"); |
| } |
| } |
| sb.append(")"); |
| return sb.toString(); |
| } |
| |
| } |