From a40e527938d76ba71f211da7e327adb50384ba69 Mon Sep 17 00:00:00 2001 From: lresende Date: Wed, 11 Nov 2009 23:26:33 +0000 Subject: Moving 1.x tags git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@835157 13f79535-47bb-0310-9956-ffa450edef68 --- .../sca/databinding/jaxb/axiom/JAXBDataSource.java | 207 +++++++++++++++++++++ 1 file changed, 207 insertions(+) create mode 100644 sca-java-1.x/tags/1.5-RC0/modules/databinding-jaxb-axiom/src/main/java/org/apache/tuscany/sca/databinding/jaxb/axiom/JAXBDataSource.java (limited to 'sca-java-1.x/tags/1.5-RC0/modules/databinding-jaxb-axiom/src/main/java/org/apache/tuscany/sca/databinding/jaxb/axiom/JAXBDataSource.java') diff --git a/sca-java-1.x/tags/1.5-RC0/modules/databinding-jaxb-axiom/src/main/java/org/apache/tuscany/sca/databinding/jaxb/axiom/JAXBDataSource.java b/sca-java-1.x/tags/1.5-RC0/modules/databinding-jaxb-axiom/src/main/java/org/apache/tuscany/sca/databinding/jaxb/axiom/JAXBDataSource.java new file mode 100644 index 0000000000..d5cd68cd70 --- /dev/null +++ b/sca-java-1.x/tags/1.5-RC0/modules/databinding-jaxb-axiom/src/main/java/org/apache/tuscany/sca/databinding/jaxb/axiom/JAXBDataSource.java @@ -0,0 +1,207 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +package org.apache.tuscany.sca.databinding.jaxb.axiom; + +import java.io.OutputStream; +import java.io.StringReader; +import java.io.StringWriter; +import java.io.UnsupportedEncodingException; +import java.io.Writer; +import java.lang.reflect.Method; +import java.security.AccessController; +import java.security.PrivilegedActionException; +import java.security.PrivilegedExceptionAction; + +import javax.xml.bind.JAXBContext; +import javax.xml.bind.JAXBException; +import javax.xml.bind.Marshaller; +import javax.xml.stream.XMLStreamException; +import javax.xml.stream.XMLStreamReader; +import javax.xml.stream.XMLStreamWriter; + +import org.apache.axiom.om.OMDataSourceExt; +import org.apache.axiom.om.OMOutputFormat; +import org.apache.axiom.om.ds.OMDataSourceExtBase; +import org.apache.axiom.om.impl.MTOMXMLStreamWriter; +import org.apache.axiom.om.util.StAXUtils; +import org.apache.tuscany.sca.databinding.jaxb.JAXBContextHelper; + +/** + * + * @version $Rev$ $Date$ + */ +public class JAXBDataSource extends OMDataSourceExtBase { + private JAXBContext context; + private Object element; + private Marshaller marshaller; + + public JAXBDataSource(Object element, JAXBContext context) { + this.element = element; + this.context = context; + } + + private Marshaller getMarshaller() throws JAXBException { + if (marshaller == null) { + // For thread safety, not sure we can cache the marshaller + marshaller = JAXBContextHelper.getMarshaller(context); + } + return marshaller; + } + + private void releaseMarshaller(Marshaller marshaller) { + JAXBContextHelper.releaseJAXBMarshaller(context, marshaller); + } + + public XMLStreamReader getReader() throws XMLStreamException { + // FIXME: [rfeng] This is a quick and dirty implementation + // We could use the fastinfoset to optimize the roundtrip + StringWriter writer = new StringWriter(); + serialize(writer, new OMOutputFormat()); + StringReader reader = new StringReader(writer.toString()); + return StAXUtils.createXMLStreamReader(reader); + } + + /** + * If the writer is backed by an OutputStream, then return the OutputStream + * @param writer + * @return OutputStream or null + */ + private static OutputStream getOutputStream(XMLStreamWriter writer) throws XMLStreamException { + if (writer.getClass() == MTOMXMLStreamWriter.class) { + return ((MTOMXMLStreamWriter)writer).getOutputStream(); + } + + try { + Method method = writer.getClass().getMethod("getOutputStream"); + return (OutputStream)method.invoke(writer); + + } catch (NoSuchMethodException e) { + return null; + } catch (Exception e) { + return null; + } + } + + public void serialize(final XMLStreamWriter xmlWriter) throws XMLStreamException { + try { + // marshaller.setProperty(Marshaller.JAXB_ENCODING, format.getCharSetEncoding()); + AccessController.doPrivileged(new PrivilegedExceptionAction() { + public Object run() throws Exception { + try { + Marshaller marshaller = getMarshaller(); + // Marshalling directly to the output stream is faster than marshalling through the + // XMLStreamWriter. Take advantage of this optimization if there is an output stream. + OutputStream os = getOutputStream(xmlWriter); + if (os != null) { + marshaller.marshal(element, os); + } else { + marshaller.marshal(element, xmlWriter); + } + } finally { + releaseMarshaller(marshaller); + } + return null; + } + }); + } catch (PrivilegedActionException e) { + throw new XMLStreamException(e.getException()); + } + } + + public void serialize(final OutputStream output, OMOutputFormat format) throws XMLStreamException { + try { + // marshaller.setProperty(Marshaller.JAXB_ENCODING, format.getCharSetEncoding()); + AccessController.doPrivileged(new PrivilegedExceptionAction() { + public Object run() throws Exception { + try { + Marshaller marshaller = getMarshaller(); + marshaller.marshal(element, output); + } finally { + releaseMarshaller(marshaller); + } + return null; + } + }); + } catch (PrivilegedActionException e) { + throw new XMLStreamException(e.getException()); + } + } + + public void serialize(final Writer writer, OMOutputFormat format) throws XMLStreamException { + try { + AccessController.doPrivileged(new PrivilegedExceptionAction() { + public Object run() throws Exception { + try { + Marshaller marshaller = getMarshaller(); + marshaller.marshal(element, writer); + } finally { + releaseMarshaller(marshaller); + } + return null; + } + }); + } catch (PrivilegedActionException e) { + throw new XMLStreamException(e.getException()); + } + } + + public Object getObject() { + return element; + } + + public void close() { + } + + public OMDataSourceExt copy() { + return new JAXBDataSource(element, context); + } + + public byte[] getXMLBytes(final String encoding) throws UnsupportedEncodingException { + try { + return AccessController.doPrivileged(new PrivilegedExceptionAction() { + public byte[] run() throws JAXBException, XMLStreamException, UnsupportedEncodingException { + try { + StringWriter sw = new StringWriter(); + Marshaller marshaller = getMarshaller(); + marshaller.marshal(element, sw); + return sw.toString().getBytes(encoding); + } finally { + releaseMarshaller(marshaller); + } + } + }); + } catch (PrivilegedActionException e) { + Throwable t = e.getCause(); + if (t instanceof UnsupportedEncodingException) { + throw (UnsupportedEncodingException)t; + } else { + throw new RuntimeException(t); + } + } + } + + public boolean isDestructiveRead() { + return false; + } + + public boolean isDestructiveWrite() { + return false; + } + +} -- cgit v1.2.3