resty_functions/scripts/lib/upyun_upload.lua

-- 指定模块引用目录，否则无法加载同目录下的其他文件
--package.path = package.path .. ';/usr/local/opt/openresty/nginx/scripts/?.lua';
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, retry)

    if retry == nil then
        retry = 1
    end

    if retry > 3 then
        ngx.log(ngx.ERR, "failed to upload file : reach max retries")
    else
        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.INFO, "Upyun Upload File Error: " .. 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