第一步安装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函数获取文件后缀。