文件压缩上传Utils

文件压缩上传utils

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Date;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

public class FileZipUtil {
    /**
     * 上传压缩文件到指定目录
     * @param sealPfxFile
     * @param path 路径
     * @return
     * @throws IOException
     */
    public static String upload(File sealPfxFile, String path) throws IOException {
        long startTime=System.currentTimeMillis();
        String fileName = sealPfxFile.getName();
        File file = new File(path);
        try {
            if (!file.exists()) {
                file.mkdirs();// 创建一个已时间戳命名的文件夹
            }
            // 获取输出流
            OutputStream os = new FileOutputStream(path + fileName);
          //  InputStream is = sealPfxFile.getInputStream();
            InputStream is =new FileInputStream(sealPfxFile);
            
            int temp;
            // 一个一个字节的读取并写入
            while ((temp = is.read()) != (-1)) {
                os.write(temp);
            }
            os.flush();
            os.close();
            is.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        
        long endTime=System.currentTimeMillis(); //获取结束时间
        
        System.out.println("上传时间： "+(startTime-endTime)+"ms");
        return path + fileName;
    }
    
    /**
     * 解压zip文件到指定的目录
     * @param file 要解压的zip文件全路径
     * @param outputDir 要解压到某个指定的目录下
     * @throws IOException
     */
    public static void unZip(String file, String outputDir) throws IOException {
        long startTime=System.currentTimeMillis();
        ZipFile zipFile = null;
        try {
            Charset gbk = Charset.forName("gbk");
            zipFile = new ZipFile(file, gbk);
            createDirectory(outputDir, null);// 创建输出目录
            Enumeration<?> enums = zipFile.entries();
            while (enums.hasMoreElements()) {
                ZipEntry entry = (ZipEntry)enums.nextElement();
                if (entry.isDirectory()) {// 是目录
                    createDirectory(outputDir, entry.getName());// 创建空目录
                } else {// 是文件
                    File tmpFile = new File(outputDir + "/" + entry.getName());
                    createDirectory(tmpFile.getParent() + "/", null);// 创建输出目录
                    InputStream in = null;
                    OutputStream out = null;
                    try {
                        in = zipFile.getInputStream(entry);;
                        out = new FileOutputStream(tmpFile);
                        int length = 0;
                        byte[] b = new byte[2048];
                        while ((length = in.read(b)) != -1) {
                            out.write(b, 0, length);
                        }
                    } catch (IOException ex) {
                        throw ex;
                    } finally {
                        if (in != null)
                            in.close();
                        if (out != null)
                            out.close();
                    }
                }
            }
        } catch (IOException e) {
            throw new IOException("解压缩文件出现异常", e);
        } finally {
            try {
                if (zipFile != null) {
                    zipFile.close();
                }
            } catch (IOException ex) {
                throw new IOException("关闭zipFile出现异常", ex);
            }
        }
        long endTime=System.currentTimeMillis();
        System.out.println("解压时间： "+(startTime-endTime)+"ms");
    }
    
    /**
     * 创建文件目录
     * @param outputDir
     * @param subDir
     */
    public static void createDirectory(String outputDir, String subDir) {
        File file = new File(outputDir);
        if (!(subDir == null || subDir.trim().equals(""))) {// 子目录不为空
            file = new File(outputDir + "/" + subDir);
        }
        if (!file.exists()) {
            if (!file.getParentFile().exists())
                file.getParentFile().mkdirs();
            file.mkdirs();
        }
    }
    
    /**
     * 获取Excel的值 第二行开始
     * @param path
     * @param mapKeys 字段依次读取后保存到map的key，用于返回后来根据key获取相关数据
     * @return
     * @throws IOException
     */
    public static List<Map<String, Object>> getListExcel(String path,String[] mapKeys) throws IOException {
        List<Map<String, Object>> listmap = new ArrayList<Map<String, Object>>();
        String excelName = getFileName(path);// 得到该文件夹下的excel名称；只取了第一个
        path = path + excelName;
        InputStream is = new FileInputStream(path);
      //  excelName.endsWith("xlsx");
        try {
            //XSSFWorkbook workbook = new XSSFWorkbook(is);
            Workbook workbook  = null;
            if(excelName.endsWith("xlsx")){
                workbook = new XSSFWorkbook(is);
            }else{
                workbook = new HSSFWorkbook(is);
            }
            Sheet sheet = workbook.getSheetAt(0);
            int rows = sheet.getLastRowNum();// 获取所有行数
            //sheet.getL
         //   String str[] = {"name","sex","cardno","address","img","work","economic"}; //定义返回的key
            String str[] = mapKeys; //定义返回的key
            // 循环EXCEL文件 获取每行的记录处理数据 从第2行开始,头行是注释
            for (int i = 1; i <= rows; i++) {
                Row row = sheet.getRow(i);// 获取第i行
                int coloumNum = sheet.getRow(i).getLastCellNum();// 获取第i行的列总数
                Map<String, Object> map = new HashMap<String, Object>();
                for (int j = 0; j < coloumNum; j++) {
                    Cell data = row.getCell(j);
                    if (data != null && !"".equals(data.toString()) ) {
                        data.setCellType(Cell.CELL_TYPE_STRING);
                        // sheet.getRow(0).getCell(j)获取该列的头部信息.
                        map.put(str[j].trim(), data);
                    } else {
                        map.put(str[j].trim(), "");
                    }
                }
                listmap.add(map);
            }
            
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            is.close();
        }
        return listmap;
    }
    
    // 获取路径下的所有文件
    public static String getFileName(String path) {
        File f = new File(path);
        if (!f.exists()) {
            return "";
        }
        File fa[] = f.listFiles();
        // for (int i = 0; i < fa.length; i++) {
        // System.out.println(fa[0].getName());
        // }
        return fa[0].getName();
    }
    
    
    public static void copyFile(File fromFile,File toFile) throws IOException{
        FileInputStream ins = new FileInputStream(fromFile);
        FileOutputStream out = new FileOutputStream(toFile);
        byte[] b = new byte[1024];
        int n=0;
        while((n=ins.read(b))!=-1){
            out.write(b, 0, n);
        }
        ins.close();
        out.close();
    }
    
    /**
     * 获取图片类型Zip数据信息
     * @param path
     * @return
     */
    public static List<Map<String, Object>> traverseFolder2(String path) {
        List<Map<String, Object>> listMap = new ArrayList<Map<String, Object>>();
        File file = new File(path);
        if (file.exists()) {
            File[] files = file.listFiles();
            if (null == files || files.length == 0) {
                System.out.println("文件夹是空的!");
                return listMap;
            } else {
                for (File file2 : files) {
                    if (file2.isDirectory()) {
                        System.out.println("文件夹:" + file2.getAbsolutePath());
                        //解决bug，如果是包含excel文件的，说明是模式1，并非 只有图片上传模式(模式2),详情看人员导入方法 TControlinfo.importZip();
                        if(file2.getAbsolutePath().indexOf("excel")>0){ 
                            listMap.clear(); 
                            return listMap;
                        }
                        listMap.addAll(traverseFolder2(file2.getAbsolutePath()));
                    } else {
                        Map<String, Object> dataMap = new HashMap<String,Object>();
                        System.out.println("文件:" + file2.getAbsolutePath());
                        String name = file2.getName();
                        if(name.indexOf("zip")>0)continue;
//                        System.out.println("文件名:" + name);
                        String control_name = "";
                        String control_cardno = ""; 
                        String contrl_address = "";
                        name = name.substring(0,name.indexOf("."));
                        String[] split = name.split("_");
                        if(split.length == 1){
                             control_name = split[0];
                        }else if(split.length == 2){
                            control_name = split[0];
                            control_cardno = split[1];
                        }else if(split.length == 3){
                            control_name = split[0];
                            control_cardno = split[1];
                            contrl_address = split[2];
                        }
//                        
                        dataMap.put("cardno", control_cardno);
                        dataMap.put("name", control_name);
                        dataMap.put("address", contrl_address);
                        dataMap.put("img", file2.getAbsolutePath());
                        listMap.add(dataMap);
                    }
                }
            }
        } else {
            System.out.println("文件不存在!");
        }
        return listMap;
    }