/*****************************************************************
* 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;
import java.net.URL;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Types;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import org.apache.cayenne.CayenneRuntimeException;
import org.apache.cayenne.access.DataNode;
import org.apache.cayenne.access.jdbc.EJBQLTranslatorFactory;
import org.apache.cayenne.access.jdbc.JdbcEJBQLTranslatorFactory;
import org.apache.cayenne.access.trans.QualifierTranslator;
import org.apache.cayenne.access.trans.QueryAssembler;
import org.apache.cayenne.access.types.ExtendedType;
import org.apache.cayenne.access.types.ExtendedTypeMap;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.map.DbJoin;
import org.apache.cayenne.map.DbRelationship;
import org.apache.cayenne.merge.MergerFactory;
import org.apache.cayenne.query.Query;
import org.apache.cayenne.query.SQLAction;
import org.apache.cayenne.resource.ClassLoaderResourceLocator;
import org.apache.cayenne.resource.Resource;
import org.apache.cayenne.resource.ResourceLocator;
import org.apache.cayenne.util.Util;
/**
* A generic DbAdapter implementation. Can be used as a default adapter or as a superclass
* of a concrete adapter implementation.
*/
public class JdbcAdapter implements DbAdapter {
final static String DEFAULT_IDENTIFIERS_START_QUOTE = "\"";
final static String DEFAULT_IDENTIFIERS_END_QUOTE = "\"";
protected PkGenerator pkGenerator;
protected TypesHandler typesHandler;
protected ExtendedTypeMap extendedTypes;
protected boolean supportsBatchUpdates;
protected boolean supportsUniqueConstraints;
protected boolean supportsGeneratedKeys;
protected EJBQLTranslatorFactory ejbqlTranslatorFactory;
protected String identifiersStartQuote;
protected String identifiersEndQuote;
protected ResourceLocator resourceLocator;
/**
* @since 3.0
*/
public String getIdentifiersStartQuote() {
return identifiersStartQuote;
}
/**
* @since 3.0
*/
public String getIdentifiersEndQuote() {
return identifiersEndQuote;
}
/**
* Creates new JdbcAdapter with a set of default parameters.
*/
public JdbcAdapter() {
// init defaults
this.setSupportsBatchUpdates(false);
this.setSupportsUniqueConstraints(true);
// TODO: andrus 05.02.2010 - ideally this should be injected
this.resourceLocator = new ClassLoaderResourceLocator();
this.pkGenerator = createPkGenerator();
this.typesHandler = TypesHandler.getHandler(findResource("/types.xml"));
this.extendedTypes = new ExtendedTypeMap();
this.configureExtendedTypes(extendedTypes);
this.ejbqlTranslatorFactory = createEJBQLTranslatorFactory();
initIdentifiersQuotes();
}
/**
* Returns default separator - a semicolon.
*
* @since 1.0.4
*/
public String getBatchTerminator() {
return ";";
}
/**
* Locates and returns a named adapter resource. A resource can be an XML file, etc.
* <p>
* This implementation is based on the premise that each adapter is located in its own
* Java package and all resources are in the same package as well. Resource lookup is
* recursive, so that if DbAdapter is a subclass of another adapter, parent adapter
* package is searched as a failover.
* </p>
*
* @since 3.0
*/
protected URL findResource(String name) {
Class<?> adapterClass = getClass();
while (adapterClass != null && JdbcAdapter.class.isAssignableFrom(adapterClass)) {
String path = Util.getPackagePath(adapterClass.getName()) + name;
Collection<Resource> resources = resourceLocator.findResources(path);
if (!resources.isEmpty()) {
return resources.iterator().next().getURL();
}
adapterClass = adapterClass.getSuperclass();
}
return null;
}
/**
* Installs appropriate ExtendedTypes as converters for passing values between JDBC
* and Java layers. Called from default constructor.
*/
protected void configureExtendedTypes(ExtendedTypeMap map) {
// noop... subclasses may override to install custom types
}
/**
* Creates and returns a primary key generator. This factory method should be
* overriden by JdbcAdapter subclasses to provide custom implementations of
* PKGenerator.
*/
protected PkGenerator createPkGenerator() {
return new JdbcPkGenerator(this);
}
/**
* Creates and returns an {@link EJBQLTranslatorFactory} used to generate visitors for
* EJBQL to SQL translations. This method should be overriden by subclasses that need
* to customize EJBQL generation.
*
* @since 3.0
*/
protected EJBQLTranslatorFactory createEJBQLTranslatorFactory() {
return new JdbcEJBQLTranslatorFactory();
}
/**
* Returns primary key generator associated with this DbAdapter.
*/
public PkGenerator getPkGenerator() {
return pkGenerator;
}
/**
* Sets new primary key generator.
*
* @since 1.1
*/
public void setPkGenerator(PkGenerator pkGenerator) {
this.pkGenerator = pkGenerator;
}
/**
* Returns true.
*
* @since 1.1
*/
public boolean supportsUniqueConstraints() {
return supportsUniqueConstraints;
}
/**
* @since 1.1
*/
public void setSupportsUniqueConstraints(boolean flag) {
this.supportsUniqueConstraints = flag;
}
/**
* @since 3.0
*/
public Collection<String> dropTableStatements(DbEntity table) {
QuotingStrategy context = getQuotingStrategy(table
.getDataMap()
.isQuotingSQLIdentifiers());
StringBuilder buf = new StringBuilder("DROP TABLE ");
buf.append(context.quoteFullyQualifiedName(table));
return Collections.singleton(buf.toString());
}
/**
* Returns a SQL string that can be used to create database table corresponding to
* <code>ent</code> parameter.
*/
public String createTable(DbEntity entity) {
boolean status;
if (entity.getDataMap() != null && entity.getDataMap().isQuotingSQLIdentifiers()) {
status = true;
}
else {
status = false;
}
QuotingStrategy context = getQuotingStrategy(status);
StringBuffer sqlBuffer = new StringBuffer();
sqlBuffer.append("CREATE TABLE ");
sqlBuffer.append(context.quoteFullyQualifiedName(entity));
sqlBuffer.append(" (");
// columns
Iterator<?> it = entity.getAttributes().iterator();
if (it.hasNext()) {
boolean first = true;
while (it.hasNext()) {
if (first) {
first = false;
}
else {
sqlBuffer.append(", ");
}
DbAttribute column = (DbAttribute) it.next();
// attribute may not be fully valid, do a simple check
if (column.getType() == TypesMapping.NOT_DEFINED) {
throw new CayenneRuntimeException("Undefined type for attribute '"
+ entity.getFullyQualifiedName()
+ "."
+ column.getName()
+ "'.");
}
createTableAppendColumn(sqlBuffer, column);
}
createTableAppendPKClause(sqlBuffer, entity);
}
sqlBuffer.append(')');
return sqlBuffer.toString();
}
/**
* @since 1.2
*/
protected void createTableAppendPKClause(StringBuffer sqlBuffer, DbEntity entity) {
boolean status;
if (entity.getDataMap() != null && entity.getDataMap().isQuotingSQLIdentifiers()) {
status = true;
}
else {
status = false;
}
QuotingStrategy context = getQuotingStrategy(status);
Iterator<DbAttribute> pkit = entity.getPrimaryKeys().iterator();
if (pkit.hasNext()) {
sqlBuffer.append(", PRIMARY KEY (");
boolean firstPk = true;
while (pkit.hasNext()) {
if (firstPk)
firstPk = false;
else
sqlBuffer.append(", ");
DbAttribute at = pkit.next();
sqlBuffer.append(context.quoteString(at.getName()));
}
sqlBuffer.append(')');
}
}
/**
* Appends SQL for column creation to CREATE TABLE buffer.
*
* @since 1.2
*/
public void createTableAppendColumn(StringBuffer sqlBuffer, DbAttribute column) {
boolean status;
if ((column.getEntity().getDataMap() != null)
&& column.getEntity().getDataMap().isQuotingSQLIdentifiers()) {
status = true;
}
else {
status = false;
}
QuotingStrategy context = getQuotingStrategy(status);
String[] types = externalTypesForJdbcType(column.getType());
if (types == null || types.length == 0) {
String entityName = column.getEntity() != null ? ((DbEntity) column
.getEntity()).getFullyQualifiedName() : "<null>";
throw new CayenneRuntimeException("Undefined type for attribute '"
+ entityName
+ "."
+ column.getName()
+ "': "
+ column.getType());
}
String type = types[0];
sqlBuffer.append(context.quoteString(column.getName()));
sqlBuffer.append(' ').append(type);
// append size and precision (if applicable)s
if (TypesMapping.supportsLength(column.getType())) {
int len = column.getMaxLength();
int scale = (TypesMapping.isDecimal(column.getType()) && column.getType() != Types.FLOAT)
? column.getScale()
: -1;
// sanity check
if (scale > len) {
scale = -1;
}
if (len > 0) {
sqlBuffer.append('(').append(len);
if (scale >= 0) {
sqlBuffer.append(", ").append(scale);
}
sqlBuffer.append(')');
}
}
sqlBuffer.append(column.isMandatory() ? " NOT NULL" : " NULL");
}
/**
* Returns a DDL string to create a unique constraint over a set of columns.
*
* @since 1.1
*/
public String createUniqueConstraint(DbEntity source, Collection<DbAttribute> columns) {
boolean status;
if (source.getDataMap() != null && source.getDataMap().isQuotingSQLIdentifiers()) {
status = true;
}
else {
status = false;
}
QuotingStrategy context = getQuotingStrategy(status);
if (columns == null || columns.isEmpty()) {
throw new CayenneRuntimeException(
"Can't create UNIQUE constraint - no columns specified.");
}
StringBuilder buf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(context.quoteFullyQualifiedName(source));
buf.append(" ADD UNIQUE (");
Iterator<DbAttribute> it = columns.iterator();
DbAttribute first = it.next();
buf.append(context.quoteString(first.getName()));
while (it.hasNext()) {
DbAttribute next = it.next();
buf.append(", ");
buf.append(context.quoteString(next.getName()));
}
buf.append(")");
return buf.toString();
}
/**
* Returns a SQL string that can be used to create a foreign key constraint for the
* relationship.
*/
public String createFkConstraint(DbRelationship rel) {
DbEntity source = (DbEntity) rel.getSourceEntity();
boolean status;
if (source.getDataMap() != null && source.getDataMap().isQuotingSQLIdentifiers()) {
status = true;
}
else {
status = false;
}
QuotingStrategy context = getQuotingStrategy(status);
StringBuilder buf = new StringBuilder();
StringBuilder refBuf = new StringBuilder();
buf.append("ALTER TABLE ");
buf.append(context.quoteFullyQualifiedName(source));
buf.append(" ADD FOREIGN KEY (");
boolean first = true;
for (DbJoin join : rel.getJoins()) {
if (!first) {
buf.append(", ");
refBuf.append(", ");
}
else
first = false;
buf.append(context.quoteString(join.getSourceName()));
refBuf.append(context.quoteString(join.getTargetName()));
}
buf.append(") REFERENCES ");
buf.append(context.quoteFullyQualifiedName((DbEntity) rel.getTargetEntity()));
buf.append(" (").append(refBuf.toString()).append(')');
return buf.toString();
}
public String[] externalTypesForJdbcType(int type) {
return typesHandler.externalTypesForJdbcType(type);
}
public ExtendedTypeMap getExtendedTypes() {
return extendedTypes;
}
public DbAttribute buildAttribute(
String name,
String typeName,
int type,
int size,
int scale,
boolean allowNulls) {
DbAttribute attr = new DbAttribute();
attr.setName(name);
attr.setType(type);
attr.setMandatory(!allowNulls);
if (size >= 0) {
attr.setMaxLength(size);
}
if (scale >= 0) {
attr.setScale(scale);
}
return attr;
}
public String tableTypeForTable() {
return "TABLE";
}
public String tableTypeForView() {
return "VIEW";
}
/**
* Creates and returns a default implementation of a qualifier translator.
*/
public QualifierTranslator getQualifierTranslator(QueryAssembler queryAssembler) {
return new QualifierTranslator(queryAssembler);
}
/**
* Uses JdbcActionBuilder to create the right action.
*
* @since 1.2
*/
public SQLAction getAction(Query query, DataNode node) {
return query
.createSQLAction(new JdbcActionBuilder(this, node.getEntityResolver()));
}
public void bindParameter(
PreparedStatement statement,
Object object,
int pos,
int sqlType,
int scale) throws SQLException, Exception {
if (object == null) {
statement.setNull(pos, sqlType);
}
else {
ExtendedType typeProcessor = getExtendedTypes().getRegisteredType(
object.getClass());
typeProcessor.setJdbcObject(statement, object, pos, sqlType, scale);
}
}
public boolean supportsBatchUpdates() {
return this.supportsBatchUpdates;
}
public void setSupportsBatchUpdates(boolean flag) {
this.supportsBatchUpdates = flag;
}
/**
* @since 1.2
*/
public boolean supportsGeneratedKeys() {
return supportsGeneratedKeys;
}
/**
* @since 1.2
*/
public void setSupportsGeneratedKeys(boolean flag) {
this.supportsGeneratedKeys = flag;
}
/**
* Returns a translator factory for EJBQL to SQL translation. This property is
* normally initialized in constructor by calling
* {@link #createEJBQLTranslatorFactory()}, and can be overridden by calling
* {@link #setEjbqlTranslatorFactory(EJBQLTranslatorFactory)}.
*
* @since 3.0
*/
public EJBQLTranslatorFactory getEjbqlTranslatorFactory() {
return ejbqlTranslatorFactory;
}
/**
* Sets a translator factory for EJBQL to SQL translation. This property is normally
* initialized in constructor by calling {@link #createEJBQLTranslatorFactory()}, so
* users would only override it if they need to customize EJBQL translation.
*
* @since 3.0
*/
public void setEjbqlTranslatorFactory(EJBQLTranslatorFactory ejbqlTranslatorFactory) {
this.ejbqlTranslatorFactory = ejbqlTranslatorFactory;
}
/**
* @since 3.0
*/
public MergerFactory mergerFactory() {
return new MergerFactory();
}
/**
* @since 3.0
*/
protected void initIdentifiersQuotes() {
this.identifiersStartQuote = DEFAULT_IDENTIFIERS_START_QUOTE;
this.identifiersEndQuote = DEFAULT_IDENTIFIERS_END_QUOTE;
}
/**
* @since 3.0
*/
public QuotingStrategy getQuotingStrategy(boolean needQuotes) {
if (needQuotes) {
return new QuoteStrategy(this.getIdentifiersStartQuote(), this
.getIdentifiersEndQuote());
}
else {
return new NoQuoteStrategy();
}
}
}