gin框架图片上传代码

gin框架图片上传代码

package manage

import (
	"github.com/gin-gonic/gin"
	"fmt"
	"os"
	"time"
	"path/filepath"
	"net/http"
)

func UploadfileIndex(c *gin.Context) {
	// 给表单限制上传大小 (默认 32 MiB)
	// router.MaxMultipartMemory = 8 << 20  // 8 MiB
		// 单文件
		fileHeader, err := c.FormFile("file")

        if err != nil {
		  responseErr(c, err)
		  return
	    }
		now := time.Now()
		fileType := "other"
	    //判断后缀为图片的文件，如果是图片我们才存入到数据库中
	    fileExt := filepath.Ext(fileHeader.Filename)
	    if fileExt == ".jpg" || fileExt == ".png" || fileExt == ".gif" || fileExt == ".jpeg" {
		   fileType = "img"
	    }
	    fileDir := fmt.Sprintf("public/upload/%s/%d/%d/%d", fileType, now.Year(), now.Month(), now.Day())
        // fileDb := fmt.Sprintf("public/upload/%s/%d/%d/%d", fileType, now.Year(), now.Month(), now.Day())
        err = os.MkdirAll(fileDir, os.ModePerm)
		if err != nil {
			responseErr(c, err)
			return
		}

		timeStamp := time.Now().Unix()
		fileName := fmt.Sprintf("%d-%s", timeStamp, fileHeader.Filename)
		filePathStr := filepath.Join(fileDir, fileName)
		// fileDbPathStr := filepath.Join(fileDb, fileName)

		c.SaveUploadedFile(fileHeader,filePathStr)

		// dst := path.Join("./public/upload/", file.Filename)

		// 上传文件到指定的路径
		// c.SaveUploadedFile(file, dst)
		c.JSON(200, gin.H{
		"status":  "200",
		"filename": fileHeader.Filename,
	   })
}
 func responseErr(c *gin.Context, err error) {
	    c.JSON(http.StatusOK, gin.H{"code": 0, "message": err})
}