liaijie 3 лет назад
Родитель
Сommit
76d190065e

+ 6 - 0
src/main/java/com/izouma/nineth/repo/UserBankCardRepo.java

@@ -4,6 +4,7 @@ import com.izouma.nineth.dto.UserBankCard;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
 import org.springframework.data.jpa.repository.Modifying;
+import org.springframework.data.jpa.repository.Query;
 
 import javax.transaction.Transactional;
 import java.util.List;
@@ -14,4 +15,9 @@ public interface UserBankCardRepo extends JpaRepository<UserBankCard, Long>, Jpa
     @Transactional
     @Modifying
     int deleteByUserId(Long userId);
+
+    @Query("select count(t) from UserBankCard t where t.userId = ?1")
+    @Modifying
+    @Transactional
+    Integer findByUserIds(Long id);
 }

+ 2 - 2
src/main/java/com/izouma/nineth/service/UserService.java

@@ -713,8 +713,8 @@ public class UserService {
         User user = userRepo.findById(id).orElseThrow(new BusinessException("用户不存在"));
         user.setPassword(null);
         user.setTradeCode(null);
-        List<UserBankCard> byUserId = userBankCardRepo.findByUserId(id);
-        if (byUserId==null || byUserId.size()==0){
+        Integer byUserId = userBankCardRepo.findByUserIds(id);
+        if (byUserId==0){
             user.setIsUserBankCard(false);
         }else {
             user.setIsUserBankCard(true);