public void testUniqueProperty() {
DefaultGrailsDomainConfiguration config =
getDomainConfig(
"class Widget {\n"
+ " Long id \n"
+ " Long version \n"
+ " String name \n"
+ " String description \n"
+ " static mapping = {\n"
+ " name unique: true\n"
+ " }\n"
+ "}");
Table tableMapping = getTableMapping("widget", config);
Column nameColumn = tableMapping.getColumn(new Column("name"));
Column descriptionColumn = tableMapping.getColumn(new Column("description"));
assertEquals(true, nameColumn.isUnique());
assertEquals(false, descriptionColumn.isUnique());
}
protected void appendCommonColumnInfo(
StringBuffer annotations, Column column, boolean insertable, boolean updatable) {
if (column.isUnique()) {
annotations.append(", unique=").append(column.isUnique());
}
if (!column.isNullable()) {
annotations.append(", nullable=").append(column.isNullable());
}
if (!insertable) {
annotations.append(", insertable=").append(insertable);
}
if (!updatable) {
annotations.append(", updatable=").append(updatable);
}
String sqlType = column.getSqlType();
if (StringHelper.isNotEmpty(sqlType)) {
annotations.append(", columnDefinition=\"").append(sqlType).append("\"");
}
}
/**
* Builds the <code>Join</code> instance for the mapped by side of a <i>OneToOne</i> association
* using a join tables.
*
* <p>Note:<br>
*
* <ul>
* <li>From the mappedBy side we should not create the PK nor the FK, this is handled from the
* other side.
* <li>This method is a dirty dupe of EntityBinder.bindSecondaryTable</i>.
*/
private Join buildJoinFromMappedBySide(
PersistentClass persistentClass, Property otherSideProperty, Join originalJoin) {
Join join = new Join();
join.setPersistentClass(persistentClass);
// no check constraints available on joins
join.setTable(originalJoin.getTable());
join.setInverse(true);
SimpleValue key =
new DependantValue(mappings, join.getTable(), persistentClass.getIdentifier());
// TODO support @ForeignKey
join.setKey(key);
join.setSequentialSelect(false);
// TODO support for inverse and optional
join.setOptional(true); // perhaps not quite per-spec, but a Good Thing anyway
key.setCascadeDeleteEnabled(false);
Iterator mappedByColumns = otherSideProperty.getValue().getColumnIterator();
while (mappedByColumns.hasNext()) {
Column column = (Column) mappedByColumns.next();
Column copy = new Column();
copy.setLength(column.getLength());
copy.setScale(column.getScale());
copy.setValue(key);
copy.setName(column.getQuotedName());
copy.setNullable(column.isNullable());
copy.setPrecision(column.getPrecision());
copy.setUnique(column.isUnique());
copy.setSqlType(column.getSqlType());
copy.setCheckConstraint(column.getCheckConstraint());
copy.setComment(column.getComment());
copy.setDefaultValue(column.getDefaultValue());
key.addColumn(copy);
}
persistentClass.addJoin(join);
return join;
}
public String sqlCreateString(
Dialect dialect, Mapping p, String defaultCatalog, String defaultSchema)
throws HibernateException {
StringBuffer buf =
new StringBuffer("create table ")
.append(getQualifiedName(dialect, defaultCatalog, defaultSchema))
.append(" (");
boolean identityColumn = idValue != null && idValue.isIdentityColumn(dialect);
// Try to find out the name of the primary key to create it as identity if the IdentityGenerator
// is used
String pkname = null;
if (hasPrimaryKey() && identityColumn) {
pkname = ((Column) getPrimaryKey().getColumnIterator().next()).getQuotedName(dialect);
}
Iterator iter = getColumnIterator();
while (iter.hasNext()) {
Column col = (Column) iter.next();
buf.append(col.getQuotedName(dialect)).append(' ');
if (identityColumn && col.getQuotedName(dialect).equals(pkname)) {
// to support dialects that have their own identity data type
if (dialect.hasDataTypeInIdentityColumn()) {
buf.append(col.getSqlType(dialect, p));
}
buf.append(' ').append(dialect.getIdentityColumnString(col.getSqlTypeCode(p)));
} else {
buf.append(col.getSqlType(dialect, p));
String defaultValue = col.getDefaultValue();
if (defaultValue != null) {
buf.append(" default ").append(defaultValue);
}
if (col.isNullable()) {
buf.append(dialect.getNullColumnString());
} else {
buf.append(" not null");
}
}
boolean useUniqueConstraint =
col.isUnique() && (!col.isNullable() || dialect.supportsNotNullUnique());
if (useUniqueConstraint) {
if (dialect.supportsUnique()) {
buf.append(" unique");
} else {
UniqueKey uk = getOrCreateUniqueKey(col.getQuotedName(dialect) + '_');
uk.addColumn(col);
}
}
if (col.hasCheckConstraint() && dialect.supportsColumnCheck()) {
buf.append(" check (").append(col.getCheckConstraint()).append(")");
}
String columnComment = col.getComment();
if (columnComment != null) {
buf.append(dialect.getColumnComment(columnComment));
}
if (iter.hasNext()) buf.append(", ");
}
if (hasPrimaryKey()) {
buf.append(", ").append(getPrimaryKey().sqlConstraintString(dialect));
}
if (dialect.supportsUniqueConstraintInCreateAlterTable()) {
Iterator ukiter = getUniqueKeyIterator();
while (ukiter.hasNext()) {
UniqueKey uk = (UniqueKey) ukiter.next();
buf.append(", ").append(uk.sqlConstraintString(dialect));
}
}
/*Iterator idxiter = getIndexIterator();
while ( idxiter.hasNext() ) {
Index idx = (Index) idxiter.next();
buf.append(',').append( idx.sqlConstraintString(dialect) );
}*/
if (dialect.supportsTableCheck()) {
Iterator chiter = checkConstraints.iterator();
while (chiter.hasNext()) {
buf.append(", check (").append(chiter.next()).append(')');
}
}
buf.append(')');
if (comment != null) buf.append(dialect.getTableComment(comment));
return buf.append(dialect.getTableTypeString()).toString();
}
public Iterator sqlAlterStrings(
Dialect dialect,
Mapping p,
TableMetadata tableInfo,
String defaultCatalog,
String defaultSchema)
throws HibernateException {
StringBuffer root =
new StringBuffer("alter table ")
.append(getQualifiedName(dialect, defaultCatalog, defaultSchema))
.append(' ')
.append(dialect.getAddColumnString());
Iterator iter = getColumnIterator();
List results = new ArrayList();
while (iter.hasNext()) {
Column column = (Column) iter.next();
ColumnMetadata columnInfo = tableInfo.getColumnMetadata(column.getName());
if (columnInfo == null) {
// the column doesnt exist at all.
StringBuffer alter =
new StringBuffer(root.toString())
.append(' ')
.append(column.getQuotedName(dialect))
.append(' ')
.append(column.getSqlType(dialect, p));
String defaultValue = column.getDefaultValue();
if (defaultValue != null) {
alter.append(" default ").append(defaultValue);
if (column.isNullable()) {
alter.append(dialect.getNullColumnString());
} else {
alter.append(" not null");
}
}
boolean useUniqueConstraint =
column.isUnique()
&& dialect.supportsUnique()
&& (!column.isNullable() || dialect.supportsNotNullUnique());
if (useUniqueConstraint) {
alter.append(" unique");
}
if (column.hasCheckConstraint() && dialect.supportsColumnCheck()) {
alter.append(" check(").append(column.getCheckConstraint()).append(")");
}
String columnComment = column.getComment();
if (columnComment != null) alter.append(dialect.getColumnComment(columnComment));
results.add(alter.toString());
}
}
return results.iterator();
}
public String sqlCreateString(
Dialect dialect, Mapping p, String defaultCatalog, String defaultSchema) {
StringBuilder buf =
new StringBuilder(
hasPrimaryKey()
? dialect.getCreateTableString()
: dialect.getCreateMultisetTableString())
.append(' ')
.append(getQualifiedName(dialect, defaultCatalog, defaultSchema))
.append(" (");
boolean identityColumn =
idValue != null && idValue.isIdentityColumn(p.getIdentifierGeneratorFactory(), dialect);
// Try to find out the name of the primary key to create it as identity if the IdentityGenerator
// is used
String pkname = null;
if (hasPrimaryKey() && identityColumn) {
pkname = ((Column) getPrimaryKey().getColumnIterator().next()).getQuotedName(dialect);
}
Iterator iter = getColumnIterator();
while (iter.hasNext()) {
Column col = (Column) iter.next();
buf.append(col.getQuotedName(dialect)).append(' ');
if (identityColumn && col.getQuotedName(dialect).equals(pkname)) {
// to support dialects that have their own identity data type
if (dialect.getIdentityColumnSupport().hasDataTypeInIdentityColumn()) {
buf.append(col.getSqlType(dialect, p));
}
buf.append(' ')
.append(
dialect.getIdentityColumnSupport().getIdentityColumnString(col.getSqlTypeCode(p)));
} else {
buf.append(col.getSqlType(dialect, p));
String defaultValue = col.getDefaultValue();
if (defaultValue != null) {
buf.append(" default ").append(defaultValue);
}
if (col.isNullable()) {
buf.append(dialect.getNullColumnString());
} else {
buf.append(" not null");
}
}
if (col.isUnique()) {
String keyName = Constraint.generateName("UK_", this, col);
UniqueKey uk = getOrCreateUniqueKey(keyName);
uk.addColumn(col);
buf.append(dialect.getUniqueDelegate().getColumnDefinitionUniquenessFragment(col));
}
if (col.hasCheckConstraint() && dialect.supportsColumnCheck()) {
buf.append(" check (").append(col.getCheckConstraint()).append(")");
}
String columnComment = col.getComment();
if (columnComment != null) {
buf.append(dialect.getColumnComment(columnComment));
}
if (iter.hasNext()) {
buf.append(", ");
}
}
if (hasPrimaryKey()) {
buf.append(", ").append(getPrimaryKey().sqlConstraintString(dialect));
}
buf.append(dialect.getUniqueDelegate().getTableCreationUniqueConstraintsFragment(this));
if (dialect.supportsTableCheck()) {
for (String checkConstraint : checkConstraints) {
buf.append(", check (").append(checkConstraint).append(')');
}
}
buf.append(')');
if (comment != null) {
buf.append(dialect.getTableComment(comment));
}
return buf.append(dialect.getTableTypeString()).toString();
}
public Iterator sqlAlterStrings(
Dialect dialect,
Mapping p,
TableInformation tableInfo,
String defaultCatalog,
String defaultSchema)
throws HibernateException {
StringBuilder root =
new StringBuilder("alter table ")
.append(getQualifiedName(dialect, defaultCatalog, defaultSchema))
.append(' ')
.append(dialect.getAddColumnString());
Iterator iter = getColumnIterator();
List results = new ArrayList();
while (iter.hasNext()) {
final Column column = (Column) iter.next();
final ColumnInformation columnInfo =
tableInfo.getColumn(Identifier.toIdentifier(column.getName(), column.isQuoted()));
if (columnInfo == null) {
// the column doesnt exist at all.
StringBuilder alter =
new StringBuilder(root.toString())
.append(' ')
.append(column.getQuotedName(dialect))
.append(' ')
.append(column.getSqlType(dialect, p));
String defaultValue = column.getDefaultValue();
if (defaultValue != null) {
alter.append(" default ").append(defaultValue);
}
if (column.isNullable()) {
alter.append(dialect.getNullColumnString());
} else {
alter.append(" not null");
}
if (column.isUnique()) {
String keyName = Constraint.generateName("UK_", this, column);
UniqueKey uk = getOrCreateUniqueKey(keyName);
uk.addColumn(column);
alter.append(dialect.getUniqueDelegate().getColumnDefinitionUniquenessFragment(column));
}
if (column.hasCheckConstraint() && dialect.supportsColumnCheck()) {
alter.append(" check(").append(column.getCheckConstraint()).append(")");
}
String columnComment = column.getComment();
if (columnComment != null) {
alter.append(dialect.getColumnComment(columnComment));
}
alter.append(dialect.getAddColumnSuffixString());
results.add(alter.toString());
}
}
if (results.isEmpty()) {
Logger.getLogger(SchemaUpdate.class)
.debugf("No alter strings for table : %s", getQuotedName());
}
return results.iterator();
}
// TODO refactor this code, there is a lot of duplication in this method
public void doSecondPass(Map persistentClasses) throws MappingException {
org.hibernate.mapping.OneToOne value =
new org.hibernate.mapping.OneToOne(
mappings, propertyHolder.getTable(), propertyHolder.getPersistentClass());
final String propertyName = inferredData.getPropertyName();
value.setPropertyName(propertyName);
String referencedEntityName =
ToOneBinder.getReferenceEntityName(inferredData, targetEntity, mappings);
value.setReferencedEntityName(referencedEntityName);
AnnotationBinder.defineFetchingStrategy(value, inferredData.getProperty());
// value.setFetchMode( fetchMode );
value.setCascadeDeleteEnabled(cascadeOnDelete);
// value.setLazy( fetchMode != FetchMode.JOIN );
if (!optional) value.setConstrained(true);
value.setForeignKeyType(
value.isConstrained()
? ForeignKeyDirection.FOREIGN_KEY_FROM_PARENT
: ForeignKeyDirection.FOREIGN_KEY_TO_PARENT);
PropertyBinder binder = new PropertyBinder();
binder.setName(propertyName);
binder.setValue(value);
binder.setCascade(cascadeStrategy);
binder.setAccessType(inferredData.getDefaultAccess());
Property prop = binder.makeProperty();
if (BinderHelper.isEmptyAnnotationValue(mappedBy)) {
/*
* we need to check if the columns are in the right order
* if not, then we need to create a many to one and formula
* but actually, since entities linked by a one to one need
* to share the same composite id class, this cannot happen in hibernate
*/
boolean rightOrder = true;
if (rightOrder) {
String path = StringHelper.qualify(propertyHolder.getPath(), propertyName);
(new ToOneFkSecondPass(
value,
joinColumns,
!optional, // cannot have nullabe and unique on certain DBs
propertyHolder.getEntityOwnerClassName(),
path,
mappings))
.doSecondPass(persistentClasses);
// no column associated since its a one to one
propertyHolder.addProperty(prop, inferredData.getDeclaringClass());
} else {
// this is a many to one with Formula
}
} else {
PersistentClass otherSide =
(PersistentClass) persistentClasses.get(value.getReferencedEntityName());
Property otherSideProperty;
try {
if (otherSide == null) {
throw new MappingException("Unable to find entity: " + value.getReferencedEntityName());
}
otherSideProperty = BinderHelper.findPropertyByName(otherSide, mappedBy);
} catch (MappingException e) {
throw new AnnotationException(
"Unknown mappedBy in: "
+ StringHelper.qualify(ownerEntity, ownerProperty)
+ ", referenced property unknown: "
+ StringHelper.qualify(value.getReferencedEntityName(), mappedBy));
}
if (otherSideProperty == null) {
throw new AnnotationException(
"Unknown mappedBy in: "
+ StringHelper.qualify(ownerEntity, ownerProperty)
+ ", referenced property unknown: "
+ StringHelper.qualify(value.getReferencedEntityName(), mappedBy));
}
if (otherSideProperty.getValue() instanceof OneToOne) {
propertyHolder.addProperty(prop, inferredData.getDeclaringClass());
} else if (otherSideProperty.getValue() instanceof ManyToOne) {
Iterator it = otherSide.getJoinIterator();
Join otherSideJoin = null;
while (it.hasNext()) {
Join otherSideJoinValue = (Join) it.next();
if (otherSideJoinValue.containsProperty(otherSideProperty)) {
otherSideJoin = otherSideJoinValue;
break;
}
}
if (otherSideJoin != null) {
// @OneToOne @JoinTable
Join mappedByJoin =
buildJoinFromMappedBySide(
(PersistentClass) persistentClasses.get(ownerEntity),
otherSideProperty,
otherSideJoin);
ManyToOne manyToOne = new ManyToOne(mappings, mappedByJoin.getTable());
// FIXME use ignore not found here
manyToOne.setIgnoreNotFound(ignoreNotFound);
manyToOne.setCascadeDeleteEnabled(value.isCascadeDeleteEnabled());
manyToOne.setEmbedded(value.isEmbedded());
manyToOne.setFetchMode(value.getFetchMode());
manyToOne.setLazy(value.isLazy());
manyToOne.setReferencedEntityName(value.getReferencedEntityName());
manyToOne.setUnwrapProxy(value.isUnwrapProxy());
prop.setValue(manyToOne);
Iterator otherSideJoinKeyColumns = otherSideJoin.getKey().getColumnIterator();
while (otherSideJoinKeyColumns.hasNext()) {
Column column = (Column) otherSideJoinKeyColumns.next();
Column copy = new Column();
copy.setLength(column.getLength());
copy.setScale(column.getScale());
copy.setValue(manyToOne);
copy.setName(column.getQuotedName());
copy.setNullable(column.isNullable());
copy.setPrecision(column.getPrecision());
copy.setUnique(column.isUnique());
copy.setSqlType(column.getSqlType());
copy.setCheckConstraint(column.getCheckConstraint());
copy.setComment(column.getComment());
copy.setDefaultValue(column.getDefaultValue());
manyToOne.addColumn(copy);
}
mappedByJoin.addProperty(prop);
} else {
propertyHolder.addProperty(prop, inferredData.getDeclaringClass());
}
value.setReferencedPropertyName(mappedBy);
String propertyRef = value.getReferencedPropertyName();
if (propertyRef != null) {
mappings.addUniquePropertyReference(value.getReferencedEntityName(), propertyRef);
}
} else {
throw new AnnotationException(
"Referenced property not a (One|Many)ToOne: "
+ StringHelper.qualify(otherSide.getEntityName(), mappedBy)
+ " in mappedBy of "
+ StringHelper.qualify(ownerEntity, ownerProperty));
}
}
ForeignKey fk = inferredData.getProperty().getAnnotation(ForeignKey.class);
String fkName = fk != null ? fk.name() : "";
if (!BinderHelper.isEmptyAnnotationValue(fkName)) value.setForeignKeyName(fkName);
}