Преглед изворни кода

update oss-console UserKey api

reghao пре 1 година
родитељ
комит
9101371359

+ 1 - 0
oss-console/src/main/java/cn/reghao/oss/console/app/db/repository/UserKeyRepository.java

@@ -9,4 +9,5 @@ import org.springframework.data.jpa.repository.JpaRepository;
  */
 public interface UserKeyRepository extends JpaRepository<UserKey, Integer> {
     UserKey findByCreateBy(int createBy);
+    UserKey findByAccessKeyId(String  accessKeyId);
 }

+ 4 - 4
oss-console/src/main/java/cn/reghao/oss/console/app/service/UserKeyService.java

@@ -45,15 +45,15 @@ public class UserKeyService {
     }
 
     public Result auth(KeyAuthDto keyAuthDto) {
-        int loginUser = UserContext.getUserId();
-        UserKey userKey = userKeyRepository.findByCreateBy(loginUser);
+        String accessKeyId = keyAuthDto.getAccessKeyId();
+        UserKey userKey = userKeyRepository.findByAccessKeyId(accessKeyId);
         if (userKey == null) {
             return Result.fail("key not exist");
         }
 
-        String accessKeyId = keyAuthDto.getAccessKeyId();
         String accessKeySecret = keyAuthDto.getAccessKeySecret();
-        if (userKey.getAccessKeyId().equals(accessKeyId) && userKey.getAccessKeySecret().equals(accessKeySecret)) {
+        if (userKey.getAccessKeySecret().equals(accessKeySecret)) {
+            int loginUser = userKey.getCreateBy();
             String token = getTokenByUserId(loginUser);
             return Result.success(token);
         }