使用ngx_lua发送文件给服务器

网上搜索了好久lua发送文件的资料，发现openresty的库里面没有post 文件的函数，没办法，只能自己实现

首先，需要了解http发送文件的原理，具体请转到http://blog.csdn.net/terry7/article/details/49464335

下面是实现的代码

步骤一：新建comm.lua，里面写httpPostFile函数

local http = require("resty.http")
local TIMEOUT = 30 --默认超时时间

--生成随机函数，需要放大后处理
function random(n,m)
math.randomseed(os.clock()*math.random(100000,9000000)*math.random(100000,900000))
return math.random(n,m)
end

--生成指定长度的随机字符串
function random_str(len)
local str = ""
for i=1,len,1 do
str = str .. string.char(random(97,122))
end
return str
end

function httpPostFile(url, filepath, timeout)
local httpc = http.new()

--如果不传入timeout，则使用默认时间
local timeout = timeout or TIMEOUT
http:set_timeout(timeout*1000)

--生成随机字符串，用于区分各个文件的内容间隔
local boundary = random_str(12)

local str = ""
--读取文件内容
local f = io.open(filepath, 'r')
local data = f:read("*all")
f:close()

--组建body
 str = str .. "------------------------------" .. boundary .. "\r\n"
str = str .. 'Content-Disposition: form-data; name="' .. filemsg .. '"; filename="' .. file .. '"\r\n'
str = str .. "Content-Type: text/plain\r\n\r\n"
str = str .. data .. "\r\n"
str = str .. "------------------------------" .. boundary .. "--\r\n"

local res, err_ = httpc:request_uri(url, {
method = "POST",
body = str,
ssl_verify = ssl_verify or false,
--设置headers
headers = {
["Accept"] = "*/*",
["Content-Type"] = "multipart/form-data; boundary=----------------------------" .. boundary,

}
})

if not res then
return res, err_
else
if res.status == 200 then
return res.body, err_
else
return nil, err_
end
end
end

步骤二：用java简单的搭建一个文件接收服务器，新建一个test的web项目，

写一个servlet，UpLoadServlet.java

package cn.terry.servlet;

import java.io.File;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

/**
*
* @author administrator
*/
@SuppressWarnings("unchecked")
public class UpLoadServlet extends HttpServlet {
private static final long serialVersionUID = 1L;
private String uploadPath = "D:\\temp"; // 上传文件的目录
File tempPathFile;

// 重写doPost方法，处理事件识别请求
protected void doPost(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
try {
// Create a factory for disk-based file items
DiskFileItemFactory factory = new DiskFileItemFactory();

// Set factory constraints
factory.setSizeThreshold(4096); // 设置缓冲区大小，这里是4kb
factory.setRepository(tempPathFile);// 设置缓冲区目录

// Create a new file upload handler
ServletFileUpload upload = new ServletFileUpload(factory);

// Set overall request size constraint
upload.setSizeMax(4194304); // 设置最大文件尺寸，这里是4MB

List<FileItem> items = upload.parseRequest(request);// 得到所有的文件
Iterator<FileItem> i = items.iterator();
while (i.hasNext()) {
FileItem fi = (FileItem) i.next();
String fileName = fi.getName();
if (fileName != null) {
File fullFile = new File(new String(fi.getName().getBytes(), "utf-8")); // 解决文件名乱码问题
File savedFile = new File(uploadPath, fullFile.getName());
fi.write(savedFile);
}
}
System.out.print("upload succeed");
} catch (Exception e) {

}
}
}

步骤三：修改WebRoot/WEB-INF/web.xml

<?xml version="1.0" encoding="UTF-8"?>
<web-app version="2.5"
xmlns="http://java.sun.com/xml/ns/javaee"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_2_5.xsd"> 
<servlet>
<servlet-name>UploadServlet</servlet-name>
<servlet-class>cn.terry.service.UpLoadServlet</servlet-class>
<load-on-startup>0</load-on-startup>
</servlet>
<servlet-mapping>
<servlet-name>UploadServlet</servlet-name>
<url-pattern>/UploadServlet</url-pattern>
</servlet-mapping>

<welcome-file-list>
<welcome-file>index.jsp</welcome-file>
</welcome-file-list>
</web-app>

代码完成，下面写一个lua测试脚本，看看效果

test_comm.lua

local timeout  = 30
local httpurl = "http://127.0.0.1:8080/test/UploadServlet"
function MAIN()
ngx.say("【begin test_httpPostFile 】")

    local filepath = "/home/nginx_pbcs/openresty/nginx/test/multipart-post.lua"
    local body = httpPostFile(httpurl,filepath,timeout)
    ngx.say("【stop test_httpPostFile 】")

end
MAIN()

执行测试脚本，

tomcat日志打印上传成功



看看文件夹中是否有该文件



OK，文件上传成功。

使用lua构建http协议发送文件成功了