gin文件上传

第一步安装gin

go get gopkg.in/gin-gonic/gin.v1

服务器代码

package main
import (
    "fmt"
    "io"
    "log"
    "net/http"
    "os"
    "gopkg.in/gin-gonic/gin.v1"
)
//文件上传处理函数
func upload(c *gin.Context) {
    //gin将het/http包的FormFile函数封装到c.Request
    file, header, err := c.Request.FormFile("file")
    if err != nil {
        c.String(http.StatusBadRequest, fmt.Sprintf("get file err : %s", err.Error()))
        return
    }
    //获取文件名
    filename := header.Filename
    //写入文件
    out, err := os.Create("public/" + filename)
    if err != nil {
        log.Fatal(err)
    }
    defer out.Close()
    _, err = io.Copy(out, file)
    if err != nil {
        log.Fatal(err)
    }
    filepath := "http://127.0.0.1:8000/file/" + filename
    //以json格式返回文件存放路径
    c.JSON(http.StatusOK, gin.H{"filepath": filepath})
}
func main() {
    router := gin.Default()
    //注册模板文件路径
    router.LoadHTMLGlob("templates/*")
    router.GET("/", func(c *gin.Context) {
        c.HTML(http.StatusOK, "chosefile.html", gin.H{})
    })
    //返回选择文件模板
    router.GET("/index", func(c *gin.Context) {
        c.HTML(http.StatusOK, "chosefile.html", gin.H{})
    })
    //注册文件上传处理器
    router.POST("/upload", upload)
    //注册静态文件路径
    router.StaticFS("/file", http.Dir("./public"))
    router.Run(":8000")
}

chosefile.html

<!DOCTYPE html>
<html>
<head>
    <title>upload</title>
</head>
<body>
    <form action="http://127.0.0.1:8000/upload/" enctype="multipart/form-data" method="POST"> 
        
        <input type="file" name="file" id="pic" accept="*" />
        <button type="submit">提交</button>
    </form>
</body>
</html>

程序结构


效果图:



如果直接使用文件名进行文件创建,很可能文件名会重复。这里对文件进行重命名,使用UUID进行命名。

安装第三方库

go get -u github.com/satori/go.uuid

对upload函数进行修改

func upload(c *gin.Context) {
    //gin将het/http包的FormFile函数封装到c.Request
    file, header, err := c.Request.FormFile("file")
    if err != nil {
        c.String(http.StatusBadRequest, fmt.Sprintf("get file err : %s", err.Error()))
        return
    }
    //获取文件名
    ext := filepath.Ext(header.Filename)
    name, _ := uuid.NewV4()
    filename := name.String() + ext
    //写入文件
    out, err := os.Create("public/" + filename)
    if err != nil {
        log.Fatal(err)
    }
    defer out.Close()
    _, err = io.Copy(out, file)
    if err != nil {
        log.Fatal(err)
    }
    filepath := "http://127.0.0.1:8000/file/" + filename
    //以json格式返回文件存放路径
    c.JSON(http.StatusOK, gin.H{"filepath": filepath})
}
此处使用path/filepath包的Ext函数获取文件后缀。
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/YYtomorrow/article/details/80322763
个人分类: go
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