同事用php寫了一個(gè)接口，要上傳文件，讓我做下測(cè)試，直接用curl命令調(diào)用成功，然后想用golang寫個(gè)示例，
源碼如下：
package main
import (
"bytes"
"fmt"
"io/ioutil"
"mime/multipart"
"net/http"
)
func main() {
uri := "http://xxxxxxxxxxxx/api/fileattr" //URL地址 xxxxxxxxxxxx由商務(wù)提供
name := "xxxxxxxxxxxx" //用戶名
pass := "xxxxxxxxxxxx" //密碼
fn := "xxxxxxxxxxxx.txt" //文件路徑
//讀出文本文件數(shù)據(jù)
file_data, _ := ioutil.ReadFile(fn)
body := new(bytes.Buffer)
w := multipart.NewWriter(body)
//取出內(nèi)容類型
content_type := w.FormDataContentType()
//將文件數(shù)據(jù)寫入
pa, _ := w.CreateFormFile("file", fn)
pa.Write(file_data)
//設(shè)置用戶名密碼
w.WriteField("name", name)
w.WriteField("pass", pass)
w.Close()
//開(kāi)始提交
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)總是調(diào)用失敗，返回文件類型不對(duì)，詢問(wèn)后得知，同事做了判斷，文件只能為text/plain類型，抓包發(fā)現(xiàn)，我提交時(shí)的文件類型為：application/octet-stream，仔細(xì)查看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，知道原因了，問(wèn)題就好解決了。
第一種方法
就是直接修改CreateFormFile，或者加個(gè)CreateFormFile2命令，這種方法將來(lái)golang升級(jí)后可能會(huì)出問(wèn)題。
第二種方法
可以自己來(lái)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，問(wèn)題解決！這種方法不侵入golang源代碼，最終代碼如下：
package main
import (
"bytes"
"fmt"
"io/ioutil"
"mime/multipart"
"net/http"
"net/textproto"
)
func main() {
uri := "http://xxxxxxxxxxxx/api/fileattr" //URL地址 xxxxxxxxxxxx由商務(wù)提供
name := "xxxxxxxxxx" //用戶名
pass := "xxxxxxx" //密碼
fn := "x:/xxx/xxx.txt" //文件路徑
//讀出文本文件數(shù)據(jù)
file_data, _ := ioutil.ReadFile(fn)
body := new(bytes.Buffer)
w := multipart.NewWriter(body)
//取出內(nèi)容類型
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") //設(shè)置文件格式
pa, _ := w.CreatePart(h)
pa.Write(file_data)
//設(shè)置用戶名密碼
w.WriteField("name", name)
w.WriteField("pass", pass)
w.Close()
//開(kāi)始提交
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)
}
補(bǔ)充：用go來(lái)玩最簡(jiǎn)單的web服務(wù)器------順便說(shuō)說(shuō)Content-Type字段
web服務(wù)端代碼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) // 注冊(cè)
err := http.ListenAndServe("localhost:8080", nil)
if err != nil {
log.Println(err)
}
}
go run s.go一下，跑起來(lái)， 然后在瀏覽器執(zhí)行http://127.0.0.1:8080/hello （或者在命令行用curl發(fā)http請(qǐng)求也可以）， 瀏覽器上的結(jié)果為：
好簡(jiǎn)單。可以在客戶端或者服務(wù)端抓包看下， 很典型的http req和rsp.
我們?cè)賮?lái)看一個(gè)有趣的問(wèn)題， 修改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) // 注冊(cè)
err := http.ListenAndServe("localhost:8080", nil)
if err != nil {
log.Println(err)
}
}
再次重啟服務(wù)并發(fā)請(qǐng)求， 瀏覽器上顯示的內(nèi)容是：
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左邊少了一個(gè)"". 我們加上后，
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) // 注冊(cè)
err := http.ListenAndServe("localhost:8080", nil)
if err != nil {
log.Println(err)
}
}
再次重啟服務(wù)，發(fā)請(qǐng)求，瀏覽器端的顯示是：
| row 1, cell 1 |
row 1, cell 2 |
| row 2, cell 1 |
row 2, cell 2 |
抓包看， 有Content-Type: text/html; charset=utf-8
可見(jiàn)， 服務(wù)端會(huì)判斷str的格式，來(lái)確定Content-Type的類型， 從而決定了瀏覽器端的展示方式。服務(wù)端的自動(dòng)判斷行為， 有點(diǎn)意思。 在我看來(lái)， 這樣不太好，應(yīng)該讓程序員來(lái)指定Content-Type.
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
您可能感興趣的文章:- golang中json小談之字符串轉(zhuǎn)浮點(diǎn)數(shù)的操作
- go浮點(diǎn)數(shù)轉(zhuǎn)字符串保留小數(shù)點(diǎn)后N位的完美解決方法
- 解決Golang中g(shù)oroutine執(zhí)行速度的問(wèn)題
- 解決golang結(jié)構(gòu)體tag編譯錯(cuò)誤的問(wèn)題
- golang 實(shí)現(xiàn)Location跳轉(zhuǎn)方式
- 對(duì)Golang中的FORM相關(guān)字段理解
- golang 打印error的堆棧信息操作
- golang 比較浮點(diǎn)數(shù)的大小方式
