/*
* JBoss, Home of Professional Open Source.
* See the COPYRIGHT.txt file distributed with this work for information
* regarding copyright ownership. Some portions may be licensed
* to Red Hat, Inc. under one or more contributor license agreements.
*
* This library 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 2.1 of the License, or (at your option) any later version.
*
* This library 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 this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA.
*/
package org.teiid.core.types.basic;
import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.events.XMLEvent;
import org.teiid.core.types.DataTypeManagerService;
import org.teiid.core.types.SQLXMLImpl;
import org.teiid.core.types.Transform;
import org.teiid.core.types.XMLType;
import org.teiid.core.types.XMLType.Type;
import org.teiid.runtime.client.Messages;
import org.teiid.runtime.client.TeiidClientException;
public class StringToSQLXMLTransform extends Transform {
/**
* @param dataTypeManager
*/
public StringToSQLXMLTransform(DataTypeManagerService dataTypeManager) {
super(dataTypeManager);
}
/**
* This method transforms a value of the source type into a value
* of the target type.
* @param value Incoming value of source type
* @return Outgoing value of target type
* @throws Exception if value is an incorrect input type or
* the transformation fails
*/
public Object transformDirect(Object value) throws Exception {
String xml = (String)value;
Reader reader = new StringReader(xml);
Type type = isXml(reader);
XMLType result = new XMLType(new SQLXMLImpl(xml));
result.setType(type);
return result;
}
public static Type isXml(Reader reader) throws Exception {
Type type = Type.ELEMENT;
XMLInputFactory inputFactory = XMLType.getXmlInputFactory();
try{
XMLStreamReader xmlReader = inputFactory.createXMLStreamReader(reader);
int event = xmlReader.getEventType();
if (event == XMLEvent.START_DOCUMENT && xmlReader.getLocation().getColumnNumber() != 1) {
type = Type.DOCUMENT;
}
while (xmlReader.hasNext()) {
xmlReader.next();
}
} catch (Exception e){
throw new TeiidClientException(e, Messages.gs(Messages.TEIID.TEIID10070));
} finally {
try {
reader.close();
} catch (IOException e) {
}
}
return type;
}
/**
* Type of the incoming value.
* @return Source type
*/
public Class<?> getSourceType() {
return DataTypeManagerService.DefaultDataTypes.STRING.getTypeClass();
}
/**
* Type of the outgoing value.
* @return Target type
*/
public Class<?> getTargetType() {
return DataTypeManagerService.DefaultDataTypes.XML.getTypeClass();
}
@Override
public boolean isExplicit() {
return true;
}
}