panlei
2019-11-06 6f4406c59ec807b5864496ca9f9bb7986949a6dd
添加一个更新方法
1个文件已添加
2个文件已修改
65 ■■■■■ 已修改文件
105.jpg 补丁 | 查看 | 原始文档 | blame | 历史
WeedFSClient.go 54 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
WeedFSClient_test.go 11 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
105.jpg
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
    }
}
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 方法名必须是以Test开头,驼峰命名
    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)
}