解決golang post文件時Content-Type出現的問題

同事用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" //文件路徑
 
    //讀出文本文件數據 
    file_data, _ := ioutil.ReadFile(fn) 
    body := new(bytes.Buffer) 
    w := multipart.NewWriter(body)
 
    //取出內容類型 
    content_type := w.FormDataContentType() 
    //將文件數據寫入 
    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) 
}

發現總是調用失敗,返回文件類型不對,詢問後得知,同事做瞭判斷,文件隻能為text/plain類型,抓包發現,我提交時的文件類型為: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"                    //文件路徑
 
    //讀出文本文件數據
    file_data, _ := ioutil.ReadFile(fn)
 
    body := new(bytes.Buffer)
    w := multipart.NewWriter(body)
 
    //取出內容類型
    content_type := w.FormDataContentType()
 
    //將文件數據寫入
    h := make(textproto.MIMEHeader)
    h.Set("Content-Disposition",
        fmt.Sprintf(`form-data; name="%s"; filename="%s"`,
            "file", //參數名為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一下,跑起來, 然後在瀏覽器執行http://127.0.0.1:8080/hello (或者在命令行用curl發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)
    }
}

再次重啟服務並發請求, 瀏覽器上顯示的內容是:

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>

抓包看一下, 發現有:Content-Type: text/plain; charset=utf-8

因此, 瀏覽器需要根據純文本顯示。 註意到, 上述的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)
    }
}

再次重啟服務,發請求,瀏覽器端的顯示是:

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.

以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方,望不吝賜教。