Electron中如何在批量文件加密完成后执行指定函数？

核心原因 #

你的encrypt是异步函数（带async），调用后会返回Promise，但点击事件里只是循环调用window.encrypt，没有等待所有异步加密操作完成，就直接执行了done()，导致它提前触发。

修复步骤 #

1. 补全encrypt函数的参数和依赖
   encrypt函数里用到了Transform但未引入，且缺少password参数，先修正encrypt.js：

   const path = require("path");
   const fs = require("fs");
   const { pipeline } = require("stream/promises");
   const { Transform } = require('stream'); // 新增：引入Transform
   const { app } = require("./app.js");
   
   // 新增password参数
   async function encrypt(file, password) {
     const fileReadStream = fs.createReadStream(file);
   
     const filePath = path.parse(file).dir;
     const fileName = path.parse(file).name;
     const fileExt = path.parse(file).ext;
     const EncFile = filePath + "/" + fileName + "_enc" + fileExt;
     const fileWriteStream = fs.createWriteStream(EncFile);
   
     await pipeline(
       fileReadStream,
       new Transform({
         transform(chunk, encoding, callback) {
           const encryptedData = app.encrypt(chunk, password);
           callback(null, encryptedData);
           console.log("Chunk encrypted"); // 修正：这里是分块加密完成，不是整个文件
         },
       }),
       fileWriteStream
     );
     console.log(`File ${file} encrypted completely`); // 新增：整个文件加密完成的日志
   }
   module.exports.encrypt = encrypt;
   

2. 修改点击事件，等待所有加密任务完成
   把事件回调改成async函数，用Promise.all等待所有encrypt返回的Promise完成后再调用done()：

   //preload.js
   const { encrypt } = require('./encrypt.js');
   contextBridge.exposeInMainWorld('encrypt', encrypt);
   
   const para = document.querySelector('#para-info');
   const btn = document.querySelector('#btn');
   // 把回调改成async函数
   btn.addEventListener("click", async () => {
     // 收集所有加密任务的Promise
     const encryptPromises = filesList.map(file => {
       return window.encrypt(file, password);
     });
     // 等待所有加密任务完成
     await Promise.all(encryptPromises);
     // 所有加密完成后再执行done
     done();
   });
   
   // function i want to run after encryption is done
   function done(){
     para.innerText = 'Encryption Done';
   }
   

关键说明 #

• Promise.all会等待传入的所有Promise都resolve后才继续执行，确保所有文件加密完成。
• 事件回调加上async才能使用await语法，否则无法等待Promise完成。
• 修正了日志的描述，原代码里的console.log("File encrypted")是在分块处理时触发，不是整个文件完成，容易造成误解。

内容的提问来源于stack exchange，提问作者Sam Seith