FaguiSheet.java 2.29 KB
package org.nafmii.admin.excel;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import io.micrometer.core.instrument.util.StringUtils;
import lombok.Data;
import org.nafmii.admin.user.dao.ImportExcelMapper;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.common.response.ApiResponse;
import org.springframework.beans.factory.annotation.Autowired;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * @author:zhoujh
 * @Function:TODO
 * @date 2021/5/6 11:05
 * @ClassName:
 * @version:
 * @remark:
 */
public class FaguiSheet extends AnalysisEventListener<Map<Integer, String>> {

    @Resource
    private ImportExcelMapper importExcelMapper;

    // 所有要插入的数据
    private List<ImportExcelDto> datas = new ArrayList<>();

    // 缓存上一次跨行数据的有效数据
    private Map<Integer, String> preContent;

    @Override
    public void invoke(Map<Integer, String> o, AnalysisContext analysisContext) {
        ImportExcelDto model = new ImportExcelDto();
        String olumn1 = o.get(0);
        String olumn2 = o.get(1);
        String olumn3 = o.get(2);

        // 如果数据是空的话,从缓存对象中取值
        if (StringUtils.isBlank(olumn1)) {
            olumn1 = preContent.get(0);
        }
        if (StringUtils.isBlank(olumn2)) {
            olumn2 = preContent.get(1);
        }
        if (StringUtils.isBlank(olumn3)) {
            olumn3 = preContent.get(2);
        }

        model.setColumn1(olumn1);
        model.setColumn2(olumn2);
        model.setColumn3(olumn3);

        // 将数据放入list
        datas.add(model);

        // 复用跨行数据的实现
        String uniqueKey = o.get(0);
        // 如果A列的值为空,那么说明这是跨行列,如果取的值不是空,那说明这是第一次读取跨行列数据
        // 如果preContent为空,说明这是第一次读取sheet表格的数据
        if (!StringUtils.isBlank(uniqueKey) || preContent == null) {
            preContent = o;
        }
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {
        for (ImportExcelDto data : datas) {
            importExcelMapper.insertExcelData(data);
        }
    }
}