java实现CSV文件导入与导出功能

年前在开发功能模块的时候用到了CSV文件导入导出，就此整理一下，便于大家参考。

导入导出功能很多时候用到的都是Excel文件，但是现在越来越多的使用了CSV文件进行此操作，它是一个纯文本文件，可以用记事本打开，也可以用Excel打开。CSV文件不像Excel那样有很多条条框框，它使用硬回车分割每条记录，用逗号分隔每条数据的字段。

CSV格式的文件就是用硬回车和文本都好实现的表格，用Excel一读就成了表格。文件名后缀就是 .csv。

直接上代码吧！

导入部分

导入的时候基于Ajax请求，js代码如下：

function importIpMac(upload) {
 var importTextInfo = document.getElementById("importTextInfo");
 importTextInfo.value="";
 $.ajaxFileUpload({
   url: ctx + "/ipmac/importIpMac",
 type: 'post',
 secureuri: false,  // 一般设置为false
 fileElementId: 'upload',  // 上传文件的id、name属性名
 dataType: 'text',  // 返回值类型，一般设置为json、application/json
   success: function(data, status){
   getIpMacBase();
   },
   error: function(data, status, e){
   alert('请求异常!');
   }
 });
}

Java代码控制层：

 /**
 * 导入
 */
 @ResponseBody
 @RequestMapping(value = "/importIpMac", method = RequestMethod.POST, headers = { "content-type=multipart/form-data" })
 public int importIpMac(HttpServletRequest request,
  HttpServletResponse response,
  @RequestParam(value = "upload") MultipartFile[] buildInfo)
  throws ServletException, IOException {

 // 得到上传文件的保存目录，将上传的文件存放于WEB-INF目录下，不允许外界直接访问，保证上传文件的安全
 String savePath = request.getSession().getServletContext().getRealPath("/WEB-INF/upload");
 savePath = savePath.replace("file:", ""); // 去掉file:
 File file1 = new File(savePath);
 // 判断上传文件的保存目录是否存在
 if (!file1.exists() && !file1.isDirectory()) {
  log.info(savePath + "目录不存在，需要创建");
  file1.mkdir();
 }
 // 删除此路径下的所有文件以及文件夹
 delAllFile(savePath);

 try {
  InputStream is = buildInfo[0].getInputStream();// 多文件也适用,我这里就一个文件
  byte[] b = new byte[(int) buildInfo[0].getSize()];
  int read = 0;
  int i = 0;
  while ((read = is.read()) != -1) {
  b[i] = (byte) read;
  i++;
  }
  is.close();
  String filePath = savePath + "/" + "temp" + "_" + buildInfo[0].getOriginalFilename();
  log.info("临时文件保存路径：" + savePath + "/" + "temp" + "_" + buildInfo[0].getOriginalFilename());
  OutputStream os = new FileOutputStream(new File(savePath + "/" + "temp" + "_" + buildInfo[0].getOriginalFilename()));// 文件原名,如a.txt
  os.write(b);
  os.flush();
  os.close();
  topologyIpMacPortRealService.importIpMac(filePath);
 } catch (Exception e) {
  if (log.isDebugEnabled())
  log.debug("系统异常", e);
 }

 return 1;
 }

Java代码实现层：

 public int importIpMac(String filePath) throws Exception {
 // List<String> dataList=CSVUtils.importCsv(new File("/Users/wjm/Desktop/testexcel.csv"));
 List<String> dataList = CSVUtils.importCsv(new File(filePath));
 if (dataList != null && !dataList.isEmpty()) {
  for (int i = 1; i < dataList.size(); i++) {
  String data = dataList.get(i);
  SiTopologyIpMacPortBase base = new SiTopologyIpMacPortBase();
  String[] source = data.split(",");
  if (source[0] != "") {
   base.setId(source[0]);
   base.setMac(source[1]);
   base.setIp(source[2]);
   base.setUpIp(source[3]);
   base.setUpName(source[4]);
   base.setUpIndex(source[5]);
   base.setModifyTime(source[6]);

   siTopologyIpMacPortBaseDao.insert(base);
  }
  }
 }
 return 1;
 }

其中CSVUtils类：

package com.one.si.toimpl.common.utils;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.List;

/**
 * CSV操作(导出和导入)
 *
 * @author wjm
 * @version 1.0 Nov 24, 2015 4:30:58 PM
 */
