解決golang post文件時Content-Type出現(xiàn)的問題

更新時間：2021年05月01日 14:03:34 作者：九江Mgx

這篇文章主要介紹了解決golang post文件時Content-Type出現(xiàn)的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

同事用php寫了一個接口，要上傳文件，讓我做下測試，直接用curl命令調用成功，然后想用golang寫個示例，

源碼如下：

package main 
import (
    "bytes" 
    "fmt" 
    "io/ioutil" 
    "mime/multipart" 
    "net/http" 
)
 
func main() { 
    uri := "http://xxxxxxxxxxxx/api/fileattr" //URL地址 xxxxxxxxxxxx由商務提供 
    name := "xxxxxxxxxxxx" //用戶名 
    pass := "xxxxxxxxxxxx" //密碼 
    fn := "xxxxxxxxxxxx.txt" //文件路徑
 
    //讀出文本文件數(shù)據(jù) 
    file_data, _ := ioutil.ReadFile(fn) 
    body := new(bytes.Buffer) 
    w := multipart.NewWriter(body)
 
    //取出內容類型 
    content_type := w.FormDataContentType() 
    //將文件數(shù)據(jù)寫入 
    pa, _ := w.CreateFormFile("file", fn) 
    pa.Write(file_data) 
    //設置用戶名密碼 
    w.WriteField("name", name) 
    w.WriteField("pass", pass) 
    w.Close() 
    //開始提交
 
    req, _ := http.NewRequest("POST", uri, body) 
    req.Header.Set("Content-Type", content_type) 
    resp, _ := http.DefaultClient.Do(req) 
    data, _ := ioutil.ReadAll(resp.Body) 
    resp.Body.Close() 
    fmt.Println(resp.StatusCode) 
    fmt.Printf("%s", data) 
}

發(fā)現(xiàn)總是調用失敗，返回文件類型不對，詢問后得知，同事做了判斷，文件只能為text/plain類型，抓包發(fā)現(xiàn)，我提交時的文件類型為：application/octet-stream，仔細查看golang源碼：mime/multipart/write.go,CreateFormFile的源碼是這樣的：

func (w *Writer) CreateFormFile(fieldname, filename string) (io.Writer, error) { 
    h := make(textproto.MIMEHeader) 
    h.Set("Content-Disposition", 
        fmt.Sprintf(`form-data; name="%s"; filename="%s"`, 
            escapeQuotes(fieldname), escapeQuotes(filename))) 
    h.Set("Content-Type", "application/octet-stream") 
    return w.CreatePart(h) 
}

可以得知Content-Type被固定為了application/octet-stream，知道原因了，問題就好解決了。

第一種方法

就是直接修改CreateFormFile，或者加個CreateFormFile2命令，這種方法將來golang升級后可能會出問題。

第二種方法

可以自己來CreatePart：

h := make(textproto.MIMEHeader)
    h.Set("Content-Disposition",
        fmt.Sprintf(`form-data; name="%s"; filename="%s"`,
            escapeQuotes(fieldname), escapeQuotes(filename)))
    h.Set("Content-Type", "text/plain")

再用 w.CreatePart(h)得到io.Writer，問題解決！這種方法不侵入golang源代碼，最終代碼如下：

package main 
import (
    "bytes"
    "fmt"
    "io/ioutil"
    "mime/multipart"
    "net/http"
    "net/textproto"
)
 
func main() {
    uri := "http://xxxxxxxxxxxx/api/fileattr" //URL地址 xxxxxxxxxxxx由商務提供
    name := "xxxxxxxxxx"                      //用戶名
    pass := "xxxxxxx"                         //密碼
    fn := "x:/xxx/xxx.txt"                    //文件路徑
 
    //讀出文本文件數(shù)據(jù)
    file_data, _ := ioutil.ReadFile(fn)
 
    body := new(bytes.Buffer)
    w := multipart.NewWriter(body)
 
    //取出內容類型
    content_type := w.FormDataContentType()
 
    //將文件數(shù)據(jù)寫入
    h := make(textproto.MIMEHeader)
    h.Set("Content-Disposition",
        fmt.Sprintf(`form-data; name="%s"; filename="%s"`,
            "file", //參數(shù)名為file
            fn))
    h.Set("Content-Type", "text/plain") //設置文件格式
    pa, _ := w.CreatePart(h)
    pa.Write(file_data)
 
    //設置用戶名密碼
    w.WriteField("name", name)
    w.WriteField("pass", pass)
 
    w.Close() 
    //開始提交
    req, _ := http.NewRequest("POST", uri, body)
    req.Header.Set("Content-Type", content_type)
    resp, _ := http.DefaultClient.Do(req)
    data, _ := ioutil.ReadAll(resp.Body)
    resp.Body.Close()
    fmt.Println(resp.StatusCode)
    fmt.Printf("%s", data)
}

補充：用go來玩最簡單的web服務器------順便說說Content-Type字段

web服務端代碼s.go：

package main 
import (
    "io"
    "log"
    "net/http"
)
 
func handlerHello(w http.ResponseWriter, r *http.Request) {
    io.WriteString(w, "hello girls")
}
 
func main() {
    http.HandleFunc("/hello", handlerHello)     // 注冊   
    err := http.ListenAndServe("localhost:8080", nil)
    if err != nil {
        log.Println(err)
    }
}

go run s.go一下，跑起來，然后在瀏覽器執(zhí)行http://127.0.0.1:8080/hello （或者在命令行用curl發(fā)http請求也可以），瀏覽器上的結果為：

hello girls

好簡單?？梢栽诳蛻舳嘶蛘叻斩俗グ聪拢?很典型的http req和rsp.

