IDEA多線程文件下載插件開發的步驟詳解

Posted on by

上周使用Java開發瞭大文件多線程下載工具類，自己平時的文件下載也在使用這個工具，下載速度確實提升不少，但是每次下載都要去打開項目運行代碼，覺得實在不是很方便；考慮到每天我們都會使用到IDEA開發工具，所以就決定把這個下載工具做成IDEA的插件，文章末尾附上插件下載地址。

Java實現大文件多線程下載

Gitee地址： https://gitee.com/silently9527/fast-download

IDEA多線程文件下載插件

Github地址： https://github.com/silently9527/FastDownloadIdeaPlugin

Gitee地址： https://gitee.com/silently9527/FastDownloadIdeaPlugin

不要忘記star喲

IDEA插件介紹

IntelliJ IDEA是目前最好用的JAVA開發IDE，它本身的功能已經非常強大瞭，但是可能我們會遇到一些定制的需求，比如說：自定義代碼生成器；這時候就需要我們自己動手來寫一個插件，如果隻是想要開發簡單的功能其實隻要掌握瞭Java Swing，那麼開發IDEA的插件是很容易的，如果想學習更多的原理和設計理念可以看 IntelliJ Platform SDK 的官方文檔。

IDEA插件開發步驟

 1. 創建Gradle的插件工程

創建完成項目之後，我們可以看一下 resource/META-INF/plugin.xml

<idea-plugin>
  <id>cn.silently9527.fast-download-idea-plugin</id>  <!-- 插件的ID -->
  <name>FastDownloadPlugin</name> <!-- 插件的名字，會在插件中心展示 -->
  <vendor email="[email protected]" url="https://silently9527">Silently9527</vendor>
  <!--插件說明-->
  <description><![CDATA[
  多線程文件下載器
  ]]></description>

  <!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/plugin_compatibility.html
     on how to target different products -->
  <!-- uncomment to enable plugin in all products
  <depends>com.intellij.modules.lang</depends>
  -->

  <extensions defaultExtensionNs="com.intellij">
    <!-- Add your extensions here -->
  </extensions>

  <actions>
    <!-- Add your actions here -->
  </actions>
</idea-plugin>

2. 創建一個Action

在IDEA的插件開發中，基本都會使用到Action，Action其實就是事件的處理器，就好比JS中的onClick方法。在IDEA中創建一個Action十分簡單，通過圖形化界面就可以完成

創建完成後就可以看到Action類

public class FastDownloadAction extends AnAction {
  @Override
  public void actionPerformed(AnActionEvent e) {

}
}

在 plugin.xml 中可以看到生成的Action信息

<action id="fast.download" class="cn.silently9527.FastDownloadAction" text="FastDownload" description="文件多線程下載">
  <add-to-group group-id="ToolsMenu" anchor="last"/>
  <keyboard-shortcut keymap="$default" first-keystroke="shift D"/>
</action>

3. 創建輸入下載信息的彈窗

IDEA插件的SDK已經對彈窗進行的封裝，隻需要繼承 DialogWrapper 即可，界面上的繪制工作都在 createCenterPanel 方法中，組件的佈局與JavaSwing類似

@Nullable
@Override
protected JComponent createCenterPanel() {
  Box verticalBox = Box.createVerticalBox();
  verticalBox.add(createUrlBox());
  verticalBox.add(Box.createVerticalStrut(10));
  verticalBox.add(createFileDirJPanel());
  verticalBox.add(Box.createVerticalStrut(10));
  verticalBox.add(createThreadNumJPanel());
  return verticalBox;
}

我們需要對輸入的下載地址和存放的路徑的參數進行校驗，判斷輸入是否正確，可以實現方法 doValidate ，校驗通過返回null，校驗不通過返回 ValidationInfo 對象

@Nullable
@Override
protected ValidationInfo doValidate() {
  if (StringUtils.isBlank(downloadUrlField.getText())) {
    return new ValidationInfo("文件下載地址必填");
  }
  if (StringUtils.isBlank(fileDirField.getText())) {
    return new ValidationInfo("文件保存目錄必填");
  }
  if (StringUtils.isBlank(threadNumField.getText())) {
    return new ValidationInfo("下載線程數必填");
  }
  return null;
}

最終界面完成後的效果

4. 在FastDownloadAction中獲取彈窗輸入的下載信息

DownloadDialog downloadDialog = new DownloadDialog();
if (downloadDialog.showAndGet()) {
  // 用戶點擊OK之後進入到這裡
}

當用戶點擊瞭OK，輸入信息檢驗通過後我們就可以開始下載文件瞭，由於之前做的下載組件是同步調用，為瞭不阻塞界面操作，需要使用線程異步下載

CompletableFuture.runAsync(() -> {
  try {
    Downloader downloader = new MultiThreadFileDownloader(threadNum, downloadProgressPrinter);
    downloader.download(downloadURL, downloadDir);
  } catch (IOException e) {
    throw new RuntimeException(e);
  }
})

在下載的過程中，需要給用戶反饋，讓用戶知道當前下載的進度是多少，以及當前下載的速度是多少

//使用SDK開啟一個後臺任務線程
ProgressManager.getInstance().run(new Task.Backgroundable(project, "File Downloading") {
  private long tmpAlreadyDownloadLength; //當前已下載字節數
  private long speed; //每秒下載速度

  public void run(@NotNull ProgressIndicator progressIndicator) {
    // start your process
    while (true) {
      long alreadyDownloadLength = downloadProgressPrinter.getAlreadyDownloadLength();
      long contentLength = downloadProgressPrinter.getContentLength();
      if (alreadyDownloadLength != 0 && alreadyDownloadLength >= contentLength) {
        // 下載已完成，進度條顯示100%
        progressIndicator.setFraction(1.0);
        progressIndicator.setText("finished");
        break;
      }
      setProgressIndicator(progressIndicator, contentLength, alreadyDownloadLength);
      sleep();
    }
  }

  private void setProgressIndicator(ProgressIndicator progressIndicator, long contentLength,
                   long alreadyDownloadLength) {
    if (alreadyDownloadLength == 0 || contentLength == 0) {
      return;
    }
    speed = alreadyDownloadLength - tmpAlreadyDownloadLength;
    tmpAlreadyDownloadLength = alreadyDownloadLength;

    double value = (double) alreadyDownloadLength / (double) contentLength;

    double fraction = Double.parseDouble(String.format("%.2f", value));
    progressIndicator.setFraction(fraction);
    String text = "already download " + fraction * 100 + "% ,speed: " + (speed / 1000) + "KB";
    progressIndicator.setText(text); //進度條顯示已下載百分比，下載速度
  }
});

測試多線程下載文件

測試下載820M的idea ，地址： https://download.jetbrains.8686c.com/idea/ideaIU-2020.3.dmg

插件安裝

下載插件之後，選擇本地安裝

總結

  • IDEA插件介紹
  • IDEA插件開發的基本步驟
  • 實現瞭多線程文件下載插件

目前測試過程中發現文件下載速度計算不太準確，個別線程的下載速度未能統計在內，後期繼續優化。

插件下載鏈接:

 https://pan.baidu.com/s/1MzYwk9tmuwA7DusN0r7n9Q 提取碼: fqr9

到此這篇關於IDEA多線程文件下載插件開發的文章就介紹到這瞭,更多相關IDEA多線程文件下載插件內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：