renun/zfile
1
0
Fork 0
mirror of https://github.com/zfile-dev/zfile.git synced 2025-04-19 05:34:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 修复多吉云令牌无法自动刷新的 bug

Browse Source
This commit is contained in:
zhaojun
2023-05-27 16:50:31 +08:00
parent ac3b4283a3
commit ecf85dfe9e
1 changed files with 34 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/main/java/im/zhaojun/zfile/module/storage/service/impl/DogeCloudServiceImpl.java
View File

@@ -11,6 +11,7 @@ import com.amazonaws.auth.AWSCredentialsProvider;
import com.amazonaws.auth.BasicSessionCredentials;
import com.amazonaws.client.builder.AwsClientBuilder;
import com.amazonaws.services.s3.AmazonS3ClientBuilder;
import com.amazonaws.util.ValidationUtils;
import im.zhaojun.zfile.module.storage.model.enums.StorageTypeEnum;
import im.zhaojun.zfile.module.storage.model.param.DogeCloudParam;
import im.zhaojun.zfile.module.storage.service.base.AbstractS3BaseFileService;
@@ -20,6 +21,8 @@ import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Service;
import java.util.function.Supplier;
/**
* @author zhaojun
*/
@@ -28,13 +31,18 @@ import org.springframework.stereotype.Service;
@Slf4j
public class DogeCloudServiceImpl extends AbstractS3BaseFileService<DogeCloudParam> implements RefreshTokenService {
private BasicSessionCredentials awsCredentials;
public void updateAwsCredentials() {
awsCredentials = new BasicSessionCredentials(param.getS3AccessKey(), param.getS3SecretKey(), param.getS3SessionToken());
}
@Override
public void init() {
refreshAccessToken();
BasicSessionCredentials awsCredentials = new BasicSessionCredentials(param.getS3AccessKey(), param.getS3SecretKey(), param.getS3SessionToken());
updateAwsCredentials();
s3Client = AmazonS3ClientBuilder.standard()
.withCredentials(new AWSStaticCredentialsProvider(awsCredentials))
.withCredentials(new DynamicAWSCredentialsProvider(() -> awsCredentials))
.withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration(
param.getEndPoint(),
"automatic"))
@@ -83,7 +91,7 @@ public class DogeCloudServiceImpl extends AbstractS3BaseFileService<DogeCloudPar
param.setBucketName(buckets.getString("s3Bucket"));
param.setEndPoint(buckets.getString("s3Endpoint"));
updateAwsCredentials();
}
@@ -94,3 +102,25 @@ public class DogeCloudServiceImpl extends AbstractS3BaseFileService<DogeCloudPar
}
class DynamicAWSCredentialsProvider implements AWSCredentialsProvider {
private final Supplier<AWSCredentials> supplier;
public DynamicAWSCredentialsProvider(Supplier<AWSCredentials> supplier) {
this.supplier = supplier;
}
@Override
public AWSCredentials getCredentials() {
return ValidationUtils.assertNotNull(supplier.get(), "credentials");
}
@Override
public void refresh() {
}
}