- refactor http.Get

This commit is contained in:
gazebo
2019-04-26 17:46:18 +08:00
parent 7e0dcbc8ec
commit d6ec47ed6c
2 changed files with 12 additions and 8 deletions

View File

@@ -9,6 +9,8 @@ import (
"net/http" "net/http"
"strings" "strings"
"git.rosy.net.cn/baseapi/platformapi"
"git.rosy.net.cn/baseapi/platformapi/autonavi" "git.rosy.net.cn/baseapi/platformapi/autonavi"
"git.rosy.net.cn/baseapi/platformapi/jdapi" "git.rosy.net.cn/baseapi/platformapi/jdapi"
"git.rosy.net.cn/baseapi/utils" "git.rosy.net.cn/baseapi/utils"
@@ -153,12 +155,16 @@ func InitSkuName(ctx *jxcontext.Context, isForce, isAsync, isContinueWhenError b
response, err := http.Get(skuName.Img) response, err := http.Get(skuName.Img)
if err == nil { if err == nil {
defer response.Body.Close() defer response.Body.Close()
if response.StatusCode == http.StatusOK {
data, err2 := ioutil.ReadAll(response.Body) data, err2 := ioutil.ReadAll(response.Body)
if err = err2; err == nil { if err = err2; err == nil {
skuName.ImgHashCode = fmt.Sprintf("%X", md5.Sum(data)) skuName.ImgHashCode = fmt.Sprintf("%X", md5.Sum(data))
db := dao.GetDB() db := dao.GetDB()
_, err = dao.UpdateEntity(db, skuName, "ImgHashCode") _, err = dao.UpdateEntity(db, skuName, "ImgHashCode")
} }
} else {
err = platformapi.ErrHTTPCodeIsNot200
}
} }
} }
return nil, err return nil, err

View File

@@ -270,9 +270,7 @@ func (s SkuList) Swap(i, j int) {
func DownloadFileByURL(fileURL string) (bodyData []byte, fileMD5 string, err error) { func DownloadFileByURL(fileURL string) (bodyData []byte, fileMD5 string, err error) {
response, err := http.Get(fileURL) response, err := http.Get(fileURL)
if err == nil { if err == nil {
defer func() { defer response.Body.Close()
response.Body.Close()
}()
if response.StatusCode == http.StatusOK { if response.StatusCode == http.StatusOK {
if bodyData, err = ioutil.ReadAll(response.Body); err == nil { if bodyData, err = ioutil.ReadAll(response.Body); err == nil {
fileMD5 = fmt.Sprintf("%X", md5.Sum(bodyData)) fileMD5 = fmt.Sprintf("%X", md5.Sum(bodyData))