李勇
140 lines
4.6 KiB
Lua
140 lines
4.6 KiB
Lua
local upload = require "upload"
|
|
local uuid = require "uuid"
|
|
|
|
local chunk_size = 5 -- should be set to 4096 or 8192
|
|
-- for real-world settings
|
|
|
|
local form, err = upload:new(chunk_size)
|
|
|
|
local output_image_filename_uuid = uuid.generate();
|
|
|
|
|
|
if not form then
|
|
ngx.log(ngx.ERR, "failed to new upload: ", err)
|
|
ngx.exit(500)
|
|
end
|
|
|
|
form:set_timeout(1000) -- 1 sec
|
|
|
|
-- String split
|
|
string.split = function(s, p)
|
|
local rt= {}
|
|
string.gsub(s, '[^'..p..']+', function(w) table.insert(rt, w) end )
|
|
return rt
|
|
end
|
|
|
|
-- Get file trim
|
|
string.trim = function(s)
|
|
return (s:gsub("^%s*(.-)%s*$", "%1"))
|
|
end
|
|
|
|
-- Upload file save path config in nginx.conf
|
|
local saveRootPath = ngx.var.store_dir
|
|
|
|
local fileToSave
|
|
local ret_save = false
|
|
local upload_image_filename
|
|
local save_image_filename
|
|
|
|
local upload_image_table = {}
|
|
local upload_image_index = 1
|
|
|
|
while true do
|
|
local typ, res, err = form:read()
|
|
if not typ then
|
|
ngx.log(ngx.ERR, "failed to read: ", err)
|
|
return
|
|
end
|
|
if typ == "header" then
|
|
-- 开始读取 http header
|
|
-- 解析出本次上传的文件名
|
|
local key = res[1]
|
|
local value = res[2]
|
|
if key == "Content-Disposition" then
|
|
-- 解析出本次上传的文件名
|
|
-- form-data; name="testFileName"; filename="testfile.txt"
|
|
local kvlist = string.split(value, ';')
|
|
for _, kv in ipairs(kvlist) do
|
|
local seg = string.trim(kv)
|
|
if seg:find("filename") then
|
|
local kvfile = string.split(seg, "=")
|
|
upload_image_filename = string.sub(kvfile[2], 2, -2)
|
|
|
|
ngx.log(ngx.INFO,"Upload origin image is ", upload_image_filename)
|
|
|
|
if upload_image_filename then
|
|
-- Save file name with generate uuid
|
|
save_image_filename = uuid:generate()
|
|
save_image_filename = save_image_filename .. '.' .. upload_image_filename:match(".+%.(%w+)$")
|
|
fileToSave = io.open(saveRootPath .. save_image_filename, "w+")
|
|
if not fileToSave then
|
|
ngx.log(ngx.ERR, " Error: Failed to open file ", save_image_filename)
|
|
return
|
|
else
|
|
ngx.log(ngx.INFO, "Open file ok ", save_image_filename)
|
|
|
|
upload_image_table[upload_image_index] = saveRootPath .. save_image_filename .. " "
|
|
ngx.log(ngx.INFO, "Insert upload image table "..upload_image_index.." with "..upload_image_table[upload_image_index])
|
|
|
|
upload_image_index = upload_image_index + 1
|
|
end
|
|
break
|
|
end
|
|
end
|
|
end
|
|
end
|
|
elseif typ == "body" then
|
|
-- 开始读取 http body
|
|
if fileToSave then
|
|
fileToSave:write(res)
|
|
end
|
|
elseif typ == "part_end" then
|
|
-- 文件写结束，关闭文件
|
|
if fileToSave then
|
|
fileToSave:close()
|
|
fileToSave = nil
|
|
end
|
|
|
|
ret_save = true
|
|
elseif typ == "eof" then
|
|
-- 文件读取结束
|
|
break
|
|
else
|
|
ngx.log(ngx.INFO, "do other things")
|
|
end
|
|
end
|
|
if ret_save then
|
|
ngx.log(ngx.INFO, "Show all upload files\n")
|
|
|
|
-- Show all upload files
|
|
upload_image_index = upload_image_index -1
|
|
for i = 1, upload_image_index do
|
|
ngx.log(ngx.INFO, "Save upload file is : " .. upload_image_table[i])
|
|
end
|
|
|
|
image_append_coroutine = coroutine.create(
|
|
function ()
|
|
-- Append images
|
|
-- ngx.log(ngx.INFO,"Upload image number is : " , upload_image_index )
|
|
local append_images_path
|
|
append_images_path = table.concat(upload_image_table)
|
|
ngx.log(ngx.INFO, "Input images list : " , append_images_path)
|
|
|
|
-- local append_output_image = saveRootPath .. uuid.generate()..".jpg ";
|
|
local append_output_image = saveRootPath .. output_image_filename_uuid..".jpg ";
|
|
ngx.log(ngx.INFO, "Output image is : "..append_output_image)
|
|
|
|
-- Debug
|
|
ngx.log(ngx.INFO, '/usr/local/ImageMagick/bin/magick convert -append '..append_images_path..' '..append_output_image)
|
|
|
|
local call_imagemagick_cmd = io.popen('/usr/local/ImageMagick/bin/magick convert -append '..append_images_path..' '..append_output_image)
|
|
local imagemagick_cmd_replay = call_imagemagick_cmd:read("*all")
|
|
ngx.log(ngx.INFO, imagemagick_cmd_replay)
|
|
|
|
|
|
end
|
|
)
|
|
|
|
coroutine.resume(image_append_coroutine)
|
|
|
|
end |