Java编程之CSV文件导入与导出的实现

来源：互联网发布：windows安装光盘下载编辑：程序博客网时间：2024/05/22 13:57

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

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

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

直接上代码吧！

导入部分

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

[javascript] view plain copy
 print?
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代码控制层：

[java] view plain copy
 print?
/** 
 * 导入 
 */  
@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代码实现层：

[java] view plain copy
 print?
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类：

[java] view plain copy
 print?
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部分：

[javascript] view plain copy
 print?
/* 
 * 导出基准表中的数据 
 */  
function exportIpMac() {  
    window.open("exportIpMac.do");  
}  

Java代码控制层：

[java] view plain copy
 print?
/** 
 * 导出的基准表信息 
 */  
@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代码实现层：

[java] view plain copy
 print?
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浏览器，下载的时候会直接在浏览器下面进行显示下载。

阅读全文

0 0