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浏览器，下载的时候会直接在浏览器下面进行显示下载。

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