src/main/java/com/hwua/erhai/controller/DoRegisterServlet.java

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

package com.hwua.erhai.controller;

import com.google.common.base.Preconditions;
import com.hwua.erhai.entity.User;
import com.hwua.erhai.service.IUserService;
import com.hwua.erhai.service.impl.MockUserService;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.lang3.StringUtils;

import javax.servlet.*;
import javax.servlet.http.*;
import javax.servlet.annotation.*;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@WebServlet(name = "DoRegisterServlet", value = "/doRegister")
public class DoRegisterServlet extends HttpServlet {

    IUserService userService = new MockUserService();
    //上传文件存储目录
    private static final String  UPLOAD_DIRECTORY ="upload";
    //上传配置
    private static final int MEMORY_THRESHOLD = 1024*1024*3;//3MB
    private static final int MAX_FILE_SIZE = 1024*1024*40;//40MB
    private  static final int MAX_REQUEST_SIZE = 1024*1024*50;//50MB
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

    }

    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

        //检测是否为多媒体上传
        if(!ServletFileUpload.isMultipartContent(request)){
            //如果不是则停止
            PrintWriter writer = response.getWriter();
            writer.println("Error:表单必须包含enctype=multipart/from-data");
            writer.flush();
            return;
        }
        //配置上传参数
        DiskFileItemFactory factory = new  DiskFileItemFactory();
        //设置内存临界值 -超过后将产生临时文件并存储于临时目录中
        factory.setSizeThreshold(MEMORY_THRESHOLD);
        //设置临时存储目录
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));

        ServletFileUpload upload = new ServletFileUpload(factory);
        //设置最大文件上传值
        upload.setFileSizeMax(MAX_FILE_SIZE);

        //设置最大请求值
        upload.setSizeMax(MAX_REQUEST_SIZE);