net/http 接收文件

代碼展現，如何使用golang 自帶net/http，將Form表單中提交上來的文件，指定位置保存。

ReadHtmlFile OutHtml(html網頁，表單測試代碼使用)

SaveFile （處理提交文件）

package main

import (
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"os"
)

//讀取html文件
func ReadHtmlFile(filePath string) ([]byte, error) {

	file, err := os.Open(filePath)
	if err != nil {
		log.Println("open file err is ", err)
		return nil, err
	}
	htmlByte, err := ioutil.ReadAll(file)

	if err != nil {
		log.Println("pase html err is ", err)
		return nil, err
	}

	return htmlByte, err
}

//將html文件輸出
func OutHtml(htmlbytes []byte) http.Handler {

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write(htmlbytes)
	})
}

//將post接收到的文件，保存到指定目錄中
func SaveFile(toFile string) http.Handler { //tofile 指定保存目錄

	return http.HandlerFunc(
		func(w http.ResponseWriter, r *http.Request) {
			if r.Method == "POST" {
				file, _, err := r.FormFile("attachment") //從 form中，接收提交來的文件
				if err != nil {
					log.Println(err)
					return
				}
				//log.Println(file)
				defer file.Close()
				f, err := os.OpenFile(toFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) //若是文件不存在，則建立
				if err != nil {
					log.Println(err)
					return
				}

				defer f.Close()
				_, err = io.Copy(f, file) //將post提交上來的文件，重定向到f中

				if err != nil {
					log.Println(err)
				}
			}
		})
}

func main() {

	htmlbyte, err := ReadHtmlFile("D:\\VsCodeProject\\NetHttpGo\\src\\static\\index.html")

	if err != nil {
		log.Println(err)
	}

	outhtml := OutHtml(htmlbyte)
	saveFile := SaveFile("D:\\VsCodeProject\\NetHttpGo\\src\\a.jpg")

	http.Handle("/", outhtml)

	http.Handle("/file", saveFile)

	fmt.Println("listening........")

	http.ListenAndServe(":8081", nil)

}