mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-10-08 14:51:49 +00:00
380d6a6f3a
Element text content was being trimmed, in sax parser's characters(...) method - but this method may be called multiple times with chunked data (due to buffering) and trim should not be applied until after the entire text has been parsed for that element. git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@119689 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
296 lines
10 KiB
Java
296 lines
10 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 StringBuilder lastText = new StringBuilder();
|
|
|
|
|
|
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);
|
|
}
|
|
|
|
// Deal with elements that contain text.
|
|
if (lastText.length() != 0)
|
|
{
|
|
if (qName.equals(XML.EL_TYPE))
|
|
{
|
|
Column column = (Column) stack.peek();
|
|
column.setType(lastText.toString());
|
|
}
|
|
else if (qName.equals(XML.EL_NULLABLE))
|
|
{
|
|
Column column = (Column) stack.peek();
|
|
column.setNullable(Boolean.parseBoolean(lastText.toString()));
|
|
}
|
|
else if (qName.equals(XML.EL_AUTOINCREMENT))
|
|
{
|
|
Column column = (Column) stack.peek();
|
|
column.setAutoIncrement(Boolean.parseBoolean(lastText.toString()));
|
|
}
|
|
else if (qName.equals(XML.EL_COLUMN_NAME))
|
|
{
|
|
if (stack.peek() instanceof PrimaryKey)
|
|
{
|
|
PrimaryKey pk = (PrimaryKey) stack.peek();
|
|
pk.getColumnNames().add(lastText.toString());
|
|
}
|
|
else if (stack.peek() instanceof Index)
|
|
{
|
|
Index index = (Index) stack.peek();
|
|
index.getColumnNames().add(lastText.toString());
|
|
}
|
|
}
|
|
else if (qName.equals(XML.EL_LOCAL_COLUMN))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setLocalColumn(lastText.toString());
|
|
}
|
|
else if (qName.equals(XML.EL_TARGET_TABLE))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setTargetTable(lastText.toString());
|
|
}
|
|
else if (qName.equals(XML.EL_TARGET_COLUMN))
|
|
{
|
|
ForeignKey fk = (ForeignKey) stack.peek();
|
|
fk.setTargetColumn(lastText.toString());
|
|
}
|
|
else if (qName.equals(XML.EL_PROPERTY))
|
|
{
|
|
String propValue = lastText.toString();
|
|
String propName = (String) stack.pop();
|
|
if (stack.peek() instanceof DbValidator)
|
|
{
|
|
@SuppressWarnings("unchecked")
|
|
DbValidator validator = (DbValidator) stack.peek();
|
|
validator.setProperty(propName, propValue);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
public void startElement(String uri, String localName, String qName, Attributes atts)
|
|
throws SAXException
|
|
{
|
|
lastText.setLength(0);
|
|
|
|
if (qName.equals(XML.EL_SCHEMA))
|
|
{
|
|
String name = atts.getValue(XML.ATTR_NAME);
|
|
String dbPrefix = atts.getValue(XML.ATTR_DB_PREFIX);
|
|
int version = Integer.parseInt(atts.getValue(XML.ATTR_VERSION));
|
|
String attrTableColumnOrder = atts.getValue(XML.ATTR_TABLE_COLUMN_ORDER);
|
|
// Should column order be checked for tables?
|
|
boolean compareTableColOrder = attrTableColumnOrder != null ? Boolean.parseBoolean(attrTableColumnOrder) : true;
|
|
schema = new Schema(name, dbPrefix, version, compareTableColOrder);
|
|
stack.push(schema);
|
|
}
|
|
else if (qName.equals(XML.EL_TABLE))
|
|
{
|
|
stack.push(new Table(atts.getValue(XML.ATTR_NAME)));
|
|
}
|
|
else if (qName.equals(XML.EL_COLUMN))
|
|
{
|
|
Column column = new Column(atts.getValue(XML.ATTR_NAME));
|
|
if (atts.getValue(XML.ATTR_ORDER) != null)
|
|
{
|
|
int order = Integer.parseInt(atts.getValue(XML.ATTR_ORDER));
|
|
column.setOrder(order);
|
|
}
|
|
column.setCompareOrder(schema.isCheckTableColumnOrder());
|
|
stack.push(column);
|
|
}
|
|
else if (qName.equals(XML.EL_COLUMN_NAME))
|
|
{
|
|
if (stack.peek() instanceof PrimaryKey && atts.getValue(XML.ATTR_ORDER) != null)
|
|
{
|
|
PrimaryKey pk = (PrimaryKey) stack.peek();
|
|
Integer columnOrder = Integer.parseInt(atts.getValue(XML.ATTR_ORDER));
|
|
pk.getColumnOrders().add(columnOrder);
|
|
}
|
|
}
|
|
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
|
|
{
|
|
String text = new String(ch, start, length);
|
|
if (text.length() > 0)
|
|
{
|
|
lastText.append(text);
|
|
}
|
|
}
|
|
}
|