golang beego 文件上传

阳光穿透心脏的1/2处 2021-07-28 12:29 389阅读 0赞

### 前端代码 ###

如果form表单中牵涉到文件上传，在form表单中就需要添加一个属性enctype="multipart/form-data"不然上传就是假上传，后台不能获取到上传的文件。`<form>`修改如下：

### 后台代码 ###

后台接收上传文件有两个函数可以用

GetFile(key string) (multipart.File, \*multipart.FileHeader, error)

作用：是获取前端传递过来的文件。  
参数：是input标签中的name值  
返回值：有三个，一个是文件流（就是我们打开文件返回的内容），第二个是文件相关信息，包括文件头，文件大小，文件名字等，第三个是错误信息。示例代码如下：

file,head,err := this.GetFile(“uploadname”)
    if err != nil{
        beego.Info(“上传图片错误，请重新添加！”)
        this.TplName = “add.html”
        return
    }

`SaveToFile(fromfile, tofile string) error`

作用：直接保存前端出过来的文件。  
参数：有两个参数，第一个参数是前端标签的name属性值，第二个参数是文件在服务器端存储的位置。注意:这个位置字符串在前面需要加一个.  
返回值：是错误信息。示例代码如下：

err := this.SaveToFile("uploadname","./static/img/1.jpg")
    if err != nil{
    	beego.Info("上传图片错误，请重新添加！")
    	this.TplName = "add.html"
    	return
    }

在我们开发过程中，如果后台接收文件并存储需要做以下几种判断

文件格式判断

我们通过GetFile可以获取到文件名，然后通过path包，可以分离出文件的后缀，即文件格式，把你需要的文件格式过滤出来，不需要的返回即可。我们根据文件名获取文件后缀，代码如下:

fileExt := path.Ext(head.Filename)
    if fileExt != ".jpg" && fileExt != ".png" && fileExt != ".jpeg"{
    	beego.Info("上传图片格式不正确，请重新添加！")
    	this.TplName = "add.html"
    	return
    }

文件大小的判断

我们获取文件之后，在存储之前，文件流一般是在内存中，所以文件不易过大，我们在这里做一个文件大小的判断。代码如下:

if head.Size > 5000000{
    	beego.Info(“上传图片太大，请重新添加！”)
    	this.TplName = “add.html”
    	return
    }

避免文件重名

获取文件之后我们要把文件存储到服务器上，但是用户可能会上传同名的文件，如果文件同名的话，后来上传的文件就把之前上传的文件给覆盖了，所以我们要给上传的文件重新确定一个名字。这里我们以上传文件时的时间作为上传文件的文件名。默认的时间格式和我们常见的时间格式不一样，所以这里我们需要对事件做一个格式化。格式化字符串为"2006-01-02-15-04-05（规定的必须是这个，方便记忆可以用6-1-2-3-4-5来记）代码如下：

fileName := time.Now().Format("2006-01-02-15-04-05")
    this.SaveToFile("uploadname","./static/img/"+fileName+fileExt)

### 完整实例 ###

模板代码如下：

上传代码实现如下：

package controllers
    import (
        "path"
        "crypto/md5"
        "time"
        "fmt"
        "github.com/astaxie/beego"
        "math/rand"
        "os"
    )
    type UploadController struct{
        beego.Controller 
    }
    
    func (this *UploadController) UpForm(){
        this.TplName = "upload.tpl"
    }
    
    func (this *UploadController) UpFile(){
    
        f, h, _ := this.GetFile("myfile")//获取上传的文件
        ext := path.Ext(h.Filename)
        //验证后缀名是否符合要求
        var AllowExtMap map[string]bool = map[string]bool{
            ".jpg":true,
            ".jpeg":true,
            ".png":true,
        }
        if _,ok:=AllowExtMap[ext];!ok{
            this.Ctx.WriteString( "后缀名不符合上传要求" )
            return 
        }
        //创建目录
        uploadDir := "static/upload/" + time.Now().Format("2006/01/02/")
        err := os.MkdirAll( uploadDir , 777)
        if err != nil {
            this.Ctx.WriteString( fmt.Sprintf("%v",err) )
            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
        //this.Ctx.WriteString(  fileName )
    
        fpath := uploadDir + fileName
        defer f.Close()//关闭上传的文件，不然的话会出现临时文件不能清除的情况
        err = this.SaveToFile("myfile", fpath)
        if err != nil {
            this.Ctx.WriteString( fmt.Sprintf("%v",err) )
        }
        this.Ctx.WriteString( "上传成功~！！！！！！！" )
    }