diff --git a/src/main/java/im/zhaojun/common/config/GlobalScheduleTask.java b/src/main/java/im/zhaojun/common/config/GlobalScheduleTask.java index b6b1166..f74bb63 100644 --- a/src/main/java/im/zhaojun/common/config/GlobalScheduleTask.java +++ b/src/main/java/im/zhaojun/common/config/GlobalScheduleTask.java @@ -80,9 +80,9 @@ public class GlobalScheduleTask { OneDriveToken refreshToken; if (Objects.equals(storageType, StorageTypeEnum.ONE_DRIVE_CHINA)) { - refreshToken = oneDriveChinaService.getRefreshToken(); + refreshToken = oneDriveChinaService.getRefreshToken(storageType); } else { - refreshToken = oneDriveService.getRefreshToken(); + refreshToken = oneDriveService.getRefreshToken(storageType); } diff --git a/src/main/java/im/zhaojun/onedrive/common/service/AbstractOneDriveService.java b/src/main/java/im/zhaojun/onedrive/common/service/AbstractOneDriveService.java index 9270b29..d00a250 100644 --- a/src/main/java/im/zhaojun/onedrive/common/service/AbstractOneDriveService.java +++ b/src/main/java/im/zhaojun/onedrive/common/service/AbstractOneDriveService.java @@ -43,9 +43,9 @@ public abstract class AbstractOneDriveService { @Resource private StorageConfigRepository storageConfigRepository; - public OneDriveToken getRefreshToken() { + public OneDriveToken getRefreshToken(StorageTypeEnum storageType) { StorageConfig refreshStorageConfig = - storageConfigRepository.findByTypeAndKey(StorageTypeEnum.ONE_DRIVE, StorageConfigConstant.REFRESH_TOKEN_KEY); + storageConfigRepository.findByTypeAndKey(storageType, StorageConfigConstant.REFRESH_TOKEN_KEY); String param = "client_id=" + getClientId() + "&redirect_uri=" + getRedirectUri() +