Matt Ward cc28ee3413 ALF-11670: Add autoincrement boolean property to columns
Columns now have a property to describe whether it has been defined as an auto-increment column, e.g. AUTO_INCREMENT for MySQL. The property is persisted in XML reference files, loaded from XML reference files and compared between schemas.

The MySQL reference file has been updated to include autoincrement values.




git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@32711 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
2011-12-12 18:28:10 +00:00

288 lines
9.6 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))
{
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))
{
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);
}
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).trim();
if (text.length() > 0)
{
lastText.append(text);
}
}
}