/*
* JBoss, the OpenSource J2EE webOS Distributable under LGPL license. See terms of license at gnu.org.
*/
package org.jboss.resteasy.plugins.providers.jaxb;
import org.jboss.resteasy.core.ExceptionAdapter;
import org.jboss.resteasy.util.Types;
import javax.ws.rs.Consumes;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.Provider;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.transform.stream.StreamSource;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
/**
* <p>
* A JAXB Provider which handles parameter and return types of {@link JAXBElement}. This provider will be
* selected when the resource is declared as:
* </p>
* <code>
* @POST<br/>
* @Consumes("applictaion/xml")<br/>
* @Produces("applictaion/xml")<br/>
* public JAXBElement<Contact> getContact(JAXBElement<Contact> value);<br/>
* </code>
*
* @author <a href="ryan@damnhandy.com">Ryan J. McDonough</a>
* @version $Revision:$
*/
@Provider
@Produces("*/*")
@Consumes("*/*")
public class JAXBElementProvider extends AbstractJAXBProvider<JAXBElement<?>>
{
@Override
protected boolean isReadWritable(Class<?> type,
Type genericType,
Annotation[] annotations,
MediaType mediaType)
{
return JAXBElement.class.equals(type);
}
/**
*
*/
public JAXBElement<?> readFrom(Class<JAXBElement<?>> type,
Type genericType,
Annotation[] annotations,
MediaType mediaType,
MultivaluedMap<String, String> httpHeaders,
InputStream entityStream) throws IOException
{
Class<?> typeArg = Object.class;
if (genericType != null) typeArg = Types.getTypeArgument(genericType);
JAXBContext jaxb = null;
try
{
jaxb = findJAXBContext(typeArg, annotations, mediaType, true);
}
catch (JAXBException e)
{
throw new RuntimeException(e);
}
JAXBElement<?> result;
try
{
Unmarshaller unmarshaller = jaxb.createUnmarshaller();
JAXBElement<?> e = unmarshaller.unmarshal(new StreamSource(entityStream), (Class<?>) typeArg);
result = e;
}
catch (JAXBException e)
{
throw new ExceptionAdapter(e);
}
JAXBElement<?> element = result;
return element;
}
@Override
public void writeTo(JAXBElement<?> t,
Class<?> type,
Type genericType,
Annotation[] annotations,
MediaType mediaType,
MultivaluedMap<String, Object> httpHeaders,
OutputStream outputStream) throws IOException
{
Class<?> typeArg = Object.class;
if (genericType != null) typeArg = Types.getTypeArgument(genericType);
super.writeTo(t, typeArg, genericType, annotations, mediaType, httpHeaders, outputStream);
}
}