/*
* Copyright (C) 2012 eXo Platform SAS.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.exoplatform.services.jcr.impl.dataflow;
import org.exoplatform.commons.utils.PrivilegedFileHelper;
import org.exoplatform.services.jcr.access.AccessControlEntry;
import org.exoplatform.services.jcr.datamodel.IllegalNameException;
import org.exoplatform.services.jcr.datamodel.IllegalPathException;
import org.exoplatform.services.jcr.datamodel.InternalQName;
import org.exoplatform.services.jcr.datamodel.QPath;
import org.exoplatform.services.jcr.datamodel.ValueData;
import org.exoplatform.services.jcr.impl.Constants;
import org.exoplatform.services.jcr.impl.dataflow.persistent.ByteArrayPersistedValueData;
import org.exoplatform.services.jcr.impl.dataflow.persistent.PersistedValueData;
import org.exoplatform.services.jcr.impl.dataflow.persistent.StreamPersistedValueData;
import org.exoplatform.services.jcr.impl.util.JCRDateFormat;
import org.exoplatform.services.jcr.impl.util.io.SpoolFile;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.nio.ByteBuffer;
import java.nio.MappedByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.WritableByteChannel;
import java.util.Arrays;
import java.util.Calendar;
import javax.jcr.ValueFormatException;
/**
* @author <a href="abazko@exoplatform.com">Anatoliy Bazko</a>
* @version $Id: StreamValueData.java 34360 2009-07-22 23:58:59Z tolusha $
*/
public abstract class StreamValueData extends AbstractValueData
{
protected InputStream stream;
protected SpoolFile spoolFile;
protected final SpoolConfig spoolConfig;
protected byte[] data;
protected FileChannel channel;
/**
* StreamValueData constructor.
*/
protected StreamValueData(int orderNumber, InputStream stream, SpoolFile spoolFile, SpoolConfig spoolConfig)
throws IOException
{
super(orderNumber);
this.stream = stream;
this.spoolFile = spoolFile;
this.spoolConfig = spoolConfig;
if (spoolFile != null)
{
spoolFile.acquire(this);
if (this.stream != null)
{
this.stream.close();
this.stream = null;
}
}
}
/**
* {@inheritDoc}
*/
@Override
public byte[] getAsByteArray() throws IOException
{
if (isByteArrayAfterSpool())
{
return data;
}
else
{
return fileToByteArray(spoolFile);
}
}
/**
* {@inheritDoc}
*/
@Override
public InputStream getAsStream() throws IOException
{
if (isByteArrayAfterSpool())
{
return new ByteArrayInputStream(data); // from bytes
}
else
{
if (spoolFile != null)
{
return PrivilegedFileHelper.fileInputStream(spoolFile); // from spool file
}
else
{
throw new IllegalArgumentException("Stream already consumed");
}
}
}
/**
* {@inheritDoc}
*/
@Override
public long getLength()
{
if (isByteArrayAfterSpool())
{
return data.length;
}
else
{
return PrivilegedFileHelper.length(spoolFile);
}
}
/**
* {@inheritDoc}
*/
@Override
public boolean isByteArray()
{
return data != null;
}
/**
* {@inheritDoc}
*/
@Override
public long read(OutputStream stream, long length, long position) throws IOException
{
if (isByteArrayAfterSpool())
{
return readFromByteArray(stream, length, position);
}
else
{
return readFromFile(stream, spoolFile, length, position);
}
}
/**
* Read <code>length</code> bytes from the binary value at <code>position</code> to the
* <code>stream</code>.
*/
protected long readFromFile(OutputStream stream, File file, long length, long position) throws IOException
{
FileInputStream in = null;
try
{
if (channel == null || !channel.isOpen())
{
in = PrivilegedFileHelper.fileInputStream(file);
channel = in.getChannel();
}
length = validateAndAdjustLenght(length, position, channel.size());
MappedByteBuffer bb = channel.map(FileChannel.MapMode.READ_ONLY, position, length);
WritableByteChannel ch = Channels.newChannel(stream);
ch.write(bb);
ch.close();
return length;
}
finally
{
if (in != null)
{
in.close();
if (channel != null)
{
channel.close();
}
}
}
}
/**
* {@inheritDoc}
*/
@Override
protected void finalize() throws Throwable
{
if (channel != null)
{
channel.close();
}
removeSpoolFile();
super.finalize();
}
/**
* Tell is this Value backed by bytes array before or after spooling.
*/
private boolean isByteArrayAfterSpool()
{
if (data != null)
{
return true;
}
else
{
spoolInputStream();
return data != null;
}
}
/**
* Spool ValueData temp InputStream to a temp File.
*/
protected void spoolInputStream()
{
if (spoolFile != null || data != null) // already spooled
{
return;
}
byte[] buffer = new byte[0];
byte[] tmpBuff = new byte[2048];
int read = 0;
int len = 0;
SpoolFile sf = null;
OutputStream sfout = null;
try
{
while ((read = stream.read(tmpBuff)) >= 0)
{
if (sfout != null)
{
// spool to temp file
sfout.write(tmpBuff, 0, read);
len += read;
}
else if (len + read > spoolConfig.maxBufferSize)
{
// threshold for keeping data in memory exceeded,
// if have a fileCleaner create temp file and spool buffer contents.
sf = SpoolFile.createTempFile("jcrvd", null, spoolConfig.tempDirectory);
sf.acquire(this);
sfout = PrivilegedFileHelper.fileOutputStream(sf);
sfout.write(buffer, 0, len);
sfout.write(tmpBuff, 0, read);
buffer = null;
len += read;
}
else
{
// reallocate new buffer and spool old buffer contents
byte[] newBuffer = new byte[len + read];
System.arraycopy(buffer, 0, newBuffer, 0, len);
System.arraycopy(tmpBuff, 0, newBuffer, len, read);
buffer = newBuffer;
len += read;
}
}
if (sf != null)
{
// spooled to file
this.spoolFile = sf;
this.data = null;
}
else
{
// ...bytes
this.spoolFile = null;
this.data = buffer;
}
}
catch (IOException e)
{
if (sf != null)
{
try
{
sf.release(this);
spoolConfig.fileCleaner.addFile(sf);
}
catch (FileNotFoundException ex)
{
if (LOG.isDebugEnabled())
{
LOG.debug("Could not remove temporary file : " + sf.getAbsolutePath());
}
}
}
throw new IllegalStateException("Error of spooling to temp file from " + stream, e);
}
finally
{
try
{
if (sfout != null)
{
sfout.close();
}
}
catch (IOException e)
{
LOG.error("Error of spool output close.", e);
}
this.stream = null;
}
}
/**
* Convert File to byte array. <br/>
* WARNING: Potential lack of memory due to call getAsByteArray() on stream data.
*
* @return byte[] bytes array
*/
protected byte[] fileToByteArray(File file) throws IOException
{
FileInputStream in = null;
try
{
if (channel == null || !channel.isOpen())
{
in = PrivilegedFileHelper.fileInputStream(file);
channel = in.getChannel();
}
ByteBuffer bb = ByteBuffer.allocate((int)channel.size());
channel.read(bb);
if (bb.hasArray())
{
return bb.array();
}
else
{
// impossible code in most cases, as we use heap backed buffer
byte[] tmpb = new byte[bb.capacity()];
bb.get(tmpb);
return tmpb;
}
}
finally
{
if (in != null)
{
in.close();
if (channel != null)
{
channel.close();
}
}
}
}
/**
* Delete current spool file.
*
* @throws IOException
* if error
*/
private void removeSpoolFile() throws IOException
{
if (spoolFile != null)
{
if (spoolFile instanceof SpoolFile)
{
(spoolFile).release(this);
}
if (PrivilegedFileHelper.exists(spoolFile))
{
if (!PrivilegedFileHelper.delete(spoolFile))
{
spoolConfig.fileCleaner.addFile(spoolFile);
if (LOG.isDebugEnabled())
{
LOG.debug("Could not remove file. Add to fileCleaner "
+ PrivilegedFileHelper.getAbsolutePath(spoolFile));
}
}
}
}
}
/**
* {@inheritDoc}
*/
protected byte[] spoolInternalValue()
{
throw new UnsupportedOperationException("Method is not supported");
}
/**
* {@inheritDoc}
*/
protected boolean internalEquals(ValueData another)
{
if (another instanceof StreamValueData)
{
StreamValueData streamValue = (StreamValueData)another;
if (isByteArray())
{
return another.isByteArray() && Arrays.equals(streamValue.data, data);
}
else
{
if (stream != null && stream == streamValue.stream)
{
return true;
}
else if (spoolFile != null && spoolFile.equals(streamValue.spoolFile))
{
return true;
}
}
}
return false;
}
/**
* {@inheritDoc}
*/
public PersistedValueData createPersistedCopy(int orderNumber) throws IOException
{
if (isByteArray())
{
return new ByteArrayPersistedValueData(orderNumber, data);
}
else if (spoolFile != null)
{
return new StreamPersistedValueData(orderNumber, spoolFile, null, spoolConfig);
}
else
{
return new StreamPersistedValueData(orderNumber, stream, null, spoolConfig);
}
}
/**
* {@inheritDoc}
*/
public TransientValueData createTransientCopy(int orderNumber) throws IOException
{
return new TransientValueData(getOrderNumber(), getAsStream(), spoolConfig);
}
/**
* {@inheritDoc}
*/
@Override
protected Long getLong() throws ValueFormatException
{
return Long.valueOf(getString());
}
/**
* {@inheritDoc}
*/
@Override
protected Boolean getBoolean() throws ValueFormatException
{
return Boolean.valueOf(getString());
}
/**
* {@inheritDoc}
*/
@Override
protected Double getDouble() throws ValueFormatException
{
return Double.valueOf(getString());
}
/**
* {@inheritDoc}
*/
protected String getString() throws ValueFormatException
{
try
{
return new String(getAsByteArray(), Constants.DEFAULT_ENCODING);
}
catch (UnsupportedEncodingException e)
{
throw new ValueFormatException("Unsupported encoding " + Constants.DEFAULT_ENCODING, e);
}
catch (IOException e)
{
throw new ValueFormatException("Can't represents data as array of bytes", e);
}
}
/**
* {@inheritDoc}
*/
@Override
protected Calendar getDate() throws ValueFormatException
{
return JCRDateFormat.parse(getString());
}
/**
* {@inheritDoc}
*/
@Override
protected InputStream getStream() throws IOException
{
return getAsStream();
}
/**
* {@inheritDoc}
*/
@Override
protected InternalQName getName() throws ValueFormatException, IllegalNameException
{
return InternalQName.parse(getString());
}
/**
* {@inheritDoc}
*/
@Override
protected QPath getPath() throws ValueFormatException, IllegalPathException
{
return QPath.parse(getString());
}
/**
* {@inheritDoc}
*/
@Override
protected String getReference() throws ValueFormatException
{
return getString();
}
/**
* {@inheritDoc}
*/
@Override
protected AccessControlEntry getPermission() throws ValueFormatException
{
return AccessControlEntry.parse(getString());
}
}