66 lines
1.9 KiB
Go
66 lines
1.9 KiB
Go
package public
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"errors"
|
||
"ferry/pkg/utils"
|
||
"ferry/tools/app"
|
||
"fmt"
|
||
"io/ioutil"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
// @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)
|
||
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)
|
||
app.OK(c, urlPerfix+singleFile, "上传成功")
|
||
return
|
||
case "2": // 多图
|
||
files := c.Request.MultipartForm.File["file"]
|
||
multipartFile := make([]string, len(files))
|
||
for _, f := range files {
|
||
guid := uuid.New().String()
|
||
multipartFileName := "static/uploadfile/" + guid + utils.GetExt(f.Filename)
|
||
_ = c.SaveUploadedFile(f, multipartFileName)
|
||
multipartFile = append(multipartFile, urlPerfix+multipartFileName)
|
||
}
|
||
app.OK(c, multipartFile, "上传成功")
|
||
return
|
||
case "3": // base64
|
||
files, _ := c.GetPostForm("file")
|
||
ddd, _ := base64.StdEncoding.DecodeString(files)
|
||
guid := uuid.New().String()
|
||
_ = ioutil.WriteFile("static/uploadfile/"+guid+".jpg", ddd, 0666)
|
||
app.OK(c, urlPerfix+"static/uploadfile/"+guid+".jpg", "上传成功")
|
||
}
|
||
}
|
||
}
|