84 lines
2.4 KiB
Lua
84 lines
2.4 KiB
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')
|
|
|
|
--- @return string|nil public_url 成功返回又拍完整 URL；失败返回 nil（已打日志）
|
|
function upyun_upload.upload_return_url(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")
|
|
return nil
|
|
end
|
|
|
|
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 nil
|
|
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, upl_err = upyun:upload_file(savePath, nil, options)
|
|
if not info then
|
|
local error_table = json.decode(upl_err)
|
|
ngx.log(ngx.INFO, "Upyun Upload File Error: " .. upl_err)
|
|
ngx.log(ngx.ERR, '[' .. error_table["code"] .. ']')
|
|
|
|
if (error_table["code"] == 40000006)
|
|
then
|
|
ngx.status = 400
|
|
ngx.log(ngx.ERR, "Retry upload file : " .. '[' .. upl_err .. ']')
|
|
return upyun_upload.upload_return_url(filepath, filename, retry + 1)
|
|
else
|
|
ngx.status = 400
|
|
ngx.log(ngx.ERR, "failed to upload file : " .. '[' .. upl_err .. ']')
|
|
return nil
|
|
end
|
|
end
|
|
|
|
ngx.status = 200
|
|
local fullpath = ngx.var.upyun_domain .. "/" .. directory .. "/" .. filename
|
|
ngx.log(ngx.INFO, "SUCCESS UPLOAD -> UPYUN URL -> ", fullpath)
|
|
return fullpath
|
|
end
|
|
|
|
function upyun_upload.upload(filepath, filename, retry)
|
|
local url = upyun_upload.upload_return_url(filepath, filename, retry)
|
|
if url then
|
|
ngx.say(url)
|
|
end
|
|
end
|
|
|
|
return upyun_upload
|