SpringBoot MultipartFile 监控上传文件进度

淩亂°似流年 2022-05-13 06:48 899阅读 0赞

＃ SpringBoot MultipartFile监控上传文件进度>引用块内容

> 在一次项目中需要监控文件上传的进度。将进度监控到之后计算百分比，存入session中session需要配置实时生效，不然监控不到进度，此进度在内网环境表现很好，公网环境因为前端body较大，接收时间较长，前端会有一段时间的0% ,所以写了一篇分块上传的文档，需要的可以留言找我要

`session实时生效`
     spring
        session:
            store-type: redis
            redis:
              flush-mode: immediate

package com.zichan360.common.Progress;
    
    import com.zichan360.common.util.DateUtil;
    import lombok.extern.slf4j.Slf4j;
    import org.apache.commons.fileupload.ProgressListener;
    import org.springframework.stereotype.Component;
    
    import javax.servlet.http.HttpSession;
    
    /** * @ClassName: FileUploadProgressListener * @ProjectName CollectorClub * @Author zhaohp * @Date 2018/9/111:10 * @Description: */
    
    
    @Slf4j
    @Component
    public class FileUploadProgressListener implements ProgressListener { 
    
        private HttpSession session;
    
        public void setSession(HttpSession session) {
            this.session = session;
            log.info("upload_percent 0--------------------------------------" + DateUtil.getTime());
            session.setAttribute("upload_percent", 0);
        }
    
        @Override
        public void update(long bytesRead, long contentLength, int items) {
            int percent = (int) (bytesRead * 100.0 / contentLength);
            session.setAttribute("upload_percent", percent);
            log.info(percent + "--------------------------------------" + DateUtil.getTime());
        }
    
    }

package com.zichan360.common.Progress;
    
    import org.apache.commons.fileupload.FileItem;
    import org.apache.commons.fileupload.FileUpload;
    import org.apache.commons.fileupload.FileUploadException;
    import org.apache.commons.fileupload.servlet.ServletFileUpload;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.multipart.MultipartException;
    import org.springframework.web.multipart.commons.CommonsMultipartResolver;
    
    import javax.servlet.http.HttpServletRequest;
    import java.util.List;
    
    
    /* * * @ClassName: CustomMultipartResolver * @ProjectName CollectorClub * @Author zhaohp * @Date 2018/9/111:11 * @Description: */
    
    
    public class CustomMultipartResolver extends CommonsMultipartResolver {
    
        @Autowired
        private FileUploadProgressListener listener;
    
    
        @Override
        protected MultipartParsingResult parseRequest(HttpServletRequest request) throws MultipartException {
            String encoding = determineEncoding(request);
            FileUpload fileUpload = prepareFileUpload(encoding);
            //fileUpload.setFileSizeMax(1024 * 1024 * 500);// 单个文件最大500M
            //fileUpload.setSizeMax(1024 * 1024 * 500);// 一次提交总文件最大500M
            //向文件上傳進度監視器設置session用於存儲上傳進度
            listener.setSession(request.getSession());
            //將文件上傳進度監視器加入到fileUpload中
            fileUpload.setProgressListener(listener);
    
            try {
                List<FileItem> fileItems = ((ServletFileUpload) fileUpload).parseRequest(request);
                return parseFileItems(fileItems, encoding);
            } catch (FileUploadException e) {
                e.printStackTrace();
            }
            return null;
        }
    }

package com.zichan360;
    import com.zichan360.common.Progress.CustomMultipartResolver;
    import com.zichan360.config.datasources.DynamicDataSourceConfig;
    import org.springframework.boot.Banner;
    import org.springframework.boot.SpringApplication;
    import org.springframework.boot.autoconfigure.SpringBootApplication;
    import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
    import org.springframework.boot.autoconfigure.web.MultipartAutoConfiguration;
    import org.springframework.boot.builder.SpringApplicationBuilder;
    import org.springframework.boot.web.servlet.ServletComponentScan;
    import org.springframework.boot.web.support.SpringBootServletInitializer;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Import;
    import org.springframework.web.multipart.MultipartResolver;
    import org.springframework.web.servlet.config.annotation.EnableWebMvc;
    
    @SpringBootApplication(exclude = {DataSourceAutoConfiguration.class,MultipartAutoConfiguration.class})
    @EnableWebMvc
    @ServletComponentScan
    @Import({DynamicDataSourceConfig.class})
    public class SpringbootApplication extends SpringBootServletInitializer {
    
        public static void main(String[] args) {
            SpringApplication application = new SpringApplication(SpringbootApplication.class);
            /* * Banner.Mode.OFF:关闭; * Banner.Mode.CONSOLE:控制台输出，默认方式; * Banner.Mode.LOG:日志输出方式; */
            application.setBannerMode(Banner.Mode.CONSOLE);
            //启动项目
            application.run(args);
        }
    
        @Override
        protected SpringApplicationBuilder configure(SpringApplicationBuilder application) {
            return application.sources(SpringbootApplication.class);
        }
    
        @Bean(name = "multipartResolver")
        public MultipartResolver multipartResolver() {
            return new CustomMultipartResolver();
        }
    }

/** * 获取实时长传进度 * * @param request * @return int percent */
        @GetMapping("getPercent")
        public Result<Integer> getUploadPercent(HttpServletRequest request) {
            HttpSession session = request.getSession();
            int percent = session.getAttribute("upload_percent") == null ? 0 : (int) session.getAttribute("upload_percent");
            return ResultUtil.success(percent);
        }
    
    
        /** * 重置上传进度 前端调用进度之前调用此接口 * * @param request * @return void */
        @PostMapping("resetPercent")
        public Result resetPercent(HttpServletRequest request) {
            HttpSession session = request.getSession();
            session.setAttribute("upload_percent", 0);
            return ResultUtil.success();
        }