/**********************************************************************
Copyright (c) 2005 Erik Bengtson and others. All rights reserved.
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.
Contributors:
2006 Andy Jefferson - moved catalog/schema setting code into newTableIdentifierForMetaData
2006 Andy Jefferson - changed to implement IdentifierFactory for RDBMS stores
...
**********************************************************************/
package org.jpox.store.rdbms.sqlidentifier;
import java.util.ArrayList;
import java.util.ListIterator;
import java.util.StringTokenizer;
import org.jpox.ClassLoaderResolver;
import org.jpox.exceptions.JPOXException;
import org.jpox.metadata.AbstractClassMetaData;
import org.jpox.metadata.AbstractMemberMetaData;
import org.jpox.metadata.FieldRole;
import org.jpox.store.mapped.DatastoreAdapter;
import org.jpox.store.mapped.DatastoreIdentifier;
import org.jpox.store.mapped.IdentifierFactory;
/**
* Factory that creates immutable instances of DatastoreIdentifier for RDBMS datastores.
* Includes a JPOX naming strategy, naming as follows
* <ul>
* <li>Class called "MyClass" will generate table name of "MYCLASS"</li>
* <li>Field called "myField" will generate column name of "MY_FIELD"</li>
* <li>Datastore id field for class "MyClass" will have the PK field "MYCLASS_ID"</li>
* <li>Join table will be named after the class and field, so "MyClass" with field "myField" will become
* a table with name "MYCLASS_MYFIELD".</li>
* <li>Columns of a join table will be named after the PK fields of the owner and element. So something
* like "MYCLASS_ID_OID" and "MYELEMENT_ID_EID"</li>
* <li>Discriminator field columns will, by default, be called "DISCRIMINATOR"</li>
* <li>Index field columns will, by default, be called "INTEGER_IDX"</li>
* <li>Version field columns will, by default, be called "OPT_VERSION"</li>
* <li>Adapter index field columns will, by default, be called "ADPT_PK_IDX"</li>
* </ul>
* Strictly speaking the naming of the table is inconsistent with the naming of the column since
* the table doesnt use the word separator whereas the column does, but JPOX has been like this for
* some time so we cant just change the default like that - an option needs adding to allow backwards
* compatibility.
*
* @version $Revision: 1.38 $
*/
public class JPOXIdentifierFactory extends AbstractRDBMSIdentifierFactory
{
/** Prefix for all generated table names. */
protected String tablePrefix = null;
/** Suffix for all generated table names. */
protected String tableSuffix = null;
/**
* Constructor.
* TODO Allow specification of multiple properties in PMF to control the id generation and just pass in a Properties into here
* @param dba Datastore adapter
* @param defaultCatalog Name of the default catalog (if any)
* @param defaultSchema Name of the default schema (if any)
* @param requiredCase The case the user requires
* @param wordSeparator Word separator for identifiers
* @param tablePrefix Prefix for table names when being generated
* @param tableSuffix Suffix for table names when being generated
*/
public JPOXIdentifierFactory(DatastoreAdapter dba, String defaultCatalog, String defaultSchema,
String requiredCase, String wordSeparator, String tablePrefix, String tableSuffix)
{
super(dba, requiredCase, defaultCatalog, defaultSchema);
if (wordSeparator != null)
{
this.wordSeparator = wordSeparator;
}
this.tablePrefix = tablePrefix;
this.tableSuffix = tableSuffix;
}
/**
* Method to return a Table identifier for the join table of the specified field.
* @param clr the ClassLoaderResolver
* @param fmd Meta data for the field
* @return The identifier for the table
**/
public DatastoreIdentifier newDatastoreContainerIdentifier(ClassLoaderResolver clr, AbstractMemberMetaData fmd)
{
String identifierName = null;
String schemaName = null;
String catalogName = null;
AbstractMemberMetaData[] relatedMmds = null;
if (fmd.getColumnMetaData().length > 0 && fmd.getColumnMetaData()[0].getName() != null)
{
// Name the table based on the column
identifierName = fmd.getColumnMetaData()[0].getName();
}
else if (fmd.hasContainer())
{
// Check for a specified join table name
if (fmd.getTable() != null)
{
// Join table name specified at this side
String specifiedName = fmd.getTable();
String[] parts = getIdentifierNamePartsFromName(specifiedName);
if (parts != null)
{
catalogName = parts[0];
schemaName = parts[1];
identifierName = parts[2];
}
if (catalogName == null)
{
catalogName = fmd.getCatalog();
}
if (schemaName == null)
{
schemaName = fmd.getSchema();
}
}
else
{
relatedMmds = fmd.getRelatedMemberMetaData(clr);
if (relatedMmds != null && relatedMmds[0].getTable() != null)
{
// Join table name specified at other side (1-N or M-N)
String specifiedName = relatedMmds[0].getTable();
String[] parts = getIdentifierNamePartsFromName(specifiedName);
if (parts != null)
{
catalogName = parts[0];
schemaName = parts[1];
identifierName = parts[2];
}
if (catalogName == null)
{
catalogName = fmd.getCatalog();
}
if (schemaName == null)
{
schemaName = fmd.getSchema();
}
}
}
}
// No schema/catalog specified in the MetaData table "name" so try alternative sources
// Note that we treat these as pairs (they both come from the same source)
if (schemaName == null && catalogName == null)
{
// Check the <class schema="..." catalog="..." >
if (fmd.getParent() instanceof AbstractClassMetaData)
{
AbstractClassMetaData ownerCmd = (AbstractClassMetaData)fmd.getParent();
if (rdba.supportsCatalogsInTableDefinitions())
{
catalogName = ownerCmd.getCatalog();
}
if (rdba.supportsSchemasInTableDefinitions())
{
schemaName = ownerCmd.getSchema();
}
}
if (schemaName == null && catalogName == null)
{
// Still no values, so try the PMF settings.
if (rdba.supportsCatalogsInTableDefinitions())
{
catalogName = this.defaultCatalogName;
}
if (rdba.supportsSchemasInTableDefinitions())
{
schemaName = this.defaultSchemaName;
}
}
}
// No user-specified name, so generate a default using the previously created fallback
if (identifierName == null)
{
// Use this field as the basis for the fallback name (CLASS_FIELD) unless
// this is a bidirectional and the other side is the owner (has mapped-by) in which case OTHERCLASS_FIELD
String fieldNameBasis = fmd.getFullFieldName();
if (relatedMmds != null && relatedMmds[0].getMappedBy() != null)
{
// Other field has mapped-by so use that
fieldNameBasis = relatedMmds[0].getFullFieldName();
}
// Generate a fallback name, based on the class/field name (CLASS_FIELD)
ArrayList name_parts = new ArrayList();
StringTokenizer tokens = new StringTokenizer(fieldNameBasis, ".");
while (tokens.hasMoreTokens())
{
name_parts.add(tokens.nextToken());
}
ListIterator li = name_parts.listIterator(name_parts.size());
String unique_name = (String)li.previous();
String full_name = (li.hasPrevious() ? (li.previous() + getWordSeparator()) : "") + unique_name;
identifierName = "";
if (tablePrefix != null && tablePrefix.length() > 0)
{
identifierName = tablePrefix;
}
if (full_name != null)
{
identifierName += full_name;
}
else
{
identifierName += unique_name;
}
if (tableSuffix != null && tableSuffix.length() > 0)
{
identifierName += tableSuffix;
}
}
// Generate the table identifier now that we have the identifier name
SQLIdentifier identifier = (SQLIdentifier)newDatastoreContainerIdentifier(identifierName);
if (schemaName != null)
{
identifier.setSchemaName(schemaName);
}
if (catalogName != null)
{
identifier.setCatalogName(catalogName);
}
return identifier;
}
/**
* Method to return a Table identifier for the specified class.
* @param clr the ClassLoaderResolver
* @param cmd Meta data for the class
* @return The identifier for the table
**/
public DatastoreIdentifier newDatastoreContainerIdentifier(ClassLoaderResolver clr, AbstractClassMetaData cmd)
{
String identifierName = null;
String schemaName = null;
String catalogName = null;
// Assign an identifier name based on the user-specified table/column name (if any)
String specifiedName = cmd.getTable();
String[] parts = getIdentifierNamePartsFromName(specifiedName);
if (parts != null)
{
catalogName = parts[0];
schemaName = parts[1];
identifierName = parts[2];
}
// No schema/catalog specified in the MetaData table "name" so try alternative sources
// Note that we treat these as pairs (they both come from the same source)
if (schemaName == null && catalogName == null)
{
// Check the <class schema="..." catalog="..." >
if (rdba.supportsCatalogsInTableDefinitions())
{
catalogName = cmd.getCatalog();
}
if (rdba.supportsSchemasInTableDefinitions())
{
schemaName = cmd.getSchema();
}
if (schemaName == null && catalogName == null)
{
// Still no values, so try the PMF settings.
if (rdba.supportsCatalogsInTableDefinitions())
{
catalogName = this.defaultCatalogName;
}
if (rdba.supportsSchemasInTableDefinitions())
{
schemaName = this.defaultSchemaName;
}
}
}
// No user-specified name, so generate a default using the previously created fallback
if (identifierName == null)
{
// Generate a fallback name, based on the last part of the class name ("MyClass" becomes "MYCLASS")
String unique_name = cmd.getFullClassName().substring(cmd.getFullClassName().lastIndexOf('.')+1);
identifierName = "";
if (tablePrefix != null && tablePrefix.length() > 0)
{
identifierName = tablePrefix;
}
identifierName += unique_name;
if (tableSuffix != null && tableSuffix.length() > 0)
{
identifierName += tableSuffix;
}
}
// Generate the table identifier now that we have the identifier name
SQLIdentifier identifier = (SQLIdentifier)newDatastoreContainerIdentifier(identifierName);
if (schemaName != null)
{
identifier.setSchemaName(schemaName);
}
if (catalogName != null)
{
identifier.setCatalogName(catalogName);
}
return identifier;
}
/**
* Method to generate an identifier name for reference field, based on the metadata for the
* field, and the ClassMetaData for the implementation.
* @param refMetaData the metadata for the reference field
* @param implMetaData the AbstractClassMetaData for this implementation
* @param implIdentifier PK identifier for the implementation
* @param embedded Whether the identifier is for a field embedded
* @param fieldRole The role to be performed by this column e.g FK, collection element ?
* @return The DatastoreIdentifier
*/
public DatastoreIdentifier newReferenceFieldIdentifier(AbstractMemberMetaData refMetaData,
AbstractClassMetaData implMetaData, DatastoreIdentifier implIdentifier, boolean embedded, int fieldRole)
{
DatastoreIdentifier identifier = null;
String key = "[" + refMetaData.getFullFieldName() + "][" + implMetaData.getFullClassName() + "][" + implIdentifier.getIdentifier() + "]";
identifier = (DatastoreIdentifier) references.get(key);
if (identifier == null)
{
// use a simple naming for now : <reference-name>_<impl_name>_<impl_type>
String referenceName = refMetaData.getName();
String implementationName = implMetaData.getFullClassName();
int dot = implementationName.lastIndexOf('.');
if (dot > -1)
{
implementationName = implementationName.substring(dot+1);
}
String name = referenceName + "." + implementationName + "." + implIdentifier.getIdentifier();
// Set the SQL identifier adding any truncation as necessary
String suffix = getColumnIdentifierSuffix(fieldRole, embedded);
String datastoreID = generateIdentifierNameForJavaName(name);
String baseID = truncate(datastoreID, getMaxLengthForIdentifierType(IdentifierFactory.COLUMN) - suffix.length());
identifier = new ColumnIdentifier(this, baseID + suffix);
references.put(key, identifier);
}
return identifier;
}
/**
* Method to generate a join-table identifier. The identifier could be for a foreign-key
* to another table (if the destinationId is provided), or could be for a simple column
* in the join table.
* @param ownerFmd MetaData for the owner field
* @param relatedFmd MetaData for the related field
* @param destinationId Identifier for the identity field of the destination (if FK)
* @param embedded Whether the identifier is for a field embedded
* @param fieldRole The role to be performed by this column e.g FK, collection element ?
* @return The identifier.
*/
public DatastoreIdentifier newJoinTableFieldIdentifier(AbstractMemberMetaData ownerFmd, AbstractMemberMetaData relatedFmd,
DatastoreIdentifier destinationId, boolean embedded, int fieldRole)
{
if (destinationId != null)
{
return newDatastoreFieldIdentifier(destinationId.getIdentifier(), embedded, fieldRole);
}
else
{
String baseName = null;
if (fieldRole == FieldRole.ROLE_COLLECTION_ELEMENT)
{
String elementType = ownerFmd.getCollection().getElementType();
baseName = elementType.substring(elementType.lastIndexOf('.') + 1);
}
else if (fieldRole == FieldRole.ROLE_ARRAY_ELEMENT)
{
String elementType = ownerFmd.getArray().getElementType();
baseName = elementType.substring(elementType.lastIndexOf('.') + 1);
}
else if (fieldRole == FieldRole.ROLE_MAP_KEY)
{
String keyType = ownerFmd.getMap().getKeyType();
baseName = keyType.substring(keyType.lastIndexOf('.') + 1);
}
else if (fieldRole == FieldRole.ROLE_MAP_VALUE)
{
String valueType = ownerFmd.getMap().getValueType();
baseName = valueType.substring(valueType.lastIndexOf('.') + 1);
}
else
{
baseName = "UNKNOWN";
}
return newDatastoreFieldIdentifier(baseName, embedded, fieldRole);
}
}
/**
* Method to generate a FK/FK-index field identifier.
* The identifier could be for the FK field itself, or for a related index for the FK.
* @param ownerFmd MetaData for the owner field
* @param relatedFmd MetaData for the related field
* @param destinationId Identifier for the identity field of the destination table (if strict FK)
* @param embedded Whether the identifier is for a field embedded
* @param fieldRole The role to be performed by this column e.g owner, index ?
* @return The identifier
*/
public DatastoreIdentifier newForeignKeyFieldIdentifier(AbstractMemberMetaData ownerFmd, AbstractMemberMetaData relatedFmd,
DatastoreIdentifier destinationId, boolean embedded, int fieldRole)
{
if (relatedFmd != null)
{
// Bidirectional
if (fieldRole == FieldRole.ROLE_OWNER)
{
return newDatastoreFieldIdentifier(relatedFmd.getName() + "." + destinationId.getIdentifier(), embedded, fieldRole);
}
else if (fieldRole == FieldRole.ROLE_INDEX)
{
return newDatastoreFieldIdentifier(relatedFmd.getName() + "." + destinationId.getIdentifier(), embedded, fieldRole);
}
else
{
throw new JPOXException("DatastoreField role " + fieldRole + " not supported by this method").setFatal();
}
}
else
{
if (fieldRole == FieldRole.ROLE_OWNER)
{
// FK field (FK collection/array/list/map)
return newDatastoreFieldIdentifier(ownerFmd.getName() + "." + destinationId.getIdentifier(), embedded, fieldRole);
}
else if (fieldRole == FieldRole.ROLE_INDEX)
{
// Order field for FK (FK list)
return newDatastoreFieldIdentifier(ownerFmd.getName() + "." + "INTEGER", embedded, fieldRole);
}
else
{
throw new JPOXException("DatastoreField role " + fieldRole + " not supported by this method").setFatal();
}
}
}
/**
* Method to return an identifier for a discriminator column.
* Returns an identifier "DISCRIMINATOR"
* @return The discriminator column identifier
*/
public DatastoreIdentifier newDiscriminatorFieldIdentifier()
{
String name = "DISCRIMINATOR";
DatastoreIdentifier identifier = (DatastoreIdentifier) columns.get(name);
if (identifier == null)
{
identifier = new ColumnIdentifier(this, name);
columns.put(name, identifier);
}
return identifier;
}
/**
* Method to return an identifier for a version datastore field.
* @return The version datastore field identifier
*/
public DatastoreIdentifier newVersionFieldIdentifier()
{
String name = "OPT_VERSION";
DatastoreIdentifier identifier = (DatastoreIdentifier) columns.get(name);
if (identifier == null)
{
identifier = new ColumnIdentifier(this, name);
columns.put(name, identifier);
}
return identifier;
}
/**
* Method to return an identifier for an adapter index datastore field.
* An "adapter index" is a column added to be part of a primary key when some other
* column cant perform that role.
* @return The index datastore field identifier
*/
public DatastoreIdentifier newAdapterIndexFieldIdentifier()
{
String name = "ADPT_PK_IDX";
DatastoreIdentifier identifier = (DatastoreIdentifier) columns.get(name);
if (identifier == null)
{
identifier = new ColumnIdentifier(this, name);
columns.put(name, identifier);
}
return identifier;
}
/**
* Method to return an identifier for an index datastore field.
* @return The index datastore field identifier
*/
public DatastoreIdentifier newIndexFieldIdentifier()
{
String name = "INTEGER_IDX";
DatastoreIdentifier identifier = (DatastoreIdentifier) columns.get(name);
if (identifier == null)
{
identifier = new ColumnIdentifier(this, name);
columns.put(name, identifier);
}
return identifier;
}
/**
* Generate a datastore identifier from a Java identifier.
*
* <p>Conversion consists of breaking the identifier into words, converting
* each word to upper-case, and separating each one with a word separator.
* Words are identified by a leading upper-case character.
* Any leading or trailing underscores are removed.</p>
*
* @param javaName the Java identifier.
* @return The datastore identifier
*/
public String generateIdentifierNameForJavaName(String javaName)
{
if (javaName == null)
{
return null;
}
StringBuffer s = new StringBuffer();
char prev = '\0';
for (int i = 0; i < javaName.length(); ++i)
{
char c = javaName.charAt(i);
if (c >= 'A' && c <= 'Z' && (identifierCase != IDENTIFIER_MIXED_CASE && identifierCase != IDENTIFIER_MIXED_CASE_QUOTED))
{
if (prev >= 'a' && prev <= 'z')
{
s.append(wordSeparator);
}
s.append(c);
}
else if (c >= 'A' && c <= 'Z' && (identifierCase == IDENTIFIER_MIXED_CASE || identifierCase == IDENTIFIER_MIXED_CASE_QUOTED))
{
s.append(c);
}
else if (c >= 'a' && c <= 'z' && (identifierCase == IDENTIFIER_MIXED_CASE || identifierCase == IDENTIFIER_MIXED_CASE_QUOTED))
{
s.append(c);
}
else if (c >= 'a' && c <= 'z' && (identifierCase != IDENTIFIER_MIXED_CASE && identifierCase != IDENTIFIER_MIXED_CASE_QUOTED))
{
s.append((char)(c - ('a' - 'A')));
}
else if (c >= '0' && c <= '9' || c=='_')
{
s.append(c);
}
else if (c == '.')
{
s.append(wordSeparator);
}
else
{
String cval = "000" + Integer.toHexString(c);
s.append(cval.substring(cval.length() - (c > 0xff ? 4 : 2)));
}
prev = c;
}
// Remove leading and trailing underscores
while (s.length() > 0 && s.charAt(0) == '_')
{
s.deleteCharAt(0);
}
if (s.length() == 0)
{
throw new IllegalArgumentException("Illegal Java identifier: " + javaName);
}
return s.toString();
}
/**
* Accessor for the suffix to add to any column identifier, based on the role type.
* @param role Datastore field role
* @param embedded Whether the DatastoreField is stored embedded
* @return The suffix (e.g _ID for id columns).
**/
protected String getColumnIdentifierSuffix(int role, boolean embedded)
{
String suffix;
switch (role)
{
case FieldRole.ROLE_NONE :
default :
suffix = !embedded ? "_ID" : "";
break;
case FieldRole.ROLE_CUSTOM :
suffix = "";
break;
case FieldRole.ROLE_OWNER :
suffix = !embedded ? "_OID" : "_OWN";
break;
case FieldRole.ROLE_FIELD :
case FieldRole.ROLE_COLLECTION_ELEMENT :
case FieldRole.ROLE_ARRAY_ELEMENT :
suffix = !embedded ? "_EID" : "_ELE";
break;
case FieldRole.ROLE_MAP_KEY :
suffix = !embedded ? "_KID" : "_KEY";
break;
case FieldRole.ROLE_MAP_VALUE :
suffix = !embedded ? "_VID" : "_VAL";
break;
case FieldRole.ROLE_INDEX :
suffix = !embedded ? "_XID" : "_IDX";
break;
}
return suffix;
}
}