diff --git a/src/main/java/com/kiisoo/ic/employee/mapper/PrivilageCpUserDOMapper.java b/src/main/java/com/kiisoo/ic/employee/mapper/PrivilageCpUserDOMapper.java index ead4263..4c5168f 100644 --- a/src/main/java/com/kiisoo/ic/employee/mapper/PrivilageCpUserDOMapper.java +++ b/src/main/java/com/kiisoo/ic/employee/mapper/PrivilageCpUserDOMapper.java @@ -14,6 +14,8 @@ public interface PrivilageCpUserDOMapper extends BaseMapper { CpUserCustomerRelationDO selectCpUserCustomerRelation(@Param("customerUserId")String custeromUserId, @Param("cpUserId")String cpUserId, @Param("cpId")Long cpId); + List selectCpUserCustomerRelationByCpId(@Param("cpId")Long cpId); + Long selectAllCount(); Long selectDistinctAllCount(); diff --git a/src/main/java/com/kiisoo/ic/job/count/CountDTOMessageGroupConsumer.java b/src/main/java/com/kiisoo/ic/job/count/CountDTOMessageGroupConsumer.java index 263b6de..fe9ff47 100644 --- a/src/main/java/com/kiisoo/ic/job/count/CountDTOMessageGroupConsumer.java +++ b/src/main/java/com/kiisoo/ic/job/count/CountDTOMessageGroupConsumer.java @@ -58,9 +58,9 @@ public class CountDTOMessageGroupConsumer implements WorkHandler addedCustomerUserIds = privilageCpUserDOMapper.selectCpUserCustomerRelationByCpId(privilageCpUserDO.getId()); for (String customerStr : customerStrs) { - CpUserCustomerRelationDO cpUserCustomerRelationDO = privilageCpUserDOMapper.selectCpUserCustomerRelation(customerStr, cpUserId, privilageCpUserDO.getId()); - if (cpUserCustomerRelationDO == null){ + if (CollectionUtils.isEmpty(addedCustomerUserIds) || !addedCustomerUserIds.contains(customerStr)){ privilageCpUserDOMapper.insertCpUserCustomerRelation(customerStr,cpUserId,privilageCpUserDO.getId()); } } diff --git a/src/main/resources/mapper/PrivilageCpUserDOMapper.xml b/src/main/resources/mapper/PrivilageCpUserDOMapper.xml index 38b1700..5fed538 100644 --- a/src/main/resources/mapper/PrivilageCpUserDOMapper.xml +++ b/src/main/resources/mapper/PrivilageCpUserDOMapper.xml @@ -14,6 +14,11 @@ limit 1 +