mirror of https://github.com/apache/openjpa.git
Adding @UniqueConstraint annotation.
git-svn-id: https://svn.apache.org/repos/asf/incubator/openjpa/trunk@492225 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
f69eb13714
commit
3e6410b75d
|
@ -36,6 +36,8 @@ import org.apache.openjpa.jdbc.schema.ColumnIO;
|
|||
import org.apache.openjpa.jdbc.schema.ForeignKey;
|
||||
import org.apache.openjpa.jdbc.schema.Schemas;
|
||||
import org.apache.openjpa.jdbc.schema.Table;
|
||||
import org.apache.openjpa.jdbc.schema.Unique;
|
||||
import org.apache.openjpa.jdbc.schema.XMLSchemaParser;
|
||||
import org.apache.openjpa.jdbc.sql.Joins;
|
||||
import org.apache.openjpa.jdbc.sql.Result;
|
||||
import org.apache.openjpa.jdbc.sql.RowManager;
|
||||
|
@ -793,6 +795,44 @@ public class ClassMapping
|
|||
_cols[i].setFlag(Column.FLAG_DIRECT_UPDATE, true);
|
||||
}
|
||||
}
|
||||
mapUniqueConstraints();
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds unique constraints to the mapped table.
|
||||
*
|
||||
*/
|
||||
void mapUniqueConstraints() {
|
||||
Log log = getRepository().getLog();
|
||||
Collection uniqueInfos = _info.getUniqueConstraints();
|
||||
if (uniqueInfos == null || uniqueInfos.isEmpty())
|
||||
return;
|
||||
Iterator iter = uniqueInfos.iterator();
|
||||
Table table = getTable();
|
||||
int i = 1;
|
||||
while (iter.hasNext()) {
|
||||
XMLSchemaParser.UniqueInfo uniqueInfo =
|
||||
(XMLSchemaParser.UniqueInfo)iter.next();
|
||||
if (uniqueInfo.cols == null || uniqueInfo.cols.isEmpty())
|
||||
continue;
|
||||
String constraintName = table.getName() + "_UNIQUE_" + i;
|
||||
i++;
|
||||
Unique uniqueConstraint = table.addUnique(constraintName);
|
||||
Iterator uniqueColumnNames = uniqueInfo.cols.iterator();
|
||||
while (uniqueColumnNames.hasNext()) {
|
||||
String uniqueColumnName = (String)uniqueColumnNames.next();
|
||||
Column uniqueColumn = table.getColumn(uniqueColumnName);
|
||||
if (uniqueColumn != null) {
|
||||
uniqueConstraint.addColumn(uniqueColumn);
|
||||
} else {
|
||||
table.removeUnique(uniqueConstraint);
|
||||
if (log.isWarnEnabled())
|
||||
log.warn(_loc.get("missing-unique-column", this,
|
||||
table.getName(), uniqueColumnName));
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -21,6 +21,9 @@ import java.util.HashMap;
|
|||
import java.util.Iterator;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Collection;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
|
||||
import org.apache.openjpa.jdbc.meta.strats.FullClassStrategy;
|
||||
import org.apache.openjpa.jdbc.schema.Column;
|
||||
|
@ -28,6 +31,7 @@ import org.apache.openjpa.jdbc.schema.ForeignKey;
|
|||
import org.apache.openjpa.jdbc.schema.Schema;
|
||||
import org.apache.openjpa.jdbc.schema.SchemaGroup;
|
||||
import org.apache.openjpa.jdbc.schema.Table;
|
||||
import org.apache.openjpa.jdbc.schema.XMLSchemaParser;
|
||||
import org.apache.openjpa.lib.meta.SourceTracker;
|
||||
import org.apache.openjpa.lib.xml.Commentable;
|
||||
|
||||
|
@ -51,6 +55,7 @@ public class ClassMappingInfo
|
|||
private File _file = null;
|
||||
private int _srcType = SRC_OTHER;
|
||||
private String[] _comments = null;
|
||||
private Collection _uniqueConstraints = null;//XMLSchemaParser.UniqueInfo
|
||||
|
||||
/**
|
||||
* The described class name.
|
||||
|
@ -308,6 +313,20 @@ public class ClassMappingInfo
|
|||
_seconds.put(key, cinfo._seconds.get(key));
|
||||
}
|
||||
}
|
||||
if (cinfo._uniqueConstraints != null)
|
||||
_uniqueConstraints = new ArrayList(cinfo._uniqueConstraints);
|
||||
}
|
||||
|
||||
public void addUniqueConstaint(String[] columnNames) {
|
||||
if (_uniqueConstraints == null)
|
||||
_uniqueConstraints = new ArrayList();
|
||||
XMLSchemaParser.UniqueInfo uniqueInfo = new XMLSchemaParser.UniqueInfo();
|
||||
uniqueInfo.cols = Arrays.asList(columnNames);
|
||||
_uniqueConstraints.add(uniqueInfo);
|
||||
}
|
||||
|
||||
public Collection getUniqueConstraints() {
|
||||
return _uniqueConstraints;
|
||||
}
|
||||
|
||||
public File getSourceFile() {
|
||||
|
|
|
@ -560,6 +560,27 @@ public class SchemaTool {
|
|||
}
|
||||
}
|
||||
|
||||
// Unique Constraints on group of columns
|
||||
Unique[] uniques;
|
||||
for (int i = 0; i < schemas.length; i++) {
|
||||
tabs = schemas[i].getTables();
|
||||
for (int j = 0; j < tabs.length; j++) {
|
||||
// create unique constraints only on new tables
|
||||
if (!newTables.contains(tabs[j]))
|
||||
continue;
|
||||
|
||||
uniques = tabs[j].getUniques();
|
||||
if (uniques == null || uniques.length == 0)
|
||||
continue;
|
||||
dbTable = db.findTable(tabs[j]);
|
||||
if (dbTable == null)
|
||||
continue;
|
||||
for (int k = 0; k < uniques.length; k++) {
|
||||
dbTable.importUnique(uniques[k]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// foreign keys
|
||||
ForeignKey[] fks;
|
||||
ForeignKey fk;
|
||||
|
|
|
@ -579,7 +579,7 @@ public class XMLSchemaParser
|
|||
/**
|
||||
* Used to hold unique constraint info before it is resolved.
|
||||
*/
|
||||
private static class UniqueInfo {
|
||||
public static class UniqueInfo {
|
||||
|
||||
public Unique unq = null;
|
||||
public Collection cols = new LinkedList();
|
||||
|
|
|
@ -401,3 +401,6 @@ untraversable-path: Result path "{2}" in result type "{1}" of mapping "{0}" \
|
|||
attempts to traverse through a non-relation field.
|
||||
num-cols-path: Result path "{2}" in result type "{1}" of mapping "{0}" \
|
||||
attempts to map a field that does not have exactly 1 column.
|
||||
missing-unique-column: A unique constraint includes a column "{2}" specified \
|
||||
in mapping of class "{0}" to table "{1}". However, the column does not \
|
||||
exist in "{1}" table. This constraint will not be defined in the schema.
|
||||
|
|
|
@ -47,6 +47,7 @@ import javax.persistence.SqlResultSetMappings;
|
|||
import javax.persistence.Table;
|
||||
import javax.persistence.TableGenerator;
|
||||
import javax.persistence.Temporal;
|
||||
import javax.persistence.UniqueConstraint;
|
||||
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.apache.openjpa.jdbc.conf.JDBCConfiguration;
|
||||
|
@ -459,10 +460,9 @@ public class AnnotationPersistenceMappingParser
|
|||
if (tableName != null)
|
||||
cm.getMappingInfo().setTableName(tableName);
|
||||
|
||||
//### EJB3
|
||||
Log log = getLog();
|
||||
if (table.uniqueConstraints().length > 0 && log.isWarnEnabled())
|
||||
log.warn(_loc.get("unique-constraints", cm));
|
||||
for (UniqueConstraint unique:table.uniqueConstraints()) {
|
||||
((ClassMappingInfo)cm.getMappingInfo()).addUniqueConstaint(unique.columnNames());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
Loading…
Reference in New Issue