/*
* $Id: FileToByteArray.java 19250 2010-08-30 16:53:14Z dirk.olmes $
* --------------------------------------------------------------------------------------
* 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.transport.file.transformers;
import org.mule.api.transformer.TransformerException;
import org.mule.transformer.simple.ObjectToByteArray;
import org.mule.transformer.types.DataTypeFactory;
import org.mule.util.ArrayUtils;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import org.apache.commons.io.IOUtils;
/**
* <code>FileToByteArray</code> reads the contents of a file as a byte array.
*/
public class FileToByteArray extends ObjectToByteArray
{
public FileToByteArray()
{
super();
registerSourceType(DataTypeFactory.create(File.class));
registerSourceType(DataTypeFactory.BYTE_ARRAY);
}
@Override
public Object doTransform(Object src, String outputEncoding) throws TransformerException
{
// Support other payload types so that this transformer can be used
// transparently both when streaming is on and off
if (src instanceof byte[])
{
return src;
}
if (src instanceof InputStream || src instanceof String)
{
return super.doTransform(src, outputEncoding);
}
else
{
File file = (File) src;
if (file == null)
{
throw new TransformerException(this, new IllegalArgumentException("null file"));
}
if (!file.exists())
{
throw new TransformerException(this, new FileNotFoundException(file.getPath()));
}
if (file.length() == 0)
{
logger.warn("File is empty: " + file.getAbsolutePath());
return ArrayUtils.EMPTY_BYTE_ARRAY;
}
FileInputStream fis = null;
byte[] bytes = null;
try
{
fis = new FileInputStream(file);
// TODO Attention: arbitrary 4GB limit & also a great way to reap
// OOMs
int length = new Long(file.length()).intValue();
bytes = new byte[length];
fis.read(bytes);
return bytes;
}
// at least try..
catch (OutOfMemoryError oom)
{
throw new TransformerException(this, oom);
}
catch (IOException e)
{
throw new TransformerException(this, e);
}
finally
{
IOUtils.closeQuietly(fis);
}
}
}
}