10

我使用以下代码来限制java中文件的下载速度：

package org;

import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

class MainClass {

    public static void main(String[] args) {
        download("https://speed.hetzner.de/100MB.bin");
    }

    public static void download(String link) {
        try {
            URL url = new URL(link);
            HttpURLConnection con = (HttpURLConnection) url.openConnection();
            con.setConnectTimeout(5000);
            con.setReadTimeout(5000);
            InputStream is = con.getInputStream();
            CustomInputStream inputStream = new CustomInputStream(is);
            byte[] buffer = new byte[2024];
            int len;
            while ((len = inputStream.read(buffer)) != -1) {
                System.out.println("downloaded : " + len);
                //save file
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public static class CustomInputStream extends InputStream {

        private static final int MAX_SPEED = 8 * 1024;
        private final long ONE_SECOND = 1000;
        private long downloadedWhithinOneSecond = 0L;
        private long lastTime = System.currentTimeMillis();

        private InputStream inputStream;

        public CustomInputStream(InputStream inputStream) {
            this.inputStream = inputStream;
            lastTime = System.currentTimeMillis();
        }

        @Override
        public int read() throws IOException {
            long currentTime;
            if (downloadedWhithinOneSecond >= MAX_SPEED
                    && (((currentTime = System.currentTimeMillis()) - lastTime) < ONE_SECOND)) {
                try {
                    Thread.sleep(ONE_SECOND - (currentTime - lastTime));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                downloadedWhithinOneSecond = 0;
                lastTime = System.currentTimeMillis();
            }
            int res = inputStream.read();
            if (res >= 0) {
                downloadedWhithinOneSecond++;
            }
            return res;
        }

        @Override
        public int available() throws IOException {
            return inputStream.available();
        }

        @Override
        public void close() throws IOException {
            inputStream.close();
        }
    }

}

成功限制了下载速度，但又出现了一个新问题。当下载正在进行时，我断开与互联网的连接，下载并没有结束并继续一段时间。当我断开互联网连接时，抛出java.net.SocketTimeoutException异常需要 10 多秒。我真的不明白后台发生了什么。

为什么会出现这个问题？

4

2 回答 2

3

您的速率限制实际上并没有像您想象的那样起作用，因为数据实际上不是逐字节发送的，而是以数据包的形式发送的。这些数据包被缓冲，你观察到的（下载继续没有连接）只是你的流读取缓冲区。一旦到达缓冲区的末尾，它会等待 5 秒，然后才会引发超时（因为这是您配置的）。

您将速率设置为 8 kB/s，正常数据包大小通常在 1 kB 左右，最高可达 64 kB，因此您仍有 8 秒的时间在读取同一个数据包。此外，可能已经发送和缓冲了多个数据包。还有一个接收缓冲区，这个缓冲区可以小到 8 - 32 kB 到几 MB。所以实际上你只是从缓冲区中读取。

[编辑]

只是为了澄清，你在做正确的事。平均而言，费率将限制在您指定的范围内。服务器将发送一堆数据，然后等待客户端清空其缓冲区以接收更多数据。

于 2018-10-02T08:15:11.400 回答
3

您显然希望限制客户端的下载速度，并且您还希望客户端立即响应关闭的连接。

AFAIK，这是不可能的......没有一些妥协。

问题是客户端应用程序可以检测到连接已关闭的唯一方法是执行read操作。该读取将传递数据。但是，如果您已经达到了当前期间的限制，那么该读数将使您超过限制。

这里有几个想法：

  • 如果您在短时间内“整合”下载速率（例如每秒 1kbytes 与每 10 秒 10kbytes），那么您可以减少sleep调用的时间长度。

  • 当您接近您的目标下载速率时，您可以退回到进行微小（例如 1 字节）读取和少量休眠。

不幸的是，这两种方法在客户端都是低效的（更多的系统调用），但如果您希望您的应用程序快速检测到连接关闭，这是您必须付出的代价。

你在评论中说：

我希望在禁用 Internet 连接后立即重置连接。

我不这么认为。通常，客户端协议栈会在告诉应用程序代码它正在读取的连接已关闭之前，传递从网络接收到的任何未完成的数据。

于 2018-10-02T11:18:28.597 回答