|
|
@@ -44,19 +44,20 @@ public class UserKeyService {
|
|
|
}
|
|
|
|
|
|
public Result auth(KeyAuthDto keyAuthDto) {
|
|
|
- String accessKeyId = keyAuthDto.getAccessKeyId();
|
|
|
- UserKey userKey = userKeyRepository.findByAccessKeyId(accessKeyId);
|
|
|
+ int loginUser = UserContext.getUserId();
|
|
|
+ UserKey userKey = userKeyRepository.findByCreateBy(loginUser);
|
|
|
if (userKey == null) {
|
|
|
return Result.fail("key not exist");
|
|
|
}
|
|
|
|
|
|
- if (!userKey.getAccessKeySecret().equals(keyAuthDto.getAccessKeySecret())) {
|
|
|
- return Result.fail("secret not matched");
|
|
|
+ String accessKeyId = keyAuthDto.getAccessKeyId();
|
|
|
+ String accessKeySecret = keyAuthDto.getAccessKeySecret();
|
|
|
+ if (userKey.getAccessKeyId().equals(accessKeyId) && userKey.getAccessKeySecret().equals(accessKeySecret)) {
|
|
|
+ String token = getToken(loginUser);
|
|
|
+ return Result.success(token);
|
|
|
}
|
|
|
|
|
|
- int loginUser = userKey.getCreateBy();
|
|
|
- String token = getToken(loginUser);
|
|
|
- return Result.success(token);
|
|
|
+ return Result.fail("secret not matched");
|
|
|
}
|
|
|
|
|
|
public String getToken(int loginUser) {
|
|
|
@@ -74,11 +75,18 @@ public class UserKeyService {
|
|
|
return ossPayload.getUserId();
|
|
|
}
|
|
|
|
|
|
- public void delete(String accessKeyId) {
|
|
|
- UserKey userKey = userKeyRepository.findByAccessKeyId(accessKeyId);
|
|
|
- if (userKey != null) {
|
|
|
- userKeyRepository.delete(userKey);
|
|
|
+ public void regenerate() {
|
|
|
+ int loginUser = UserContext.getUserId();
|
|
|
+ UserKey userKey = userKeyRepository.findByCreateBy(loginUser);
|
|
|
+ if (userKey == null) {
|
|
|
+ return;
|
|
|
}
|
|
|
+
|
|
|
+ String accessKeyId = RandomString.getString(8);
|
|
|
+ String accessKeySecret = RandomString.getString(18);
|
|
|
+ userKey.setAccessKeyId(accessKeyId);
|
|
|
+ userKey.setAccessKeySecret(accessKeySecret);
|
|
|
+ userKeyRepository.save(userKey);
|
|
|
}
|
|
|
|
|
|
public List<UserKeyVo> getUserKeys() {
|