golang实现文件上传api接口（基于net/http），不使用三方框架

实现一个简单的web服务器，为文件上传提供一个接口

1. 实现一个简单的web服务

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) { io.WriteString(w, "Hello from a HandleFunc #1!\n") })
	log.Println("web server is star")
	http.ListenAndServe(":80",nil)
	if err != nil {
		log.Println(err)
	}
}

启动

go run main.go

效果图

启动成功

2.编写文件上传接口

文件上传需要把文件保存在我们的服务器上,而且可以对上传的文件格式做一个验证，你想要什么格式的，对文件的大小做了限制
upload函数

var channels = make(chan struct{})
var successChan = make(chan struct{})


func save(m *multipart.FileHeader) {
	suffix := strings.Replace(path.Ext(m.Filename), ".", "", 1)
	fileMap := map[string]bool{
		"mp4": true, "png": true, "jpg": true, "jpeg": true, "gif": true, "docx": true,
	}
	fmt.Println(strings.ToLower(suffix))
	//判断是否在fileMap
	if _, ok := fileMap[strings.ToLower(suffix)]; !ok {
		//不存在
		channels <- struct{}{}
		log.Println("文件格式错误")
		return
	}
	if m.Size > 20<<10<<10 {
		fmt.Println(m.Size, 20<<10<<10)
		channels <- struct{}{}
		log.Println("文件太大")
		return
	}
	var file *os.File
	var err error
	if strings.ToLower(suffix) == "docx" {
		file, err = os.Create("./" + m.Filename)
		if err != nil {
			channels <- struct{}{}
			log.Println(err)
			return
		}
	} else {
		file, err = os.Create("./static/" + m.Filename)
		if err != nil {
			channels <- struct{}{}
			log.Println(err)
			return
		}
	}
	log.Println(m.Filename)

	open, err := m.Open()
	if err != nil {
		channels <- struct{}{}
		log.Println(err)
		return
	}
	defer open.Close()
	_, err = io.Copy(file, open)
	if err != nil {
		channels <- struct{}{}
		log.Println(err)
		return
	}
	successChan <- struct{}{}
}

main函数

func main() {
	http.HandleFunc("/upload", func(w http.ResponseWriter, r *http.Request) {
		err := r.ParseMultipartForm(20 << 20)
		if err != nil {
			w.WriteHeader(http.StatusInternalServerError)
			log.Println(err)
			return
		}
		file := r.MultipartForm.File["file"]
		for _, val := range file {
			go save(val)
		}
		i := 0
		fmt.Println(i)
		for i < len(file) {
			fmt.Println(i)
			select {
			case <-channels:
				i++
			case <-successChan:
				i++
			}
			fmt.Println(i)
		}
		fmt.Println("结束")
		w.WriteHeader(http.StatusOK)

	})
	log.Println("web server is star")
	http.ListenAndServe(":80",nil)
	if err != nil {
		log.Println(err)
	}
}

启动

go  run main.go

测试

3. 上传成功

可以多文件上传，用goroutine处理多个文件上传的问题