一个java流量控制小程序

在开发中,常常会遇到文件上传下载流量控制的问题,这是一个小程序,简单的实现了这个功能,可以参考一下

public class FlowControlInputStream extends FilterInputStream{

public static void main(String[] args){
try{
byte[] buffer = new byte[8092];
int n;
long start = System.currentTimeMillis();
FileInputStream in = new FileInputStream("E:\\pic.rar");
//限制流量为10000也就是10000=10kbps
FlowControlInputStream fin = new FlowControlInputStream(in, 40000);
// fin.update(1000);
System.out.println("当前速率:"+fin.check()/1000+"kbps");
System.out.println("当前文件大小:"+fin.available()/1024+"kb");
while( (n = fin.read(buffer)) > 0);
fin.close();
long end = System.currentTimeMillis();
System.out.println("读取文件完成,共花掉"+(end-start)/600+"秒");
}catch(IOException e){
e.printStackTrace();
}
}


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;
}
}

代码很简单!
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值