Opencsv将csv文件转成Bean

Maven依赖

    <dependency>
      <groupId>com.opencsv</groupId>
      <artifactId>opencsv</artifactId>
      <version>5.2</version>
    </dependency>

CsvToBeanBuilder

优雅的解析文档中的字段。将CSV文件转换为Bean对象。

opencsv提供了基于"策略"的映射,将CSV绑定到bean。


16013479-954cb1d0f5887cab.png
接口名 策略
MappingStrategy 顶级的映射接口
headerColumnNameMappingStrategy 基于列名的映射策略,读取csv文件的第一行作为header,比如header1,header2,header3然后调用bean的setHeader1方法,setHeader2方法,setHeader3方法分别设置值。**所以这种策略要求,列名与bean中的属性名完全一致,如果不一致,则值为空,不会出错。使用注解时,注解名字必须与csv中列名一致。
ColumnPositionMappingStrategy 列位置映射策略,他没有header的概念,所以会输出取所有行。在columnMapping数组中指定bean的属性,第一个值对应csv的第一列,第二个值对应csv的第二列……
HeaderColumnNameTranslateMappingStrategy 列头名字翻译映射策略,与HeaderColumnNameMappintStrategy相比,bean的属性名可以与csv列头不一样。通过指定map来映射。

注:bean的类型只能为基本数据类型以及String类型,若是BigDecimal类型,那么将会抛出异常。

Bean

@Data
@AllArgsConstructor
@NoArgsConstructor
public class CsvBean {
    //@CsvBindByName(column = "PLEG_FLTNO")
    @CsvBindByPosition(position = 0)
    private String PLEG_FLTNO;

    //@CsvBindByName(column = "PLEG_AP_THR_DEP")
    @CsvBindByPosition(position = 1)
    private String PLEG_AP_THR_DEP;

    //@CsvBindByName(column = "PLEG_AP_THR_ARR")
    @CsvBindByPosition(position = 2)
    private String PLEG_AP_THR_ARR;

    //@CsvBindByName(column = "PLEG_DATE")
    @CsvBindByPosition(position = 3)
    private String PLEG_DATE;

    //@CsvBindByName(column = "PLEG_LON")
    @CsvBindByPosition(position = 4)
    private String PLEG_LON;
}

csv文件:

"PLEG_FLTNO","PLEG_AP_THR_DEP","PLEG_AP_THR_ARR","PLEG_DATE","PLEG_LON","PLEG_LAT","PLEG_ALT","PLEG_SPE","PLEG_DIR","PLEG_TM_SEN","PLEG_UPDATE_TIME"
CA1658,SHE,PEK,"2019-12-01","122.388744","41.6808833","7193.28","618.966","278.53","2019-12-01 15:52:11","2019-12-01 15:54:15"
CA1658,SHE,PEK,"2019-12-01","122.255966","41.695625","7193.28","613.44","278.44","2019-12-01 15:53:14","2019-12-01 15:55:10"
CA1658,SHE,PEK,"2019-12-01","122.128377","41.7096805","7193.28","608.292","277.65","2019-12-01 15:54:16","2019-12-01 15:55:30"
CA1658,SHE,PEK,"2019-12-01","121.994802","41.7067388","7193.28","596.826","266.89","2019-12-01 15:55:21","2019-12-01 15:57:15"

基于列索引的映射

通俗点就是列位置映射,csv文件中列位置对应到bean中的列。
需要注意的是,该策略会输出所有的行,故,我们需要跳过某些行。

    @Test
    public void readCsvByPosition() throws Exception
    {
        String fileName="C:\\Users\\lyh\\Desktop\\CA1658SHE-PEK.csv";
        InputStreamReader is = new InputStreamReader(new  FileInputStream(fileName), StandardCharsets.UTF_8);
        ColumnPositionMappingStrategy<CsvBean> mappingStrategy = new ColumnPositionMappingStrategy<>();
        mappingStrategy.setType(CsvBean.class);
        CsvToBean<CsvBean> build = new CsvToBeanBuilder<CsvBean>(is)
                .withMappingStrategy(mappingStrategy)
                .withSeparator(CSVWriter.DEFAULT_SEPARATOR)
                .withSkipLines(1)   //跳过第一行
                .build();
        List<CsvBean> personList = build.parse();
        personList.forEach(System.out::println);
    }

结果

CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.388744)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.255966)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.128377)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=121.994802)

