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

更新時間：2020年12月30日 09:44:05 投稿：mrr

這篇文章主要介紹了IDEA多線程文件下載插件開發(fā)的步驟詳解,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

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

Java實現(xiàn)大文件多線程下載

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開發(fā)IDE，它本身的功能已經非常強大了，但是可能我們會遇到一些定制的需求，比如說：自定義代碼生成器；這時候就需要我們自己動手來寫一個插件，如果只是想要開發(fā)簡單的功能其實只要掌握了Java Swing，那么開發(fā)IDEA的插件是很容易的，如果想學習更多的原理和設計理念可以看 IntelliJ Platform SDK 的官方文檔。

IDEA插件開發(fā)步驟

1. 創(chuàng)建Gradle的插件工程

創(chuàng)建完成項目之后，我們可以看一下 resource/META-INF/plugin.xml

<idea-plugin>
  <id>cn.silently9527.fast-download-idea-plugin</id>  <!-- 插件的ID -->
  <name>FastDownloadPlugin</name> <!-- 插件的名字，會在插件中心展示 -->
  <vendor email="380303318@qq.com" 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. 創(chuàng)建一個Action

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

創(chuàng)建完成后就可以看到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. 創(chuàng)建輸入下載信息的彈窗

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

我們需要對輸入的下載地址和存放的路徑的參數(shù)進行校驗，判斷輸入是否正確，可以實現(xiàn)方法 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("下載線程數(shù)必填");
  }
  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; //當前已下載字節(jié)數(shù)
  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); //進度條顯示已下載百分比，下載速度
  }
});