Java中用Socket实现HTTP文件上传实例

我想做过web开发的程序员大部分都做过文件上传的功能，大多数时候我们都是借助于commons-fileupload这样的jar包实现的。下面我试着通过读取Socket的输入流来实现一个文件上传的功能。

在做文件上传之前我们需要先了解一下HTTP POST的附件上传协议。HTTP附件上传协议是RFC1876协议，RFC1876协议是在HTTP协议的基础上为INPUT标签增加了file属性，同时限定了Form的method必须为POST，ENCTYPE必须为multipart/form-data。RFC1867协议对HTTP头作了适当地变更，content-type头由以前的：content-type:application/x-www-form-urlencoded变为content-type:multipart/form-data;+空格+boundary=字符串。RFC1867增加了文件上传得功能，而上传文件内容自然也会被加入到HTTP的实体中。现在因为既有HTTP一般的参数实体，又有上传文件的实体，所以用boundary把每种实体进行了分割。具体的看下图：

接下来就开始我们的代码部分吧。

我在前面的文章中写过创建一个自己的Web服务器，现在我们的重点要放在对socket的输入流的解析中。具体代码如下：

public void parseRequest() {
  LineNumberReader br = new LineNumberReader(new InputStreamReader(inputStream));
  StringBuffer sb = new StringBuffer();
  String str = null;
  try {
    //读取请求行
    String requestLine = br.readLine();
    if (!StringUtils.isEmpty(requestLine)) {
      sb.append(requestLine);
      String[] reqs = requestLine.split(" ");
      if (reqs != null && reqs.length > 0) {
        if ("GET".equals(reqs[0])) {
          method = "GET";
        } else {
          method = "POST";
        }
      }
    }
    //读取请求头
    while ((str = br.readLine()) != null) {
      if ("".equals(str)) {
        break;
      }
      if (!StringUtils.isEmpty(str)) {
        if (str.indexOf(":") > 0) {
          String[] strs = str.split(":");
          headers.put(strs[0].toLowerCase(), strs[1].trim());
        }
      }
      sb.append(str).append("\n");
    }
    //POST请求，Content-type为 multipart/form-data
    String contentType = null;
    if ("POST".equals(method) && ((contentType = headers.get("content-type")) != null
        && headers.get("content-type").startsWith("multipart/form-data"))) {
      //文件上传的分割位 这里只处理单个文件的上传
      String boundary = contentType.substring(contentType.indexOf("boundary") +
          "boundary=".length());
      //解析消息体
      while ((str = br.readLine()) != null) {
        //解析结束的标记
        do {
          //读取boundary中的内容
          //读取Content-Disposition
          str = br.readLine();
          //说明是文件上传
          if (str.indexOf("Content-Disposition:") >= 0 && str.indexOf("filename") > 0) {
            str = str.substring("Content-Disposition:".length());
            String[] strs = str.split(";");
            String fileName = strs[strs.length - 1].replace("\"", "").split("=")[1];
            System.out.println("fileName = " + fileName);
            //这一行是Content-Type
            br.readLine();
            //这一行是换行
            br.readLine();
            //正式去读文件的内容
            BufferedWriter bw = null;
            try {
              bw = new BufferedWriter(new OutputStreamWriter(new
                  FileOutputStream("G:\\LearnVideo\\fileLoad" +
                  File.separator + fileName), "UTF-8"));
              while (true) {
                str = br.readLine();
                if (str.startsWith("--" + boundary)) {
                  break;
                }
                bw.write(str);
                bw.newLine();
              }
              bw.flush();
            } catch (Exception e) { 

            } finally {
              if (bw != null) {
                bw.close();
              }
            }
          }
          if (str.indexOf("Content-Disposition:") >= 0) {
            str = str.substring("Content-Disposition:".length());
            String[] strs = str.split(";");
            String name = strs[strs.length - 1].replace("\"", "").split("=")[1];
            br.readLine();
            StringBuilder stringBuilder = new StringBuilder();
            while (true) {
              str = br.readLine();
              if (str.startsWith("--" + boundary)) {
                break;
              }
              stringBuilder.append(str);
            }
            parameters.put(name, stringBuilder.toString());
          }
        } while (("--" + boundary).equals(str));
        //解析结束
        if (str.equals("--" + boundary + "--")) {
          break;
        }
      }
    }
    //System.out.println(sb.toString());
    //获取URI
    uri = StringUtils.parserUri(sb.toString(), " ");
    int flag = -1;
    //说明有参数
    if ((flag = uri.indexOf('?')) >= 0) {
      String oldUri = uri;
      uri = uri.substring(0,flag);
      String parameterPath = oldUri.substring(flag+1);
      String[] parameter = parameterPath.split("&");
      if (parameter != null && parameter.length > 0) {
        for (int i = 0; i < parameter.length; i++) {
          String str1 = parameter[i];
          if((flag = str1.indexOf('=')) >= 0){
            String key = str1.substring(0,flag);
            String value = str1.substring(flag+1);
            parameters.put(key,value);
          }else{
            parameters.put(str,null);
          }
        }
      }
    }
  } catch (IOException e) {
    e.printStackTrace();
  }
}

我们启动自己创建的Web服务器，然后在浏览器中输入：http://localhost:8004/static/uploadPage.html，页面如下：

选择我们要上次的文件，然后点击上传按钮，我们会发现我们的功能已经被上传到G:\LearnVideo\fileLoad这个目录下了。示例如下：

完整的代码请从这里下载：FullStackTraining_jb51.rar

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。