org.apache.ddlutils.platform.postgresql.PostgreSqlBuilder Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of snappydata-store-tools Show documentation
Show all versions of snappydata-store-tools Show documentation
SnappyData store based off Pivotal GemFireXD
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: 712002 $
*/
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
{
StringBuilder result = new StringBuilder();
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));
}
}
}