From 6f4406c59ec807b5864496ca9f9bb7986949a6dd Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 06 十一月 2019 09:57:13 +0800 Subject: [PATCH] 添加一个更新方法 --- 105.jpg | 0 WeedFSClient.go | 54 +++++++++++++++++++++++++++ WeedFSClient_test.go | 11 +++++ 3 files changed, 65 insertions(+), 0 deletions(-) diff --git a/105.jpg b/105.jpg new file mode 100644 index 0000000..c5e3e68 --- /dev/null +++ b/105.jpg Binary files differ diff --git a/WeedFSClient.go b/WeedFSClient.go index 7c662b5..5264404 100644 --- a/WeedFSClient.go +++ b/WeedFSClient.go @@ -80,4 +80,58 @@ return "",err } return "http://"+m["url"].(string)+"/"+m["fid"].(string),err +} + +func UpDateFile(uri string, fileName string, fileData []byte) (m map[string]interface{}, err error) { + body := &bytes.Buffer{} + writer := multipart.NewWriter(body) + _, err = writer.CreateFormFile("file", fileName) + if err != nil { + return nil, err + } + boundary := writer.Boundary() + //close_string := fmt.Sprintf("\r\n--%s--\r\n", boundary) + close_buf := bytes.NewBufferString(fmt.Sprintf("\r\n--%s--\r\n", boundary)) + file := bytes.NewBuffer(fileData) + request_reader := io.MultiReader(body, file, close_buf) + //_, err = io.Copy(part, file) + //writer.WriteField(key, val) + request, err := http.NewRequest("POST", uri, request_reader) + request.Header.Add("Content-Type", writer.FormDataContentType()) + timeout := time.Duration(5 * time.Second) //瓒呮椂鏃堕棿50ms + client := &http.Client{Timeout: timeout} + resp, err := client.Do(request) + if err != nil { + fmt.Println("UploadFile client.Do(request) err:", err) + return nil, err + } + defer func() { + if r := recover(); r != nil { + fmt.Printf("panic鐨勫唴瀹�%v\n", r) + msg := "涓婁紶鍥剧墖鏈嶅姟鍣ㄥ紓甯�" + if _, ok := r.(error); ok { + msg = r.(error).Error() + fmt.Println("panic--recover()寰楀埌鐨勬槸error绫诲瀷") + } + if _, ok := r.(string); ok { + msg = r.(string) + fmt.Println("panic--recover()寰楀埌鐨勬槸string绫诲瀷") + } + err = errors.New(msg) + } + }() + defer resp.Body.Close() + { + body := &bytes.Buffer{} + _, err := body.ReadFrom(resp.Body) + if err != nil { + fmt.Println("UploadFile body.ReadForm(resp.Body) err:", err) + } + + m := make(map[string]interface{}) + if err := json.Unmarshal([]byte(body.String()), &m); err != nil { + return nil, err + } + return m,nil + } } \ No newline at end of file diff --git a/WeedFSClient_test.go b/WeedFSClient_test.go index 27dbb36..e1567e6 100644 --- a/WeedFSClient_test.go +++ b/WeedFSClient_test.go @@ -2,6 +2,7 @@ import ( "fmt" + "io/ioutil" "testing" ) @@ -12,4 +13,14 @@ t.Error("娴嬭瘯澶辫触") } fmt.Println(path) +} + +func TestUpDateFile(t *testing.T) { //1 鏂规硶蹇呴』浼犲叆testing.T 2 鏂规硶鍚嶅繀椤绘槸浠est寮�澶达紝椹煎嘲鍛藉悕 + url := "http://192.168.5.23:6120/78,22eae6b7875b67" + data,err := ioutil.ReadFile("./105.jpg") + path,err := UpDateFile(url,"",data) + if err != nil { + t.Error("娴嬭瘯澶辫触") + } + fmt.Println(path) } \ No newline at end of file -- Gitblit v1.8.0