Java程序如何限速
java1234
共 13303字,需浏览 27分钟
· 2021-06-13
点击上方蓝色字体,选择“标星公众号”
优质文章,第一时间送达
作者 | 小码哥
这里简单的讨论一下java设计网络程序中如何控制上传和下载速度,我们常见的FTP,HTTP,BT等协议都是TCP的,但是现在流行的utorrent却基于UDP实现了自己UTP协议(UDP+拥塞控制),不管使用什么协议,站在I/O的角度来说,限速的控制思路都是一样的。
思路很简单,如下:
1.假设下载或者上传速度上限是m (KB/s),那么发送一个固定的字节数据(假设是n字节)的时间花费是:n/m;
2.假设现在要发送n字节的数据,那么理论所需的时间应该是n/m,而在实际情况下,发送n字节的数据只花费了t秒,那么发送该发送线程就应该睡眠n/m-t秒,这样就基本实现了速度的控制。
代码以TCP为例
速度控制
1 package com.actiontec.net.bandwidth;
2
3 /**
4 *
5 * @author Le
6 *
7 */
8 public class BandwidthLimiter {
9
10 /* KB */
11 private static Long KB = 1024l;
12
13 /* The smallest count chunk length in bytes */
14 private static Long CHUNK_LENGTH = 1024l;
15
16 /* How many bytes will be sent or receive */
17 private int bytesWillBeSentOrReceive = 0;
18
19 /* When the last piece was sent or receive */
20 private long lastPieceSentOrReceiveTick = System.nanoTime();
21
22 /* Default rate is 1024KB/s */
23 private int maxRate = 1024;
24
25 /* Time cost for sending CHUNK_LENGTH bytes in nanoseconds */
26 private long timeCostPerChunk = (1000000000l * CHUNK_LENGTH)
27 / (this.maxRate * KB);
28
29 /**
30 * Initialize a BandwidthLimiter object with a certain rate.
31 *
32 * @param maxRate
33 * the download or upload speed in KBytes
34 */
35 public BandwidthLimiter(int maxRate) {
36 this.setMaxRate(maxRate);
37 }
38
39 /**
40 * Set the max upload or download rate in KB/s. maxRate must be grater than
41 * 0. If maxRate is zero, it means there is no bandwidth limit.
42 *
43 * @param maxRate
44 * If maxRate is zero, it means there is no bandwidth limit.
45 * @throws IllegalArgumentException
46 */
47 public synchronized void setMaxRate(int maxRate)
48 throws IllegalArgumentException {
49 if (maxRate < 0) {
50 throw new IllegalArgumentException("maxRate can not less than 0");
51 }
52 this.maxRate = maxRate < 0 ? 0 : maxRate;
53 if (maxRate == 0)
54 this.timeCostPerChunk = 0;
55 else
56 this.timeCostPerChunk = (1000000000l * CHUNK_LENGTH)
57 / (this.maxRate * KB);
58 }
59
60 /**
61 * Next 1 byte should do bandwidth limit.
62 */
63 public synchronized void limitNextBytes() {
64 this.limitNextBytes(1);
65 }
66
67 /**
68 * Next len bytes should do bandwidth limit
69 *
70 * @param len
71 */
72 public synchronized void limitNextBytes(int len) {
73 this.bytesWillBeSentOrReceive += len;
74
75 /* We have sent CHUNK_LENGTH bytes */
76 while (this.bytesWillBeSentOrReceive > CHUNK_LENGTH) {
77 long nowTick = System.nanoTime();
78 long missedTime = this.timeCostPerChunk
79 - (nowTick - this.lastPieceSentOrReceiveTick);
80 if (missedTime > 0) {
81 try {
82 Thread.sleep(missedTime / 1000000,
83 (int) (missedTime % 1000000));
84 } catch (InterruptedException e) {
85 e.printStackTrace();
86 }
87 }
88 this.bytesWillBeSentOrReceive -= CHUNK_LENGTH;
89 this.lastPieceSentOrReceiveTick = nowTick
90 + (missedTime > 0 ? missedTime : 0);
91 }
92 }
93 }
94
下载控制
1 package com.actiontec.net.bandwidth;
2
3 import java.io.IOException;
4 import java.io.InputStream;
5
6 /**
7 * @author Le
8 *
9 */
10 public class DownloadLimiter extends InputStream {
11 private InputStream is = null;
12 private BandwidthLimiter bandwidthLimiter = null;
13
14 public DownloadLimiter(InputStream is, BandwidthLimiter bandwidthLimiter)
15 {
16 this.is = is;
17 this.bandwidthLimiter = bandwidthLimiter;
18 }
19 @Override
20 public int read() throws IOException {
21 if(this.bandwidthLimiter != null)
22 this.bandwidthLimiter.limitNextBytes();
23 return this.is.read();
24 }
25
26 public int read(byte b[], int off, int len) throws IOException
27 {
28 if (bandwidthLimiter != null)
29 bandwidthLimiter.limitNextBytes(len);
30 return this.is.read(b, off, len);
31 }
32 }
同样,上传控制
1 package com.actiontec.net.bandwidth;
2
3 import java.io.IOException;
4 import java.io.OutputStream;
5
6 /**
7 * @author Le
8 *
9 */
10 public class UploadLimiter extends OutputStream {
11 private OutputStream os = null;
12 private BandwidthLimiter bandwidthLimiter = null;
13
14 public UploadLimiter(OutputStream os, BandwidthLimiter bandwidthLimiter)
15 {
16 this.os = os;
17 this.bandwidthLimiter = bandwidthLimiter;
18 }
19
20 @Override
21 public void write(int b) throws IOException {
22 if (bandwidthLimiter != null)
23 bandwidthLimiter.limitNextBytes();
24 this.os.write(b);
25 }
26
27 public void write(byte[] b, int off, int len) throws IOException {
28 if (bandwidthLimiter != null)
29 bandwidthLimiter.limitNextBytes(len);
30 this.os.write(b, off, len);
31 }
32
33 }
对于一个TCP socket
1 ServerSocket socket = new ServerSocket();
2 //其它初始化略
1 //从socket中以一定的速率读数据
2 //```java
3 DownloadLimiter dl = new DownloadLimiter(socket.getInputStream(), new BandwidthLimiter(6250));
4 is = new DataInputStream(dl);
5
6 //读数据
7 int len = is.readInt();
8 ByteBuffer buffer = ByteBuffer.allocate(4 + len);
9 buffer.putInt(len);
10 is.readFully(buffer.array(), 4, buffer.remaining());
11 //```
12
13 //以一定的速率写数据到socket
14 //```java
15 UploadLimiter ul = new UploadLimiter(socket.getOutputStream(), new BandwidthLimiter(6250));
16 ul.write();
17 //```
在多线程环境下也可以使用上述的方法。最后附图是任务管理器的网络利用率图6250KB/s(也就是50000kb/s,附图中网络利用率也在5%左右,所以应该这个做法还算准确)
评论
偷偷告诉你如何一台电脑开多个微信!
大家好,我是轩辕。前几天在粉丝群里,有人问我是怎么在一台电脑上同时登录两个微信的?正好之前写过一篇文章,分析过原理,分享给没看过的小伙伴学习一下。手机端多开微信估计很多人都知道,像华为、小米等手机系统都对此做了支持,不过在运行Windows系统的电脑上怎么启动两个微信呢?其实很简单,你只需要写一个批
编程技术宇宙
0
豆瓣9.7,这部Java神作第3版重磅上市!
文末赠书Java 程序员们开年就有重磅好消息,《Effective Java 中文版(原书第 3 版)》要上市啦!该书的第1版出版于 2001 年,当时就在业界流传开来,受到广泛赞誉。时至今日,已热销近20年,本书第 3 版已是 Java 程序员的必读神书,被誉为“Java 四大名著之一”,甚至连
编码之外
0
测试新人,如何快速上手一个陌生的系统!
大家好,我是狂师!作为刚入行不久的测试新人,面对一个陌生的系统时,可能会感到有些手足无措。面对一个全新的系统系统,如何快速上手并展开有效的测试工作是一个重要的挑战。本文将探讨测试新人如何通过一系列步骤和策略,快速熟悉并掌握新系统的测试要点,从而提高测试效率和质量。本文旨在为测试新手提供一份指导,帮助
测试开发技术
0
光纤详解:光纤跳线如何分类,多向单模转换?
本文来自“光纤详解:光纤跳线如何分类,多向单模转换?”,光纤跳线作为光网络布线最基础的元件之一,被广泛应用于光纤链路的搭建中。如今,光纤制造商根据应用场景的不同推出众多类型的光纤跳线,如MPO/LC/SC/FC/ST光纤跳线,单工/双工光纤跳线,单模/多模光纤跳线等,它们之间各有特色,且不可替代。本
架构师技术联盟
0
如何计算数据中心的冷却需求?
今日分享 【导读】数据中心的冷却要求受多种因素影响,包括设备的热量输出、占地面积、设施设计和电气系统功率额定值等等……众所周知,环境因素会严重影响数据中心设备。过多的热量积聚会损坏服务器,可能导致其自动关闭。经常在高于可接受的温度下运行服务器会缩短其使用
数据中心运维管理
0
5000w+ 的大表如何拆?亿级别大表拆分实战复盘
前言笔者是在两年前接手公司的财务系统的开发和维护工作。在系统移交的初期,笔者和团队就发现,系统内有一张5000W+的大表。跟踪代码发现,该表是用于存储资金流水的表格,关联着众多功能点,同时也有众多的下游系统在使用这张表的数据。进一步的观察发现,这张表还在以每月600W+的数据持续增长,也就是说,不超
码农编程进阶笔记
0
如何做到无感刷新Token?
来源:juejin.cn/post/7316797749517631515为什么需要无感刷新Token?自动刷新token前端token续约疑问及思考图片为什么需要无感刷新Token?「最近浏览到一个文章里面的提问,是这样的:」当我在系统页面上做业务操作的时候会出现突然闪退的情况,然后跳转到登录页面
Java专栏
2
大量 Java 开源项目停更...
点击关注公众号,Java 干货及时推送↓推荐阅读:投了 100 多份简历后…出品 | OSC开源社区(ID:oschina2013)Sonatype 发布了最新的一份《软件供应链状况》报告,深入探讨了如何在充满选择的世界中定义更好的软件,并探讨人工智能 (AI) 对软件开发的深远
Java技术栈
0