org.apache.openjpa.jdbc.schema.SchemaGroup Maven / Gradle / Ivy
The newest version!
/*
* 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.openjpa.jdbc.schema;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import org.apache.openjpa.jdbc.identifier.DBIdentifier;
import org.apache.openjpa.jdbc.identifier.QualifiedDBIdentifier;
/**
* Represents a grouping of schemas used in a database.
*
* @author Abe White
*/
public class SchemaGroup
extends NameSet
implements Cloneable {
private static final long serialVersionUID = 1L;
private Map _schemaMap = null;
// cache
private Schema[] _schemas = null;
/**
* Return all schemas.
*/
public Schema[] getSchemas() {
if (_schemas == null)
_schemas = (_schemaMap == null) ? new Schema[0] : (Schema[])
_schemaMap.values().toArray(new Schema[_schemaMap.size()]);
return _schemas;
}
/**
* Return the schema with the given name, or null if none.
* @deprecated
*/
@Deprecated
public Schema getSchema(String name) {
if (_schemaMap == null)
return null;
return getSchema(DBIdentifier.toUpper(DBIdentifier.newSchema(name)));
}
public Schema getSchema(DBIdentifier name) {
if (_schemaMap == null)
return null;
DBIdentifier sName = DBIdentifier.toUpper(name);
Schema schema = _schemaMap.get(sName);
return schema;
}
/**
* Add a schema to the group.
*/
public Schema addSchema() {
return addSchema(DBIdentifier.NULL);
}
/**
* Add a schema to the group.
*/
public Schema addSchema(DBIdentifier name) {
addName(name, false);
Schema schema = newSchema(name);
DBIdentifier sName = DBIdentifier.toUpper(name);
if (_schemaMap == null)
_schemaMap = new HashMap<>();
_schemaMap.put(sName, schema);
_schemas = null;
return schema;
}
/**
* @deprecated
* @param name
*/
@Deprecated
public Schema addSchema(String name) {
return addSchema(DBIdentifier.newSchema(name));
}
/**
* Remove the given schema from the group.
*
* @return true if the schema was removed, false if not in the group
*/
public boolean removeSchema(Schema schema) {
if (schema == null)
return false;
DBIdentifier name = DBIdentifier.toUpper(schema.getIdentifier());
Schema rem = _schemaMap.get(name);
if (schema.equals(rem)) {
_schemaMap.remove(name);
removeName(schema.getIdentifier());
_schemas = null;
schema.remove();
return true;
}
return false;
}
/**
* Import a schema from another group. Foreign keys are not imported.
*/
public Schema importSchema(Schema schema) {
if (schema == null)
return null;
Schema copy = addSchema(schema.getIdentifier());
Sequence[] seqs = schema.getSequences();
for (Sequence seq : seqs) {
copy.importSequence(seq);
}
Table[] tables = schema.getTables();
Index[] idxs;
Unique[] unqs;
Table tab;
for (Table table : tables) {
tab = copy.importTable(table);
idxs = table.getIndexes();
for (Index idx : idxs) {
tab.importIndex(idx);
}
unqs = table.getUniques();
for (Unique unq : unqs) {
tab.importUnique(unq);
}
}
return copy;
}
/**
* Return true if the given table is known to exist. While
* {@link #findTable} may exhibit dynamic behavior in some schema group
* implementations, this method only returns true if the table has been
* added to this group or is known to exist in the database.
*/
public boolean isKnownTable(Table table) {
return findTable(table) != null;
}
/**
* Return true if the given table is known to exist. While
* {@link #findTable} may exhibit dynamic behavior in some schema group
* implementations, this method only returns true if the table has been
* added to this group or is known to exist in the database.
* @deprecated
*/
@Deprecated
public boolean isKnownTable(String name) {
return findTable(name) != null;
}
public boolean isKnownTable(QualifiedDBIdentifier path) {
return findTable(path) != null;
}
/**
* Find the equivalent of the given table in this schema group. The
* given table that may have come from another schema group.
*/
public Table findTable(Table table) {
return findTable(table.getQualifiedPath());
}
/**
* Find the table with the given name in the group, using '.' as the
* catalog separator. Returns null if no table found.
* @deprecated
*/
@Deprecated
public Table findTable(String name) {
if (name == null)
return null;
return findTable(QualifiedDBIdentifier.getPath(DBIdentifier.newTable(name)));
}
public Table findTable(QualifiedDBIdentifier path) {
if (DBIdentifier.isNull(path)) {
return null;
}
if (!DBIdentifier.isNull(path.getSchemaName())) {
Schema schema = getSchema(path.getSchemaName());
if (schema != null)
return schema.getTable(path.getUnqualifiedName());
} else {
Schema[] schemas = getSchemas();
Table tab;
for (Schema schema : schemas) {
tab = schema.getTable(path.getIdentifier());
if (tab != null)
return tab;
}
}
return null;
}
/**
* Find the table with the given name in the group, using '.' as the catalog
* separator. Returns null if no table found.
* @deprecated
*/
@Deprecated
public Table findTable(Schema inSchema, String name) {
if (name == null)
return null;
return findTable(inSchema, DBIdentifier.newTable(name), DBIdentifier.NULL);
}
public Table findTable(Schema inSchema, DBIdentifier name) {
if (DBIdentifier.isNull(name))
return null;
return findTable(inSchema, QualifiedDBIdentifier.getPath(name), DBIdentifier.NULL);
}
public Table findTable(Schema inSchema, DBIdentifier name, DBIdentifier defaultSchemaName) {
if (DBIdentifier.isNull(name))
return null;
return findTable(inSchema, QualifiedDBIdentifier.getPath(name), defaultSchemaName);
}
public Table findTable(Schema inSchema, QualifiedDBIdentifier path, DBIdentifier defaultSchemaName) {
if (path == null)
return null;
if (!DBIdentifier.isNull(path.getSchemaName())) {
Schema schema = getSchema(path.getSchemaName());
if (schema != null)
return schema.getTable(path.getIdentifier());
} else {
Schema[] schemas = getSchemas();
for (Schema schema : schemas) {
Table tab = schema.getTable(path.getIdentifier());
// if a table is found and it has the same schema
// as the input schema , it means that the table
// exists. However, if the input schema is null,
// then we assume that there is only one table for the
// db default schema, in this case, table exists..
// We can't handle the case that one entity has schema name
// and other entity does not have schema name but both entities
// map to the same table.
boolean isDefaultSchema = DBIdentifier.isNull(inSchema.getIdentifier()) &&
!DBIdentifier.isNull(defaultSchemaName) &&
DBIdentifier.equalsIgnoreCase(defaultSchemaName, schema.getIdentifier());
boolean hasNoDefaultSchema = DBIdentifier.isNull(inSchema.getIdentifier()) &&
DBIdentifier.isNull(defaultSchemaName);
if (tab != null &&
(schema == inSchema || isDefaultSchema || hasNoDefaultSchema))
return tab;
}
}
return null;
}
/**
* Return true if the given sequence is known to exist. While
* {@link #findSequence} may exhibit dynamic behavior in some schema group
* implementations, this method only returns true if the sequence has been
* added to this group or is known to exist in the database.
*/
public boolean isKnownSequence(Sequence seq) {
return findSequence(seq) != null;
}
/**
* Return true if the given sequence is known to exist. While
* {@link #findSequence} may exhibit dynamic behavior in some schema group
* implementations, this method only returns true if the sequence has been
* added to this group or is known to exist in the database.
* @deprecated
*/
@Deprecated
public boolean isKnownSequence(String name) {
return findSequence(name) != null;
}
public boolean isKnownSequence(DBIdentifier name) {
return findSequence(QualifiedDBIdentifier.getPath(name)) != null;
}
public boolean isKnownSequence(QualifiedDBIdentifier path) {
return findSequence(path) != null;
}
/**
* Find the equivalent of the given sequence in this schema group. The
* given sequence that may have come from another schema group.
*/
public Sequence findSequence(Sequence seq) {
return findSequence(seq.getQualifiedPath());
}
/**
* Find the sequence with the given name in the group, using '.' as the
* catalog separator. Returns null if no sequence found.
* @deprecated
*/
@Deprecated
public Sequence findSequence(String name) {
if (name == null)
return null;
return findSequence(DBIdentifier.newSequence(name));
}
public Sequence findSequence(DBIdentifier name) {
if (DBIdentifier.isNull(name))
return null;
return findSequence(QualifiedDBIdentifier.getPath(name));
}
public Sequence findSequence(QualifiedDBIdentifier path) {
if (path == null)
return null;
if (!DBIdentifier.isNull(path.getSchemaName())) {
Schema schema = getSchema(path.getSchemaName());
if (schema != null)
return schema.getSequence(path.getIdentifier());
} else {
Schema[] schemas = getSchemas();
Sequence seq;
for (Schema schema : schemas) {
seq = schema.getSequence(path.getIdentifier());
if (seq != null)
return seq;
}
}
return null;
}
/**
* Find the sequence with the given name in the group, using '.' as the
* catalog separator. Returns null if no sequence found.
* @deprecated
*/
@Deprecated
public Sequence findSequence(Schema inSchema, String name) {
if (name == null)
return null;
return findSequence(inSchema, QualifiedDBIdentifier.getPath(DBIdentifier.newSequence(name)));
}
public Sequence findSequence(Schema inSchema, QualifiedDBIdentifier path) {
if (path == null)
return null;
if (!DBIdentifier.isNull(path.getSchemaName())) {
Schema schema = getSchema(path.getSchemaName());
if (schema != null)
return schema.getSequence(path.getIdentifier());
} else {
Schema[] schemas = getSchemas();
Sequence seq;
for (Schema schema : schemas) {
seq = schema.getSequence(path.getIdentifier());
if ((seq != null) &&
(schema == inSchema || DBIdentifier.isNull(inSchema.getIdentifier())))
return seq;
}
}
return null;
}
/**
* Find all foreign keys exported by a given primary key (all foreign keys
* that link to the primary key).
*/
public ForeignKey[] findExportedForeignKeys(PrimaryKey pk) {
if (pk == null)
return new ForeignKey[0];
Schema[] schemas = getSchemas();
Table[] tabs;
ForeignKey[] fks;
Collection exports = new LinkedList<>();
for (Schema schema : schemas) {
tabs = schema.getTables();
for (Table tab : tabs) {
fks = tab.getForeignKeys();
for (ForeignKey fk : fks) {
if (fk.getPrimaryKeyTable() != null
&& pk.equals(fk.getPrimaryKeyTable().
getPrimaryKey()))
exports.add(fk);
}
}
}
return exports.toArray(new ForeignKey[exports.size()]);
}
/**
* Remove unreferenced or emtpy components from the schema.
*/
public void removeUnusedComponents() {
Schema[] schemas = getSchemas();
Table[] tabs;
Column[] cols;
Sequence[] seqs;
PrimaryKey pk;
ForeignKey[] fks;
for (Schema schema : schemas) {
seqs = schema.getSequences();
for (Sequence seq : seqs)
if (seq.getRefCount() == 0)
schema.removeSequence(seq);
tabs = schema.getTables();
for (Table tab : tabs) {
pk = tab.getPrimaryKey();
fks = tab.getForeignKeys();
cols = tab.getColumns();
if (pk != null && pk.getRefCount() == 0)
tab.removePrimaryKey();
for (ForeignKey fk : fks)
if (fk.getRefCount() == 0)
tab.removeForeignKey(fk);
for (Column col : cols)
if (col.getRefCount() == 0)
tab.removeColumn(col);
if (tab.getColumns().length == 0)
schema.removeTable(tab);
}
if (schema.getTables().length == 0)
removeSchema(schema);
}
}
@Override
public Object clone() {
SchemaGroup clone = newInstance();
clone.copy(this);
return clone;
}
/**
* Create a new instance of this class.
*/
protected SchemaGroup newInstance() {
return new SchemaGroup();
}
/**
* Copy cloneable state from the given instance.
*/
protected void copy(SchemaGroup group) {
Schema[] schemas = group.getSchemas();
for (Schema value : schemas) {
importSchema(value);
}
// have to do fks after all schemas are imported
Table[] tabs;
ForeignKey[] fks;
for (Schema schema : schemas) {
tabs = schema.getTables();
for (Table tab : tabs) {
fks = tab.getForeignKeys();
for (ForeignKey fk : fks)
getSchema(schema.getIdentifier()).getTable
(tab.getIdentifier()).importForeignKey(fk);
}
}
}
/**
* Return a new schema with the given name.
* @deprecated
*/
@Deprecated
protected Schema newSchema(String name) {
return new Schema(name, this);
}
protected Schema newSchema(DBIdentifier name) {
return new Schema(name, this);
}
/**
* Return a new sequence with the given name and owner schema.
* @deprecated
*/
@Deprecated
protected Sequence newSequence(String name, Schema schema) {
return new Sequence(name, schema);
}
protected Sequence newSequence(DBIdentifier name, Schema schema) {
return new Sequence(name, schema);
}
/**
* Return a new table with the given name and owner schema.
* @deprecated
*/
@Deprecated
protected Table newTable(String name, Schema schema) {
return new Table(name, schema);
}
protected Table newTable(DBIdentifier name, Schema schema) {
return new Table(name, schema);
}
/**
* Return a new column with the given name and owner table.
* @deprecated
*/
@Deprecated
protected Column newColumn(String name, Table table) {
return new Column(name, table);
}
protected Column newColumn(DBIdentifier name, Table table) {
return new Column(name, table);
}
/**
* Return a new primary key with the given name and owner table.
* @deprecated
*/
@Deprecated
protected PrimaryKey newPrimaryKey(String name, Table table) {
return new PrimaryKey(name, table);
}
protected PrimaryKey newPrimaryKey(DBIdentifier name, Table table) {
return new PrimaryKey(name, table);
}
/**
* Return a new index with the given name and owner table.
* @deprecated
*/
@Deprecated
protected Index newIndex(String name, Table table) {
return new Index(name, table);
}
protected Index newIndex(DBIdentifier name, Table table) {
return new Index(name, table);
}
/**
* Return a new unique constraint with the given name and owner table.
* @deprecated
*/
@Deprecated
protected Unique newUnique(String name, Table table) {
return new Unique(name, table);
}
protected Unique newUnique(DBIdentifier name, Table table) {
return new Unique(name, table);
}
/**
* Return a new foreign key with the given name and owner table.
* @deprecated
*/
@Deprecated
protected ForeignKey newForeignKey(String name, Table table) {
return new ForeignKey(name, table);
}
protected ForeignKey newForeignKey(DBIdentifier name, Table table) {
return new ForeignKey(name, table);
}
}