org.dbtools.renderer.PostgreSQLRenderer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of dbtools-gen Show documentation
Show all versions of dbtools-gen Show documentation
DBTools ORM Class Generator.
/*
* PostgreSQLRenderer.java
*
* Created on October 21, 2002
*
* Copyright 2006 Jeff Campbell. All rights reserved. Unauthorized reproduction
* is a violation of applicable law. This material contains certain
* confidential or proprietary information and trade secrets of Jeff Campbell.
*/
package org.dbtools.renderer;
import org.dbtools.schema.ForeignKey;
import org.dbtools.schema.schemafile.*;
import java.util.ArrayList;
import java.util.List;
/**
* @author Jeff Campbell
*/
public class PostgreSQLRenderer extends SchemaRenderer {
public static final String RENDERER_NAME = "postgresql";
private static final String NOW = "CURRENT_TIMESTAMP";
public PostgreSQLRenderer() {
super();
this.setDbVendorName(RENDERER_NAME);
}
@Override
public String generateSchema(SchemaDatabase database, String[] tablesToGenerate, String[] viewsToGenerate, boolean dropTables, boolean createInserts) {
showProgress("Generating SQL schema using PostgreSQL renderer ...", true);
StringBuilder schema = new StringBuilder();
List foreignKeysToCreate = new ArrayList<>();
List requestedTables = getTablesToGenerate(database, tablesToGenerate);
List requestedViews = getViewsToGenerate(database, viewsToGenerate);
// drop schema
if (dropTables) {
generateDropSchema(false, true, schema, requestedTables, requestedViews);
}
// create tables
for (SchemaTable requestedTable : requestedTables) {
// add table header
SchemaTable table = requestedTable;
List fields = table.getFields();
// determine sequence name
String sequencerName = null;
int sequencerStartValue = 1;
for (SchemaTableField field : fields) {
String fieldSeqName = field.getSequencerName();
if (fieldSeqName != null && fieldSeqName.length() > 0) {
sequencerName = fieldSeqName;
sequencerStartValue = field.getSequencerStartValue();
break;
}
}
// reset values for new table
SchemaTableField primaryKey = null;
List indexFields = new ArrayList<>();
schema.append("CREATE TABLE ");
schema.append(table.getName());
schema.append(" (\n");
// add fields
SchemaTableField enumPKField = null;
SchemaTableField enumValueField = null;
for (int j = 0; j < fields.size(); j++) {
SchemaTableField field = fields.get(j);
if (field.isPrimaryKey()) {
primaryKey = field;
}
// add field
// name
schema.append("\t");
schema.append(field.getName());
// datatype
schema.append(" ");
schema.append(getSqlType(field.getJdbcDataType()));
//check for size for datatype
if (field.getSize() > 0) {
schema.append("(");
schema.append(field.getSize());
// check for decimals
if (field.getDecimals() > 0) {
schema.append(", ").append(field.getDecimals());
}
schema.append(")");
}
String defaultValue = field.getDefaultValue();
if (defaultValue != null && !defaultValue.equals("")) {
String newDefaultValue = formatDefaultValue(field);
if (newDefaultValue.toUpperCase().equals("'NOW'")) {
newDefaultValue = NOW;
}
schema.append(" DEFAULT ").append(formatDefaultValue(field));
} else if (field.isNotNull()) {
schema.append(" NOT NULL");
}
if (field.isUnique()) {
schema.append(" UNIQUE");
}
if (field.isIndex()) {
indexFields.add(field);
}
// add foreign key
if (!field.getForeignKeyField().equals("")) {
foreignKeysToCreate.add(new ForeignKey(requestedTable.getName(), field.getName(), field.getForeignKeyTable(), field.getForeignKeyField()));
}
schema.append("");
// if this is the last one, then don't put a ','
if (j == fields.size() - 1) {
// add the primary key
if (primaryKey != null) {
schema.append(",\n");
schema.append("\tPRIMARY KEY(");
schema.append(primaryKey.getName());
schema.append(")");
} else {
//schema.append("\n");
}
} else {
schema.append(",\n");
}
// check for enumFields
if (enumPKField == null && field.isPrimaryKey()) {
enumPKField = field;
}
if (enumValueField == null && field.getJdbcDataType() == SchemaFieldType.VARCHAR) {
enumValueField = field;
}
}
// check for uniqueDeclarations
generateUniqueConstraints(schema, table);
// check for indexDeclarations
generateIndexes(schema, table);
// add table footer
schema.append("\n);");
// create indexes
for (SchemaTableField indexField : indexFields) {
schema.append("\nCREATE INDEX ").append(requestedTable.getName()).append(indexField.getName()).append("_IDX ON ").append(requestedTable.getName()).append(" (").append(indexField.getName()).append(");");
}
schema.append("\n");
generateEnumSchema(schema, requestedTable, getAlreadyCreatedEnum(), enumPKField, enumValueField, createInserts);
// check to see if we need to create a sequence
if (sequencerName != null && sequencerName.length() > 0) {
if (requestedTable.isEnumerationTable() && super.isCreateEnumInserts()) {
schema.append("CREATE SEQUENCE ").append(sequencerName).append(" START WITH ").append(requestedTable.getEnumerations().length()).append(";\n");
} else {
schema.append("CREATE SEQUENCE ").append(sequencerName).append(" START WITH ").append(sequencerStartValue).append(";\n");
}
}
schema.append("\n\n");
}
// create foreign keys
for (ForeignKey fk : foreignKeysToCreate) {
schema.append("ALTER TABLE ").append(fk.getPrimaryKeyTable()).append("\n");
schema.append("ADD CONSTRAINT ").append(fk.getPrimaryKeyTable()).append("_").append(fk.getPrimaryKeyField()).append("_FK\n");
schema.append("FOREIGN KEY (").append(fk.getPrimaryKeyField()).append(")\n");
schema.append("REFERENCES ").append(fk.getForeignKeyTable()).append(" (").append(fk.getForeignKeyField()).append(");\n");
schema.append("\n");
}
// create views
// for (SchemaView view : requestedViews) {
//
// // header
// schema.append("CREATE VIEW ").append(view.getName()).append(" AS \n");
//
// // get fields
// String aliases = "";
// String selectItems = "";
//
// Iterator vfItr = view.getViewFields().iterator();
// while (vfItr.hasNext()) {
// SchemaViewField viewField = (SchemaViewField) vfItr.next();
//
// aliases += viewField.getName();
// selectItems += "\t" + viewField.getExpression();
//
// if (vfItr.hasNext()) {
// aliases += ", ";
// selectItems += ",\n";
// } else {
// selectItems += "\n";
// }
// }
//
// // aliases
// schema.append("(").append(aliases).append(")");
//
// // AS SELECT
// schema.append(" AS\n SELECT \n").append(selectItems);
//
// // POSTSELECT
// schema.append(" ").append(view.getViewPostSelectClause()).append(";");
//
// // end
// schema.append("\n\n");
// } // end of views
return schema.toString();
}
@Override
public String generatePostSchema(SchemaDatabase database, String[] tablesToGenerate) {
showProgress("Generating Post SQL schema using PostgreSQL renderer ...", true);
StringBuilder postSchema = new StringBuilder();
postSchema.append("\n\n");
for (SchemaTable table : getTablesToGenerate(database, tablesToGenerate)) {
for (SchemaTableField field : table.getFields()) {
String sequencerName = field.getSequencerName();
int sequencerStartValue = field.getSequencerStartValue();
if (sequencerName != null && sequencerName.length() > 0) {
if (sequencerStartValue == SchemaTableField.DEFAULT_INITIAL_INCREMENT) {
// SELECT SETVAL ('MYTABLE_SEQ', (SELECT MAX(ID)+1 FROM MYTABLE), false)
postSchema.append("SELECT SETVAL ('").append(sequencerName).append("', (SELECT MAX(").append(field.getName()).append(")+1 FROM ").append(table.getName()).append("), false);\n\n");
} else {
postSchema.append("SELECT SETVAL ('").append(sequencerName).append("', ").append(sequencerStartValue).append(", false);\n\n");
}
}
}
}
return postSchema.toString();
}
@Override
public void generateDropSchema(boolean addIfExists, boolean ifExistsAtEndOfStmnt, StringBuilder schema, List tablesToGenerate, List viewsToGenerate) {
for (SchemaTable table : tablesToGenerate) {
for (SchemaTableField field : table.getFields()) {
String sequencerName = field.getSequencerName();
if (sequencerName != null && sequencerName.length() > 0) {
schema.append("DROP SEQUENCE ").append(sequencerName).append(";\n");
}
}
}
super.generateDropSchema(false, true, schema, tablesToGenerate, viewsToGenerate);
}
}