golang beego 文件上传

前端代码

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

<form method="post" action="/addArticle" enctype="multipart/form-data">
    <input type="file" class="input_file"  name="uploadname">

后台代码

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

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)

完整实例

模板代码如下：

<form id="fform" method="POST" enctype="multipart/form-data" action="/upload">   
    <input id="myfile" name="myfile" type="file" />  
    <input type="submit" value="保存"  />
</form>

上传代码实现如下：

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( "上传成功~！！！！！！！" )
}