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