| package org.apache.ddlutils.platform.postgresql; |
| |
| /* |
| * 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. |
| */ |
| |
| import java.io.IOException; |
| import java.util.Map; |
| |
| import org.apache.ddlutils.Platform; |
| import org.apache.ddlutils.alteration.ColumnDefinitionChange; |
| import org.apache.ddlutils.model.Column; |
| import org.apache.ddlutils.model.Database; |
| import org.apache.ddlutils.model.Index; |
| import org.apache.ddlutils.model.Table; |
| import org.apache.ddlutils.platform.SqlBuilder; |
| |
| /** |
| * The SQL Builder for PostgresSql. |
| * |
| * @version $Revision$ |
| */ |
| public class PostgreSqlBuilder extends SqlBuilder |
| { |
| /** |
| * Creates a new builder instance. |
| * |
| * @param platform The plaftform this builder belongs to |
| */ |
| public PostgreSqlBuilder(Platform platform) |
| { |
| super(platform); |
| // we need to handle the backslash first otherwise the other |
| // already escaped sequences would be affected |
| addEscapedCharSequence("\\", "\\\\"); |
| addEscapedCharSequence("'", "\\'"); |
| addEscapedCharSequence("\b", "\\b"); |
| addEscapedCharSequence("\f", "\\f"); |
| addEscapedCharSequence("\n", "\\n"); |
| addEscapedCharSequence("\r", "\\r"); |
| addEscapedCharSequence("\t", "\\t"); |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public void dropTable(Table table) throws IOException |
| { |
| print("DROP TABLE "); |
| printIdentifier(getTableName(table)); |
| print(" CASCADE"); |
| printEndOfStatement(); |
| |
| Column[] columns = table.getAutoIncrementColumns(); |
| |
| for (int idx = 0; idx < columns.length; idx++) |
| { |
| dropAutoIncrementSequence(table, columns[idx]); |
| } |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public void dropIndex(Table table, Index index) throws IOException |
| { |
| print("DROP INDEX "); |
| printIdentifier(getIndexName(index)); |
| printEndOfStatement(); |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public void createTable(Database database, Table table, Map parameters) throws IOException |
| { |
| for (int idx = 0; idx < table.getColumnCount(); idx++) |
| { |
| Column column = table.getColumn(idx); |
| |
| if (column.isAutoIncrement()) |
| { |
| createAutoIncrementSequence(table, column); |
| } |
| } |
| super.createTable(database, table, parameters); |
| } |
| |
| /** |
| * Creates the auto-increment sequence that is then used in the column. |
| * |
| * @param table The table |
| * @param column The column |
| */ |
| private void createAutoIncrementSequence(Table table, Column column) throws IOException |
| { |
| print("CREATE SEQUENCE "); |
| printIdentifier(getConstraintName(null, table, column.getName(), "seq")); |
| printEndOfStatement(); |
| } |
| |
| /** |
| * Creates the auto-increment sequence that is then used in the column. |
| * |
| * @param table The table |
| * @param column The column |
| */ |
| private void dropAutoIncrementSequence(Table table, Column column) throws IOException |
| { |
| print("DROP SEQUENCE "); |
| printIdentifier(getConstraintName(null, table, column.getName(), "seq")); |
| printEndOfStatement(); |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| protected void writeColumnAutoIncrementStmt(Table table, Column column) throws IOException |
| { |
| print("UNIQUE DEFAULT nextval('"); |
| printIdentifier(getConstraintName(null, table, column.getName(), "seq")); |
| print("')"); |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public String getSelectLastIdentityValues(Table table) |
| { |
| Column[] columns = table.getAutoIncrementColumns(); |
| |
| if (columns.length == 0) |
| { |
| return null; |
| } |
| else |
| { |
| StringBuffer result = new StringBuffer(); |
| |
| result.append("SELECT "); |
| for (int idx = 0; idx < columns.length; idx++) |
| { |
| if (idx > 0) |
| { |
| result.append(", "); |
| } |
| result.append("currval('"); |
| result.append(getDelimitedIdentifier(getConstraintName(null, table, columns[idx].getName(), "seq"))); |
| result.append("') AS "); |
| result.append(getDelimitedIdentifier(columns[idx].getName())); |
| } |
| return result.toString(); |
| } |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public void addColumn(Database model, Table table, Column newColumn) throws IOException |
| { |
| if (newColumn.isAutoIncrement()) |
| { |
| createAutoIncrementSequence(table, newColumn); |
| } |
| super.addColumn(model, table, newColumn); |
| } |
| |
| /** |
| * Writes the SQL to drop a column. |
| * |
| * @param table The table |
| * @param column The column to drop |
| */ |
| public void dropColumn(Table table, Column column) throws IOException |
| { |
| print("ALTER TABLE "); |
| printlnIdentifier(getTableName(table)); |
| printIndent(); |
| print("DROP COLUMN "); |
| printIdentifier(getColumnName(column)); |
| printEndOfStatement(); |
| if (column.isAutoIncrement()) |
| { |
| dropAutoIncrementSequence(table, column); |
| } |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| protected void writeCastExpression(Column sourceColumn, Column targetColumn) throws IOException |
| { |
| boolean sizeChanged = ColumnDefinitionChange.isSizeChanged(getPlatformInfo(), sourceColumn, targetColumn); |
| boolean typeChanged = ColumnDefinitionChange.isTypeChanged(getPlatformInfo(), sourceColumn, targetColumn); |
| |
| if (sizeChanged || typeChanged) |
| { |
| print("CAST("); |
| printIdentifier(getColumnName(sourceColumn)); |
| print(" AS "); |
| print(getSqlType(targetColumn)); |
| print(")"); |
| } |
| else |
| { |
| printIdentifier(getColumnName(sourceColumn)); |
| } |
| } |
| } |