org.hsqldb.SchemaObjectSet Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of hsqldb Show documentation
Show all versions of hsqldb Show documentation
HSQLDB - Lightweight 100% Java SQL Database Engine
/* Copyright (c) 2001-2021, The HSQL Development Group
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of the HSQL Development Group nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG,
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.hsqldb;
import org.hsqldb.HsqlNameManager.HsqlName;
import org.hsqldb.error.Error;
import org.hsqldb.error.ErrorCode;
import org.hsqldb.lib.HashMap;
import org.hsqldb.lib.HsqlArrayList;
import org.hsqldb.lib.Iterator;
import org.hsqldb.lib.OrderedHashMap;
import org.hsqldb.lib.OrderedHashSet;
/**
* Collection of SQL schema objects of a specific type in a schema
*
* @author Fred Toussi (fredt@users dot sourceforge.net)
* @version 2.5.1
* @since 1.9.0
*/
public class SchemaObjectSet {
HashMap map;
int type;
SchemaObjectSet(int type) {
this.type = type;
switch (type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.FUNCTION :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE :
map = new OrderedHashMap();
break;
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX :
map = new HashMap();
break;
default :
throw Error.runtimeError(ErrorCode.U_S0500, "SchemaObjectSet");
}
}
HsqlName getName(String name) {
switch (type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.FUNCTION :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE :
SchemaObject object = ((SchemaObject) map.get(name));
return object == null ? null
: object.getName();
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX : {
return (HsqlName) map.get(name);
}
default :
return (HsqlName) map.get(name);
}
}
public SchemaObject getObject(String name) {
switch (type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.FUNCTION :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE :
return (SchemaObject) map.get(name);
default :
throw Error.runtimeError(ErrorCode.U_S0500, "SchemaObjectSet");
}
}
public boolean contains(String name) {
return map.containsKey(name);
}
void checkAdd(HsqlName name) {
if (map.containsKey(name.name)) {
int code = getAddErrorCode(name.type);
throw Error.error(code, name.name);
}
}
boolean isEmpty() {
return map.isEmpty();
}
void checkExists(String name) {
if (!map.containsKey(name)) {
int code = getGetErrorCode(type);
throw Error.error(code, name);
}
}
public void add(SchemaObject object, boolean replace) {
HsqlName name = object.getName();
if (type == SchemaObject.SPECIFIC_ROUTINE) {
name = ((Routine) object).getSpecificName();
}
if (!replace && map.containsKey(name.name)) {
int code = getAddErrorCode(name.type);
throw Error.error(code, name.name);
}
Object value = object;
switch (name.type) {
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX :
value = name;
break;
default :
}
map.put(name.name, value);
}
void remove(String name) {
map.remove(name);
}
void removeParent(HsqlName parent) {
Iterator it = map.values().iterator();
while (it.hasNext()) {
if (type == SchemaObject.TRIGGER
|| type == SchemaObject.SPECIFIC_ROUTINE) {
SchemaObject object = (SchemaObject) it.next();
if (object.getName().parent == parent) {
it.remove();
}
} else {
HsqlName name = (HsqlName) it.next();
if (name.parent == parent) {
it.remove();
}
}
}
}
void rename(HsqlName name, HsqlName newName) {
if (map.containsKey(newName.name)) {
int code = getAddErrorCode(name.type);
throw Error.error(code, newName.name);
}
switch (newName.type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.FUNCTION :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.ROUTINE :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE : {
int i = ((OrderedHashMap) map).getIndex(name.name);
if (i == -1) {
int code = getGetErrorCode(name.type);
throw Error.error(code, name.name);
}
SchemaObject object =
(SchemaObject) ((OrderedHashMap) map).get(i);
object.getName().rename(newName);
((OrderedHashMap) map).setKeyAt(i, newName.name);
break;
}
case SchemaObject.SPECIFIC_ROUTINE : {
int i = ((OrderedHashMap) map).getIndex(name.name);
if (i == -1) {
int code = getGetErrorCode(name.type);
throw Error.error(code, name.name);
}
Routine routine = (Routine) ((OrderedHashMap) map).get(i);
routine.getSpecificName().rename(newName);
((OrderedHashMap) map).setKeyAt(i, newName.name);
break;
}
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX : {
map.remove(name.name);
name.rename(newName);
map.put(name.name, name);
break;
}
default :
}
}
static int getAddErrorCode(int type) {
int code;
switch (type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.FUNCTION :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE :
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX :
code = ErrorCode.X_42504;
break;
default :
throw Error.runtimeError(ErrorCode.U_S0500, "SchemaObjectSet");
}
return code;
}
static int getGetErrorCode(int type) {
int code;
switch (type) {
case SchemaObject.VIEW :
case SchemaObject.TABLE :
case SchemaObject.SEQUENCE :
case SchemaObject.CHARSET :
case SchemaObject.DOMAIN :
case SchemaObject.TYPE :
case SchemaObject.COLLATION :
case SchemaObject.PROCEDURE :
case SchemaObject.FUNCTION :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.ASSERTION :
case SchemaObject.TRIGGER :
case SchemaObject.REFERENCE :
case SchemaObject.EXCEPTION :
case SchemaObject.MODULE :
case SchemaObject.CONSTRAINT :
case SchemaObject.INDEX :
code = ErrorCode.X_42501;
break;
default :
throw Error.runtimeError(ErrorCode.U_S0500, "SchemaObjectSet");
}
return code;
}
public static String getName(int type) {
switch (type) {
case SchemaObject.VIEW :
return Tokens.T_VIEW;
case SchemaObject.TABLE :
return Tokens.T_TABLE;
case SchemaObject.SEQUENCE :
return Tokens.T_SEQUENCE;
case SchemaObject.CHARSET :
return Tokens.T_CHARACTER + ' ' + Tokens.T_SET;
case SchemaObject.DOMAIN :
return Tokens.T_DOMAIN;
case SchemaObject.TYPE :
return Tokens.T_TYPE;
case SchemaObject.COLLATION :
return Tokens.T_COLLATION;
case SchemaObject.PROCEDURE :
return Tokens.T_PROCEDURE;
case SchemaObject.FUNCTION :
return Tokens.T_FUNCTION;
case SchemaObject.SPECIFIC_ROUTINE :
return Tokens.T_SPECIFIC + ' ' + Tokens.T_ROUTINE;
case SchemaObject.ASSERTION :
return Tokens.T_ASSERTION;
case SchemaObject.TRIGGER :
return Tokens.T_TRIGGER;
case SchemaObject.EXCEPTION :
return Tokens.T_EXCEPTION;
case SchemaObject.MODULE :
return Tokens.T_MODULE;
case SchemaObject.REFERENCE :
return Tokens.T_SYNONYM;
case SchemaObject.PARAMETER :
return Tokens.T_PARAMETER;
case SchemaObject.CONSTRAINT :
return Tokens.T_CONSTRAINT;
case SchemaObject.INDEX :
return Tokens.T_INDEX;
case SchemaObject.PERIOD :
return Tokens.T_PERIOD;
default :
throw Error.runtimeError(ErrorCode.U_S0500, "SchemaObjectSet");
}
}
void getSQL(HsqlArrayList list, OrderedHashSet resolved,
OrderedHashSet unresolved) {
// HashMap lists are not persisted with this method
if (!(map instanceof OrderedHashMap)) {
return;
}
if (map.isEmpty()) {
return;
}
Iterator it = map.values().iterator();
if (type == SchemaObject.FUNCTION || type == SchemaObject.PROCEDURE) {
OrderedHashSet set = new OrderedHashSet();
while (it.hasNext()) {
RoutineSchema routineSchema = (RoutineSchema) it.next();
for (int i = 0; i < routineSchema.routines.length; i++) {
Routine routine = routineSchema.routines[i];
set.add(routine);
}
}
it = set.iterator();
}
addAllSQL(resolved, unresolved, list, it, null);
}
static void addAllSQL(OrderedHashSet resolved, OrderedHashSet unresolved,
HsqlArrayList list, Iterator it,
OrderedHashSet newResolved) {
while (it.hasNext()) {
SchemaObject object = (SchemaObject) it.next();
boolean isResolved = true;
OrderedHashSet references;
if (object.getType() == SchemaObject.TABLE) {
((Table) object).setForwardConstraints(resolved);
references = ((Table) object).getReferencesForScript();
} else {
references = object.getReferences();
}
for (int j = 0; j < references.size(); j++) {
HsqlName name = (HsqlName) references.get(j);
if (SqlInvariants.isSchemaNameSystem(name)) {
continue;
}
switch (name.type) {
case SchemaObject.TABLE :
if (!resolved.contains(name)) {
isResolved = false;
}
break;
case SchemaObject.COLUMN : {
if (object.getType() == SchemaObject.TABLE) {
int index = ((Table) object).findColumn(name.name);
ColumnSchema column =
((Table) object).getColumn(index);
if (!isChildObjectResolved(column, resolved)) {
isResolved = false;
}
break;
}
if (!resolved.contains(name.parent)) {
isResolved = false;
}
break;
}
case SchemaObject.CONSTRAINT : {
if (name.parent == object.getName()) {
Constraint constraint =
((Table) object).getConstraint(name.name);
if (constraint.getConstraintType()
== SchemaObject.ConstraintTypes.CHECK) {
if (!isChildObjectResolved(constraint,
resolved)) {
isResolved = false;
}
}
}
// only UNIQUE constraint referenced by FK in table
break;
}
case SchemaObject.CHARSET :
if (name.schema == null) {
continue;
}
// fall through
case SchemaObject.TYPE :
case SchemaObject.DOMAIN :
case SchemaObject.COLLATION :
case SchemaObject.FUNCTION :
case SchemaObject.PROCEDURE :
case SchemaObject.SPECIFIC_ROUTINE :
case SchemaObject.SEQUENCE :
if (!resolved.contains(name)) {
isResolved = false;
}
break;
default :
}
}
if (!isResolved) {
unresolved.add(object);
continue;
}
HsqlName name;
if (object.getType() == SchemaObject.FUNCTION
|| object.getType() == SchemaObject.PROCEDURE) {
name = ((Routine) object).getSpecificName();
} else {
name = object.getName();
}
resolved.add(name);
if (newResolved != null) {
newResolved.add(object);
}
switch (object.getType()) {
case SchemaObject.TABLE : {
list.addAll(((Table) object).getSQL(resolved, unresolved));
String comment =
object.getName().getCommentSQL(Tokens.T_TABLE);
if (comment != null) {
list.add(comment);
}
for (int j = 0; j < ((Table) object).getColumnCount();
j++) {
ColumnSchema column = ((Table) object).getColumn(j);
comment =
column.getName().getCommentSQL(Tokens.T_COLUMN);
if (comment != null) {
list.add(comment);
}
}
break;
}
case SchemaObject.FUNCTION :
case SchemaObject.PROCEDURE : {
Routine routine = ((Routine) object);
if (routine.isRecursive) {
list.add(((Routine) object).getSQLDeclaration());
list.add(((Routine) object).getSQLAlter());
} else {
list.add(object.getSQL());
}
String comment =
object.getName().getCommentSQL(Tokens.T_ROUTINE);
if (comment != null) {
list.add(comment);
}
break;
}
case SchemaObject.TRIGGER : {
TriggerDef trigger = ((TriggerDef) object);
list.add(object.getSQL());
String comment =
object.getName().getCommentSQL(Tokens.T_TRIGGER);
if (comment != null) {
list.add(comment);
}
break;
}
case SchemaObject.SEQUENCE : {
list.add(object.getSQL());
String comment =
object.getName().getCommentSQL(Tokens.T_SEQUENCE);
if (comment != null) {
list.add(comment);
}
break;
}
case SchemaObject.CONSTRAINT : {
list.add(object.getSQL());
String added = ((Constraint) object).getAlterSQL();
if (!added.isEmpty()) {
list.add(added);
}
break;
}
default :
list.add(object.getSQL());
}
}
}
static boolean isChildObjectResolved(SchemaObject object,
OrderedHashSet resolved) {
OrderedHashSet refs = object.getReferences();
for (int i = 0; i < refs.size(); i++) {
HsqlName name = (HsqlName) refs.get(i);
if (SqlInvariants.isSchemaNameSystem(name)) {
continue;
}
if (!resolved.contains(name)) {
return false;
}
}
return true;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy