2020-12-17

【Electron Playground 系列】文件下载篇

作者：long.woo

文件下载是我们开发中比较常见的业务需求，比如：导出 excel。

web 应用文件下载存在一些局限性，通常是让后端将响应的头信息改成 Content-Disposition: attachment; filename=xxx.pdf，触发浏览器的下载行为。

在 electron 中的下载行为，都会触发 session 的 will-download 事件。在该事件里面可以获取到 downloadItem 对象，通过 downloadItem 对象实现一个简单的文件下载管理器：
demo.gif

1. 如何触发下载

由于 electron 是基于 chromium 实现的，通过调用 webContents 的 downloadURL 方法，相当于调用了 chromium 底层实现的下载，会忽略响应头信息，触发 will-download 事件。

// 触发下载win.webContents.downloadURL(url)// 监听 will-downloadsession.defaultSession.on('will-download', (event, item, webContents) => {})

2. 下载流程

flow_chart.png

3. 功能设计

实现一个简单的文件下载管理器功能包含：

  • 设置保存路径
  • 暂停/恢复和取消
  • 下载进度
  • 下载速度
  • 下载完成
  • 打开文件和打开文件所在位置
  • 文件图标
  • 下载记录

3.1 设置保存路径

如果没有设置保存路径，electron 会自动弹出系统的保存对话框。不想使用系统的保存对话框，可以使用 setSavePath 方法，当有重名文件时，会直接覆盖下载。

item.setSavePath(path)

为了更好的用户体验，可以让用户自己选择保存位置操作。当点击位置输入框时，渲染进程通过 ipc 与主进程通信，打开系统文件选择对话框。
select_path.gif
主进程实现代码：

/** * 打开文件选择框 * @param oldPath - 上一次打开的路径 */const openFileDialog = async (oldPath: string = app.getPath('downloads')) => { if (!win) return oldPath const { canceled, filePaths } = await dialog.showOpenDialog(win, { title: '选择保存位置', properties: ['openDirectory', 'createDirectory'], defaultPath: oldPath, }) return !canceled ? filePaths[0] : oldPath}ipcMain.handle('openFileDialog', (event, oldPath?: string) => openFileDialog(oldPath))

渲染进程代码：

const path = await ipcRenderer.invoke('openFileDialog', 'PATH')

3.2 暂停/恢复和取消

拿到 downloadItem 后，暂停、恢复和取消分别调用 pause、resume 和 cancel 方法。当我们要删除列表中正在下载的项，需要先调用 cancel 方法取消下载。

3.3 下载进度

在 DownloadItem 中监听 updated 事件，可以实时获取到已下载的字节数据，来计算下载进度和每秒下载的速度。

// 计算下载进度const progress = item.getReceivedBytes() / item.getTotalBytes()

download_progress.png
在下载的时候，想在 Mac 系统的程序坞和 Windows 系统的任务栏展示下载信息，比如：

  • 下载数：通过 app 的 badgeCount 属性设置，当为 0 时，不会显示。也可以通过 dock 的 setBadge 方法设置，该方法支持的是字符串，如果不要显示，需要设置为 ''。
  • 下载进度：通过窗口的 setProgressBar 方法设置。

由于 Mac 和 Windows 系统差异，下载数仅在 Mac 系统中生效。加上 process.platform === 'darwin' 条件，避免在非 Mac、Linux 系统下出现异常错误。

下载进度（Windows 系统任务栏、Mac 系统程序坞）显示效果：
windows_progress.png
mac_download_progress.png

// mac 程序坞显示下载数：// 方式一app.badgeCount = 1// 方式二app.dock.setBadge('1')// mac 程序坞、windows 任务栏显示进度win.setProgressBar(progress)

3.4 下载速度

由于 downloadItem 没有直接为我们提供方法或属性获取下载速度，需要自己实现。

思路：在 updated 事件里通过 getReceivedBytes 方法拿到本次下载的字节数据减去上一次下载的字节数据。

// 记录上一次下载的字节数据let prevReceivedBytes = 0item.on('updated', (e, state) => { const receivedBytes = item.getReceivedBytes() // 计算每秒下载的速度 downloadItem.speed = receivedBytes - prevReceivedBytes prevReceivedBytes = receivedBytes})

需要注意的是，updated 事件执行的时间约 500ms 一次。

updated_event.png

3.5 下载完成

当一个文件下载完成、中断或者被取消，需要通知渲染进程修改状态，通过监听 downloadItem 的 done 事件。

item.on('done', (e, state) => { downloadItem.state = state downloadItem.receivedBytes = item.getReceivedBytes() downloadItem.lastModifiedTime = item.getLastModifiedTime() // 通知渲染进程，更新下载状态 webContents.send('downloadItemDone', downloadItem)})

3.6 打开文件和打开文件所在位置

使用 electron 的 shell 模块来实现打开文件（openPath）和打开文件所在位置（showItemInFolder）。

由于 openPath 方法支持返回值 Promise<string>，当不支持打开的文件，系统会有相应的提示，而 showItemInFolder 方法返回值是 void。如果需要更好的用户体验，可使用 nodejs 的 fs 模块，先检查文件是否存在。

import fs from 'fs'// 打开文件const openFile = (path: string): boolean => { if (!fs.existsSync(path)) return false shell.openPath(path) return true}// 打开文件所在位置const openFileInFolder = (path: string): boolean => { if (!fs.existsSync(path)) return false shell.showItemInFolder(path) return true}

3.7 文件图标

很方便的是使用 app 模块的 getFileIcon 方法来获取系统关联的文件图标，返回的是 Promise<NativeImage> 类型，我们可以用 toDataURL 方法转换成 base64，不需要我们去处理不同文件类型显示不同的图标。

const getFileIcon = async (path: string) => { const iconDefault = './icon_default.png' if (!path) Promise.resolve(iconDefault) const icon = await app.getFileIcon(path, { size: 'normal' }) return icon.toDataURL()}

3.8 下载记录

随着下载的历史数据越来越多，使用 electron-store 将下载记录保存在本地。

对 Electron 感兴趣？请关注我们的开源项目 Electron Playground，带你极速上手 Electron。

我们每周五会精选一些有意思的文章和消息和大家分享，来掘金关注我们的 晓前端周刊。

我们是好未来 · 晓黑板前端技术团队。
我们会经常与大家分享最新最酷的行业技术知识。
欢迎来 知乎、掘金、Segmentfault、CSDN、简书、开源中国、博客园 关注我们。









原文转载：http://www.shaoqun.com/a/501156.html

易佰：https://www.ikjzd.com/w/2023

达方物流：https://www.ikjzd.com/w/2562

败欧洲运费：https://www.ikjzd.com/w/1555


作者：long.woo文件下载是我们开发中比较常见的业务需求，比如：导出excel。web应用文件下载存在一些局限性，通常是让后端将响应的头信息改成Content-Disposition:attachment;filename=xxx.pdf，触发浏览器的下载行为。在electron中的下载行为，都会触发session的will-download事件。在该事件里面可以获取到downloadItem
首信易支付：首信易支付
Sunrate：Sunrate
成都离峨眉山有多远 ：成都离峨眉山有多远
冬天去九寨沟有什么好玩的民间活动？：冬天去九寨沟有什么好玩的民间活动？
去昆明游玩大概需要多少钱？2人去云南旅游5天多少钱呢：去昆明游玩大概需要多少钱？2人去云南旅游5天多少钱呢
Posted by at

No comments:

Post a Comment

Subscribe to: Post Comments (Atom)