iris多文件上传示例

目录结构

主目录uploadFiles

  1.     —— templates
  2.         —— upload_form.html
  3.     —— uploads
  4.     —— main.go

示例代码

/templates/upload_form.html

  2. <html>
  3. <head>
  4. <title>Upload file</title>
  5. </head>
  6. <body>
  7.     <form enctype="multipart/form-data"
  8.         action="http://127.0.0.1:8080/upload" method="POST">
  9.         <input type="file" name="uploadfile" multiple/> <input type="hidden"
  10.             name="token" value="{{.}}" /> <input type="submit" value="upload" />
  11.     </form>
  12. </body>
  13. </html>

main.go

  1. package main
  3. import (
  4.     "crypto/md5"
  5.     "fmt"
  6.     "io"
  7.     "mime/multipart"
  8.     "os"
  9.     "path/filepath"
  10.     "strconv"
  11.     "strings"
  12.     "time"
  13.     "github.com/kataras/iris"
  14. )
  16. func main() {
  17.     app := iris.New()
  18.     app.RegisterView(iris.HTML("./templates", ".html"))
  19.     //将upload_form.html提供给客户端
  20.     app.Get("/upload", func(ctx iris.Context) {
  21.         //创建一个令牌（可选）。
  22.         now := time.Now().Unix()
  23.         h := md5.New()
  24.         io.WriteString(h, strconv.FormatInt(now, 10))
  25.         token := fmt.Sprintf("%x", h.Sum(nil))
  26.         //使用令牌渲染表单以供您使用
  27.         ctx.View("upload_form.html", token)
  28.     })
  29.     //处理来自upload_form.html的请求
  30.     app.Post("/upload", func(ctx iris.Context) {
  31.         //上传任意数量的文件（表单输入中的multiple属性）。
  32.         //第二个参数完全是可选的,它可以用来根据请求更改文件的名称，
  33.         //在这个例子中，我们将展示如何使用它,通过在上传文件前加上当前用户的ip前缀
  34.         ctx.UploadFormFiles("./uploads", beforeSave)
  35.     })
  36.     app.Post("/upload_manual", func(ctx iris.Context) {
  37.         //获取通过iris.WithPostMaxMemory获取的最大上传值大小。
  38.         maxSize := ctx.Application().ConfigurationReadOnly().GetPostMaxMemory()
  39.         err := ctx.Request().ParseMultipartForm(maxSize)
  40.         if err != nil {
  41.             ctx.StatusCode(iris.StatusInternalServerError)
  42.             ctx.WriteString(err.Error())
  43.             return
  44.         }
  45.         form := ctx.Request().MultipartForm
  46.         files := form.File["files[]"]
  47.         failures := 0
  48.         for _, file := range files {
  49.             _, err = saveUploadedFile(file, "./uploads")
  50.             if err != nil {
  51.                 failures++
  52.                 ctx.Writef("failed to upload: %s\n", file.Filename)
  53.             }
  54.         }
  55.         ctx.Writef("%d files uploaded", len(files)-failures)
  56.     })
  57.     //在http://localhost:8080启动服务器，上传限制为32 MB。
  58.     app.Run(iris.Addr(":8080"), iris.WithPostMaxMemory(32<<20))
  59. }
  61. func saveUploadedFile(fh *multipart.FileHeader, destDirectory string) (int64, error) {
  62.     src, err := fh.Open()
  63.     if err != nil {
  64.         return 0, err
  65.     }
  66.     defer src.Close()
  67.     out, err := os.OpenFile(filepath.Join(destDirectory, fh.Filename),
  68.         os.O_WRONLY|os.O_CREATE, os.FileMode(0666))
  69.     if err != nil {
  70.         return 0, err
  71.     }
  72.     defer out.Close()
  73.     return io.Copy(out, src)
  74. }
  76. func beforeSave(ctx iris.Context, file *multipart.FileHeader) {
  77.     ip := ctx.RemoteAddr()
  78.     //确保以某种方式格式化ip
  79.     //可以用于文件名（简单情况）：
  80.     ip = strings.Replace(ip, ".", "_", -1)
  81.     ip = strings.Replace(ip, ":", "_", -1)
  82.     //你可以使用time.Now，为文件添加前缀或后缀
  83.     //基于当前时间戳。
  84.     //即unixTime := time.Now().Unix()
  85.     //使用$IP-为Filename添加前缀
  86.     //不需要更多动作，内部上传者将使用此功能
  87.     //将文件保存到"./uploads"文件夹中的名称。
  88.     file.Filename = ip + "-" + file.Filename
  89. }