/*
* 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.simple;
import org.mule.api.MuleEvent;
import org.mule.api.transformer.DiscoverableTransformer;
import org.mule.api.transformer.TransformerException;
import org.mule.api.transport.OutputHandler;
import org.mule.config.i18n.MessageFactory;
import org.mule.transformer.AbstractTransformer;
import org.mule.transformer.types.DataTypeFactory;
import org.mule.util.IOUtils;
import org.mule.util.SerializationUtils;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.Serializable;
/** <code>ObjectToOutputHandler</code> converts a byte array into a String. */
public class ObjectToOutputHandler extends AbstractTransformer implements DiscoverableTransformer
{
/** Give core transformers a slighty higher priority */
private int priorityWeighting = DiscoverableTransformer.DEFAULT_PRIORITY_WEIGHTING + 1;
public ObjectToOutputHandler()
{
registerSourceType(DataTypeFactory.BYTE_ARRAY);
registerSourceType(DataTypeFactory.STRING);
registerSourceType(DataTypeFactory.INPUT_STREAM);
registerSourceType(DataTypeFactory.create(Serializable.class));
setReturnDataType(DataTypeFactory.create(OutputHandler.class));
}
@Override
public Object doTransform(final Object src, final String encoding) throws TransformerException
{
if (src instanceof String)
{
return new OutputHandler()
{
public void write(MuleEvent event, OutputStream out) throws IOException
{
out.write(((String) src).getBytes(encoding));
}
};
}
else if (src instanceof byte[])
{
return new OutputHandler()
{
public void write(MuleEvent event, OutputStream out) throws IOException
{
out.write((byte[]) src);
}
};
}
else if (src instanceof InputStream)
{
return new OutputHandler()
{
public void write(MuleEvent event, OutputStream out) throws IOException
{
InputStream is = (InputStream) src;
try
{
IOUtils.copyLarge(is, out);
}
finally
{
is.close();
}
}
};
}
else if (src instanceof Serializable)
{
return new OutputHandler()
{
public void write(MuleEvent event, OutputStream out) throws IOException
{
SerializationUtils.serialize((Serializable) src, out);
}
};
}
else
{
throw new TransformerException(MessageFactory
.createStaticMessage("Unable to convert " + src.getClass() + " to OutputHandler."));
}
}
public int getPriorityWeighting()
{
return priorityWeighting;
}
public void setPriorityWeighting(int priorityWeighting)
{
this.priorityWeighting = priorityWeighting;
}
}