git-svn-id: https://svn.berlios.de/svnroot/repos/yacy/trunk@6813 6c8d7289-2bf4-0310-a012-ef5d649a1542pull/1/head
orbiter
15 years ago
parent
48b9371735
commit
f204076d25
@ -1,157 +0,0 @@
|
||||
// FileFallbackByteArrayOutputStream.java
|
||||
// -------------------------------------
|
||||
// part of YACY
|
||||
// (C) by Michael Peter Christen; mc@yacy.net
|
||||
// first published on http://www.anomic.de
|
||||
// Frankfurt, Germany, 2004
|
||||
//
|
||||
// This file ist contributed by Franz Brausze
|
||||
//
|
||||
// This program is free software; you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// the Free Software Foundation; either version 2 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program 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 General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program; if not, write to the Free Software
|
||||
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
|
||||
package net.yacy.kelondro.io;
|
||||
|
||||
import java.io.BufferedInputStream;
|
||||
import java.io.BufferedOutputStream;
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
|
||||
import net.yacy.kelondro.util.FileUtils;
|
||||
|
||||
|
||||
public class CachedFileOutputStream extends ByteArrayOutputStream {
|
||||
|
||||
protected File fallbackFile;
|
||||
protected long fallbackSize;
|
||||
protected boolean buffered;
|
||||
|
||||
protected long size = 0;
|
||||
protected boolean isFallback = false;
|
||||
protected OutputStream fallback = null;
|
||||
|
||||
public CachedFileOutputStream(final long fallbackSize) throws IOException {
|
||||
this(fallbackSize, null, true, 32);
|
||||
}
|
||||
|
||||
public CachedFileOutputStream(final long fallbackSize, final File fallback, final boolean buffered)
|
||||
throws IOException {
|
||||
this(fallbackSize, fallback, buffered, 32);
|
||||
}
|
||||
|
||||
public CachedFileOutputStream(final long fallbackSize, final File fallback, final boolean buffered,
|
||||
final long size) throws IOException {
|
||||
this.fallbackSize = fallbackSize;
|
||||
this.fallbackFile = (fallback == null) ? File.createTempFile(
|
||||
CachedFileOutputStream.class.getName(),
|
||||
Long.toString(System.currentTimeMillis())) : fallback;
|
||||
this.buffered = buffered;
|
||||
checkFallback(size);
|
||||
}
|
||||
|
||||
public CachedFileOutputStream(final long fallbackSize, final File fallback, final boolean buffered,
|
||||
final byte[] data) throws IOException {
|
||||
this(fallbackSize, fallback, buffered, 0);
|
||||
super.buf = data;
|
||||
super.count = data.length;
|
||||
checkFallback(this.size = data.length);
|
||||
}
|
||||
|
||||
protected boolean checkFallback(final long size) {
|
||||
if (size > this.fallbackSize) try {
|
||||
fallback();
|
||||
return true;
|
||||
} catch (final IOException e) {
|
||||
throw new RuntimeException("error falling back to file", e);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
public void fallback() throws IOException {
|
||||
if (this.isFallback) return;
|
||||
this.isFallback = true;
|
||||
if (!this.fallbackFile.exists()) {
|
||||
this.fallbackFile.createNewFile();
|
||||
} else if (this.fallbackFile.isDirectory()) {
|
||||
throw new IOException("cannot write on a directory");
|
||||
}
|
||||
final OutputStream os = new FileOutputStream(this.fallbackFile);
|
||||
this.fallback = (this.buffered) ? new BufferedOutputStream(os) : os;
|
||||
FileUtils.copy(new ByteArrayInputStream(super.buf), this.fallback);
|
||||
super.buf = new byte[0];
|
||||
super.count = 0;
|
||||
super.reset();
|
||||
}
|
||||
|
||||
public boolean isFallback() {
|
||||
return this.isFallback;
|
||||
}
|
||||
|
||||
public synchronized void write(final int b) {
|
||||
if (checkFallback(++this.size)) try {
|
||||
this.fallback.write(b);
|
||||
} catch (final IOException e) {
|
||||
throw new RuntimeException("error writing to fallback", e);
|
||||
} else {
|
||||
super.write(b);
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void write(final byte[] b, final int off, final int len) {
|
||||
if (checkFallback(this.size += len)) try {
|
||||
this.fallback.write(b, off, len);
|
||||
} catch (final IOException e) {
|
||||
throw new RuntimeException("error writing to fallback", e);
|
||||
} else {
|
||||
super.write(b, off, len);
|
||||
}
|
||||
}
|
||||
|
||||
public void close() throws IOException {
|
||||
if (this.fallback != null)
|
||||
this.fallback.close();
|
||||
super.close();
|
||||
}
|
||||
|
||||
public InputStream getContent() throws IOException {
|
||||
close();
|
||||
if (this.isFallback) {
|
||||
final InputStream is = new FileInputStream(this.fallbackFile);
|
||||
return (this.buffered) ? new BufferedInputStream(is) : is;
|
||||
}
|
||||
return new ByteArrayInputStream(this.buf);
|
||||
}
|
||||
|
||||
public byte[] getContentBAOS() {
|
||||
if (this.isFallback)
|
||||
throw new RuntimeException("underlying ByteArrayOutputStream not available, already fell back to file");
|
||||
return super.buf;
|
||||
}
|
||||
|
||||
public File getContentFile() {
|
||||
if (!this.isFallback)
|
||||
throw new RuntimeException("haven't fallen back yet, fallback file has no content");
|
||||
return this.fallbackFile;
|
||||
}
|
||||
|
||||
public long getLength() {
|
||||
return this.size;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue