Non puoi selezionare più di 25 argomenti
Gli argomenti devono iniziare con una lettera o un numero, possono includere trattini ('-') e possono essere lunghi fino a 35 caratteri.
148 righe
3.9 KiB
148 righe
3.9 KiB
package controllers
|
|
|
|
import (
|
|
"beeblog/models"
|
|
"beeblog/service"
|
|
"crypto/md5"
|
|
"fmt"
|
|
"github.com/beego/beego/v2/core/config"
|
|
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, _ := config.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, _ := config.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, _ := config.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, _ := config.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
|
|
}
|
|
|