package com.ard.sample.utils;
|
|
import com.ard.oss.model.CredentialsDTO;
|
import io.minio.credentials.AssumeRoleProvider;
|
import lombok.extern.slf4j.Slf4j;
|
import okhttp3.ConnectionPool;
|
import okhttp3.OkHttpClient;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Nonnull;
|
import javax.annotation.Nullable;
|
import java.security.NoSuchAlgorithmException;
|
import java.security.ProviderException;
|
import java.util.concurrent.TimeUnit;
|
|
import static com.ard.sample.config.DockConfiguration.*;
|
|
/**
|
* 获取http的STS凭证的工具类，用于简化STS临时凭证的获取过程。
|
*/
|
@Slf4j
|
@Component
|
public class MinioStsUtils {
|
|
public CredentialsDTO getCredentials() {
|
try {
|
// 构建适配HTTP协议的OkHttpClient（移除所有SSL相关配置）
|
OkHttpClient httpClient = new OkHttpClient.Builder()
|
// 保留原代码的连接池配置（禁用连接复用，如需复用可删除此行）
|
.connectionPool(new ConnectionPool(5, 30, TimeUnit.SECONDS))
|
.connectTimeout(10, TimeUnit.SECONDS) // 连接超时
|
.readTimeout(30, TimeUnit.SECONDS) // 读取超时
|
.writeTimeout(30, TimeUnit.SECONDS) // 写入超时
|
.build();
|
|
// 构建 AssumeRoleProvider，使用简化后的HTTP客户端
|
String roleSessionName = "dji-sts-session-" + System.currentTimeMillis(); // 唯一会话名称（必需，避免重复）
|
AssumeRoleProvider provider = new AssumeRoleProvider(
|
endpoint, // 注意：此处endpoint需改为HTTP地址（如 http://sts.example.com）
|
accessKey, // 主账号AccessKey
|
secretKey, // 主账号SecretKey
|
Math.toIntExact(expire), // 凭证过期时间（转int）
|
null, // 可选参数（如RoleArn，根据实际SDK调整）
|
region, // 地域（如cn-hangzhou）
|
null,
|
roleSessionName,
|
null, // 其他可选参数
|
httpClient // 传入HTTP版本的OkHttpClient
|
);
|
|
log.info("生成临时凭证条件：{}",provider);
|
|
// 获取 STS 临时凭证
|
var stsCredentials = provider.fetch();
|
return new CredentialsDTO(stsCredentials, expire);
|
|
} catch (ProviderException e) {
|
log.error("Failed to fetch/parse STS response (HTTP access)", e);
|
throw new RuntimeException("STS response parse failed", e);
|
} catch (NoSuchAlgorithmException e) {
|
log.error("Failed to fetch/parse STS response (HTTP access)", e);
|
throw new RuntimeException(e);
|
}
|
}
|
}
|
