org.plasma.text.ddl.DefaultDDLFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plasma-provisioning Show documentation
Show all versions of plasma-provisioning Show documentation
The Plasma Provisioning Module
/**
* Copyright 2017 TerraMeta Software, Inc.
*
* 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 org.plasma.text.ddl;
import org.plasma.runtime.DataAccessProvider;
import org.plasma.runtime.DataAccessProviderName;
import org.plasma.runtime.PlasmaRuntime;
import org.plasma.runtime.SequenceConfiguration;
public abstract class DefaultDDLFactory {
public abstract String getType(Table table, Column column);
public String createSchema(Schema schema) {
StringBuilder buf = new StringBuilder();
buf.append("CREATE SCHEMA ");
buf.append(schema.getName());
buf.append(";\n");
return buf.toString();
}
public String dropSchema(Schema schema) {
StringBuilder buf = new StringBuilder();
buf.append("DROP SCHEMA ");
buf.append(schema.getName());
buf.append(";\n");
return buf.toString();
}
public String createTable(Schema schema, Table table) {
StringBuilder buf = new StringBuilder();
buf.append("CREATE TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" ( ");
int i = 0;
for (Column column : table.getColumns()) {
if (i > 0)
buf.append(", ");
buf.append(column.getName());
buf.append(" ");
buf.append(getType(table, column));
if (!column.isNullable())
buf.append(" NOT NULL");
i++;
}
if (table.getPk() != null) {
buf.append(", PRIMARY KEY (");
i = 0;
for (On on : table.getPk().getOns()) {
if (i > 0)
buf.append(", ");
buf.append(on.getColumn());
i++;
}
buf.append(" )");
}
buf.append(" );\n");
return buf.toString();
}
protected boolean isPk(Table table, Column column) {
if (table.getPk() != null) {
for (On on : table.getPk().getOns()) {
if (on.getColumn().equals(column.getName()))
return true;
}
}
return false;
}
protected boolean isFk(Table table, Column column) {
if (table.getFks() != null) {
for (Fk fk : table.getFks()) {
if (fk.getColumn().equals(column.getName()))
return true;
}
}
return false;
}
public String dropTable(Schema schema, Table table) {
StringBuilder buf = new StringBuilder();
buf.append("DROP TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(";\n");
return buf.toString();
}
public String truncateTable(Schema schema, Table table) {
StringBuilder buf = new StringBuilder();
buf.append("TRUNCATE TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(";\n");
return buf.toString();
}
public String createView(Schema schema, Table table, Behavior create) {
StringBuilder buf = new StringBuilder();
String ddl = create.getValue();
if (ddl != null) {
ddl = ddl.trim();
buf.append(ddl);
if (!ddl.endsWith(";"))
buf.append(";");
}
return buf.toString();
}
public String dropView(Schema schema, Table table, Behavior drop) {
StringBuilder buf = new StringBuilder();
String ddl = drop.getValue();
if (ddl != null) {
ddl = ddl.trim();
buf.append(ddl);
if (!ddl.endsWith(";"))
buf.append(";");
}
return buf.toString();
}
private DataAccessProvider getProvider() {
DataAccessProvider provider = PlasmaRuntime.getInstance().findDataAccessProvider(
DataAccessProviderName.JPA);
if (provider == null)
provider = PlasmaRuntime.getInstance().findDataAccessProvider(DataAccessProviderName.JDO);
if (provider == null)
provider = PlasmaRuntime.getInstance().findDataAccessProvider(DataAccessProviderName.JDBC);
return provider;
}
private SequenceConfiguration getSequenceConfiguration() {
DataAccessProvider provider = getProvider();
if (provider != null) {
return provider.getSequenceConfiguration();
}
return null;
}
public String createSequence(Schema schema, Table table) {
StringBuilder buf = new StringBuilder();
SequenceConfiguration config = getSequenceConfiguration();
if (config != null) {
buf.append("CREATE SEQUENCE ");
if (config.getPrefix() != null && config.getPrefix().trim().length() > 0) {
buf.append(config.getPrefix());
}
buf.append(table.getName());
if (config.getSuffix() != null && config.getSuffix().trim().length() > 0) {
buf.append(config.getSuffix());
}
buf.append(" MINVALUE ");
buf.append(config.getMinValue());
buf.append(" MAXVALUE ");
buf.append(config.getMaxValue());
buf.append(" START WITH ");
buf.append(config.getStartValue());
buf.append(" INCREMENT BY ");
buf.append(config.getIncrementBy());
buf.append(" CACHE ");
buf.append(config.getCache());
buf.append(";\n");
}
return buf.toString();
}
public String dropSequence(Schema schema, Table table) {
StringBuilder buf = new StringBuilder();
SequenceConfiguration config = getSequenceConfiguration();
if (config != null) {
buf.append("DROP SEQUENCE ");
if (config.getPrefix() != null && config.getPrefix().trim().length() > 0) {
buf.append(config.getPrefix());
}
buf.append(table.getName());
if (config.getSuffix() != null && config.getSuffix().trim().length() > 0) {
buf.append(config.getSuffix());
}
buf.append(";\n");
}
return buf.toString();
}
public String createIndex(Schema schema, Table table, Index index) {
StringBuilder buf = new StringBuilder();
buf.append("CREATE INDEX ");
buf.append(index.getName());
buf.append(" ON ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" ( ");
buf.append(index.getColumn());
buf.append(" );\n");
return buf.toString();
}
public String dropIndex(Schema schema, Table table, Index index) {
StringBuilder buf = new StringBuilder();
buf.append("DROP INDEX ");
buf.append(index.getName());
buf.append(";\n");
return buf.toString();
}
public String createCheckConstraint(Schema schema, Table table, Check check) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" ADD CONSTRAINT ");
buf.append(check.getName());
buf.append(" CHECK ( ");
buf.append(check.getColumn());
buf.append(" IN ( ");
int i = 0;
for (String value : check.getValues()) {
if (i > 0)
buf.append(", ");
buf.append("'");
buf.append(value);
buf.append("'");
i++;
}
buf.append(" ) ) ENABLE;\n");
return buf.toString();
}
public String dropCheckConstraint(Schema schema, Table table, Check check) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" DROP CONSTRAINT ");
buf.append(check.getName());
buf.append(";\n");
return buf.toString();
}
public String enableCheckConstraint(Schema schema, Table table, Check check, boolean enable) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
if (enable)
buf.append(" ENABLE CONSTRAINT ");
else
buf.append(" DISABLE CONSTRAINT ");
buf.append(check.getName());
buf.append(";\n");
return buf.toString();
}
public String createUniqueConstraint(Schema schema, Table table, Unique unique) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" ADD CONSTRAINT ");
buf.append(unique.getName());
buf.append(" UNIQUE ( ");
int i = 0;
for (On on : unique.getOns()) {
if (i > 0)
buf.append(", ");
buf.append(on.getColumn());
i++;
}
buf.append(" ) ENABLE;\n");
return buf.toString();
}
public String dropUniqueConstraint(Schema schema, Table table, Unique unique) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" DROP CONSTRAINT ");
buf.append(unique.getName());
buf.append(";\n");
return buf.toString();
}
public String enableUniqueConstraint(Schema schema, Table table, Unique unique, boolean enable) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
if (enable)
buf.append(" ENABLE CONSTRAINT ");
else
buf.append(" DISABLE CONSTRAINT ");
buf.append(unique.getName());
buf.append(";\n");
return buf.toString();
}
public String createForeignKeyConstraint(Schema schema, Table table, Fk fk, Table toTable) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" ADD CONSTRAINT ");
buf.append(fk.getName());
buf.append(" FOREIGN KEY ( ");
buf.append(fk.getColumn());
buf.append(" ) REFERENCES ");
buf.append(schema.getName());
buf.append(".");
buf.append(toTable.getName());
buf.append(" ( ");
int i = 0;
for (On on : toTable.getPk().getOns()) {
if (i > 0)
buf.append(", ");
buf.append(on.getColumn());
i++;
}
buf.append(" ) DEFERRABLE;\n");
return buf.toString();
}
public String dropForeignKeyConstraint(Schema schema, Table table, Fk fk, Table toTable) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
buf.append(" DROP CONSTRAINT ");
buf.append(fk.getName());
buf.append(";\n");
return buf.toString();
}
public String enableForeignKeyConstraint(Schema schema, Table table, Fk fk, Table toTable,
boolean enable) {
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(schema.getName());
buf.append(".");
buf.append(table.getName());
if (enable)
buf.append(" ENABLE CONSTRAINT ");
else
buf.append(" DISABLE CONSTRAINT ");
buf.append(fk.getName());
buf.append(";\n");
return buf.toString();
}
protected Column getColumn(Table table, String name) {
for (Column col : table.getColumns()) {
if (col.getName().equals(name))
return col;
}
throw new DDLException("could not find column '" + name + "' for table. " + table.getName());
}
}