ziwen 1 year ago
parent
commit
d6de6b7328
3 changed files with 32 additions and 9 deletions
  1. 25 9
      controllers/resource.go
  2. 1 0
      controllers/sys_role.go
  3. 6 0
      utils/constants.go

+ 25 - 9
controllers/resource.go

@@ -722,16 +722,32 @@ func (this *ResourceController) OssSTSToken() {
 		this.Data["json"] = br
 		this.ServeJSON()
 	}()
-	resp, err := services.GetOssSTSToken()
-	if err != nil {
-		br.Msg = "获取失败"
-		br.ErrMsg = "获取STSToken失败, Err: " + err.Error()
-		return
+
+	source, _ := this.GetInt("StorageSource")
+
+	if source == utils.STORAGESOURCE_OSS {
+		resp, err := services.GetOssSTSToken()
+		if err != nil {
+			br.Msg = "获取失败"
+			br.ErrMsg = "获取STSToken失败, Err: " + err.Error()
+			return
+		}
+		br.Data = resp
+		br.Msg = "获取成功"
+		br.Ret = 200
+		br.Success = true
+	} else if source == utils.STORAGESOURCE_MINIO {
+		resp, err := services.GetMinIOSTSToken()
+		if err != nil {
+			br.Msg = "获取失败"
+			br.ErrMsg = "获取STSToken失败, Err: " + err.Error()
+			return
+		}
+		br.Data = resp
+		br.Msg = "获取成功"
+		br.Ret = 200
+		br.Success = true
 	}
-	br.Msg = "获取成功"
-	br.Ret = 200
-	br.Success = true
-	br.Data = resp
 }
 
 // WechatWarning 小程序前端预警提示

+ 1 - 0
controllers/sys_role.go

@@ -3,6 +3,7 @@ package controllers
 import (
 	"eta/eta_mobile/models"
 	"eta/eta_mobile/models/system"
+	"eta/eta_mobile/utils"
 )
 
 type SysRoleController struct {

+ 6 - 0
utils/constants.go

@@ -371,3 +371,9 @@ const (
 
 // BusinessCodeSalt 商家编码盐值
 const BusinessCodeSalt = "dr7WY0OZgGR7upw1"
+
+// 对象存储客户端
+const (
+	STORAGESOURCE_OSS   = 1 //阿里云OSS
+	STORAGESOURCE_MINIO = 2 //MinIo
+)