105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
|
package public
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
imgType "github.com/shamsher31/goimgtype"
|
|||
|
|
|
|||
|
|
"go-admin/pkg/utils"
|
|||
|
|
"go-admin/tools"
|
|||
|
|
"go-admin/tools/app"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type FileResponse struct {
|
|||
|
|
Size int64 `json:"size"`
|
|||
|
|
Path string `json:"path"`
|
|||
|
|
FullPath string `json:"full_path"`
|
|||
|
|
Name string `json:"name"`
|
|||
|
|
Type string `json:"type"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// @Summary 上传图片
|
|||
|
|
// @Description 获取JSON
|
|||
|
|
// @Tags 公共接口
|
|||
|
|
// @Accept multipart/form-data
|
|||
|
|
// @Param type query string true "type" (1：单图，2：多图, 3：base64图片)
|
|||
|
|
// @Param file formData file true "file"
|
|||
|
|
// @Success 200 {string} string "{"code": 200, "message": "添加成功"}"
|
|||
|
|
// @Success 200 {string} string "{"code": -1, "message": "添加失败"}"
|
|||
|
|
// @Router /api/v1/public/uploadFile [post]
|
|||
|
|
func UploadFile(c *gin.Context) {
|
|||
|
|
tag, _ := c.GetPostForm("type")
|
|||
|
|
urlPerfix := fmt.Sprintf("http://%s/", c.Request.Host)
|
|||
|
|
var fileResponse FileResponse
|
|||
|
|
if tag == "" {
|
|||
|
|
app.Error(c, 200, errors.New(""), "缺少标识")
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
switch tag {
|
|||
|
|
case "1": // 单图
|
|||
|
|
files, err := c.FormFile("file")
|
|||
|
|
if err != nil {
|
|||
|
|
app.Error(c, 200, errors.New(""), "图片不能为空")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 上传文件至指定目录
|
|||
|
|
guid := uuid.New().String()
|
|||
|
|
|
|||
|
|
singleFile := "static/uploadfile/" + guid + utils.GetExt(files.Filename)
|
|||
|
|
_ = c.SaveUploadedFile(files, singleFile)
|
|||
|
|
fileType, _ := imgType.Get(singleFile)
|
|||
|
|
fileResponse = FileResponse{
|
|||
|
|
Size: tools.GetFileSize(singleFile),
|
|||
|
|
Path: singleFile,
|
|||
|
|
FullPath: urlPerfix + singleFile,
|
|||
|
|
Name: files.Filename,
|
|||
|
|
Type: fileType,
|
|||
|
|
}
|
|||
|
|
app.OK(c, fileResponse, "上传成功")
|
|||
|
|
return
|
|||
|
|
case "2": // 多图
|
|||
|
|
files := c.Request.MultipartForm.File["file"]
|
|||
|
|
var multipartFile []FileResponse
|
|||
|
|
for _, f := range files {
|
|||
|
|
guid := uuid.New().String()
|
|||
|
|
multipartFileName := "static/uploadfile/" + guid + utils.GetExt(f.Filename)
|
|||
|
|
e := c.SaveUploadedFile(f, multipartFileName)
|
|||
|
|
fileType, _ := imgType.Get(multipartFileName)
|
|||
|
|
if e == nil {
|
|||
|
|
multipartFile = append(multipartFile, FileResponse{
|
|||
|
|
Size: tools.GetFileSize(multipartFileName),
|
|||
|
|
Path: multipartFileName,
|
|||
|
|
FullPath: urlPerfix + multipartFileName,
|
|||
|
|
Name: f.Filename,
|
|||
|
|
Type: fileType,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
app.OK(c, multipartFile, "上传成功")
|
|||
|
|
return
|
|||
|
|
case "3": // base64
|
|||
|
|
files, _ := c.GetPostForm("file")
|
|||
|
|
file2list := strings.Split(files, ",")
|
|||
|
|
ddd, _ := base64.StdEncoding.DecodeString(file2list[1])
|
|||
|
|
guid := uuid.New().String()
|
|||
|
|
base64File := "static/uploadfile/" + guid + ".jpg"
|
|||
|
|
_ = ioutil.WriteFile(base64File, ddd, 0666)
|
|||
|
|
typeStr := strings.Replace(strings.Replace(file2list[0], "data:", "", -1), ";base64", "", -1)
|
|||
|
|
fileResponse = FileResponse{
|
|||
|
|
Size: tools.GetFileSize(base64File),
|
|||
|
|
Path: base64File,
|
|||
|
|
FullPath: urlPerfix + base64File,
|
|||
|
|
Name: "",
|
|||
|
|
Type: typeStr,
|
|||
|
|
}
|
|||
|
|
app.OK(c, fileResponse, "上传成功")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|