From 83bf74a3a5430b5f2aef3a0d03c558c73d8366e1 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Thu, 6 Feb 2025 21:37:57 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0=E4=BF=A1=E4=BB=BB?=
 =?UTF-8?q?=E8=AE=BE=E5=A4=87=E9=80=BB=E8=BE=91=EF=BC=8C=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E5=A4=9A=E8=AE=BE=E5=A4=87=E6=A3=80=E6=9F=A5=E5=8F=8A=E7=94=A8?=
 =?UTF-8?q?=E6=88=B7=E9=AA=8C=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../coal/session/service/SessionService.java  | 13 ++++--
 .../service/TrustDeviceService.java           | 41 +++++++++----------
 2 files changed, 30 insertions(+), 24 deletions(-)

diff --git a/src/main/java/cn/lihongjie/coal/session/service/SessionService.java b/src/main/java/cn/lihongjie/coal/session/service/SessionService.java
index d84c3212..a55e95a7 100644
--- a/src/main/java/cn/lihongjie/coal/session/service/SessionService.java
+++ b/src/main/java/cn/lihongjie/coal/session/service/SessionService.java
@@ -251,6 +251,11 @@ public class SessionService {
                         userService
                                 .findByUsername(dto.getUsername())
                                 .orElseThrow(() -> new BizException("用户名或者密码错误"));
+
+
+                if (!userService.isValidPassword(dto.getPassword(), user.getPassword())) {
+                    throw new BizException("用户名或者密码错误");
+                }
             }
 
             if (StringUtils.equalsIgnoreCase(dto.getLoginType(), "1")) {
@@ -290,6 +295,10 @@ public class SessionService {
                 user = trustDevice.get().getUser();
             }
 
+            if (user == null){
+                throw new BizException("用户不存在");
+            }
+
             if (user.isDisabled()) {
                 throw new BizException("用户被禁用");
             }
@@ -304,9 +313,7 @@ public class SessionService {
                 throw new BizException("用户所在机构已过期");
             }
 
-            if (!userService.isValidPassword(dto.getPassword(), user.getPassword())) {
-                throw new BizException("用户名或者密码错误");
-            }
+
 
             LoginUserEntity entity = new LoginUserEntity();
             entity.setEnv(dto.getEnv());
diff --git a/src/main/java/cn/lihongjie/coal/trustDevice/service/TrustDeviceService.java b/src/main/java/cn/lihongjie/coal/trustDevice/service/TrustDeviceService.java
index 06ab2adc..6e03e138 100644
--- a/src/main/java/cn/lihongjie/coal/trustDevice/service/TrustDeviceService.java
+++ b/src/main/java/cn/lihongjie/coal/trustDevice/service/TrustDeviceService.java
@@ -32,6 +32,7 @@ import org.springframework.data.jpa.domain.Specification;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.List;
 import java.util.Optional;
 
 @Service
@@ -94,8 +95,8 @@ public class TrustDeviceService extends BaseService<TrustDeviceEntity, TrustDevi
 
         UserEntity userEntity = userService.get(user.getId());
 
-        Optional<TrustDeviceEntity> one =
-                this.repository.findOne(
+        List<TrustDeviceEntity> all =
+                this.repository.findAll(
                         new Specification<TrustDeviceEntity>() {
                             @Override
                             public Predicate toPredicate(
@@ -104,14 +105,17 @@ public class TrustDeviceService extends BaseService<TrustDeviceEntity, TrustDevi
                                     CriteriaBuilder criteriaBuilder) {
 
                                 return criteriaBuilder.and(
-                                        criteriaBuilder.equal(
-                                                root.get("user").get("id"), userEntity.getId()),
                                         criteriaBuilder.equal(root.get("code"), dto.getCode()));
                             }
                         });
 
-        if (one.isPresent()) {
-            return;
+        if (!all.isEmpty()) {
+
+            if (all.stream().allMatch(x -> x.getUser().getId().equals(userEntity.getId()))) {
+                return;
+            } else {
+                throw new BizException("设备已被其他人添加，无法重复添加");
+            }
         }
 
         TrustDeviceEntity entity = new TrustDeviceEntity();
@@ -125,20 +129,15 @@ public class TrustDeviceService extends BaseService<TrustDeviceEntity, TrustDevi
 
     public Optional<TrustDeviceEntity> getByCode(String trustDeviceCode) {
 
-
-
-        return this.repository
-                .findOne(
-                        new Specification<TrustDeviceEntity>() {
-                            @Override
-                            public Predicate toPredicate(
-                                    Root<TrustDeviceEntity> root,
-                                    CriteriaQuery<?> query,
-                                    CriteriaBuilder criteriaBuilder) {
-                                return criteriaBuilder.equal(root.get("code"), trustDeviceCode);
-                            }
-                        });
-
-
+        return this.repository.findOne(
+                new Specification<TrustDeviceEntity>() {
+                    @Override
+                    public Predicate toPredicate(
+                            Root<TrustDeviceEntity> root,
+                            CriteriaQuery<?> query,
+                            CriteriaBuilder criteriaBuilder) {
+                        return criteriaBuilder.equal(root.get("code"), trustDeviceCode);
+                    }
+                });
     }
 }