package org.jvnet.mimepull;
import java.nio.ByteBuffer;
import java.io.File;
import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;
final class MemoryData implements Data {
private static final Logger LOGGER = Logger.getLogger(MemoryData.class.getName());
private final byte[] data;
private final int len;
private final MIMEConfig config;
MemoryData(ByteBuffer buf, MIMEConfig config) {
data = buf.array();
len = buf.limit();
this.config = config;
}
@Override
public int size() {
return len;
}
@Override
public byte[] read() {
return data;
}
@Override
public long writeTo(DataFile file) {
return file.writeTo(data, 0, len);
}
@Override
public Data createNext(DataHead dataHead, ByteBuffer buf) {
if (!config.isOnlyMemory() && dataHead.inMemory >= config.memoryThreshold) {
try {
String prefix = config.getTempFilePrefix();
String suffix = config.getTempFileSuffix();
File tempFile = TempFiles.createTempFile(prefix, suffix, config.getTempDir());
tempFile.deleteOnExit();
if (LOGGER.isLoggable(Level.FINE)) {
LOGGER.log(Level.FINE, "Created temp file = {0}", tempFile);
}
dataHead.dataFile = new DataFile(tempFile);
} catch (IOException ioe) {
throw new MIMEParsingException(ioe);
}
if (dataHead.head != null) {
for (Chunk c = dataHead.head; c != null; c = c.next) {
long pointer = c.data.writeTo(dataHead.dataFile);
c.data = new FileData(dataHead.dataFile, pointer, len);
}
}
return new FileData(dataHead.dataFile, buf);
} else {
return new MemoryData(buf, config);
}
}
}