基於 java 註解的 csv 文件讀寫框架

csv

基於 java 註解生成加簽驗籤 csv。

開源地址: github csv

創作原由

以前覺得 csv 文件的多寫非常簡單,就懶得封裝。

最近一個月寫了兩次 csv 文件相關的東西,發現要處理的細節還是有的,還浪費比較多的時間。

比如:

  1. UTF-8 中文編碼使用 excel 打開亂碼,因爲缺少 BOM 頭。

  2. 不同類型字段轉化爲字符串,順序的指定,head 頭的指定,如果手寫都會很繁瑣。

  3. 讀取的時候最後 , 後無元素,split 會缺失等。

爲了解決上述問題,此框架應運而生。

特性

  • Fluent 流式寫法

  • 基於 java 註解

  • 字段類型轉換的靈活支持,內置 8 大基本類型以及 String 類型轉換

快速開始

環境

jdk7+

maven 3.x

maven 引入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>csv</artifactId>
    <version>0.0.2</version>
</dependency>

示例代碼

  • User.java

演示基本類型的轉換

public class User {

    private String name;

    private int age;

    private float score;

    private double money;

    private boolean sex;

    private short level;

    private long id;

    private char status;

    private byte coin;

    //Getter & Setter & toString()
}
  • 對象列表構建
    /**
     * 構建通用測試列表
     * @return 列表
     */
    private List<User> buildCommonList() {
        User user = new User();
        short s = 4;
        byte b = 1;
        user.age(10)
        .name("你好")
        .id(1L)
        .score(60)
        .coin(b)
        .level(s)
        .money(200)
        .sex(true)
        .status('Y');
        return Arrays.asList(user);
    }

寫入

  • 測試代碼
public void commonTest() {
    final String path = "src\\test\\resources\\common.csv";
    CsvWriteBs.newInstance(path)
            .write(buildCommonList());
}
  • 文件生成
name,age,score,money,sex,level,id,status,coin
你好,10,60.0,200.0,true,4,1,Y,1

讀取

public void commonTest() {
    final String path = "src\\test\\resources\\common.csv";
    List<User> userList = CsvReadBs.newInstance(path)
            .read(User.class);
    System.out.println(userList);
}
  • 日誌信息
[User{name='你好', age=10, score=60.0, money=200.0, sex=true, level=4, id=1, status=Y, coin=1}]

CSV 引導類

爲了用戶使用的便利性,和後期拓展的靈活性。

引導類

CSV 有兩個引導類:

名稱 作用
CsvWriteBs csv 文件寫入引導類
CsvReadBs csv 文件讀取引導類

CsvWriteBs

方法 默認值 說明
newInstance(final String path) 必填 創建實例,並且指定待寫入文件路徑。
writeBom(boolean writeBom) true 是否寫入 UTF8 BOM 頭,建議第一次寫入指定,避免中文亂碼
charset(String charset) UTF-8 指定文件編碼
sort(ISort sort) NoSort 默認不進行字段排序
write(List<T> list) 待寫入的文件列表

CsvReadBs

方法 默認值 說明
newInstance(final String path) 必填 創建實例,並且指定待讀取文件路徑。
charset(String charset) UTF-8 指定文件編碼
sort(ISort sort) NoSort 默認不進行字段排序
startIndex(int startIndex) 1 文件的第二行,默認第一行是 head
endIndex(int endIndex) 文件的最後一行

Csv 註解

註解屬性說明

用於待處理對象的字段上。

    /**
     * 字段顯示名稱
     * 1. 默認使用 field.name
     * @return 顯示名稱
     */
    String label() default "";

    /**
     * 讀取是否需要
     * @return 是
     */
    boolean readRequire() default true;

    /**
     * 寫入是否需要
     * @return 是
     */
    boolean writeRequire() default true;

    /**
     * 讀取轉換
     * @return 處理實現類
     */
    Class<? extends IReadConverter> readConverter() default CommonReadConverter.class;

    /**
     * 寫入轉換
     * @return 處理實現類
     */
    Class<? extends IWriteConverter> writeConverter() default StringWriteConverter.class;

屬性概覽表

屬性 默認值 說明
label 字段名稱 用於 csv 頭生成
readRequire true 是否需要從 csv 文件讀取
writeRequire true 當前字段是否需要寫入 csv 文件
readConverter CommonReadConverter 將 csv 中的字符串轉化爲當前字段類型,支持 8 大基本類型+String
writeConverter StringWriteConverter 直接調用當前字段值 toString() 方法,null 直接爲空字符串

其中 readConverter/writeConverter 支持用戶自定義

註解使用代碼示例

對象定義

public class UserAnnotation {

    @Csv(label = "名稱")
    private String name;

    @Csv(label = "密碼", readRequire = false, writeRequire = false)
    private String password;

    @Csv(label = "生日", readConverter = ReadDateConvert.class, writeConverter = WriteDateConvert.class)
    private Date birthday;

    //Getter & Setter & toString()
}

ReadDateConvert/WriteDateConvert

使我們自定義的針對 Date 的轉換實現。

  • Write
public class WriteDateConvert implements IWriteConverter<Date> {

    @Override
    public String convert(Date value) {
        DateFormat dateFormat = new SimpleDateFormat("yyyyMMdd");
        return dateFormat.format(value);
    }

}
  • ReadDateConvert
public class ReadDateConvert implements IReadConverter<Date> {

    @Override
    public Date convert(String value, Class fieldType) {
        try {
            DateFormat dateFormat = new SimpleDateFormat("yyyyMMdd");
            return dateFormat.parse(value);
        } catch (ParseException e) {
            throw new RuntimeException(e);
        }
    }

}

寫入文件

public void annotationTest() {
    final String path = "src\\test\\resources\\annotation.csv";
    CsvWriteBs.newInstance(path)
            .write(buildAnnotationList());
}

其中列表構建:

/**
 * 構建基於註解的測試列表
 * @return 列表
 */
private List<UserAnnotation> buildAnnotationList() {
    UserAnnotation user = new UserAnnotation();
    user.name("你好")
            .password("123")
            .birthday(new Date());
    return Arrays.asList(user);
}
  • 生成文件內容
名稱,生日
你好,20190603

讀取文件測試

public void annotationTest() {
     final String path = "src\\test\\resources\\annotation.csv";
     List<UserAnnotation> userList = CsvReadBs.newInstance(path)
             .read(UserAnnotation.class);
     System.out.println(userList);
}
  • 日誌信息
[UserAnnotation{name='你好', password='null', birthday=Mon Jun 03 00:00:00 CST 2019}]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章