feat: add retry option

xiaweiss · Dec 8, 2024 · 69ae091 · 69ae091
1 parent f50b48e
commit 69ae091
Show file tree

Hide file tree

Showing 5 changed files with 41 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -86,5 +86,6 @@ The plugin will upload files of `outDir` path after bundle.
 | headers         | Request headers setting, more information: https://www.alibabacloud.com/help/en/doc-detail/31978.html | object | {} |
 | test            | Only test path, no files upload                                           | boolean | false         |
 | enabled         | Enable the ali oss plugin                                                 | boolean | true          |
+| retry           | Number of retries when upload fails                                       | number  | 0             |
 | ...             | Other init oss options, more information: https://www.alibabacloud.com/help/en/doc-detail/64097.html | any | |
 
diff --git a/README_CN.md b/README_CN.md
@@ -84,5 +84,6 @@ pnpm run build
 | headers         | 请求头设置，详细信息请见 https://www.alibabacloud.com/help/zh/doc-detail/31978.html | object| {} |
 | test            | 仅测试路径，不会有文件上传                                  | boolean | false         |
 | enabled         | 是否启用本插件                                            | boolean | true          |
+| retry           | 上传失败时，重试次数                                       | number  | 0             |
 | ...             | 其他初始化 oss 的参数，详细信息请见 https://www.alibabacloud.com/help/zh/doc-detail/64097.html | any | |
 
diff --git a/example/vite-6/vite.config.ts b/example/vite-6/vite.config.ts
@@ -15,6 +15,7 @@ const options = {
   // overwrite: false,
   // enabled: false,
   // test: true,
+  // retry: 3
 }
 
 const prod = process.env.NODE_ENV === 'production'

diff --git a/index.d.ts b/index.d.ts
@@ -19,6 +19,8 @@ interface Options {
   test?: boolean
   /** Enable the ali oss plugin. Default true */
   enabled?: boolean
+  /** Number of retries when upload (default 0) */
+  retry?: number
   /** The temporary Security Token Service (STS) token used to access OSS. */
   stsToken?: string
   /** The endpoint that is used to access your OSS bucket. */

diff --git a/index.js b/index.js
@@ -6,6 +6,28 @@ import { URL } from 'node:url'
 
 import { normalizePath } from 'vite'
 
+const retry = async (fn, time) => {
+  while (true) {
+    try {
+      await fn()
+      break
+    } catch (error) {
+
+      if (time > 0) {
+        time -= 1
+        console.error(color.red(error))
+        await new Promise(resolve => setTimeout(resolve, 500))
+        console.log('')
+        console.log('')
+        console.log(`[vite-plugin-ali-oss] retry upload after 0.5s, ${time} times left`)
+        console.log('')
+      } else {
+        throw new Error(error)
+      }
+    }
+  }
+}
+
 export default function vitePluginAliOss (options) {
   let baseConfig = '/'
   let buildConfig = {}
@@ -41,6 +63,7 @@ export default function vitePluginAliOss (options) {
         delete createOssOption.headers
         delete createOssOption.test
         delete createOssOption.enabled
+        delete createOssOption.retry
 
         const client = new OSS(createOssOption)
         const ssrClient = buildConfig.ssrManifest
@@ -85,22 +108,24 @@ export default function vitePluginAliOss (options) {
           }
 
           if (options.overwrite) {
-            await client.put(
-              ossFilePath,
-              fileFullPath,
-              {
-                headers: options.headers || {}
-              }
-            )
-            console.log(`upload complete: ${output}`)
+            await retry(async () => {
+              await client.put(
+                ossFilePath,
+                fileFullPath,
+                {
+                  headers: options.headers || {}
+                }
+              )
+              console.log(`upload complete: ${output}`)
+            }, Number(options.retry || 0))
 
           } else {
             try {
               await client.head(ossFilePath);
               console.log(`${color.gray('files exists')}: ${output}`)
 
-            }  catch (error) {
-              if (error.code === 'NoSuchKey') {
+            } catch (error) {
+              await retry(async () => {
                 await client.put(
                   ossFilePath,
                   fileFullPath,
@@ -109,9 +134,7 @@ export default function vitePluginAliOss (options) {
                   }
                 )
                 console.log(`upload complete: ${output}`)
-              } else {
-                throw new Error(error)
-              }
+              }, Number(options.retry || 0))
             }
           }
         }