org.apache.cayenne.dba.postgres.PostgresAdapter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cayenne-client-nodeps
Show all versions of cayenne-client-nodeps
Cayenne Object Persistence Framework
/*****************************************************************
* 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.
****************************************************************/
package org.apache.cayenne.dba.postgres;
import java.sql.Types;
import java.util.Iterator;
import org.apache.cayenne.CayenneRuntimeException;
import org.apache.cayenne.access.DataNode;
import org.apache.cayenne.access.trans.QualifierTranslator;
import org.apache.cayenne.access.trans.QueryAssembler;
import org.apache.cayenne.access.types.CharType;
import org.apache.cayenne.access.types.ExtendedTypeMap;
import org.apache.cayenne.dba.JdbcAdapter;
import org.apache.cayenne.dba.PkGenerator;
import org.apache.cayenne.dba.TypesMapping;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.map.DerivedDbEntity;
import org.apache.cayenne.query.Query;
import org.apache.cayenne.query.SQLAction;
/**
* DbAdapter implementation for PostgreSQL RDBMS .
* Sample connection settings to
* use with PostgreSQL are shown below:
*
*
*
* test-postgresql.cayenne.adapter = org.apache.cayenne.dba.postgres.PostgresAdapter
* test-postgresql.jdbc.username = test
* test-postgresql.jdbc.password = secret
* test-postgresql.jdbc.url = jdbc:postgresql://serverhostname/cayenne
* test-postgresql.jdbc.driver = org.postgresql.Driver
*
*
*
* @author Dirk Olmes
* @author Holger Hoffstaette
* @author Andrus Adamchik
*/
public class PostgresAdapter extends JdbcAdapter {
public PostgresAdapter() {
setSupportsBatchUpdates(true);
}
/**
* Uses PostgresActionBuilder to create the right action.
*
* @since 1.2
*/
public SQLAction getAction(Query query, DataNode node) {
return query.createSQLAction(new PostgresActionBuilder(this, node
.getEntityResolver()));
}
/**
* Installs appropriate ExtendedTypes as converters for passing values between JDBC
* and Java layers.
*/
protected void configureExtendedTypes(ExtendedTypeMap map) {
super.configureExtendedTypes(map);
map.registerType(new CharType(true, false));
map.registerType(new PostgresByteArrayType(true, true));
}
public DbAttribute buildAttribute(
String name,
String typeName,
int type,
int size,
int precision,
boolean allowNulls) {
// "bytea" maps to pretty much any binary type, so
// it is up to us to select the most sensible default.
// And the winner is LONGVARBINARY
if ("bytea".equalsIgnoreCase(typeName)) {
type = Types.LONGVARBINARY;
}
// oid is returned as INTEGER, need to make it BLOB
else if ("oid".equals(typeName)) {
type = Types.BLOB;
}
// somehow the driver reverse-engineers "text" as VARCHAR, must be CLOB
else if ("text".equalsIgnoreCase(typeName)) {
type = Types.CLOB;
}
return super.buildAttribute(name, typeName, type, size, precision, allowNulls);
}
/**
* Customizes table creating procedure for PostgreSQL. One difference with generic
* implementation is that "bytea" type has no explicit length unlike similar binary
* types in other databases.
*
* @since 1.0.2
*/
public String createTable(DbEntity ent) {
// later we may support view creation
// for derived DbEntities
if (ent instanceof DerivedDbEntity) {
throw new CayenneRuntimeException("Can't create table for derived DbEntity '"
+ ent.getName()
+ "'.");
}
StringBuffer buf = new StringBuffer();
buf.append("CREATE TABLE ").append(ent.getFullyQualifiedName()).append(" (");
// columns
Iterator it = ent.getAttributes().iterator();
boolean first = true;
while (it.hasNext()) {
if (first) {
first = false;
}
else {
buf.append(", ");
}
DbAttribute at = (DbAttribute) it.next();
// attribute may not be fully valid, do a simple check
if (at.getType() == TypesMapping.NOT_DEFINED) {
throw new CayenneRuntimeException("Undefined type for attribute '"
+ ent.getFullyQualifiedName()
+ "."
+ at.getName()
+ "'.");
}
String[] types = externalTypesForJdbcType(at.getType());
if (types == null || types.length == 0) {
throw new CayenneRuntimeException("Undefined type for attribute '"
+ ent.getFullyQualifiedName()
+ "."
+ at.getName()
+ "': "
+ at.getType());
}
String type = types[0];
buf.append(at.getName()).append(' ').append(type);
// append size and precision (if applicable)
if (typeSupportsLength(at.getType())) {
int len = at.getMaxLength();
int prec = TypesMapping.isDecimal(at.getType()) ? at.getPrecision() : -1;
// sanity check
if (prec > len) {
prec = -1;
}
if (len > 0) {
buf.append('(').append(len);
if (prec >= 0) {
buf.append(", ").append(prec);
}
buf.append(')');
}
}
if (at.isMandatory()) {
buf.append(" NOT NULL");
}
else {
buf.append(" NULL");
}
}
// primary key clause
Iterator pkit = ent.getPrimaryKey().iterator();
if (pkit.hasNext()) {
if (first)
first = false;
else
buf.append(", ");
buf.append("PRIMARY KEY (");
boolean firstPk = true;
while (pkit.hasNext()) {
if (firstPk)
firstPk = false;
else
buf.append(", ");
DbAttribute at = (DbAttribute) pkit.next();
buf.append(at.getName());
}
buf.append(')');
}
buf.append(')');
return buf.toString();
}
private boolean typeSupportsLength(int type) {
// "bytea" type does not support length
String[] externalTypes = externalTypesForJdbcType(type);
if (externalTypes != null && externalTypes.length > 0) {
for (int i = 0; i < externalTypes.length; i++) {
if ("bytea".equalsIgnoreCase(externalTypes[i])) {
return false;
}
}
}
return TypesMapping.supportsLength(type);
}
/**
* Adds the CASCADE option to the DROP TABLE clause.
*
* @see JdbcAdapter#dropTable(DbEntity)
*/
public String dropTable(DbEntity ent) {
return super.dropTable(ent) + " CASCADE";
}
/**
* Returns a trimming translator.
*/
public QualifierTranslator getQualifierTranslator(QueryAssembler queryAssembler) {
return new PostgresQualifierTranslator(queryAssembler);
}
/**
* @see JdbcAdapter#createPkGenerator()
*/
protected PkGenerator createPkGenerator() {
return new PostgresPkGenerator();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy