如何使用golang自帶net/http接收文件

這篇文章主要介紹“如何使用golang自帶net/http接收文件”，在日常操作中，相信很多人在如何使用golang自帶net/http接收文件問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”如何使用golang自帶net/http接收文件”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

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)

}

到此，關于“如何使用golang自帶net/http接收文件”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！