Golang編寫客戶端对蒲、服務(wù)端并實現(xiàn)文件傳輸

建立Server

server := &http.Server{
        Addr:    ":" + port,
        Handler: mux}

注意mux是一個實現(xiàn)了路由的對象屎即。其帶有ServeHttp方法。其實也可以是一般的HandlerFunc著角,只要是實現(xiàn)了ServeHTTP的對象都行揪漩。
handler函數(shù)原型為:
func HandlerDownload(w http.ResponseWriter, r *http.Request)
通過http.HandleFunc()函數(shù),可將其改變?yōu)閹в蠸erveHTTP方法的對象吏口。

建立client request

client := &http.Client{}//建立客戶端
    req, _ := http.NewRequest("GET", "http://baidu.com", nil)//建立request對象奄容,第三個參數(shù)是body內(nèi)容,建立之后可以設(shè)置head參數(shù)等产徊。
    httputil.DumpRequest(req, false)//dump出遠(yuǎn)程服務(wù)器返回的信息
    resp, _ := client.Do(req)//客戶端發(fā)出request請求
    body, _ := ioutil.ReadAll(resp.Body)
    defer resp.Body.Close()
    log.Println(string(body))

客戶端與服務(wù)端信息傳輸嫩海,可以是二進(jìn)制或字符串。字符串可以自定義編碼囚痴,二進(jìn)制就是文件或圖片等叁怪。不同的content-type含義:
x-www-form-urlencoded:客戶端提交表單時候的編碼方式,類似于url編碼
form-data:也是表單深滚,但是支持文件奕谭,不同key-value之間要用boundary分隔
binary:文件,只能是單個文件
json/html/plain text/xml等:都是字符串痴荐,交由瀏覽器解析

解析json

序列化

json.Marshal

反序列化為map

func ParseJSONFromBody(input io.Reader) (map[string]interface{}, error) {
    var m map[string]interface{}
    decoder := json.NewDecoder(input)
    err := decoder.Decode(&m)
    if err != nil {
        return nil, err
    }
    return m, nil
}

測試Handler

測試server的Handler函數(shù)血柳,首先需要構(gòu)建request對象,然后構(gòu)建responseRecorder對象生兆,記錄handler的返回值难捌。

func testHandler(t *testing.T) {
    param := url.Values{}
    param["page"] = []string{"1"}
    param["page_size"] = []string{"100"}
    //創(chuàng)建 request
    req, _ := http.NewRequest("GET", "/"+param.Encode(), nil)
    req.Header.Set("Authorization", "Token abc123")
    //創(chuàng)建 response
    rr := httptest.NewRecorder()
    //只測試函數(shù),跳過mux路由
    HandlerTest(rr, req)

    log.Println(rr.Body.String())
    log.Println(rr.Result())
    log.Println(rr.Code)

}

文件上傳

func testUpload(t *testing.T) {
    var client *http.Client
    var remoteURL string

        //構(gòu)建server 對象
        ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            b, err := httputil.DumpRequest(r, true)
            if err != nil {
                panic(err)
            }
            fmt.Printf("%s", b)
        }))
        defer ts.Close()
        client = ts.Client()
        remoteURL = ts.URL

    //模擬客戶端提交表格
    values := map[string]io.Reader{
        "file":   mustOpen("123.txt"), // 123.txt是本地文件
        "key":  strings.NewReader("hello world!"),
        "key2": strings.NewReader("hello world2!"),
    }
    err := Upload(client, remoteURL, values)
    if err != nil {
        panic(err)
    }
}
func Upload(client *http.Client, url string, values map[string]io.Reader) (err error) {
    // 構(gòu)建multipart鸦难,然后post給服務(wù)端
    var b bytes.Buffer
    w := multipart.NewWriter(&b)
    for key, r := range values {
        var fw io.Writer
        if x, ok := r.(io.Closer); ok {
            defer x.Close()
        }
        // 添加文件
        if x, ok := r.(*os.File); ok {
            if fw, err = w.CreateFormFile(key, x.Name()); err != nil {
                return
            }
        } else {//添加字符串
            // Add other fields
            if fw, err = w.CreateFormField(key); err != nil {
                return
            }
        }
        if _, err = io.Copy(fw, r); err != nil {
            return err
        }

    }
    // close動作會在末端寫入 boundary.
    w.Close()

    req, err := http.NewRequest("POST", url, &b)
    if err != nil {
        return
    }
    // form-data格式根吁,且自動生成分隔符.
//例Content-Type: multipart/form-data; boundary=d76....d29
    req.Header.Set("Content-Type", w.FormDataContentType())

    // 提交 request
    res, err := client.Do(req)
    if err != nil {
        return
    }

    // 檢查返回狀態(tài)嗎
    if res.StatusCode != http.StatusOK {
        err = fmt.Errorf("bad status: %s", res.Status)
    }
    return
}

