Browse Source

fix:常量拆分出来称为配置项

Roc 1 year ago
parent
commit
539417a9c6
2 changed files with 3 additions and 1 deletions
  1. 1 1
      services/oss.go
  2. 2 0
      utils/config.go

+ 1 - 1
services/oss.go

@@ -186,7 +186,7 @@ func NewSTSToken() (item *STSToken, err error) {
 	}
 	request := sts.CreateAssumeRoleRequest()
 	request.Scheme = "https"
-	request.RegionId = "cn-shanghai"
+	request.RegionId = utils.RegionId
 	request.RoleArn = utils.RoleArn
 	now := time.Now().Format(utils.FormatDateTimeUnSpace)
 	request.RoleSessionName = utils.RoleSessionName + now

+ 2 - 0
utils/config.go

@@ -129,6 +129,7 @@ var (
 
 // 阿里云oss前端上传用
 var (
+	RegionId           string
 	RoleArn            string
 	RoleSessionName    string
 	RAMAccessKeyId     string
@@ -309,6 +310,7 @@ func init() {
 
 	// OSS相关(前端使用)
 	{
+		RegionId = config["region_id"]
 		RoleArn = config["role_arn"]
 		RoleSessionName = config["role_session_name"]
 		RAMAccessKeyId = config["ram_access_key_id"]