add upyun upload rety

2023-07-23 23:09:13 +08:00
parent 55487dd68f
commit 95878ab1d3
2 changed files with 57 additions and 38 deletions
@@ -4,49 +4,68 @@ package.path = package.path .. ';<SCRIPT_PATH>/?.lua';

 local upyun_upload = {}

+local json = require("cjson")
+
 local strip_path = require("lib.strip_path")
 local Upyun = require('lib.upyun')

-function upyun_upload.upload(filepath, filename)
-    local upyun, err = Upyun:new({
-        user = ngx.var.upyun_operator,
-        passwd = ngx.var.upyun_password,
-        localFilePath = filepath
-    })
+function upyun_upload.upload(filepath, filename, retry)

-    if not upyun then
-        ngx.status = 500
-        ngx.log(ngx.ERR, "failed to initialize upyun: " .. err)
-        return
-    end
-
-    local bucket = ngx.var.upyun_bucket
-    local directory = ngx.var.upyun_directory
-    if not filename then
-        filename = strip_path.strip_path(filepath)
-    end
-
-    ngx.log(ngx.INFO, 'bucket -> ', bucket)
-    ngx.log(ngx.INFO, 'directory -> ', directory)
-
-    local savePath = bucket .. "/" .. directory .. "/" .. filename
-    ngx.log(ngx.INFO, " savePath " , savePath)
-
-    local options = {
-        md5 = true
-    }
-
-    local info, err = upyun:upload_file(savePath, nil, options)
-    if not info then
-        ngx.status = 400
-        ngx.log(ngx.ERR, "failed to upload file : " .. err)
-        return
+    if retry > 3 then
+        ngx.log(ngx.ERR, "failed to upload file : reach max retries")
    else
-        ngx.status = 200
-        local fullpath = ngx.var.upyun_domain .. "/" .. directory .. "/" .. filename
-        ngx.log(ngx.INFO, "SUCCESS UPLOAD -> UPYUN URL -> ", fullpath)
-        ngx.say(fullpath)
+        local upyun, err = Upyun:new({
+            user = ngx.var.upyun_operator,
+            passwd = ngx.var.upyun_password,
+            localFilePath = filepath
+        })
+
+        if not upyun then
+            ngx.status = 500
+            ngx.log(ngx.ERR, "failed to initialize upyun: " .. err)
+            return
+        end
+
+        local bucket = ngx.var.upyun_bucket
+        local directory = ngx.var.upyun_directory
+        if not filename then
+            filename = strip_path.strip_path(filepath)
+        end
+
+        ngx.log(ngx.INFO, 'bucket -> ', bucket)
+        ngx.log(ngx.INFO, 'directory -> ', directory)
+
+        local savePath = bucket .. "/" .. directory .. "/" .. filename
+        ngx.log(ngx.INFO, " savePath " , savePath)
+
+        local options = {
+            md5 = true
+        }
+
+        local info, err = upyun:upload_file(savePath, nil, options)
+        if not info then
+
+            local error_table = json.decode(err)
+            ngx.log(ngx.ERR, '[' .. error_table["code"] .. ']')
+
+            if (error_table["code"] == 40000006)
+            then
+                ngx.status = 400
+                ngx.log(ngx.ERR, "Retry upload file : " .. '[' .. err .. ']')
+                upyun_upload.upload(filepath, filename, retry+1)
+            else
+                ngx.status = 400
+                ngx.log(ngx.ERR, "failed to upload file : " .. '[' .. err .. ']')
+            end
+
+        else
+            ngx.status = 200
+            local fullpath = ngx.var.upyun_domain .. "/" .. directory .. "/" .. filename
+            ngx.log(ngx.INFO, "SUCCESS UPLOAD -> UPYUN URL -> ", fullpath)
+            ngx.say(fullpath)
+        end
    end
+
 end

 return upyun_upload
@@ -21,4 +21,4 @@ local fullpath = file_dir .. "/" .. file_to_upload .. ".png";

 ngx.log(ngx.INFO, 'UPLOAD COMBINED IMAGE -> fullpath -> ', fullpath)

-upyun_upload.upload(fullpath, nil)
+upyun_upload.upload(fullpath, nil, 0)