文件上传和下载

水深无声 2022-04-10 06:36 654阅读 0赞

/\*\*  
     \* 实现文件上传  
     \*\*/  
    @RequestMapping(value = "/fileUpload",method = RequestMethod.POST)  
    @ResponseBody  
    public ResultDO fileurl(@RequestParam("fileUpload") MultipartFile fileUpload,HttpSession session )\{  
        ResultDO resultDO = new ResultDO();  
        if(fileUpload.isEmpty())\{  
            resultDO.setCode(StatusCode.HTTP\_FAILURE);  
            resultDO.setMsg("上传文件不能为空");  
            return null;  
        \}  
        //获取文件名称  
        String fileName = fileUpload.getOriginalFilename();  
        session.setAttribute("filename",fileName);  
        //文件大小  
        int size=(int) fileUpload.getSize();

String suffix = fileName.substring(fileName.lastIndexOf(".")+1);//后缀  
       // fileName = productlogoName+"."+suffix;  
//        fileName = productlogoName+".png";  
//        String savePath = Constants.FILE\_PATH;  
//        本地路径  
        String savaPath="/home/ubuntu/bankground/products";  
        File filesave = new File(savaPath);  
        //判断上传文件的保存目录是否存在  
        if (!filesave.exists() && !filesave.isDirectory()) \{  
            //创建目录  
            filesave.mkdirs();  
        \}  
        String fileurl=savaPath;

try \{  
//            //获取item中的上传文件的输入流  
            InputStream in = fileUpload.getInputStream();  
            //创建一个文件输出流  
            // FileOutputStream out = new FileOutputStream(savePath + "\\\\" + file\_name);  
            FileOutputStream out = new FileOutputStream("/home/ubuntu/bankground/products/"+fileName);  
            //创建一个缓冲区  
            byte buffer\[\] = new byte\[1024\];  
            //判断输入流中的数据是否已经读完的标识  
            int len = 0;  
            //循环将输入流读入到缓冲区当中，(len=in.read(buffer))>0就表示in里面还有数据  
            while((len=in.read(buffer))>0)\{  
                //使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\\\" + filename)当中  
                out.write(buffer, 0, len);  
            \}  
            //关闭输入流  
            in.close();  
            //关闭输出流  
            out.close();  
        \} catch (Exception e) \{  
            e.printStackTrace();  
            resultDO.setCode(StatusCode.HTTP\_FAILURE);  
            resultDO.setMsg("上传失败");  
            return null;  
        \}  
        resultDO.setData("上传成功");  
        return resultDO;  
    \}

/\*\*  
 \* 下载文件  
 \* @return  
 \*/  
@RequestMapping(value = "/file",method = RequestMethod.GET)  
@ResponseBody  
public ResultDO file(int id, HttpServletResponse resp) throws IOException \{  
    ResultDO resultDO=new ResultDO();  
    try\{  //根据文件id提取文件信息  
        FileDO fileDO=fileService.select(id);  
        //获取文件路径  
        String filename=fileDO.getFileurl();

//获取文件路径  
        String fileurl=fileDO.getFileurl();  
        //截取文件名  
        String sb=filename;  
        //name是从文件路径第33位开始截取 剩下的文件名  
        String name = sb.substring(33);

FileInputStream input = new FileInputStream(fileurl);  
        //设置响应头，对文件进行url编码  
        filename = URLEncoder.encode(name, "UTF-8");  
        resp.setHeader("Content-Disposition", "attachment;filename="+filename);  
        resp.setContentLength(input.available());

ServletOutputStream output = resp.getOutputStream();  
        IOUtils.copy(input,output);  
        input.close();  
        resultDO.setData("下载");

\}catch (Exception e)\{  
        e.printStackTrace();  
        resultDO.setMsg("文件不存在,请联系总部");  
        return resultDO;  
    \}  
    return resultDO;  
\}