diff --git a/ai-root-cloud-statistics/src/main/java/com/qniao/rootcloudstatistics/until/OSSUtils.java b/ai-root-cloud-statistics/src/main/java/com/qniao/rootcloudstatistics/until/OSSUtils.java index 437ad44..96f1541 100644 --- a/ai-root-cloud-statistics/src/main/java/com/qniao/rootcloudstatistics/until/OSSUtils.java +++ b/ai-root-cloud-statistics/src/main/java/com/qniao/rootcloudstatistics/until/OSSUtils.java @@ -2,6 +2,9 @@ package com.qniao.rootcloudstatistics.until; import com.aliyun.oss.OSS; import com.aliyun.oss.OSSClient; +import com.ctrip.framework.apollo.build.ApolloInjector; +import com.ctrip.framework.apollo.core.enums.Env; +import com.ctrip.framework.apollo.util.ConfigUtil; import java.io.IOException; import java.io.InputStream; @@ -39,9 +42,12 @@ public class OSSUtils { String[] list = url.split("\\."); String type = list[list.length - 1]; - - String folderName = "cloudprint"; + Env apolloEnv = ApolloInjector.getInstance(ConfigUtil.class).getApolloEnv(); String bucketName = "qncloudprintfiletest"; + if (Env.PRO.equals(apolloEnv)) { + bucketName = "qncloudprintfile"; + } + String folderName = "cloudprint"; String objectName = folderName + "/" + UUID.randomUUID() + "." + type; OSS ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret); // 填写网络流地址。