如何使用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接收文件”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！