[fix]异常处理

2026-04-23 02:48:34 +08:00 · 2024-06-06 20:46:35 +08:00
parent 41b61ebf1e
commit 4cee5ae537
24 changed files with 801 additions and 600 deletions
--- a/pkg/tool/file.go
+++ b/pkg/tool/file.go
@@ -2,11 +2,189 @@ package tool

 import (
 	"crypto/md5"
+	"encoding/base64"
 	"encoding/hex"
+	"encoding/json"
+	"errors"
 	"io"
+	"mime/multipart"
 	"os"
+	"pandax/pkg/global"
+	"path"
+	"strings"
+	"time"
 )

+type Local struct {
+	Path string
+}
+
+const (
+	DevicePath        = "DevicePath"        //设备相关
+	RulePath          = "RulePath"          //规则链截图及json
+	VisualPath        = "VisualPath"        //组态截图及json
+	ThreePath         = "ThreePath"         //3D组态截图及json
+	VisualGalleryPath = "VisualGalleryPath" //组态图库
+	ThreeModelPath    = "ThreeModelPath"    //3D模型和缩略图
+	ThreeMapPath      = "ThreeMapPath"      //3d贴图
+	OtherPath         = "OtherPath"         //其他
+)
+
+var PathMap = map[string]string{
+	DevicePath:        "uploads/device",
+	RulePath:          "uploads/rule",
+	VisualPath:        "uploads/visual",
+	ThreePath:         "uploads/three",
+	VisualGalleryPath: "uploads/visual/gallery",
+	ThreeModelPath:    "uploads/three/model",
+	ThreeMapPath:      "uploads/three/map",
+	OtherPath:         "uploads/other",
+}
+
+func init() {
+	for _, path := range PathMap {
+		go os.MkdirAll(path, 0755)
+	}
+}
+
+func GetFilePath(fielType string) string {
+	if path, ok := PathMap[fielType]; ok {
+		return path
+	}
+	return ""
+}
+
+//@object: *Local
+//@function: UploadFile
+//@description: 上传文件
+//@param: file *multipart.FileHeader
+//@return: string, string, error
+
+func (local *Local) UploadFile(file *multipart.FileHeader) (string, string, error) {
+	// 读取文件后缀
+	ext := path.Ext(file.Filename)
+	// 读取文件名并加密
+	name := strings.TrimSuffix(file.Filename, ext)
+	name = MD5V([]byte(name))
+	// 拼接新文件名
+	filename := name + "_" + time.Now().Format("20060102150405") + ext
+	// 尝试创建此路径
+	mkdirErr := os.MkdirAll(local.Path, os.ModePerm)
+	if mkdirErr != nil {
+		global.Log.Error("function os.MkdirAll() Filed", mkdirErr.Error())
+		return "", "", errors.New("function os.MkdirAll() Filed, err:" + mkdirErr.Error())
+	}
+	// 拼接路径和文件名
+	p := local.Path + "/" + filename
+
+	f, openError := file.Open() // 读取文件
+	if openError != nil {
+		global.Log.Error("function file.Open() Filed", openError.Error())
+		return "", "", errors.New("function file.Open() Filed, err:" + openError.Error())
+	}
+	defer f.Close() // 创建文件 defer 关闭
+
+	out, createErr := os.Create(p)
+	if createErr != nil {
+		global.Log.Error("function os.Create() Filed", createErr.Error())
+		return "", "", errors.New("function os.Create() Filed, err:" + createErr.Error())
+	}
+	defer out.Close() // 创建文件 defer 关闭
+
+	_, copyErr := io.Copy(out, f) // 传输（拷贝）文件
+	if copyErr != nil {
+		global.Log.Error("function io.Copy() Filed", copyErr.Error())
+		return "", "", errors.New("function io.Copy() Filed, err:" + copyErr.Error())
+	}
+	return p, filename, nil
+}
+
+func (local *Local) Base64ToFile(name, base64Str string) (path string, fileName string, err error) {
+	if strings.Contains(base64Str, "data:image") {
+		base64Str = strings.TrimPrefix(base64Str, "data:image/png;base64,")
+	}
+
+	imgData, err := base64.StdEncoding.DecodeString(base64Str)
+	filename := name + ".png"
+	path = local.Path + "/" + filename
+	_, err = os.Stat(path)
+	if err == nil {
+		// 文件存在,删除原有文件
+		os.Remove(fileName)
+	}
+
+	file, err := os.Create(path)
+	if err != nil {
+		return path, "", errors.New("function os.Create() Filed, err:" + err.Error())
+	}
+	defer file.Close() // 创建文件 defer 关闭
+
+	_, err = file.Write(imgData)
+	if err != nil {
+		return path, "", errors.New("写文件错误:" + err.Error())
+	}
+	return path, filename, nil
+}
+
+func (local *Local) JsonToFile(name, jsonStr string) (path string, fileName string, err error) {
+	var data map[string]interface{}
+	err = json.Unmarshal([]byte(jsonStr), &data)
+	if err != nil {
+		return "", "", errors.New("json 格式错误:" + err.Error())
+	}
+	jsonData, err := json.MarshalIndent(data, "", "  ")
+	if err != nil {
+		return "", "", errors.New("json 格式化错误:" + err.Error())
+	}
+
+	filename := name + ".json"
+	path = local.Path + "/" + filename
+	_, err = os.Stat(path)
+	if err == nil {
+		// 文件存在,删除原有文件
+		os.Remove(fileName)
+	}
+
+	file, err := os.Create(path)
+	if err != nil {
+		return path, "", errors.New("function os.Create() Filed, err:" + err.Error())
+	}
+	defer file.Close() // 创建文件 defer 关闭
+
+	_, err = file.Write(jsonData)
+	if err != nil {
+		return path, "", errors.New("写文件错误:" + err.Error())
+	}
+	return path, filename, nil
+}
+
+func (local *Local) GetJsonFile(name string) (map[string]any, error) {
+	path := local.Path + "/" + name + ".json"
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+	decoder := json.NewDecoder(file)
+	var data = make(map[string]any)
+	err = decoder.Decode(&data)
+	return data, err
+}
+
+//@object: *Local
+//@function: DeleteFile
+//@description: 删除文件
+//@param: key string
+//@return: error
+
+func (local *Local) DeleteFile(key string) error {
+	p := local.Path + "/" + key
+	if err := os.Remove(p); err != nil {
+		return errors.New("本地文件删除失败, err:" + err.Error())
+	}
+	return nil
+}
+
 func GetFileMd5(filePath string) (string, error) {
 	// 打开文件
 	file, err := os.Open(filePath)
@@ -26,3 +204,9 @@ func GetFileMd5(filePath string) (string, error) {
 	md5String := hex.EncodeToString(hashBytes)
 	return md5String, nil
 }
+
+func MD5V(str []byte) string {
+	h := md5.New()
+	h.Write(str)
+	return hex.EncodeToString(h.Sum(nil))
+}
--- a/pkg/tool/local.go
+++ b/pkg/tool/local.go
@@ -1,85 +0,0 @@
-package tool
-
-import (
-	"crypto/md5"
-	"encoding/hex"
-	"errors"
-	"io"
-	"mime/multipart"
-	"os"
-	"pandax/pkg/global"
-	"path"
-	"strings"
-	"time"
-)
-
-type Local struct {
-	Path string
-}
-
-//@object: *Local
-//@function: UploadFile
-//@description: 上传文件
-//@param: file *multipart.FileHeader
-//@return: string, string, error
-
-func (local *Local) UploadFile(file *multipart.FileHeader) (string, string, error) {
-	// 读取文件后缀
-	ext := path.Ext(file.Filename)
-	// 读取文件名并加密
-	name := strings.TrimSuffix(file.Filename, ext)
-	name = MD5V([]byte(name))
-	// 拼接新文件名
-	filename := name + "_" + time.Now().Format("20060102150405") + ext
-	// 尝试创建此路径
-	mkdirErr := os.MkdirAll(local.Path, os.ModePerm)
-	if mkdirErr != nil {
-		global.Log.Error("function os.MkdirAll() Filed", mkdirErr.Error())
-		return "", "", errors.New("function os.MkdirAll() Filed, err:" + mkdirErr.Error())
-	}
-	// 拼接路径和文件名
-	p := local.Path + "/" + filename
-
-	f, openError := file.Open() // 读取文件
-	if openError != nil {
-		global.Log.Error("function file.Open() Filed", openError.Error())
-		return "", "", errors.New("function file.Open() Filed, err:" + openError.Error())
-	}
-	defer f.Close() // 创建文件 defer 关闭
-
-	out, createErr := os.Create(p)
-	if createErr != nil {
-		global.Log.Error("function os.Create() Filed", createErr.Error())
-		return "", "", errors.New("function os.Create() Filed, err:" + createErr.Error())
-	}
-	defer out.Close() // 创建文件 defer 关闭
-
-	_, copyErr := io.Copy(out, f) // 传输（拷贝）文件
-	if copyErr != nil {
-		global.Log.Error("function io.Copy() Filed", copyErr.Error())
-		return "", "", errors.New("function io.Copy() Filed, err:" + copyErr.Error())
-	}
-	return p, filename, nil
-}
-
-//@object: *Local
-//@function: DeleteFile
-//@description: 删除文件
-//@param: key string
-//@return: error
-
-func (local *Local) DeleteFile(key string) error {
-	p := local.Path + "/" + key
-	if strings.Contains(p, local.Path) {
-		if err := os.Remove(p); err != nil {
-			return errors.New("本地文件删除失败, err:" + err.Error())
-		}
-	}
-	return nil
-}
-
-func MD5V(str []byte) string {
-	h := md5.New()
-	h.Write(str)
-	return hex.EncodeToString(h.Sum(nil))
-}