105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
|
package coryCommon
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/gogf/gf/v2/os/gfile"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"math/rand"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Base64ToImgFunc 将base64转成图片保存在本地
|
|||
|
|
func Base64ToImgFunc(base64Str string, numTyoe string, cdPath string) (outputPath string, err error) {
|
|||
|
|
// 获取当前时间+随机数得到文件名
|
|||
|
|
currentTime := time.Now()
|
|||
|
|
timestamp := currentTime.UnixNano() / int64(time.Millisecond)
|
|||
|
|
randomNum := rand.Intn(1000)
|
|||
|
|
uniqueFileName := fmt.Sprintf("%d_%d", timestamp, randomNum)
|
|||
|
|
// 用户指定的本地文件路径
|
|||
|
|
//ynr := Ynr(Portrait + "/")
|
|||
|
|
ynr := Ynr(cdPath + "/")
|
|||
|
|
path := ynr + uniqueFileName + ".png"
|
|||
|
|
path = filepath.ToSlash(path)
|
|||
|
|
// Base64编码的图像字符串
|
|||
|
|
b64 := "data:image/png;"
|
|||
|
|
base64Image := ""
|
|||
|
|
if strings.Contains(base64Str, "base64,") { // 判断是否有【base64,】如果有就替换
|
|||
|
|
base64Image = strings.Replace(base64Str, base64Str[:strings.Index(base64Str, "base64,")], b64, 1)
|
|||
|
|
} else {
|
|||
|
|
base64Image = b64 + "base64," + base64Str
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用函数将Base64图像保存到指定路径
|
|||
|
|
err = SaveBase64ImageToFile(base64Image, path)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
if numTyoe == "1" {
|
|||
|
|
outputPath = strings.Replace(path, "resource/public", "file", 1)
|
|||
|
|
return
|
|||
|
|
} else if numTyoe == "2" {
|
|||
|
|
outputPath = strings.Replace(path, "resource/public", "wxfile", 1)
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
err = errors.New("第二参数只能为1 or 2！")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Base64ToFileFunc(base64Str string, filePath string, suffix string, numTyoe string) (outputPath string, err error) {
|
|||
|
|
|
|||
|
|
// 获取当前时间+随机数得到文件名
|
|||
|
|
timestamp := time.Now().UnixNano() / int64(time.Millisecond)
|
|||
|
|
uniqueFileName := fmt.Sprintf("%d_%d", timestamp, rand.Intn(1000))
|
|||
|
|
// 用户指定的本地文件路径,filePath路径最后必须是/
|
|||
|
|
path := filepath.ToSlash(filePath + uniqueFileName + suffix)
|
|||
|
|
// 调用函数将Base64图像保存到指定路径
|
|||
|
|
err = SaveBase64ImageToFile(base64Str, path)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
if numTyoe == "1" {
|
|||
|
|
outputPath = strings.Replace(path, "resource/public", "file", 1)
|
|||
|
|
return
|
|||
|
|
} else if numTyoe == "2" {
|
|||
|
|
outputPath = strings.Replace(path, "resource/public", "wxfile", 1)
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
err = errors.New("第二参数只能为1 or 2！")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SaveBase64ImageToFile 将Base64编码的图像保存到指定的本地文件路径
|
|||
|
|
func SaveBase64ImageToFile(base64Image string, outputPath string) error {
|
|||
|
|
if len(outputPath) > 0 && outputPath[0] == '/' {
|
|||
|
|
outputPath = outputPath[1:]
|
|||
|
|
}
|
|||
|
|
getwd, _ := os.Getwd()
|
|||
|
|
outputPath = gfile.Join(getwd, outputPath)
|
|||
|
|
outputPath = strings.ReplaceAll(outputPath, "\\", "/")
|
|||
|
|
// 1. 解码Base64字符串
|
|||
|
|
parts := strings.Split(base64Image, ",")
|
|||
|
|
if len(parts) != 2 {
|
|||
|
|
return errors.New("Base64字符串格式不正确！")
|
|||
|
|
}
|
|||
|
|
data, err := base64.StdEncoding.DecodeString(parts[1])
|
|||
|
|
if err != nil {
|
|||
|
|
return errors.New("转码错误！")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 将字节数组保存为图像文件
|
|||
|
|
err = ioutil.WriteFile(outputPath, data, 0644)
|
|||
|
|
if err != nil {
|
|||
|
|
return errors.New("报错图像失败！")
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|