1. 文件上传

  1. package main
  3. import (
  4.     "io"
  5.     "io/ioutil"
  6.     "log"
  7.     "net/http"
  9.     "github.com/julienschmidt/httprouter"
  10. )
  12. const (
  13.     MAX_UPLOAD_SIZE = 1024 * 1024 * 20 //50MB
  14. )
  16. func main() {
  17.     r := RegisterHandlers()
  19.     http.ListenAndServe(":8080", r)
  20. }
  22. //RegisterHandlers ...
  23. func RegisterHandlers() *httprouter.Router {
  24.     router := httprouter.New()
  26.     router.POST("/upload", uploadHandler)
  28.     return router
  29. }
  30. func uploadHandler(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
  31.     r.Body = http.MaxBytesReader(w, r.Body, MAX_UPLOAD_SIZE)
  32.     if err := r.ParseMultipartForm(MAX_UPLOAD_SIZE); err != nil {
  33.         log.Printf("File is too big")
  34.         return
  35.     }
  36.     file, headers, err := r.FormFile("file")
  37.     if err != nil {
  38.         log.Printf("Error when try to get file: %v", err)
  39.         return
  40.     }
  41.     //获取上传文件的类型
  42.     if headers.Header.Get("Content-Type") != "image/png" {
  43.         log.Printf("只允许上传png图片")
  44.         return
  45.     }
  46.     data, err := ioutil.ReadAll(file)
  47.     if err != nil {
  48.         log.Printf("Read file error: %v", err)
  49.         return
  50.     }
  51.     fn := headers.Filename
  52.     err = ioutil.WriteFile("./video/"+fn, data, 0666)
  53.     if err != nil {
  54.         log.Printf("Write file error: %v", err)
  55.         return
  56.     }
  57.     w.WriteHeader(http.StatusCreated)
  58.     io.WriteString(w, "Uploaded successfully")
  59. }

此上传函数同样也适用于gin框架w http.ResponseWriter等同于c.Writer，r *http.Request等同于c.Request