beeblog/controllers/FileController.go

package controllers

import (
	"beeblog/models"
	"beeblog/service"
	"crypto/md5"
	"fmt"
	beego "github.com/beego/beego/v2/server/web"
	"math/rand"
	"os"
	"path"
	"time"
)

type FileController struct {
	beego.Controller
}

func (this *FileController) Upload() {
	filename := ""
	for filename, _ = range this.Ctx.Request.MultipartForm.File {
	}
	f, h, filerr := this.GetFile(filename) //获取上传的文件
	if filerr != nil {
		fmt.Println("err:", filerr)
		this.Data["json"] = models.ReurnError(500, "后缀名不符合上传要求")
		this.ServeJSON()
		return
	}
	ext := path.Ext(h.Filename)
	//验证后缀名是否符合要求
	var AllowExtMap map[string]bool = map[string]bool{
		".jpg":  true,
		".jpeg": true,
		".gif":  true,
		".png":  true,
		".GIF":  true,
		".JPG":  true,
		".PNG":  true,
	}
	if _, ok := AllowExtMap[ext]; !ok {
		this.Data["json"] = models.ReurnError(500, "后缀名不符合上传要求")
		this.ServeJSON()
		return
	}
	//创建目录
	urlDir := time.Now().Format("2006/01/02/")
	uploadDir, _ := beego.AppConfig.String("uploadDir")
	uploadDir = uploadDir + urlDir
	err := os.MkdirAll(uploadDir, 777)
	if err != nil {
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	//构造文件名称
	rand.Seed(time.Now().UnixNano())
	randNum := fmt.Sprintf("%d", rand.Intn(9999)+1000)
	hashName := md5.Sum([]byte(time.Now().Format("2006_01_02_15_04_05_") + randNum))
	fileName := fmt.Sprintf("%x", hashName) + ext
	fpath := uploadDir + fileName
	urlDir += fileName
	defer f.Close() //关闭上传的文件，不然的话会出现临时文件不能清除的情况
	err = this.SaveToFile(filename, fpath)
	if err != nil {
		fmt.Println(err)
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	imgUrlPrefix, _ := beego.AppConfig.String("imgUrlPrefix")
	urlDir = imgUrlPrefix + urlDir
	this.Data["json"] = models.ReurnData("", urlDir)
	this.ServeJSON()
	return
}

func (this *FileController) HeadImgUpload() {
	uid := this.GetSession("userid")
	if uid == nil {
		this.Data["json"] = models.ReurnError(401, "")
		this.ServeJSON()
		return
	}
	filename := "imgfile"
	f, h, filerr := this.GetFile(filename) //获取上传的文件
	if filerr != nil {
		fmt.Println("err:", filerr)
		this.Data["json"] = models.ReurnError(500, "后缀名不符合上传要求")
		this.ServeJSON()
		return
	}
	ext := path.Ext(h.Filename)
	//验证后缀名是否符合要求
	var AllowExtMap map[string]bool = map[string]bool{
		".jpg":  true,
		".jpeg": true,
		".gif":  true,
		".png":  true,
		".GIF":  true,
		".JPG":  true,
		".PNG":  true,
	}
	if _, ok := AllowExtMap[ext]; !ok {
		this.Data["json"] = models.ReurnError(500, "后缀名不符合上传要求")
		this.ServeJSON()
		return
	}
	//创建目录
	urlDir := time.Now().Format("2006/01/02/")
	uploadDir, _ := beego.AppConfig.String("uploadDir")
	uploadDir = uploadDir + urlDir
	err := os.MkdirAll(uploadDir, 777)
	if err != nil {
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	//构造文件名称
	rand.Seed(time.Now().UnixNano())
	randNum := fmt.Sprintf("%d", rand.Intn(9999)+1000)
	hashName := md5.Sum([]byte(time.Now().Format("2006_01_02_15_04_05_") + randNum))
	fileName := fmt.Sprintf("%x", hashName) + ext
	fpath := uploadDir + fileName
	urlDir += fileName
	defer f.Close() //关闭上传的文件，不然的话会出现临时文件不能清除的情况
	err = this.SaveToFile(filename, fpath)
	if err != nil {
		fmt.Println(err)
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	imgUrlPrefix, _ := beego.AppConfig.String("imgUrlPrefix")
	urlDir = imgUrlPrefix + urlDir
	user := &models.User{Id: uid.(int64), Headimg: urlDir}
	userSerivce := service.UserService{}
	if _, err = userSerivce.EditHeadImg(user); err != nil {
		this.Data["json"] = models.ReurnError(500, "")
		this.ServeJSON()
		return
	}
	this.SetSession("headimg", urlDir)
	this.Data["json"] = models.ReurnData("", urlDir)
	this.ServeJSON()
	return
}