diff --git a/service/wechatminiprogram/save_img.go b/service/wechatminiprogram/save_img.go index e97feb94..e3c1e960 100644 --- a/service/wechatminiprogram/save_img.go +++ b/service/wechatminiprogram/save_img.go @@ -5,7 +5,12 @@ import ( "os" ) -func (app *App) SaveImg(resp gohttp.Response, dir, saveName string) string { +type SaveImgResponse struct { + Path string + Name string +} + +func (app *App) SaveImg(resp gohttp.Response, dir, saveName string) SaveImgResponse { // 返回是二进制图片,或者json错误 if resp.Header.Get("Content-Type") == "image/jpeg" || resp.Header.Get("Content-Type") == "image/png" { // 保存在output目录 @@ -24,8 +29,10 @@ func (app *App) SaveImg(resp gohttp.Response, dir, saveName string) string { } f.Write(resp.Body) f.Close() - return dir + outputFileName - } else { - return "" + return SaveImgResponse{ + Path: dir + outputFileName, + Name: outputFileName, + } } + return SaveImgResponse{} }