测评系统外部代码

linzhijie

2021-03-11 c33914ba0a98c823c4b4d7da21cdd476906c9924

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package com.ots.project.exam.service.impl;
import com.ots.common.enums.TestMemberStatusEnum;
import com.ots.project.exam.domain.*;
import com.ots.project.exam.mapper.EntTestMemberMapper;
import com.ots.project.exam.mapper.EntTestPackageMapper;
import com.ots.project.exam.mapper.SysUserExtendMapper;
import com.ots.project.exam.service.CountCalculateService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.*;
import java.util.stream.Collectors;
 
@Component
public class CountCalculateServiceImpl implements CountCalculateService {
    @Autowired
    private EntTestMemberMapper entTestMemberMapper;
    @Autowired
    private EntTestPackageMapper entTestPackageMapper;
    @Autowired
    private SysUserExtendMapper sysUserExtendMapper;
    @Override
    public NumberOfUsersByCompany calculateProductPackagesMemberCount(EntTestPackage entTestPackage) {
        Long userId = entTestPackage.getUserId();
        NumberOfUsersByCompany result = new NumberOfUsersByCompany();
        if (Objects.nonNull(userId)) {
            
            SysUserExtend sysUserExtend = sysUserExtendMapper.selectSysUserExtendById(userId);
            if (Objects.isNull(sysUserExtend)) {
                return result;
            }
            Long testNum = sysUserExtend.getTestNum();
            if (Objects.isNull(testNum)) {
                return result;
            }
            result.setTestNum(testNum.intValue());
        }
        List<EntTestPackage> entTestPackages = findEntTestPackages(entTestPackage);
        
        if (Objects.nonNull(entTestPackages) && !entTestPackages.isEmpty()) {
            result = constructResult(userId, entTestPackages);
        }
        return result;
    }
    private List<EntTestPackage> findEntTestPackages(EntTestPackage entTestPackage) {
        return entTestPackageMapper.selectEntTestPackageList(entTestPackage);
    }
    private NumberOfUsersByCompany constructResult(Long userId, List<EntTestPackage> entTestPackages) {
        List<Long> collect = entTestPackages.stream().map(p -> p.getId()).collect(Collectors.toList());
        List<EntTestMember> entTestMembers = findEntTestMembers(collect);
        NumberOfUsersByCompany result = new NumberOfUsersByCompany();
        if (Objects.nonNull(userId)) {
            result.setUserId(userId);
        }
        if (Objects.nonNull(result.getTestNum())) {
            result.setUsedTimes(entTestMembers.size());
        }
        
        Map<Long, List<EntTestMember>> testPackageUsageTimesMap = entTestMembers.stream().collect(Collectors.groupingBy(EntTestMember::getTestId));
        entTestPackages.stream().forEach(p -> {
            NumberOfUsersByPackage numberOfUsersByPackage = new NumberOfUsersByPackage();
            List<EntTestMember> testMembers = testPackageUsageTimesMap.get(p.getId());
            long total = Long.valueOf(0);
            long finishNum = Long.valueOf(0);
            long unTestNum = Long.valueOf(0);
            long testIngNum = Long.valueOf(0);
            if (Objects.nonNull(testMembers)) {
                total = Long.valueOf(testMembers.size());
                finishNum = testMembers.stream().filter(testMember -> TestMemberStatusEnum.completed.getCode().equals(testMember.getStatus())).mapToLong(x -> 1).sum();
                testIngNum = testMembers.stream().filter(testMember -> TestMemberStatusEnum.answering.getCode().equals(testMember.getStatus())).mapToLong(x -> 1).sum();
                unTestNum = total - testIngNum - finishNum;
            }
            numberOfUsersByPackage.setTestNum(total);
            numberOfUsersByPackage.setFinishNum(finishNum);
            numberOfUsersByPackage.setUnTestNum(unTestNum);
            numberOfUsersByPackage.setTestingNum(testIngNum);
            if (Objects.isNull(result.getTestPackageUsageTimesMap())) {
                result.setTestPackageUsageTimesMap(new HashMap<>());
            }
            result.getTestPackageUsageTimesMap().put(p.getId(), numberOfUsersByPackage);
        });
        return result;
    }
    private List<EntTestMember> findEntTestMembers(List<Long> collect) {
        Map map = new HashMap();
        map.put("testIds", collect);
        
        List<String> recyclingStatus = Arrays.asList(TestMemberStatusEnum.unprocessed.getCode(), TestMemberStatusEnum.completed.getCode(), TestMemberStatusEnum.answering.getCode());
        map.put("status", recyclingStatus);
        return entTestMemberMapper.selectEntTestMemberListByMap(map);
    }
}