public class CSVUtils {

<span style="white-space:pre"> </span>/**
   * 导出
   *
   * @param file csv文件(路径+文件名)，csv文件不存在会自动创建
   * @param dataList 数据
   * @return
   */
  public static boolean exportCsv(File file, List<String> dataList){
    boolean isSucess=false;

    FileOutputStream out=null;
    OutputStreamWriter osw=null;
    BufferedWriter bw=null;
    try {
//     OutputStreamWriter in_=new OutputStreamWriter(new FileOutputStream("文件名"), "gbk");
      out = new FileOutputStream(file);
      osw = new OutputStreamWriter(out, "gbk");
      bw =new BufferedWriter(osw);
      if(dataList!=null && !dataList.isEmpty()){
        for(String data : dataList){
          bw.append(data).append("\r");
        }
      }
      isSucess=true;
    } catch (Exception e) {
      isSucess=false;
    }finally{
      if(bw!=null){
        try {
          bw.close();
          bw=null;
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
      if(osw!=null){
        try {
          osw.close();
          osw=null;
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
      if(out!=null){
        try {
          out.close();
          out=null;
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }

    return isSucess;
  }

  /**
   * 导入
   *
   * @param file csv文件(路径+文件)
   * @return
   */
  public static List<String> importCsv(File file){
    List<String> dataList=new ArrayList<String>();

    BufferedReader br=null;
    try {
      br = new BufferedReader(new FileReader(file));
      String line = "";
      while ((line = br.readLine()) != null) {
        dataList.add(line);
      }
    }catch (Exception e) {
    }finally{
      if(br!=null){
        try {
          br.close();
          br=null;
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }

    return dataList;
  }
}

导出部分

js部分：

/*
 * 导出基准表中的数据
 */
function exportIpMac() {
 window.open("exportIpMac.do");
}

Java代码控制层：

 /**
 * 导出的基准表信息
 */
 @ResponseBody
 @RequestMapping("/exportIpMac")
 public String exportIpMac(HttpServletRequest request, HttpServletResponse response) throws Exception {
 List<String> dataList = topologyIpMacPortRealService.exportIpMac();
 response.setCharacterEncoding("GBK");
 SimpleDateFormat dfs = new SimpleDateFormat("yyyyMMddHHmmss");// 设置日期格式
 Date time = new Date();
 String tStamp = dfs.format(time);
 String filename = "IpMacPortExport"+tStamp + ".csv";
 response.setHeader("contentType", "text/html; charset=GBK");
 response.setContentType("application/octet-stream");
 response.addHeader("Content-Disposition", "attachment; filename="+filename);
 String cp=request.getSession().getServletContext().getRealPath("/");
 String path = cp+"download/"+filename;
 File file = new File(path);
 BufferedInputStream bis = null;
 BufferedOutputStream out = null;
 FileWriterWithEncoding fwwe =new FileWriterWithEncoding(file,"GBK");
 BufferedWriter bw = new BufferedWriter(fwwe);
 if(dataList!=null && !dataList.isEmpty()){
      for(String data : dataList){
        bw.write(data);
        bw.write("\n");
      }
    }
 bw.close();
 fwwe.close();
 try {
  bis = new BufferedInputStream(new FileInputStream(file));
  out = new BufferedOutputStream(response.getOutputStream());
  byte[] buff = new byte[2048];
  while (true) {
   int bytesRead;
   if (-1 == (bytesRead = bis.read(buff, 0, buff.length))){
   break;
   }
   out.write(buff, 0, bytesRead);
  }
  file.deleteOnExit();
 }
 catch (IOException e) {
  throw e;
 }
 finally{
  try {
  if(bis != null){
   bis.close();
  }
  if(out != null){
   out.flush();
   out.close();
  }
  }
  catch (IOException e) {
  throw e;
  }
 }
 delAllFile(cp+"download/");
 return null;

 }

Java代码实现层：

public List<String> exportIpMac() throws Exception {
 List<String> dataList = new ArrayList<String>();
 try {
  List<SiTopologyIpMacPortReal> list = siTopologyIpMacPortRealdao.selectAllData();
  dataList.add("ID,地址,IP地址,设备,设备名称,端口,更新时间");
  for (int i = 0; i < list.size(); i++) {
  dataList.add(list.get(i).getId() + "," + list.get(i).getMac()
   + "," + list.get(i).getIp() + ","
   + list.get(i).getUpIp() + ","
   + list.get(i).getUpName() + ","
   + list.get(i).getUpIfIndex() + ","
   + list.get(i).getModifyTime());
  }
 } catch (Exception e) {
  e.printStackTrace();
 }
 return dataList;
 }

使用的是Chrome浏览器，下载的时候会直接在浏览器下面进行显示下载。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。