java實現解析csv文件方法二：解析csv文件並轉成bean

1.pom.xml引入jar包

       <!--csv-->
        <dependency>
            <groupId>com.opencsv</groupId>
            <artifactId>opencsv</artifactId>
            <version>4.4</version>
        </dependency>

2.重寫映射規則

public class CustomMappingStrategy<T> extends ColumnPositionMappingStrategy<T> {

    @Override
    public String[] generateHeader(T bean) throws CsvRequiredFieldEmptyException {
        super.setColumnMapping(new String[ FieldUtils.getAllFields(bean.getClass()).length]);
        final int numColumns = findMaxFieldIndex();
        if (!isAnnotationDriven() || numColumns == -1) {
            return super.generateHeader(bean);
        }

        String[] header = new String[numColumns + 1];

        BeanField beanField;
        for (int i = 0; i <= numColumns; i++) {
            beanField = findField(i);
            String columnHeaderName = extractHeaderName(beanField);
            header[i] = columnHeaderName;
        }
        return header;
    }

    private String extractHeaderName(final BeanField beanField) {
        if (beanField == null || beanField.getField() == null || beanField.getField().getDeclaredAnnotationsByType(CsvBindByName.class).length == 0) {
            return StringUtils.EMPTY;
        }

        final CsvBindByName bindByNameAnnotation = beanField.getField().getDeclaredAnnotationsByType(CsvBindByName.class)[0];
        return bindByNameAnnotation.column();
    }
}

3.實體類（bean）添加注解 @CsvBindByPosition 和 @CsvBindByName

public class WftTestEntity implements Serializable{
    private static final long serialVersionUID = 1L;
    /**
     *交易時間
     */
    @CsvBindByPosition(position = 0)
    @CsvBindByName(column = "交易時間")
    private String test1;


    /**
     *公眾賬號ID
     */
    @CsvBindByPosition(position = 1)
    @CsvBindByName(column = "公眾賬號ID")
    private String test2;

    public String getTest1() {
        return test1;
    }

    public void setTest1(String test1) {
        this.test1 = test1;
    }

    public String getTest2() {
        return test2;
    }

    public void setTest2(String test2) {
        this.test2 = test2;
    }
}

4.解析並轉換為bean

4.1:單一的轉換（非泛型）

     /**
     * 解析csv文件並轉成bean
     * @return
     * @throws IOException
     */
    public static List<WftTestEntity> getWftTestEntityList(String fileName) throws IOException{
        InputStreamReader is = new InputStreamReader(new  FileInputStream(fileName),"GBK");
        HeaderColumnNameMappingStrategy<WftTestEntity> mappingStrategy = new HeaderColumnNameMappingStrategy<>();
        mappingStrategy.setType(WftTestEntity.class);
        CsvToBean<WftTestEntity> build = new CsvToBeanBuilder<WftTestEntity>(is).withMappingStrategy(mappingStrategy).withQuoteChar('`').withSeparator('"').withSeparator(',').withSeparator('"').build();
        List<WftTestEntity> wftTestEntityList = build.parse();
        return wftTestEntityList;
    }

4.2：通用的轉換（泛型）

　　

    /**
     * 解析csv文件並轉成bean
     * @param fileName
     * @param clazz
     * @param <T>
     * @return
     * @throws IOException
     */
    public static <T> List<T> getCsvData(String fileName, Class<T> clazz) throws IOException{
        InputStreamReader is = new InputStreamReader(new  FileInputStream(fileName),"gbk");

        HeaderColumnNameMappingStrategy<T> strategy = new HeaderColumnNameMappingStrategy<>();
        strategy.setType(clazz);

        CsvToBean<T> csvToBean = new CsvToBeanBuilder<T>(is)
                .withSeparator(',')
                .withQuoteChar('\'')
                .withMappingStrategy(strategy).build();
        return csvToBean.parse();
    }

5.調用測試mian

public static void main(String[] args) throws IOException{

   //調用
   List<WftTestEntity> wftTestEntities = getWftTestEntityList("路徑+csv文件名");
   List<WftTestEntity> wftTestEntityList = getCsvData("路徑+csv文件名",WftTestEntity.class);
}