package cn.lili.utils;
|
|
import cn.lili.cos.CDNConfigProperty;
|
import com.tencentcloudapi.cdn.v20180606.CdnClient;
|
import com.tencentcloudapi.cdn.v20180606.models.DescribePushTasksRequest;
|
import com.tencentcloudapi.cdn.v20180606.models.DescribePushTasksResponse;
|
import com.tencentcloudapi.cdn.v20180606.models.PushUrlsCacheRequest;
|
import com.tencentcloudapi.cdn.v20180606.models.PushUrlsCacheResponse;
|
import com.tencentcloudapi.common.Credential;
|
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Arrays;
|
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class TencentCdnUtil {
|
private final CDNConfigProperty cdnConfigProperty;
|
|
public void pushUrlsCache(String[] urls) {
|
try {
|
// 预处理URLs，添加CDN域名
|
String[] processedUrls = processUrlsWithCdnDomain(urls);
|
|
log.info("cdn链接：{}", Arrays.toString(processedUrls));
|
// 为了保护密钥安全，建议将密钥设置在环境变量中或者配置文件中，请参考本文凭证管理章节。
|
// 硬编码密钥到代码中有可能随代码泄露而暴露，有安全隐患，并不推荐。
|
Credential cred = new Credential(cdnConfigProperty.getSecretId(), cdnConfigProperty.getSecretKey());
|
CdnClient client = new CdnClient(cred, "ap-guangzhou");
|
|
PushUrlsCacheRequest req = new PushUrlsCacheRequest();
|
req.setUrls(processedUrls);
|
|
PushUrlsCacheResponse resp = client.PushUrlsCache(req);
|
|
log.info("cdn预热返回：{}", client.gson.toJson(resp));
|
|
} catch (TencentCloudSDKException e) {
|
log.warn("cdn预热失败：{}", e.toString());
|
}
|
}
|
|
/**
|
* 处理URL列表，在每个URL前添加CDN域名
|
*
|
* @param urls 原始URL列表
|
* @return 添加了CDN域名的完整URL列表
|
*/
|
public String[] processUrlsWithCdnDomain(String[] urls) {
|
if (urls == null || urls.length == 0) {
|
return urls;
|
}
|
|
String cdnDomain = cdnConfigProperty.getCdnDomain();
|
if (cdnDomain == null || cdnDomain.isEmpty()) {
|
return urls;
|
}
|
|
String[] processedUrls = new String[urls.length];
|
for (int i = 0; i < urls.length; i++) {
|
// 如果URL已经包含协议，则直接拼接
|
if (urls[i].startsWith("http://") || urls[i].startsWith("https://")) {
|
processedUrls[i] = urls[i];
|
} else {
|
// 确保cdnDomain末尾没有斜杠，urls开头没有斜杠
|
String domain = cdnDomain.endsWith("/") ? cdnDomain.substring(0, cdnDomain.length() - 1) : cdnDomain;
|
String path = urls[i].startsWith("/") ? urls[i].substring(1) : urls[i];
|
processedUrls[i] = domain + "/" + path;
|
}
|
}
|
|
return processedUrls;
|
}
|
|
}
|
