/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.transformer.wire;
import org.mule.api.DefaultMuleException;
import org.mule.api.MuleContext;
import org.mule.api.MuleException;
import org.mule.api.transformer.Transformer;
import org.mule.api.transformer.TransformerException;
import org.mule.api.transformer.wire.WireFormat;
import org.mule.config.i18n.CoreMessages;
import org.mule.transformer.types.DataTypeFactory;
import org.mule.util.IOUtils;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import org.apache.commons.io.output.ByteArrayOutputStream;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* A pairing of an outbound transformer and an inbound transformer that can be used to serialize and deserialize data.
* THis is used when marshalling requests over the wire. IN Mule the MuleClient RemoteDispatcher uses wire formats to
* communicate with the server.
*/
public class TransformerPairWireFormat implements WireFormat
{
/**
* logger used by this class
*/
protected transient Log logger = LogFactory.getLog(getClass());
protected Transformer inboundTransformer;
protected Transformer outboundTransformer;
protected MuleContext muleContext;
public void setMuleContext(MuleContext context)
{
this.muleContext = context;
inboundTransformer.setMuleContext(muleContext);
outboundTransformer.setMuleContext(muleContext);
}
public Object read(InputStream in) throws MuleException
{
if (inboundTransformer == null)
{
throw new IllegalArgumentException(CoreMessages.objectIsNull("inboundTransformer").getMessage());
}
if (inboundTransformer.isSourceDataTypeSupported(DataTypeFactory.INPUT_STREAM))
{
return inboundTransformer.transform(in);
}
else
{
try
{
ByteArrayOutputStream baos = new ByteArrayOutputStream();
IOUtils.copy(in, baos);
return inboundTransformer.transform(baos.toByteArray());
}
catch (IOException e)
{
throw new DefaultMuleException(CoreMessages.failedToReadPayload(), e);
}
}
}
public void write(OutputStream out, Object o, String encoding) throws MuleException
{
if (outboundTransformer == null)
{
throw new IllegalArgumentException(CoreMessages.objectIsNull("outboundTransformer").getMessage());
}
try
{
Class returnClass = outboundTransformer.getReturnClass();
if (returnClass == null)
{
logger.warn("No return class was set on transformer: " + outboundTransformer
+ ". Attempting to work out how to treat the result transformation");
Object result = outboundTransformer.transform(o);
byte[] bytes;
if (result instanceof byte[])
{
bytes = (byte[]) result;
}
else
{
bytes = result.toString().getBytes(encoding);
}
out.write(bytes);
}
else if (returnClass.equals(byte[].class))
{
byte[] b = (byte[]) outboundTransformer.transform(o);
out.write(b);
}
else if (returnClass.equals(String.class))
{
String s = (String) outboundTransformer.transform(o);
out.write(s.getBytes(encoding));
}
else
{
throw new TransformerException(CoreMessages.transformFailedFrom(o.getClass()));
}
}
catch (IOException e)
{
throw new TransformerException(CoreMessages.transformFailedFrom(o.getClass()), e);
}
}
public Transformer getInboundTransformer()
{
return inboundTransformer;
}
public void setInboundTransformer(Transformer inboundTransformer)
{
this.inboundTransformer = inboundTransformer;
}
public Transformer getOutboundTransformer()
{
return outboundTransformer;
}
public void setOutboundTransformer(Transformer outboundTransformer)
{
this.outboundTransformer = outboundTransformer;
}
}