diff --git a/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/hidden/HiddenPenaltyRepository.java b/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/hidden/HiddenPenaltyRepository.java index 8aaf695..a1f73c7 100644 --- a/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/hidden/HiddenPenaltyRepository.java +++ b/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/hidden/HiddenPenaltyRepository.java @@ -20,4 +20,8 @@ public interface HiddenPenaltyRepository extends BaseRepository HiddenPenaltyFeedbackDO getFeedbackByPenaltyId(String hiddenPenaltyId); void saveFeedback(HiddenPenaltyFeedbackDO feedbackDO); + + Long countByUserId(Long userId); + + Long countFeedbackByUserId(Long userId); } diff --git a/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/impl/hidden/HiddenPenaltyRepositoryImpl.java b/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/impl/hidden/HiddenPenaltyRepositoryImpl.java index cbce764..190d6e4 100644 --- a/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/impl/hidden/HiddenPenaltyRepositoryImpl.java +++ b/web-infrastructure/src/main/java/com/zcloud/key/project/persistence/repository/impl/hidden/HiddenPenaltyRepositoryImpl.java @@ -57,4 +57,14 @@ public class HiddenPenaltyRepositoryImpl public void saveFeedback(HiddenPenaltyFeedbackDO feedbackDO) { hiddenPenaltyFeedbackMapper.insert(feedbackDO); } + + @Override + public Long countByUserId(Long userId) { + return hiddenPenaltyMapper.countByUserId(userId); + } + + @Override + public Long countFeedbackByUserId(Long userId) { + return hiddenPenaltyMapper.countFeedbackByUserId(userId); + } }