构建multipart/form-data实现文件上传

构建multipart/form-data实现文件上传

通常文件上传都是通过form表单中的file控件，并将form中的content-type设置为multipart/form-data。现在我们通过java来构建这部分请求内容实现文件上传功能。

一、关于multipart/form-data

文件上传本质上是一个POST请求。只不过请求头以及请求内容遵循一定的规则（协议）

• 请求头（Request Headers）中需要设置 Content-Type 为 multipart/form-data; boundary=${boundary}。其中${boundary}分割线，需要在代码中替换，且尽量复杂，不易重复

• 请求正文（Request Body）需要使用在 Header中设置的 ${boundary}来分割当前正文中的FormItem，内容格式如下

  --${boundary}
  Content-Disposition: form-data; name="id"
   
  testCodeUpload
  --${boundary}
  Content-Disposition: form-data; name="file";filename="xx.txt"
  Content-Type: application/octet-stream
   
  {{这里写入文件流}}
  --${boundary}--

  正文开始以前缀+${boundary}开始，以 前缀 +${boundary}+前缀结束。中间每个FormItem 以 前缀+${boundary}开始，以一个空白的换行结束。

二、代码实现

实例代码采用HttpURLConnection实现一个简单POST请求

• 建立http请求，设置基本参数

URL postUrl = new URL(url);
HttpURLConnection conn = (HttpURLConnection) postUrl.openConnection();
conn.setRequestMethod("POST");
conn.setDoInput(true);
conn.setDoOutput(true);
conn.setUseCaches(false);
conn.setRequestProperty("connection", "Keep-Alive");
conn.setRequestProperty("Charset", "UTF-8");

• 添加文件上传必须的请求信息，获取http请输出流

String boundary = "----" + UUID.randomUUID().toString();
conn.setRequestProperty("Content-Type",
"multipart/form-data; boundary=" + boundary);
OutputStream out = conn.getOutputStream();
StringBuilder sb = new StringBuilder();

• 一组FormItem

sb.append(boundaryPrefix);
sb.append(boundary);
sb.append(newLine);
sb.append("Content-Disposition: form-data; name=\"id\"");
sb.append(newLine);
sb.append(newLine);
sb.append("testCodeUpload");
sb.append(newLine);

• 文件写人

sb.append(boundaryPrefix);
sb.append(boundary);
sb.append(newLine);
sb.append("Content-Disposition: form-data; name=\"file\"; filename=\""
+ fileName + "\"");
sb.append("Content-Type: application/octet-stream");
sb.append(newLine);
sb.append(newLine);
out.write(sb.toString().getBytes());
File file = new File(file1);
FileInputStream in = new FileInputStream(file);
byte[] bufferOut = new byte[1024];
int bytes = 0;
while ((bytes = in.read(bufferOut)) != -1) {
out.write(bufferOut, 0, bytes);
}
out.write(newLine.getBytes());
in.close();

 

• 结束标志 前缀+boundary +前缀

byte[] end_data = (newLine + boundaryPrefix + boundary + boundaryPrefix + newLine)
.getBytes();
out.write(end_data);
out.flush();
out.close();

三、文件接收

• 文件接收端通过迭代每个FileItem获取不同的数据

FileItemFactory factory = new DiskFileItemFactory();
ServletFileUpload upload = new ServletFileUpload(factory);
upload.setHeaderEncoding("UTF-8");
try {
  items = upload.parseRequest(request);
} catch (FileUploadException ex) {
  ex.printStackTrace();
  out.println(ex.getMessage());
  return;
}
Iterator<FileItem> itr = items.iterator();
String id = "", fileName = "";
int chunks = 1, chunk = 0;
FileItem tempFileItem = null;
while (itr.hasNext()) {
FileItem item = (FileItem) itr.next();
if (item.getFieldName().equals("id")) {
  id = item.getString();
} else if (item.getFieldName().equals("name")) {
  fileName = new String(item.getString().getBytes("ISO-8859-1"), "UTF-8");
} else if (item.getFieldName().equals("file")) {
  tempFileItem = item;
}

四、总结

 

通过代码实现一遍文件上传，了解其运行机制，解开了以前在写文件上传代码中item.getFieldName().equals("name")等相关判断的疑惑。所以，对于已有的基础代码，还是多看，多写，多实践。

附完整代码

@Test
    public void buildUploadStream() throws IOException {
        String url ="uploadurl";
        file1 = "D:\\test.xls";
        fileName = "test.xls";
        
        String newLine = "\r\n";
        String boundaryPrefix = "--";
        String boundary = "----" + UUID.randomUUID().toString();

        URL postUrl = new URL(url);
        HttpURLConnection conn = (HttpURLConnection) postUrl.openConnection();

        conn.setRequestMethod("POST");
        conn.setDoInput(true);
        conn.setDoOutput(true);
        conn.setUseCaches(false);

        conn.setRequestProperty("connection", "Keep-Alive");
        conn.setRequestProperty("Charset", "UTF-8");
        conn.setRequestProperty("Content-Type",
                "multipart/form-data; boundary=" + boundary);

        OutputStream out = conn.getOutputStream();

        
        StringBuilder sb = new StringBuilder();

        sb.append(boundaryPrefix);
        sb.append(boundary);
        sb.append(newLine);

        sb.append("Content-Disposition: form-data; name=\"id\"");
        sb.append(newLine);
        sb.append(newLine);
        sb.append("testCodeUpload");
        sb.append(newLine);

        sb.append(boundaryPrefix);
        sb.append(boundary);
        sb.append(newLine);

        sb.append("Content-Disposition: form-data; name=\"name\"");
        sb.append(newLine);
        sb.append(newLine);
        sb.append(fileName);
        sb.append(newLine);

        sb.append(boundaryPrefix);
        sb.append(boundary);
        sb.append(newLine);

        sb.append("Content-Disposition: form-data; name=\"file\"; filename=\""
                + fileName + "\"");
        sb.append("Content-Type: application/octet-stream");
        sb.append(newLine);
        sb.append(newLine);

        out.write(sb.toString().getBytes());
        
        File file = new File(file1);
        FileInputStream in = new FileInputStream(file);
        byte[] bufferOut = new byte[1024];
        int bytes = 0;
        while ((bytes = in.read(bufferOut)) != -1) {
            out.write(bufferOut, 0, bytes);
        }
        out.write(newLine.getBytes());
        in.close();
        byte[] end_data = (newLine + boundaryPrefix + boundary + boundaryPrefix + newLine)
                .getBytes();
        out.write(end_data);
        out.flush();
        out.close();

        BufferedReader reader = new BufferedReader(new InputStreamReader(
                conn.getInputStream()));
        String line = null;
        while ((line = reader.readLine()) != null) {
            System.out.println(line);
        }
    }