我們再來看一個有趣的問題，修改s.go為：

package main 
import (
    "io"
    "log"
    "net/http"
)
 
func handlerHello(w http.ResponseWriter, r *http.Request) {
    str := `
        table border="1">
        <tr>
        <td>row 1, cell 1</td>
        <td>row 1, cell 2</td>
        </tr>
        <tr>
        <td>row 2, cell 1</td>
        <td>row 2, cell 2</td>
        </tr>
        </table>
        ` 
    io.WriteString(w, str)
}
 
func main() {
    http.HandleFunc("/hello", handlerHello)     // 注冊   
    err := http.ListenAndServe("localhost:8080", nil)
    if err != nil {
        log.Println(err)
    }
}

再次重啟服務并發(fā)請求，瀏覽器上顯示的內容是：

table border="1">
 <tr>
	 <td>row 1, cell 1</td>
	 <td>row 1, cell 2</td>
 </tr>
 <tr>
	 <td>row 2, cell 1</td>
	 <td>row 2, cell 2</td>
 </tr>
</table>

抓包看一下，發(fā)現(xiàn)有：Content-Type: text/plain; charset=utf-8

因此，瀏覽器需要根據(jù)純文本顯示。注意到，上述的table左邊少了一個"<". 我們加上后，

s.go的代碼如下：

package main 
import (
    "io"
    "log"
    "net/http"
)
 
func handlerHello(w http.ResponseWriter, r *http.Request) {
    str := `
        <table border="1">
        <tr>
        <td>row 1, cell 1</td>
        <td>row 1, cell 2</td>
        </tr>
        <tr>
        <td>row 2, cell 1</td>
        <td>row 2, cell 2</td>
        </tr>
        </table>
        ` 
    io.WriteString(w, str)
}
 
func main() {
    http.HandleFunc("/hello", handlerHello)     // 注冊   
    err := http.ListenAndServe("localhost:8080", nil)
    if err != nil {
        log.Println(err)
    }
}

再次重啟服務，發(fā)請求，瀏覽器端的顯示是：

row 1, cell 1	row 1, cell 2
row 2, cell 1	row 2, cell 2

抓包看，有Content-Type: text/html; charset=utf-8

可見，服務端會判斷str的格式，來確定Content-Type的類型，從而決定了瀏覽器端的展示方式。服務端的自動判斷行為，有點意思。在我看來，這樣不太好，應該讓程序員來指定Content-Type.

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章:

Go多線程中數(shù)據(jù)不一致問題的解決方案(sync鎖機制)
在Go語言的并發(fā)編程中,如何確保多個goroutine安全地訪問共享資源是一個關鍵問題,Go語言提供了sync包,其中包含了多種同步原語,用于解決并發(fā)編程中的同步問題,本文將詳細介紹sync包中的鎖機制,需要的朋友可以參考下
2024-10-10
Golang中Channel實戰(zhàn)技巧與一些說明
channel是Go語言內建的first-class類型,也是Go語言與眾不同的特性之一,下面這篇文章主要給大家介紹了關于Golang中Channel實戰(zhàn)技巧與一些說明的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下
2022-11-11
關于Gin框架中的Cookie和Session的使用方法
為了實現(xiàn)跨請求的數(shù)據(jù)共享,我們可以使用Cookie和Session,本文將結合實際案例,詳細介紹在Go語言的Gin框架中如何使用Cookie和Session,并通過代碼示例介紹的非常詳細,需要的朋友可以參考下
2024-10-10
Go語言二進制文件的讀寫操作
本文主要介紹了Go語言二進制文件的讀寫操作，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2023-04-04
golang time包下定時器的實現(xiàn)方法
定時器的實現(xiàn)大家應該都遇到過，最近在學習golang，所以下面這篇文章主要給大家介紹了關于golang time包下定時器的實現(xiàn)方法，文中通過示例代碼介紹的非常詳細，需要的朋友可以參考借鑒，下面來一起看看吧。
2017-12-12
Golang連接并操作PostgreSQL數(shù)據(jù)庫基本操作
PostgreSQL是常見的免費的大型關系型數(shù)據(jù)庫,具有豐富的數(shù)據(jù)類型,也是軟件項目常用的數(shù)據(jù)庫之一,下面這篇文章主要給大家介紹了關于Golang連接并操作PostgreSQL數(shù)據(jù)庫基本操作的相關資料,需要的朋友可以參考下
2022-09-09
Go1.18都出泛型了速來圍觀
泛型允許程序員在強類型程序設計語言中編寫代碼時使用一些以后才指定的類型，在實例化時作為參數(shù)指明這些類型，本文通過例子給大家介紹下如何使用泛型，對Go1.18泛型相關知識感興趣的朋友一起看看吧
2022-03-03
深入理解Go語言的容器包
Go語言的container標準包包含了堆（heap）、鏈表（list）和環(huán)（ring）三種數(shù)據(jù)結構,本文就來詳細的介紹一下這三種的使用,感興趣的可以了解一下
2024-10-10
Golang10進制轉16進制的幾種方法代碼示例
這篇文章主要給大家介紹了關于Golang10進制轉16進制的幾種方法,進制轉換是Golang的一些基本操作,文中通過實例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考借鑒價值,需要的朋友可以參考下
2023-07-07
go語言中數(shù)據(jù)接口set集合的實現(xiàn)
set集合是一種常見的數(shù)據(jù)結構,它代表了一個唯一元素的集合,本文主要介紹了set的基本特性,包括唯一性、無序性、可變性和集合運算,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
2024-10-10