文件上传和下载（可批量上传）——Spring（三）

在文件上传和下载（可批量上传）——Spring（二）的基础上，发现了文件下载时，只有在Chrome浏览器下文件名正常显示，还有发布到服务器后，不能上传到指定的文件夹目录，如上传20160310.txt文件，上传到正是服务器后，并没有在upload文件夹下，而是以upload/20160310.txt这样的文加名的文件。针对以上问题，进行了修正。最后一个错误是，在显示文件列表的时候如果对应项目下没有文件夹会报空指针异常。

处理空指针异常的。

处理文件名中文显示正常和处理无法上传到正确的文件夹

完整代码如下

package com.tommy.business;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by OnlyOne on 2016/3/4.
 */
@Controller
@RequestMapping("/background/uploadAndDownload/")
public class UploadAndDownloadController {
    /**
     *
     * @param req
     * @param resp
     * @throws ServletException
     * @throws IOException
     * @throws FileUploadException
     */
    @RequestMapping("uploadFile")
    public String uploadFile(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException, FileUploadException {
        /*//防止中文乱码，与页面字符集一致
        req.setCharacterEncoding("UTF-8");*/
        //得到上传文件的保存目录，将上传的文件存放于WEB-INF目录下，不允许外界直接访问，保证上传文件的安全
        String savePath = req.getServletContext().getRealPath("/upload");
//                            this.getServletContext().getRealPath("/WEB-INF/upload");
        //创建保存目录的文件
        File saveFile = new File(savePath);
        //判断保存目录文件是否存在,不存在则创建一个文件夹
        if(!saveFile.exists()){
            System.out.println("文件目录创建中。。。");
            saveFile.mkdir();
        }
        //消息提示
        //将req转换成Spring的request
        MultipartHttpServletRequest multipartHttpServletRequest = (MultipartHttpServletRequest) req;
        //获取上传文件
        List<MultipartFile> list = multipartHttpServletRequest.getFiles("file");
        /*//获取普通输入项的数据
        String map = multipartHttpServletRequest.getParameter("username");
        System.out.println(map);*/
        for(MultipartFile multipartFile: list){
            if(!multipartFile.isEmpty()){
                //得到上传的文件名称
                String fileName = multipartFile.getOriginalFilename();
                System.out.println("文件名是："+ fileName);
                if(fileName == null || fileName.trim().equals("")){
                    continue;
                }
                //获取item中的上传输入流
                BufferedInputStream bis = new BufferedInputStream(multipartFile.getInputStream());
                //创建一个文件输出流
                BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(savePath + "/" + fileName));
                //创建一个缓冲区
                byte[] buffer = new byte[1024*8];
                //循环将缓冲输入流读入到缓冲区当中
                while(true){
                    //循环将缓冲输入流读入到缓冲区当中
                    int length = bis.read(buffer);
                    //判断是否读取到文件末尾
                    if(length == -1){
                        break;
                    }
                    //使用BufferedOutputStream缓冲输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
                    bos.write(buffer,0,length);
                }
                //关闭输入流
                bis.close();
                //关闭输出流
                bos.close();
            }
        }
            return "redirect:listFile.html";
    }

    /**
     * @ClassName: ListFileServlet
     * @Description: 列出Web系统中所有下载文件
     * @param req
     * @param resp
     */
    @RequestMapping("listFile")
    public String listFile(HttpServletRequest req, HttpServletResponse resp) {
        String flag = req.getParameter("flag");
        //获取上传文件的目录
        String uploadFilePath = req.getServletContext().getRealPath("/upload");
        //创建保存目录的文件
        File saveFile = new File(uploadFilePath);
        //判断保存目录文件是否存在,不存在则创建一个文件夹
        if(!saveFile.exists()){
            System.out.println("文件目录创建中。。。");
            saveFile.mkdir();
        }
        //存储要下载的文件名
        Map<String, String> fileNameMap = new HashMap<String, String>();
        //地鬼遍历filePath目录下的所有文件和目录，将文件的文件名存储到map集合中
        getListfile(new File(uploadFilePath), fileNameMap);
        if(flag != null){
            req.setAttribute("flag",flag);
        }
        req.setAttribute("fileNameMap", fileNameMap);
        return "/background/uploadAndDownload/resourcesList";
    }