基于列名的映射

    @Test
    public void readCsvByName() throws Exception
    {
        String fileName="C:\\Users\\lyh\\Desktop\\CA1658SHE-PEK.csv";
        InputStreamReader is = new InputStreamReader(getInputStream(new  FileInputStream(fileName)), StandardCharsets.UTF_8);
        HeaderColumnNameMappingStrategy<CsvBean> mappingStrategy = new HeaderColumnNameMappingStrategy<>();
        mappingStrategy.setType(CsvBean.class);
        CsvToBean<CsvBean> build = new CsvToBeanBuilder<CsvBean>(is)
                .withMappingStrategy(mappingStrategy)
                .withSeparator(CSVWriter.DEFAULT_SEPARATOR)
                .build();
        List<CsvBean> personList = build.parse();
        personList.forEach(System.out::println);
    }

    /**
     * 读取流中前面的字符,看是否有bom,如果有bom,将bom头先读掉丢弃
     *  (opencsv 按列名获取bean对象,第一列缺失的情况)
     */
    public static InputStream getInputStream(InputStream in) throws IOException {

        PushbackInputStream testin = new PushbackInputStream(in);
        int ch = testin.read();
        if (ch != 0xEF) {
            testin.unread(ch);
        } else if ((ch = testin.read()) != 0xBB) {
            testin.unread(ch);
            testin.unread(0xef);
        } else if (testin.read() != 0xBF) {
            throw new IOException("错误的UTF-8格式文件");
        }
        return testin;
    }

结果

CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.388744)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.255966)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=122.128377)
CsvBean(PLEG_FLTNO=CA1658, PLEG_AP_THR_DEP=SHE, PLEG_AP_THR_ARR=PEK, PLEG_DATE=2019-12-01, PLEG_LON=121.994802)

转换器

在csv获取的都是字符串,这种情况下应该使用转换器。将csv中的字段转换为对应的bean中的字段类型。

@Data
@AllArgsConstructor
@NoArgsConstructor
public class CsvBean {
    //@CsvBindByName(column = "PLEG_FLTNO")
    @CsvBindByPosition(position = 0)
    private String PLEG_FLTNO;

    //@CsvBindByName(column = "PLEG_AP_THR_DEP")
    @CsvBindByPosition(position = 1)
    private String PLEG_AP_THR_DEP;

    //@CsvBindByName(column = "PLEG_AP_THR_ARR")
    @CsvBindByPosition(position = 2)
    private String PLEG_AP_THR_ARR;

    //@CsvBindByName(column = "PLEG_DATE")
    //@CsvBindByPosition(position = 3)
    @CsvCustomBindByPosition(position = 3,converter = ConvertToDate.class)
    private LocalDate PLEG_DATE;

    //@CsvBindByName(column = "PLEG_LON")
    @CsvBindByPosition(position = 4)
    private String PLEG_LON;
}
public class ConvertToDate extends AbstractBeanField {
    @Override
    protected Object convert(String s) throws CsvDataTypeMismatchException, CsvConstraintViolationException {
        return LocalDate.parse(s, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
    }
}

注:若是列映射策略,则要使用@CsvCustomBindByPosition()注解。
注:若是名字映射策略,则要使用@CsvCustomBindByName()注解。

过滤器

opencsv提供了过滤器,可以过滤某些行,比如page header、page footer等

所有的过滤器必须实现CsvToBeanFilter 接口

public class CsvFilter implements CsvToBeanFilter {
    @Override
    public boolean allowLine(String[] strings) {
        return strings[4].equalsIgnoreCase("122.388744");
    }
}

CsvToBeanFilter 是一个函数式接口所以也可以直接简化为lambda表达式

    @Test
    public void readCsvByPosition() throws Exception
    {
        String fileName="C:\\Users\\lyh\\Desktop\\CA1658SHE-PEK.csv";
        InputStreamReader is = new InputStreamReader(new  FileInputStream(fileName), StandardCharsets.UTF_8);
        ColumnPositionMappingStrategy<CsvBean> mappingStrategy = new ColumnPositionMappingStrategy<>();
        mappingStrategy.setType(CsvBean.class);
        CsvToBean<CsvBean> build = new CsvToBeanBuilder<CsvBean>(is)
                .withMappingStrategy(mappingStrategy)
                .withSeparator(CSVWriter.DEFAULT_SEPARATOR)
                .withSkipLines(1)
                .withFilter(param-> param[4].equalsIgnoreCase("122.388744"))  //增加过滤器
                //.withFilter(new CsvFilter())
                .build();
        List<CsvBean> personList = build.parse();
        personList.forEach(System.out::println);
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350