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