mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-06-30 18:15:39 +00:00
f5b08ab073
Reference schemas (e.g. classpath:org/alfresco/util/schemacomp/reference/PostgreSQLDialect-Reference.xml) are loaded (if present) and the database is compared to that reference schema. git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@31991 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
264 lines
8.5 KiB
Java
264 lines
8.5 KiB
Java
/*
|
|
* Copyright (C) 2005-2011 Alfresco Software Limited.
|
|
*
|
|
* This file is part of Alfresco
|
|
*
|
|
* Alfresco is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Alfresco is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package org.alfresco.util.schemacomp;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.util.Stack;
|
|
|
|
import javax.xml.parsers.SAXParser;
|
|
import javax.xml.parsers.SAXParserFactory;
|
|
|
|
import org.alfresco.util.schemacomp.model.Column;
|
|
import org.alfresco.util.schemacomp.model.DbObject;
|
|
import org.alfresco.util.schemacomp.model.ForeignKey;
|
|
import org.alfresco.util.schemacomp.model.Index;
|
|
import org.alfresco.util.schemacomp.model.PrimaryKey;
|
|
import org.alfresco.util.schemacomp.model.Schema;
|
|
import org.alfresco.util.schemacomp.model.Sequence;
|
|
import org.alfresco.util.schemacomp.model.Table;
|
|
import org.alfresco.util.schemacomp.validator.DbValidator;
|
|
import org.xml.sax.Attributes;
|
|
import org.xml.sax.SAXException;
|
|
import org.xml.sax.helpers.DefaultHandler;
|
|
|
|
/**
|
|
* Parse an XML document representing a database schema.
|
|
*
|
|
* @author Matt Ward
|
|
*/
|
|
public class XMLToSchema extends DefaultHandler
|
|
{
|
|
private SAXParser parser;
|
|
private InputStream in;
|
|
private Schema schema;
|
|
private Stack<Object> stack = new Stack<Object>();
|
|
private String lastTag;
|
|
private String lastText;
|
|
|
|
|
|
public XMLToSchema(InputStream in, SAXParserFactory saxParserFactory)
|
|
{
|
|
this.in = in;
|
|
try
|
|
{
|
|
parser = saxParserFactory.newSAXParser();
|
|
}
|
|
catch (Throwable e)
|
|
{
|
|
throw new RuntimeException("Cannot create parser.", e);
|
|
}
|
|
}
|
|
|
|
public XMLToSchema(InputStream in)
|
|
{
|
|
this(in, SAXParserFactory.newInstance());
|
|
}
|
|
|
|
public void parse()
|
|
{
|
|
try
|
|
{
|
|
parser.parse(in, this);
|
|
}
|
|
catch (SAXException error)
|
|
{
|
|
throw new RuntimeException("Unable to parse input stream.", error);
|
|
}
|
|
catch (IOException error)
|
|
{
|
|
throw new RuntimeException("Unable to parse input stream.", error);
|
|
}
|
|
}
|
|
|
|
public Schema getSchema()
|
|
{
|
|
return this.schema;
|
|
}
|
|
|
|
|
|
@Override
|
|
public void endElement(String uri, String localName, String qName) throws SAXException
|
|
{
|
|
if (qName.equals(XML.EL_TABLE))
|
|
{
|
|
Table table = (Table) stack.pop();
|
|
schema.add(table);
|
|
}
|
|
else if (qName.equals(XML.EL_COLUMN))
|
|
{
|
|
Column column = (Column) stack.pop();
|
|
Table table = (Table) stack.peek();
|
|
column.setParent(table);
|
|
table.getColumns().add(column);
|
|
}
|
|
else if (qName.equals(XML.EL_PRIMARY_KEY))
|
|
{
|
|
PrimaryKey pk = (PrimaryKey) stack.pop();
|
|
Table table = (Table) stack.peek();
|
|
table.setPrimaryKey(pk);
|
|
}
|
|
else if (qName.equals(XML.EL_FOREIGN_KEY))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.pop();
|
|
Table table = (Table) stack.peek();
|
|
fk.setParent(table);
|
|
table.getForeignKeys().add(fk);
|
|
}
|
|
else if (qName.equals(XML.EL_INDEX))
|
|
{
|
|
Index index = (Index) stack.pop();
|
|
Table table = (Table) stack.peek();
|
|
index.setParent(table);
|
|
table.getIndexes().add(index);
|
|
}
|
|
else if (qName.equals(XML.EL_SEQUENCE))
|
|
{
|
|
Sequence seq = (Sequence) stack.pop();
|
|
seq.setParent(schema);
|
|
schema.add(seq);
|
|
}
|
|
else if (qName.equals(XML.EL_VALIDATOR))
|
|
{
|
|
DbValidator validator = (DbValidator) stack.pop();
|
|
DbObject dbo = (DbObject) stack.peek();
|
|
dbo.getValidators().add(validator);
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
public void startElement(String uri, String localName, String qName, Attributes atts)
|
|
throws SAXException
|
|
{
|
|
lastTag = qName;
|
|
|
|
if (qName.equals(XML.EL_SCHEMA))
|
|
{
|
|
schema = new Schema(atts.getValue(XML.ATTR_NAME));
|
|
}
|
|
else if (qName.equals(XML.EL_TABLE))
|
|
{
|
|
stack.push(new Table(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_COLUMN))
|
|
{
|
|
stack.push(new Column(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_PRIMARY_KEY))
|
|
{
|
|
stack.push(new PrimaryKey(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_FOREIGN_KEY))
|
|
{
|
|
stack.push(new ForeignKey(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_INDEX))
|
|
{
|
|
Index index = new Index(atts.getValue(XML.ATTR_NAME));
|
|
boolean unique = Boolean.parseBoolean(atts.getValue(XML.ATTR_UNIQUE));
|
|
index.setUnique(unique);
|
|
stack.push(index);
|
|
}
|
|
else if (qName.equals(XML.EL_SEQUENCE))
|
|
{
|
|
stack.push(new Sequence(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_VALIDATOR))
|
|
{
|
|
String className = atts.getValue(XML.ATTR_CLASS);
|
|
DbValidator validator = null;
|
|
try
|
|
{
|
|
validator = (DbValidator) Class.forName(className).newInstance();
|
|
}
|
|
catch (Throwable e)
|
|
{
|
|
throw new RuntimeException("Couldn't create validator, class: " + className, e);
|
|
}
|
|
|
|
stack.push(validator);
|
|
}
|
|
else if (qName.equals(XML.EL_PROPERTY))
|
|
{
|
|
String name = atts.getValue(XML.ATTR_NAME);
|
|
stack.push(name);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void characters(char[] ch, int start, int length) throws SAXException
|
|
{
|
|
lastText = new String(ch, start, length).trim();
|
|
|
|
if (lastText.length() != 0)
|
|
{
|
|
if (lastTag.equals(XML.EL_TYPE))
|
|
{
|
|
Column column = (Column) stack.peek();
|
|
column.setType(lastText);
|
|
}
|
|
else if (lastTag.equals(XML.EL_NULLABLE))
|
|
{
|
|
Column column = (Column) stack.peek();
|
|
column.setNullable(Boolean.parseBoolean(lastText));
|
|
}
|
|
else if (lastTag.equals(XML.EL_COLUMN_NAME))
|
|
{
|
|
if (stack.peek() instanceof PrimaryKey)
|
|
{
|
|
PrimaryKey pk = (PrimaryKey) stack.peek();
|
|
pk.getColumnNames().add(lastText);
|
|
}
|
|
else if (stack.peek() instanceof Index)
|
|
{
|
|
Index index = (Index) stack.peek();
|
|
index.getColumnNames().add(lastText);
|
|
}
|
|
}
|
|
else if (lastTag.equals(XML.EL_LOCAL_COLUMN))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setLocalColumn(lastText);
|
|
}
|
|
else if (lastTag.equals(XML.EL_TARGET_TABLE))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setTargetTable(lastText);
|
|
}
|
|
else if (lastTag.equals(XML.EL_TARGET_COLUMN))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setTargetColumn(lastText);
|
|
}
|
|
else if (lastTag.equals(XML.EL_PROPERTY))
|
|
{
|
|
String propValue = lastText;
|
|
String propName = (String) stack.pop();
|
|
if (stack.peek() instanceof DbValidator)
|
|
{
|
|
@SuppressWarnings("unchecked")
|
|
DbValidator validator = (DbValidator) stack.peek();
|
|
validator.setProperty(propName, propValue);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|