浏览代码

兼容申万系统缩略图

kobe6258 2 天之前
父节点
当前提交
14a3b87c82
共有 2 个文件被更改,包括 9 次插入8 次删除
  1. 3 3
      controllers/data_manage/edb_info.go
  2. 6 5
      services/file.go

+ 3 - 3
controllers/data_manage/edb_info.go

@@ -6746,14 +6746,14 @@ func (this *EdbInfoController) ChartImageSetBySvg() {
 		br.Ret = 408
 		return
 	}
-	picType := this.GetString("PicType")
+	backendGenerate,_ := this.GetBool("BackendGenerate",false)
 	imgData := this.GetString("Img")
 	if imgData == "" {
 		br.Msg = "图片参数错误"
 		br.ErrMsg = "图片参数错误,Img Is Empty"
 		return
 	}
-	if strings.ToLower(picType) == "png" {
+	if !backendGenerate {
 		b, _ := regexp.MatchString(`^data:\s*image\/(\w+);base64,`, imgData)
 		if !b {
 			br.Msg = "图片格式不正确"
@@ -6773,7 +6773,7 @@ func (this *EdbInfoController) ChartImageSetBySvg() {
 	}
 
 	// 通过svg图片生成图片资源地址
-	resourceUrl, err, errMsg := services.GetResourceUrlBySvgImg(imgData,picType)
+	resourceUrl, err, errMsg := services.GetResourceUrlBySvgImg(imgData,backendGenerate)
 	if err != nil {
 		br.Msg = errMsg
 		br.ErrMsg = err.Error()

+ 6 - 5
services/file.go

@@ -108,7 +108,7 @@ func saveToFileByHeader(fileHeader *multipart.FileHeader, tofile string) error {
 // @return resourceUrl string
 // @return err error
 // @return errMsg string
-func GetResourceUrlBySvgImg(imgData string, picType ...string) (resourceUrl string, err error, errMsg string) {
+func GetResourceUrlBySvgImg(imgData string, backendGenerate ...bool) (resourceUrl string, err error, errMsg string) {
 	errMsg = "图表保存失败"
 	uploadDir := "static/images/"
 	if !utils.FileIsExist(uploadDir) {
@@ -118,15 +118,16 @@ func GetResourceUrlBySvgImg(imgData string, picType ...string) (resourceUrl stri
 			return
 		}
 	}
-	var fileName, outFileName, picTypeStr string
+	var fileName, outFileName string
+	var generte bool
 	uploadDir = "static/images/"
 	randStr := utils.GetRandStringNoSpecialChar(28)
 	outFileName = randStr + ".png"
-	if len(picType) > 0 {
-		picTypeStr = picType[0]
+	if len(backendGenerate) > 0 {
+		generte = backendGenerate[0]
 	}
 
-	if strings.ToLower(picTypeStr) == "png" {
+	if !generte {
 		err = utils.SaveBase64ToFile(imgData, outFileName)
 		if err != nil {
 			err = errors.New("图片保存失败,Err:" + err.Error())