【Java後端】解析csv文件並轉成bean對象

爲了批量導入數據，採用了csv文件存儲數據，後端寫csv解析工具對數據進行讀取和解析並導入數據庫。
pom依賴：

<!-- csv文件解析依賴 -->
<dependency>
    <groupId>com.opencsv</groupId>
    <artifactId>opencsv</artifactId>
    <version>4.3.2</version>
</dependency>

1、解析csv文件工具類（通用）

public class CsvUtil {
    /**
     * 日誌對象
     */
    private static final Logger LOGGER = LoggerFactory.getLogger(CsvUtil.class);

    /**
     * 解析csv文件並轉成bean
     * @param file csv文件
     * @param clazz 類
     * @param <T> 泛型
     * @return 泛型bean集合
     */
    public <T> List<T> getCsvData(MultipartFile file, Class<T> clazz) {
        InputStreamReader in;
        try {
            in = new InputStreamReader(file.getInputStream(), "gbk");
        } catch (Exception e) {
            LOGGER.error(Constant.LOG_FOMAT_TYPE_TWO,
                    ErrorCode.FILE_LOAD_ERROR.getErrorCode(),
                    ErrorCode.FILE_LOAD_ERROR.getMessage(),
                    e.getMessage());
            throw BaseException.of(ErrorCode.FILE_LOAD_ERROR.of());
        }

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

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

2、csv文件

3、csv文件對應的bean對象DataAndTypeCsv.java

public class DataAndTypeCsv {

    /**
     * 字典代碼
     */
    @CsvBindByName(column = "code")
    private String code;

    /**
     * 簡寫
     */
    @CsvBindByName(column = "short_name")
    private String shortName;

    /**
     * 名稱
     */
    @CsvBindByName(column = "name", required = true)  //是否可以爲null 否
    private String name;

    /**
     * 拼音或英文描述
     */
    @CsvBindByName(column = "remark")
    private String remark;

    /**
     * 父類型id
     */
    @CsvBindByName(column = "parent_id")
    private Integer parentId;

    /**
     * 類型名稱
     */
    @CsvBindByName(column = "type_name", required = true)  //是否可以爲null 否
    private String typeName;

    /**
     * 類型id
     */
    @CsvBindByName(column = "type_id", required = true)  //是否可以爲null 否
    private Integer typeId;
    
	……
	get、set方法
	……

bean對象通過註解@CsvBindByName(column = “”, required = true) 映射csv文件的列名，並檢查列數據是否爲空，若是csv文件中該列存在值爲空的情況會報錯。

4、應用
controller層定義restful接口：

@Transactional(rollbackFor = BaseException.class)
@PostMapping
public Map<String, String> batchInsert(MultipartFile file) {
    CsvUtil csvUtil = new CsvUtil();

    // 將csv文件內容轉成bean
    List<DataAndTypeCsv> csvData = csvUtil.getCsvData(file, DataAndTypeCsv.class);
    ……
}

postman調用接口測試：

擴展
withQuoteChar()方法的作用