package liquibase.structure.core;
import liquibase.CatalogAndSchema;
import liquibase.structure.AbstractDatabaseObject;
import liquibase.structure.DatabaseObject;
import liquibase.util.StringUtils;
import java.util.*;
public class Schema extends AbstractDatabaseObject {
@Override
public DatabaseObject[] getContainingObjects() {
return null;
}
public Schema() {
setAttribute("objects", new HashMap<Class<? extends DatabaseObject>, Set<DatabaseObject>>());
}
public Schema(String catalog, String schemaName) {
this(new Catalog(catalog), schemaName);
}
public Schema(Catalog catalog, String schemaName) {
schemaName = StringUtils.trimToNull(schemaName);
setAttribute("name", schemaName);
setAttribute("catalog", catalog);
setAttribute("objects", new HashMap<Class<? extends DatabaseObject>, Set<DatabaseObject>>());
}
@Override
public String getName() {
return getAttribute("name", String.class);
}
@Override
public Schema setName(String name) {
setAttribute("name", name);
return this;
}
public boolean isDefault() {
return getAttribute("default", false);
}
public Schema setDefault(Boolean isDefault) {
setAttribute("default", isDefault);
return this;
}
@Override
public Schema getSchema() {
return this;
}
public Catalog getCatalog() {
return getAttribute("catalog", Catalog.class);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Schema schema = (Schema) o;
if (getCatalog() != null ? !getCatalog().equals(schema.getCatalog()) : schema.getCatalog() != null) return false;
if (getName() != null ? !getName().equals(schema.getName()) : schema.getName() != null) return false;
return true;
}
@Override
public int hashCode() {
int result = getCatalog() != null ? getCatalog().hashCode() : 0;
result = 31 * result + (getName() != null ? getName().hashCode() : 0);
return result;
}
public String getCatalogName() {
if (getCatalog() == null) {
return null;
}
return getCatalog().getName();
}
@Override
public String toString() {
String catalogName = getCatalogName();
String schemaName = getName();
if (schemaName == null) {
schemaName = "DEFAULT";
}
if (catalogName == null) {
return schemaName;
} else {
return catalogName +"."+ schemaName;
}
}
public CatalogAndSchema toCatalogAndSchema() {
String catalogName;
if (getCatalog() != null && getCatalog().isDefault()) {
catalogName = null;
} else {
catalogName = getCatalogName();
}
String schemaName;
if (isDefault()) {
schemaName = null;
} else {
schemaName = getName();
}
return new CatalogAndSchema(catalogName, schemaName);
}
protected Map<Class<? extends DatabaseObject>, Set<DatabaseObject>> getObjects() {
return getAttribute("objects", Map.class);
}
public <DatabaseObjectType extends DatabaseObject> List<DatabaseObjectType> getDatabaseObjects(Class<DatabaseObjectType> type) {
Set<DatabaseObjectType> databaseObjects = (Set<DatabaseObjectType>) getObjects().get(type);
if (databaseObjects == null) {
return new ArrayList<DatabaseObjectType>();
}
return new ArrayList<DatabaseObjectType>(databaseObjects);
}
public void addDatabaseObject(DatabaseObject databaseObject) {
if (databaseObject == null) {
return;
}
Set<DatabaseObject> objects = this.getObjects().get(databaseObject.getClass());
if (objects == null) {
objects = new HashSet<DatabaseObject>();
this.getObjects().put(databaseObject.getClass(), objects);
}
objects.add(databaseObject);
}
}