package com.ettrema.httpclient;
import com.ettrema.httpclient.Utils.CancelledException;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
/**
*
* @author brad
*/
public class NotifyingFileOutputStream extends FileOutputStream {
private final ProgressListener listener;
private final String fileName;
private final Long length;
long pos;
// the system time we last notified the progress listener
long timeLastNotify = System.currentTimeMillis();
long bytesSinceLastNotify;
public NotifyingFileOutputStream(java.io.File f, ProgressListener listener, Long length) throws FileNotFoundException {
super(f);
this.length = length;
this.listener = listener;
this.fileName = f.getAbsolutePath();
}
public NotifyingFileOutputStream(java.io.File f, boolean append, ProgressListener listener, Long length) throws FileNotFoundException {
super(f, append);
this.length = length;
this.listener = listener;
this.fileName = f.getAbsolutePath();
}
@Override
public void write(int b) throws IOException {
increment(1);
super.write(b);
}
@Override
public void write(byte[] b) throws IOException {
increment(b.length);
super.write(b);
}
@Override
public void write(byte[] b, int off, int len) throws IOException {
increment(len);
super.write(b, off, len);
}
private void increment(int len) throws IOException {
pos += len;
if (listener != null) {
notifyListener(len);
}
}
void notifyListener(int numBytes) throws IOException{
if( listener.isCancelled() ) {
throw new CancelledException();
}
bytesSinceLastNotify += numBytes;
if (bytesSinceLastNotify < 1000) {
return;
}
int timeDiff = (int) (System.currentTimeMillis() - timeLastNotify);
if (timeDiff > 10) {
timeLastNotify = System.currentTimeMillis();
if (length == null || length <= 0) {
listener.onProgress(100, fileName);
} else {
int percent = (int) ((pos * 100 / length));
if (percent > 100) {
percent = 100;
}
listener.onProgress(percent, fileName);
}
bytesSinceLastNotify = 0;
}
}
}