将超时设置为 fs.copy 函数

问题

1. global.resultArr = {};
   
2.   global.failedArr = [];
   
3.   global.successArr = [];   
   
4.     const writeFile = async function (dirResult, queryResult) {
   
5.   queryResult.recordset.forEach((ip_vch, index) => {
   
6.     for (file of dirResult) {
   
7.       if (
   
8.         file.substring(file.length - 3) == "wav" &&
   
9.         global.failedArr.indexOf(ip_vch.ip_vch) == -1
   
10.       ) {
    
11.         try {
    
12.           writeResult[index] = await timeout(
    
13.             fs.copy(
    
14.               dir + "//" + file,
    
15.               "//" +
    
16.                 ip_vch.ip_vch +
    
17.                 "//RXWaveFiles//DynamicLibraries" +
    
18.                 "//" +
    
19.                 libid +
    
20.                 "//" +
    
21.                 file
    
22.             ),
    
23.             8000
    
24.           );
    
25.           Promise.all([writeResult])
    
26.             .then(function (values) {
    
27.               console.log(values);
    
28.               if (
    
29.                 writeResult &&
    
30.                 global.failedArr.indexOf(ip_vch.ip_vch) == -1
    
31.               ) {
    
32.                 console.log(ip_vch.ip_vch);
    
33.                 global.failedArr.push(ip_vch.ip_vch);
    
34.                 sql.query`update opower..dialers set fileMoveResult_int=0 where ip_vch =${ip_vch.ip_vch}`;
    
35.               } else if (
    
36.                 global.successArr.indexOf(ip_vch.ip_vch) == -1 &&
    
37.                 global.failedArr.indexOf(ip_vch.ip_vch) == -1
    
38.               ) {
    
39.                 global.successArr.push(ip_vch.ip_vch);
    
40.                 sql.query`update opower..dialers set fileMoveResult_int=1 where ip_vch =${ip_vch.ip_vch}`;
    
41.                 console.log("success!" + ip_vch.ip_vch);
    
42.               }
    
43.             })
    
44.             .catch((err) => console.error(err));
    
45.         } catch (error) {
    
46.           console.error(error);
    
47.           if (global.failedArr.indexOf(ip_vch.ip_vch) == -1) {
    
48.             global.failedArr.push(ip_vch).ip_vch;
    
49.             sql.query`update opower..dialers set fileMoveResult_int=0 where ip_vch =${ip_vch.ip_vch}`;
    
50.           }
    
51.         }
    
52.       }
    
53.     }
    
54.   });
    
55. 
    
56.   global.resultArr.success = successArr;
    
57.   global.resultArr.failed = failedArr;
    
58.   return global.resultArr;
    
59. };
    
60. 
    
61. // utility function that creates a promise that rejects after a certain time
    
62. function timeoutPromise(t, errMsg = "timeout") {
    
63.     // create possible error object here to get appropriate stack trace
    
64.     let e = new Error(errMsg);
    
65.     e.timeout = true;
    
66.     return new Promise((resolve, reject) => {
    
67.         setTimeout(reject, t, e);
    
68.     });
    
69. }
    
70. 
    
71. // wrap a promise with a timeout, pass promise, time in ms and
    
72. // optional timeout error message
    
73. function timeout(p, t, errMsg = "timeout") {
    
74.     return Promise.race(p, timeoutPromise(t, errMsg));
    
75. }

复制代码

我在 for 循环中使用这个 await 函数，我需要将一些文件从一个源目录复制到多个网络目录，但是 await 的问题是，解决失败的目录需要将近一分钟的时间，然后将控制权交还给下一次迭代，有没有办法在 5 秒后停止当前迭代。

回答
您可以为以下任何 Promise 添加错误超时：

1. // utility function that creates a promise that rejects after a certain time
   
2. function timeoutPromise(t, errMsg = "timeout") {
   
3.     // create possible error object here to get appropriate stack trace
   
4.     let e = new Error(errMsg);
   
5.     e.timeout = true;
   
6.     return new Promise((resolve, reject) => {
   
7.         setTimeout(reject, t, e);
   
8.     });
   
9. }
   
10. 
    
11. // wrap a promise with a timeout, pass promise, time in ms and
    
12. // optional timeout error message
    
13. function timeout(p, t, errMsg = "timeout") {
    
14.     return Promise.race([p, timeoutPromise(t, errMsg)]);
    
15. }

复制代码

，您可以将它与 fs.copy() 一起使用，如下所示：

const writeResult = await timeout(fs.copy(...), 5000);

因此，如果 fs.copy() 花费的时间超过 5 秒，您正在等待的承诺将被拒绝，您可以在 catch 处理程序中捕获它并采取相应的行动。您将能够看到错误对象具有 .timeout 属性。

它的工作方式是 timeout() 函数在传入的 Promise 和另一个将在超时后被拒绝的 Promise 之间创建竞争。第一个完成的控件控制 Promise.race() 的输出，而后者又控制您在其上使用的 await 的内容。如果超时成功，则 Promise 将被拒绝。