Java学习-015-CSV 文件写入实例源代码

在日常的自动化测试脚本编写的过程中，有时要将获取的测试结果或者测试数据存放在数据文件中，以用作后续的参数化测试。常用的文件文件类型无非 txt、csv、xls、properties、xml 这五种文件类型，前文已讲述过了txt文件相关的读写操作（读、写），此文主要讲述 CSV 文件的写入操作。此文中 CSV 文件的写入方法和最终的测试方法，各位小主们在实际参考应用时，需要进行相应的修改（例如文件路径等）才可。若有不足之处，敬请各位大神告知，不胜感激！

闲话少述，小二上码勒

CSV文件写入方法源码如下所示，敬请参阅！

     /**

      * @function 文件写入: CSV文件

      *

      * @author Aaron.ffp

      * @version V1.0.0: autoUISelenium main.java.aaron.java.tools FileUtils.java csvWrite, 2014-11-25 15:58:20 Exp $

      *

      * @param filename  : CSV文件名

      * @param delimiter : CSV文件分隔符

      * @param encoding  : CSV文件编码格式

      * @param csvdata   : CSV文件内容, 二维数组

      *

      * @return boolean

      */

     public boolean csvWrite(String filename, char delimiter, String encoding, ArrayList<String[]> csvdata){

         boolean success = false;

         /* csv写对象 */

         CsvWriter csvwriter = null;

         /* 参数校验: 为null或空字符串时, 抛出参数非法异常 */

         if (filename == null || "".equals(filename) || !assertFileType(filename, "CSV")) {

             throw new IllegalArgumentException();

         }

         /* 分隔符默认处理, 若分割符号为空, 则默认 ‘,’ */

         if ("".equals(delimiter)) {

             delimiter = ',';

         }

         /* 参数校验: 编码格式是否在格式范围内 */

         if(!this.constantslist.ENCODING_TYPE.contains(encoding)){

             this.message = "参数非法:第三个参数{CSV编码}非法, 编码格式{" + encoding + "}不在CSV文件编码列表[" + this.constantslist.ENCODING_TYPE.toString() + "]中, 不支持!";

             this.logger.error(this.message);

             throw new UnsupportedCharsetException(this.message);

         }

         try {

             // delete file

             this.deleteFile(filename);

             csvwriter = new CsvWriter(filename, delimiter, Charset.forName(encoding));

             /* 数据写入 */

             for (String[] rowdata : csvdata) {

                 csvwriter.writeRecord(rowdata);

             }

             csvwriter.close();

             success = true;

         } catch (IOException ioe) {

             this.message = "文件 {" + filename + "} 写入失败！";

             this.logger.error(this.message, ioe);

             success = false;

         } catch (UnsupportedCharsetException uce){

             this.message = "文件 {" + filename + "} 编码格式设置失败！";

             this.logger.error(this.message, uce);

             success = false;

         }

         return success;

     }

CSV 文件写入方法源码

CSV文件写入方法测试源码如下所示，敬请参阅！

     /**

      * Test : read|write CSV file

      *

      * @author Aaron.ffp

      * @version V1.0.0: autoUISelenium test.java.aaron.java.tools FileUtilsTest.java test_csvWrite_and_csvRead, 2014-11-25 16:28:20 Exp $

      *

      */

     @Test

     public void test_csvWrite_and_csvRead(){

         this.message = "\n\n\nTEST：FileUtils.csvWrite(String filename, char delimiter, String encoding, ArrayList<String[]> csvdata)";

         this.logger.debug(this.message);

         this.fu = new FileUtils();

         // change to you local path

         String filename = this.constantslist.PROJECTHOME + this.constantslist.FILESEPARATOR +

                           "testng-temp" + this.constantslist.FILESEPARATOR + "createfile.csv";

         ArrayList<String[]> csvdata = new ArrayList<String[]>();

         String[] rows;

         // init csv data

         for (int i = 0; i < 20; i++) {

             rows = new String[10];

             for (int j = 0; j < rows.length; j++) {

                 rows[j] = i + " = " + j;

             }

             csvdata.add(rows);

         }

         boolean w = this.fu.csvWrite(filename, ',', "UTF-8", csvdata);

         Assert.assertEquals(w, true, "Test case failed.");

     }

CSV 文件写入方法测试源码

至此， Java学习-015-CSV 文件写入实例源代码 顺利完结，希望此文能够给初学 Java 的您一份参考。

最后，非常感谢亲的驻足，希望此文能对亲有所帮助。热烈欢迎亲一起探讨，共同进步。非常感谢！ ^_^

巴特西

Java学习-015-CSV 文件写入实例源代码

最新文章

热门文章