Skip to content

Commit 35af33f

Browse files
authored
feat(media): add UploadImgFromReader api (#802)
* feat(media): add getTempFile api add getTempFile api * feat(media): handle error in GetTempFile handle error in GetTempFile * feat(media): add UploadImgFromReader api add UploadImgFromReader api * fix(media): fixed not return common error fixed not return common error
1 parent 4a8371e commit 35af33f

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed

work/material/media.go

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,30 @@ func (r *Client) UploadImg(filename string) (*UploadImgResponse, error) {
5959
return result, err
6060
}
6161

62+
// UploadImgFromReader 从 io.Reader 上传图片
63+
// @see https://developer.work.weixin.qq.com/document/path/90256
64+
func (r *Client) UploadImgFromReader(filename string, reader io.Reader) (*UploadImgResponse, error) {
65+
var (
66+
accessToken string
67+
err error
68+
)
69+
if accessToken, err = r.GetAccessToken(); err != nil {
70+
return nil, err
71+
}
72+
var byteData []byte
73+
byteData, err = io.ReadAll(reader)
74+
if err != nil {
75+
return nil, err
76+
}
77+
var response []byte
78+
if response, err = util.PostFileByStream("media", filename, fmt.Sprintf(uploadImgURL, accessToken), byteData); err != nil {
79+
return nil, err
80+
}
81+
result := &UploadImgResponse{}
82+
err = util.DecodeWithError(response, result, "UploadImg")
83+
return result, err
84+
}
85+
6286
// UploadTempFile 上传临时素材
6387
// @see https://developer.work.weixin.qq.com/document/path/90253
6488
// @mediaType 媒体文件类型,分别有图片(image)、语音(voice)、视频(video),普通文件(file)
@@ -166,5 +190,13 @@ func (r *Client) GetTempFile(mediaID string) ([]byte, error) {
166190
if err != nil {
167191
return nil, err
168192
}
193+
194+
// 检查响应是否为错误信息
195+
err = util.DecodeWithCommonError(response, "GetTempFile")
196+
if err != nil {
197+
return nil, err
198+
}
199+
200+
// 如果不是错误响应,则返回原始数据
169201
return response, nil
170202
}

0 commit comments

Comments
 (0)