/* The contents of this file are subject to the license and copyright terms
* detailed in the license directory at the root of the source tree (also
* available online at http://fedora-commons.org/license/).
*/
package org.fcrepo.server.utilities;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
/**
* A DDLConverter that works with Postgres.
*
* <p>This class is based on Hubert Stigler's contribution to the fedora-users
* mailing list on April 4th, 2006. It has been modified to create indexes.
*
* @author Hubert Stigler
*/
public class PostgresDDLConverter
implements DDLConverter {
public PostgresDDLConverter() {
}
public List<String> getDDL(TableSpec spec) {
ArrayList<String> l = new ArrayList<String>();
StringBuffer out = new StringBuffer();
StringBuffer end = new StringBuffer();
out.append("CREATE TABLE " + spec.getName() + " (\n");
Iterator<ColumnSpec> csi = spec.columnSpecIterator();
int csNum = 0;
while (csi.hasNext()) {
if (csNum > 0) {
out.append(",\n");
}
csNum++;
ColumnSpec cs = csi.next();
out.append(" ");
out.append(cs.getName());
out.append(' ');
if (cs.isAutoIncremented()) {
out.append(" bigserial");
} else {
if (cs.getType().toLowerCase().indexOf("int(") == 0) {
// if precision was specified for int, use postgres's default int precision
out.append("int");
} else if (cs.getType().toLowerCase().indexOf("smallint(") == 0) {
out.append("smallint");
} else {
out.append(cs.getType());
}
if (cs.isNotNull()) {
out.append(" NOT NULL");
}
if (cs.getDefaultValue() != null) {
out.append(" DEFAULT ");
if (cs.getType().toLowerCase().indexOf("char(") > -1
|| cs.getType().toLowerCase().indexOf("text") > -1) {
out.append("'");
}
out.append(cs.getDefaultValue());
if (cs.getType().toLowerCase().indexOf("char(") > -1
|| cs.getType().toLowerCase().indexOf("text") > -1) {
out.append("'");
}
}
}
if (cs.isUnique()) {
if (end.length() != 0) {
end.append(",\n");
}
end.append(" UNIQUE");
end.append(" (");
end.append(cs.getName());
end.append(")");
}
if (cs.getIndexName() != null) {
l.add("CREATE INDEX " + spec.getName() + "_" + cs.getName()
+ " ON " + spec.getName() + " (" + cs.getName() + ")");
}
if (cs.getForeignTableName() != null) {
if (end.length() != 0) {
end.append(",\n");
}
end.append(" FOREIGN KEY ");
end.append(cs.getName());
end.append(" (");
end.append(cs.getName());
end.append(") REFERENCES ");
end.append(cs.getForeignTableName());
end.append(" (");
end.append(cs.getForeignColumnName());
end.append(")");
if (cs.getOnDeleteAction() != null) {
end.append(" ON DELETE ");
end.append(cs.getOnDeleteAction());
}
}
}
if (spec.getPrimaryColumnName() != null) {
out.append(",\n PRIMARY KEY (");
out.append(spec.getPrimaryColumnName());
out.append(")");
}
if (end.length() != 0) {
out.append(",\n");
out.append(end);
}
out.append("\n");
out.append(")");
l.add(0, out.toString());
return l;
}
}