java生成csv文件

转自:http://www.jb51.net/article/52724.htm,直接拷到代码里就可以了

package test_csv;

	import java.io.BufferedWriter;
	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.io.OutputStreamWriter;
	import java.net.URLEncoder;
	import java.util.ArrayList;
	import java.util.Iterator;
	import java.util.LinkedHashMap;
	import java.util.List;
	import java.util.Map;
	 
	import javax.servlet.http.HttpServletResponse;
	 
	import org.apache.commons.beanutils.BeanUtils;
	 
	/**
	 * 文件操作
	 */
	public class CSVUtils {
	 
	  /**
	   * 生成为CVS文件 
	   * @param exportData
	   *       源数据List
	   * @param map
	   *       csv文件的列表头map
	   * @param outPutPath
	   *       文件路径
	   * @param fileName
	   *       文件名称
	   * @return
	   */
	  @SuppressWarnings("rawtypes")
	  public static File createCSVFile(List exportData, LinkedHashMap map, String outPutPath,
	                   String fileName) {
	    File csvFile = null;
	    BufferedWriter csvFileOutputStream = null;
	    try {
	      File file = new File(outPutPath);
	      if (!file.exists()) {
	        file.mkdir();
	      }
	      //定义文件名格式并创建
	      csvFile = File.createTempFile(fileName, ".csv", new File(outPutPath));
	      System.out.println("csvFile:" + csvFile);
	      // UTF-8使正确读取分隔符"," 
	      csvFileOutputStream = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(
	        csvFile), "UTF-8"), 8192);
	      System.out.println("csvFileOutputStream:" + csvFileOutputStream);
	      // 写入文件头部 
	      for (Iterator propertyIterator = map.entrySet().iterator(); propertyIterator.hasNext();) {
	        java.util.Map.Entry propertyEntry = (java.util.Map.Entry) propertyIterator.next();
	        csvFileOutputStream
	          .write("" + (String) propertyEntry.getValue() != null ? (String) propertyEntry.getValue() : "" + "");
	        if (propertyIterator.hasNext()) {
	          csvFileOutputStream.write(",");
	        }
	      }
	      csvFileOutputStream.newLine();
	      // 写入文件内容 
	      for (Iterator iterator = exportData.iterator(); iterator.hasNext();) {
	        Object row = (Object) iterator.next();
	        for (Iterator propertyIterator = map.entrySet().iterator(); propertyIterator
	          .hasNext();) {
	          java.util.Map.Entry propertyEntry = (java.util.Map.Entry) propertyIterator
	            .next();
	          csvFileOutputStream.write((String) BeanUtils.getProperty(row,
	            (String) propertyEntry.getKey()));
	          if (propertyIterator.hasNext()) {
	            csvFileOutputStream.write(",");
	          }
	        }
	        if (iterator.hasNext()) {
	          csvFileOutputStream.newLine();
	        }
	      }
	      csvFileOutputStream.flush();
	    } catch (Exception e) {
	      e.printStackTrace();
	    } finally {
	      try {
	        csvFileOutputStream.close();
	      } catch (IOException e) {
	        e.printStackTrace();
	      }
	    }
	    return csvFile;
	  }
	 
	  /**
	   * 下载文件
	   * @param response
	   * @param csvFilePath
	   *       文件路径
	   * @param fileName
	   *       文件名称
	   * @throws IOException
	   */
	  public static void exportFile(HttpServletResponse response, String csvFilePath, String fileName)
	                                                  throws IOException {
	    response.setContentType("application/csv;charset=UTF-8");
	    response.setHeader("Content-Disposition",
	      "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8"));
	 
	    InputStream in = null;
	    try {
	      in = new FileInputStream(csvFilePath);
	      int len = 0;
	      byte[] buffer = new byte[1024];
	      response.setCharacterEncoding("UTF-8");
	      OutputStream out = response.getOutputStream();
	      while ((len = in.read(buffer)) > 0) {
	    	//utf-8的文件在头部有三个字段是用来标识该文件的格式的
	        out.write(new byte[] { (byte) 0xEF, (byte) 0xBB, (byte) 0xBF });//输出那三个字节的标示信息
	        out.write(buffer, 0, len);
	      }
	    } catch (FileNotFoundException e) {
	      System.out.println(e);
	    } finally {
	      if (in != null) {
	        try {
	          in.close();
	        } catch (Exception e) {
	          throw new RuntimeException(e);
	        }
	      }
	    }
	  }
	 
	  /**
	   * 删除该目录filePath下的所有文件
	   * @param filePath
	   *      文件目录路径
	   */
	  public static void deleteFiles(String filePath) {
	    File file = new File(filePath);
	    if (file.exists()) {
	      File[] files = file.listFiles();
	      for (int i = 0; i < files.length; i++) {
	        if (files[i].isFile()) {
	          files[i].delete();
	        }
	      }
	    }
	  }
	 
	  /**
	   * 删除单个文件
	   * @param filePath
	   *     文件目录路径
	   * @param fileName
	   *     文件名称
	   */
	  public static void deleteFile(String filePath, String fileName) {
	    File file = new File(filePath);
	    if (file.exists()) {
	      File[] files = file.listFiles();
	      for (int i = 0; i < files.length; i++) {
	        if (files[i].isFile()) {
	          if (files[i].getName().equals(fileName)) {
	            files[i].delete();
	            return;
	          }
	        }
	      }
	    }
	  }
	 
	  /**
	   * 测试数据
	   * @param args
	   */
	  @SuppressWarnings({ "rawtypes", "unchecked" })
	  public static void main(String[] args) {
	    List exportData = new ArrayList<Map>();
	    Map row1 = new LinkedHashMap<String, String>();
	    row1.put("1", "11");
	    row1.put("2", "12");
	    row1.put("3", "13");
	    row1.put("4", "14");
	    exportData.add(row1);
	    row1 = new LinkedHashMap<String, String>();
	    row1.put("1", "21");
	    row1.put("2", "22");
	    row1.put("3", "23");
	    row1.put("4", "24");
	    exportData.add(row1);
	    LinkedHashMap map = new LinkedHashMap();
	    map.put("1", "第一列");
	    map.put("2", "第二列");
	    map.put("3", "第三列");
	    map.put("4", "第四列");
	 
	    String path = "d:/export/";
	    String fileName = "文件导出";
	    File file = CSVUtils.createCSVFile(exportData, map, path, fileName);
	    String fileName2 = file.getName();
	    System.out.println("文件名称:" + fileName2);
	  }
}

猜你喜欢

转载自sky-xin.iteye.com/blog/2297512