A operação de loop for do mongoTemplate

Por causa dos critérios, o mesmo parâmetro não pode ser adicionado repetidamente. Por exemplo, o printDateStr a seguir relatará um erro quando o loop for atingir a segunda vez.

for (String certificateDate : dataList) {
            criteria.and("printDateStr").is(certificateDate);
            query.addCriteria(criteria);
            List<UserCertificatePrintRecordVo> userCertificatePrintRecordVos = mongoTemplate.find(query, UserCertificatePrintRecordVo.class);
            for (UserCertificatePrintRecordVo userCertificatePrintRecordVo : userCertificatePrintRecordVos) {
                classHours += userCertificatePrintRecordVo.getClassScores();
            }
            dataMap.put(certificateDate, classHours);
            classHours = 0.0;
        }

Podemos usar Agregação para resolvê-lo, da seguinte forma:

    private Map<String, Object> setUserCertificateTotalClassHours(List<String> dataList) {
        Criteria criteria = new Criteria();
        Map<String, Object> dataMap = new HashMap<>(16);
        double classHours = 0.0;
        for (String certificateDate : dataList) {
// 设置条件
            Aggregation aggregation = Aggregation.newAggregation(
                    Aggregation.match(new Criteria().andOperator(criteria.where("printDateStr").is(certificateDate), criteria))
            );
// 查询
            AggregationResults<UserCertificatePrintRecordVo> carFlowResultVOs = mongoTemplate.aggregate(aggregation,
                    "UserCertificatePrintRecordVo", UserCertificatePrintRecordVo.class);
            List<UserCertificatePrintRecordVo> userCertificatePrintRecordVos = carFlowResultVOs.getMappedResults();
            for (UserCertificatePrintRecordVo userCertificatePrintRecordVo : userCertificatePrintRecordVos) {
                classHours += userCertificatePrintRecordVo.getClassScores();
            }
            dataMap.put(certificateDate, classHours);
            classHours = 0.0;
        }
        return dataMap;
    }

Acho que você gosta

Origin blog.csdn.net/Ciel_Y/article/details/122468255
Recomendado
Clasificación