func mustOpen(f string) *os.File {
    r, err := os.Open(f)
    if err != nil {
        panic(err)
    }
    return r
}

form-data的body樣例:
//文件部分
--d6...9
Content-Disposition: form-data; name="file"; filename="123.txt"
Content-Type: application/octet-stream
//key value部分:w.CreateFormField(key);
--d6.....9
Content-Disposition: form-data; name="file"
//最末端還有分隔符
--d6.....9--

server端處理文件

保存客戶端上傳的文件
Request.FormFile("FormFile")得到file對象和file頭
然后用os.File創(chuàng)建本地文件
最后用io.Copy將request文件保存到目標(biāo)文件。

向客戶端推送文件
os.File打開文件合蔽,
用file.Stat()獲取文件屬性

設(shè)置文件頭
w.Header().Set("Content-Type", "application/x-zip-compressed")
//也可用http.DetectContentType獲得文件類型:application/zip
w.Header().Set("Content-Disposition", "attachment; filename="+fileSta.Name())
w.Header().Set("Content-Length", strconv.FormatInt(fileSta.Size(), 10))
推送文件
io.Copy(ResponseWriter,file)

參考:
https://www.alexedwards.net/blog/a-recap-of-request-handling
http://www.reibang.com/p/8fe93a14754c

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末击敌,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子拴事,更是在濱河造成了極大的恐慌沃斤,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,997評論 6 502
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件刃宵,死亡現(xiàn)場離奇詭異衡瓶,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)牲证,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,603評論 3 392
  • 文/潘曉璐 我一進(jìn)店門哮针,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事诚撵$择桑” “怎么了?”我有些...
    開封第一講書人閱讀 163,359評論 0 353
  • 文/不壞的土叔 我叫張陵寿烟,是天一觀的道長澈驼。 經(jīng)常有香客問我,道長筛武,這世上最難降的妖魔是什么缝其? 我笑而不...
    開封第一講書人閱讀 58,309評論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮徘六,結(jié)果婚禮上内边,老公的妹妹穿的比我還像新娘。我一直安慰自己待锈,他們只是感情好漠其,可當(dāng)我...
    茶點故事閱讀 67,346評論 6 390
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著竿音,像睡著了一般和屎。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上春瞬,一...
    開封第一講書人閱讀 51,258評論 1 300
  • 那天柴信,我揣著相機(jī)與錄音,去河邊找鬼宽气。 笑死随常,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的萄涯。 我是一名探鬼主播绪氛,決...
    沈念sama閱讀 40,122評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼窃判!你這毒婦竟也來了钞楼?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,970評論 0 275
  • 序言:老撾萬榮一對情侶失蹤袄琳,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后燃乍,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體唆樊,經(jīng)...
    沈念sama閱讀 45,403評論 1 313
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,596評論 3 334
  • 正文 我和宋清朗相戀三年刻蟹,在試婚紗的時候發(fā)現(xiàn)自己被綠了逗旁。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,769評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖片效,靈堂內(nèi)的尸體忽然破棺而出红伦,到底是詐尸還是另有隱情,我是刑警寧澤淀衣,帶...
    沈念sama閱讀 35,464評論 5 344
  • 正文 年R本政府宣布昙读,位于F島的核電站,受9級特大地震影響膨桥,放射性物質(zhì)發(fā)生泄漏蛮浑。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,075評論 3 327
  • 文/蒙蒙 一只嚣、第九天 我趴在偏房一處隱蔽的房頂上張望沮稚。 院中可真熱鬧,春花似錦册舞、人聲如沸蕴掏。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,705評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽囚似。三九已至,卻和暖如春线得,著一層夾襖步出監(jiān)牢的瞬間饶唤,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,848評論 1 269
  • 我被黑心中介騙來泰國打工贯钩, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留募狂,地道東北人。 一個月前我還...
    沈念sama閱讀 47,831評論 2 370
  • 正文 我出身青樓角雷,卻偏偏與公主長得像祸穷,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子勺三,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,678評論 2 354

推薦閱讀更多精彩內(nèi)容