如何实现Java限制的下载速率？

Z时代
2024-01-10
分类：问答

我将作为个人练习用Java实现一个（简单的）下载器应用程序。它将在不同的线程中运行多个作业，以这种方式，我将在执行期间始终同时下载几个文件。

我希望能够定义在所有下载作业之间共享的下载速率限制，但是我不知道如何执行单个下载任务。我应该怎么做呢？我应该尝试实施哪些解决方案？

谢谢。

回答：

我将从管理所有下载的DownloadManager开始。

interface DownloadManager
{
    public InputStream registerDownload(InputStream stream);
}

希望参与托管带宽的所有代码都将在开始从下载管理器中读取之前将其流注册到下载管理器中。在它的registerDownload（）方法中，管理器将给定的输入流包装在中ManagedBandwidthStream。

   public class ManagedBandwidthStream extends InputStream
   {
      private DownloadManagerImpl owner;
      public ManagedBandwidthStream(
            InputStream original,
            DownloadManagerImpl owner
         )
      {
         super(original);
         this.owner = owner;
      }
      public int read(byte[] b, int offset, int length)
      {
          owner.read(this, b, offset, length);
      }
      // used by DownloadManager to actually read from the stream
      int actuallyRead(byte[] b, int offset, int length)
      {
          super.read(b, offset, length);
      }
      // also override other read() methods to delegate to the read() above
   }

该流确保将对read（）的所有调用都定向回下载管理器。

class DownloadManagerImpl implements DownloadManager
{
   public InputStream registerDownload(InputStream in)
   {
       return new ManagedDownloadStream(in);
   }
   void read(ManagedDownloadStream source, byte[] b, int offset, int len)
   {
      // all your streams now call this method.
      // You can decide how much data to actually read.
      int allowed = getAllowedDataRead(source, len);
      int read = source.actuallyRead(b, offset, len);
      recordBytesRead(read);  // update counters for number of bytes read
   }
}