    /**
     *
     * @param file 即代表一个文件，也代表一个文件目录
     * @param map  存储文件名的Map集合
     * @Method: listfile
     * @Description: 递归遍历指定目录下的所有文件
     */
    private void getListfile(File file, Map<String, String> map) {
        //如果file代表的不是一个文件，而是一个目录
        if (!file.isFile()) {
            //列出该目录下的所有文件和目录
            File files[] = file.listFiles();
            //遍历files[]数组
            for (File f : files) {
                //递归
                getListfile(f, map);
            }
        } else {
            /**
             * 处理文件名，上传后的文件是以uuid_文件名的形式去重新命名的，去除文件名的uuid_部分
             file.getName().indexOf("_")检索字符串中第一次出现"_"字符的位置，如果文件名类似于：9349249849-88343-8344_阿_凡_达.avi
             那么file.getName().substring(file.getName().indexOf("_")+1)处理之后就可以得到阿_凡_达.avi部分
             */
            //String realName = file.getName().substring(file.getName().indexOf("_") + 1);
            //file.getName()得到的是文件的原始名称，这个名称是唯一的，因此可以作为key，realName是处理过后的名称，有可能会重复
            map.put(file.getName(), file.getName());
        }
    }

    /**
     * 下载文件
     * @param req
     * @param resp
     * @throws ServletException
     * @throws IOException
     */
    @RequestMapping("downLoadFile")
    public void downLoadFile(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        /*//得到要下载的文件名
        req.setCharacterEncoding("UTF-8");*/
        String fileName = req.getParameter("fileName");
//        fileName = new String(fileName.getBytes("iso8859-1"),"UTF-8");
        //上传的文件都是保存在/WEN-INF/upload目录党徽宗
        String fileSavePath = req.getServletContext().getRealPath("/upload");
        //得到要下载的文件
        File file = new File(fileSavePath + "//" + fileName);
        //如果文件不存在
        if(!file.exists()){
            req.setAttribute("message", "资源已被删除！");
            req.getRequestDispatcher("/message.jsp").forward(req, resp);
        }
        //处理文件名
        String realName = fileName.substring(fileName.indexOf("_")+1);
        String agent = req.getHeader("user-agent");
        resp.setContentType("application/x-download");
        //处理下载时文件名乱码，针对不同浏览器做处理
        if(agent.contains("Firefox")){
            realName = new String(realName.getBytes("GB2312"),"ISO-8859-1");
            //resp.addHeader("Content-Disposition","attachment;filename="+ new String(realName.getBytes("GB2312"),"ISO-8859-1"));
        }else {
            realName = URLEncoder.encode(realName, "UTF-8");
            //resp.addHeader("Content-Disposition","attachment;filename=" + URLEncoder.encode(realName, "UTF-8"));
        }
        //设置响应头，控制浏览器下载该文件
        resp.setHeader("content-disposition", "attachment;filename="+realName);
        //读取要下载的文件，保存到文件输入流
        BufferedInputStream bis = new BufferedInputStream(new FileInputStream(fileSavePath + "/" + fileName));
        //以前的写法，本地可以，发布到服务器后就不能上传到指定的文件夹而是已upload\***的形式作为文件名，应该是转义符"\\"在linux转义的不是"/"
        // BufferedInputStream bis = new BufferedInputStream(new FileInputStream(fileSavePath + "\\" + fileName));
        //创建输出流
        BufferedOutputStream bos = new BufferedOutputStream(resp.getOutputStream());
        //创建一个缓冲区
        byte[] buffer = new byte[1024*8];
        //循环将缓冲输入流读入到缓冲区当中
        while(true){
            //循环将缓冲输入流读入到缓冲区当中
            int length = bis.read(buffer);
            //判断是否读取到文件末尾
            if(length == -1){
                break;
            }
            //使用BufferedOutputStream缓冲输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
            bos.write(buffer,0,length);
        }
        //关闭文件输入流
        bis.close();
        //刷新此输入流并强制写出所有缓冲的输出字节数
        bos.flush();
        //关闭文件输出流
        bos.close();
    }

}