アップロードとダウンロードの制限
5424 ワード
あまり言わないで、自分で見ましょう.
public class FlowControlInputStream extends FilterInputStream{
private long timestamp;
private int maxbps;
private int currentbps;
private int bytesread;
//----------------------------------------------------------
//constructor
public FlowControlInputStream(InputStream in, int maxbps){
super(in);
this.maxbps = maxbps;
this.currentbps = 0;
this.bytesread = 0;
this.timestamp = System.currentTimeMillis();
}
//----------------------------------------------------------
//decorated methods
public int read() throws IOException{
synchronized(in){
int avaliable = check();
if(avaliable == 0){
waitForAvailable();
avaliable = check();
}
int value = in.read();
update(1);
return value;
}
}
public int read(byte[] b) throws IOException{
return read(b, 0, b.length);
}
public int read(byte[] b, int off, int len) throws IOException{
synchronized(in){
int avaliable = check();
if(avaliable == 0){
waitForAvailable();
avaliable = check();
}
int n = in.read(b, off, Math.min(len, avaliable));
update(n);
return n;
}
}
private int check(){
long now = System.currentTimeMillis();
if(now - timestamp >= 1000){
timestamp = now;
currentbps = bytesread;
bytesread = 0;
return maxbps;
}else{
return maxbps - bytesread;
}
}
private void waitForAvailable(){
long time = System.currentTimeMillis() - timestamp;
boolean isInterrupted = false;
while(time < 1000){
try{
Thread.sleep(1000 - time);
}catch(InterruptedException e){
isInterrupted = true;
}
time = System.currentTimeMillis() - timestamp;
}
if(isInterrupted)
Thread.currentThread().interrupt();
return;
}
private void update(int n){
bytesread += n;
}
public int getCurrentbps(){
return currentbps;
}
}
public class FlowControlOutputStream extends FilterOutputStream {
private long timestamp;
private int maxbps;
private int currentbps;
private int byteswrite;
public FlowControlOutputStream(OutputStream out, int maxbps) {
super(out);
this.maxbps = maxbps;
this.currentbps = 0;
this.byteswrite = 0;
this.timestamp = System.currentTimeMillis();
}
public void setMaxbps(int maxbps){
this.maxbps = maxbps;
}
@Override
public void write(int oneByte) throws IOException {
synchronized(out){
int avaliable = check();
if(avaliable <= 0){
waitForAvailable();
}
out.write(oneByte);
update(1);
}
}
@Override
public void write(byte[] buffer) throws IOException {
write(buffer, 0, buffer.length);
}
@Override
public void write(byte[] buffer, int offset, int length) throws IOException {
synchronized(out){
int avaliable = check();
if(avaliable <= 0){
waitForAvailable();
}
out.write(buffer, offset, length);
update(length);
}
}
private int check(){
long now = System.currentTimeMillis();
if(now - timestamp >= 1000){
timestamp = now;
currentbps = byteswrite;
byteswrite = 0;
return maxbps;
}else{
return maxbps - byteswrite;
}
}
private void waitForAvailable(){
long time = System.currentTimeMillis() - timestamp;
boolean isInterrupted = false;
while(time < 1000){
try{
Thread.sleep(1000 - time);
}catch(InterruptedException e){
isInterrupted = true;
}
time = System.currentTimeMillis() - timestamp;
}
if(isInterrupted)
Thread.currentThread().interrupt();
return;
}
private void update(int n){
byteswrite += n;
}
public int getCurrentbps(){
return currentbps;
}
}