createmodel-java/src/main/java/ieven/server/webapp/service/impl/MongoExcelServiceImpl.java

package ieven.server.webapp.service.impl;

import ieven.server.webapp.domain.data.Fields;
import ieven.server.webapp.service.IExcelRowReader;
import ieven.server.webapp.util.Properties;
import org.apache.commons.lang3.StringUtils;
import org.bson.types.ObjectId;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class MongoExcelServiceImpl implements IExcelRowReader {
    private final Map<String, Integer> headerDefined = new LinkedHashMap<>();
    private final LinkedHashMap<String, String> headerTransfer = new LinkedHashMap<>();
    private final MongoTemplate mongoTemplate;
    private final String collectionData = Properties.COLLECTION_DATA;
    private final String fileId;
    private final List<Map<String, String>> dataList = new ArrayList<>(1000);

    public MongoExcelServiceImpl(MongoTemplate mongoTemplate, String fileId) {
        this.mongoTemplate = mongoTemplate;
        this.fileId = fileId;
    }

    @Override
    public void getRows(int sheetIndex, int curRow, List<String> valueList) {
        if (sheetIndex == 0 && curRow == 0) {
            for (int i = 0; i < valueList.size(); i++) {
                String value = valueList.get(i);
                if (StringUtils.isNotBlank(value)) {
                    //存入数据库的字段名
                    String aliasField = "f" + i;
                    headerDefined.put(aliasField, i);
                    headerTransfer.put(aliasField, value);
                } else {
                    String undefined = "undefined" + i;
                    //存入数据库的字段名
                    String aliasField = "f" + i;
                    headerDefined.put(aliasField, i);
                    headerTransfer.put(aliasField, undefined);
                }
            }
            Fields fields = new Fields();
            fields.setFileId(fileId);
            fields.setFields(headerTransfer);
            mongoTemplate.insert(fields);
        } else {
            Map<String, String> data = new LinkedHashMap<>(headerDefined.size());
            data.put("fileId", fileId);
            for (Map.Entry<String, Integer> entry : headerDefined.entrySet()) {
                String field = entry.getKey();
                Integer index = entry.getValue();
                String value = index < valueList.size() ? valueList.get(index) : "";
                data.put(field, value);
            }
            dataList.add(data);
            if (dataList.size() == 1000) {
                mongoTemplate.insert(dataList, collectionData);
                dataList.clear();
            }
        }
    }

    public void insertRest() {
        if (dataList.size() > 0) {
            mongoTemplate.insert(dataList, collectionData);
            dataList.clear();
        }
    }
}