SpringMvc上传txt、csv并解析

末蓝、 2022-05-17 11:06 278阅读 0赞

> applicationContext.xml

<bean id="multipartResolver" class="org.springframework.web.multipart.commons.CommonsMultipartResolver">
            
            <property name="defaultEncoding" value="utf-8"></property>
            
            <property name="maxUploadSize" value="50000000"></property>
            
            <property name="maxInMemorySize" value="10240"></property>
        </bean>

> Controller

@ResponseBody
        @RequestMapping(value = "addLabel", method = RequestMethod.POST)
        public ResponseEntity addLabel(@RequestParam(value = "files") MultipartFile files) {
    
            String sourceName = files.getOriginalFilename(); // 原始文件名
            String fileType = sourceName.substring(sourceName.lastIndexOf("."));
            if (files.isEmpty() || StringUtils.isBlank(fileType)) {
                return new ResponseEntity(ResponseEntity.STATUS_FAIL, null, "文件不能为空", null);
            }
            if (!".txt".equals(fileType.toLowerCase()) && !".csv".equals(fileType.toLowerCase())) {
                return new ResponseEntity(ResponseEntity.STATUS_FAIL, null, "文件暂时只支持txt,csv格式", null);
            }
    
            // 存放文件临时路径
            String base = request.getSession().getServletContext().getRealPath("/upload//");  //获取文件上传的路径，在webapp下的upload中
            File file = new File(base);
            if (!file.exists()) {
                file.mkdirs();
            }
    
            // 讲文件上传到临时目录
            String path = base + File.separator + sourceName;
            File upload = new File(path);
            try {
                files.transferTo(upload);
            } catch (IOException e) {
                return new ResponseEntity(ResponseEntity.STATUS_FAIL, null, "文件上传失败，请联系管理员", null);
            }
    
            // 解析文件
            BufferedReader br = null;
            FileReader reader = null;
            try {
                reader = new FileReader(upload);
                br = new BufferedReader(reader);
                String line = "";
                while ((line = br.readLine()) != null) {
                    if (line.trim().length() == 11) {
                        phones.add(line.trim());
                    }
                }
            } catch (Exception e) {
                log.error("上传文件失败", e);
                return new ResponseEntity(ResponseEntity.STATUS_FAIL, null, "读取文件失败，请联系管理员", null);
            } finally {
                try {
                    if (reader != null) {
                        reader.close();
                        reader = null;
                    }
                    if (br != null) {
                        br.close();
                        br = null;
                    }
                } catch (Exception e) {
                    return new ResponseEntity(ResponseEntity.STATUS_FAIL, null, "关闭流失败，请联系管理员", null);
                }
                // 删除临时文件
                if (upload.isFile()) {
                    upload.delete();
                }
            }
            return new ResponseEntity(ResponseEntity.STATUS_OK);
        }