mirror of
https://github.com/httprunner/httprunner.git
synced 2026-09-05 23:51:25 +08:00
change: request vedem service
This commit is contained in:
@@ -4,7 +4,7 @@ import (
|
|||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"image"
|
"image"
|
||||||
"io/ioutil"
|
"io"
|
||||||
"mime/multipart"
|
"mime/multipart"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -261,23 +261,31 @@ func (s *veDEMImageService) GetImage(imageBuf *bytes.Buffer, options ...ActionOp
|
|||||||
start := time.Now()
|
start := time.Now()
|
||||||
resp, err = client.Do(req)
|
resp, err = client.Do(req)
|
||||||
elapsed := time.Since(start)
|
elapsed := time.Since(start)
|
||||||
var logID string
|
if err != nil {
|
||||||
if resp != nil {
|
log.Error().Err(err).
|
||||||
logID = getLogID(resp.Header)
|
Int("imageBufSize", size).
|
||||||
|
Msgf("request veDEM OCR service error, retry %d", i)
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
if err == nil && resp.StatusCode == http.StatusOK {
|
|
||||||
log.Debug().
|
logID := getLogID(resp.Header)
|
||||||
|
statusCode := resp.StatusCode
|
||||||
|
if statusCode != http.StatusOK {
|
||||||
|
log.Error().
|
||||||
Str("X-TT-LOGID", logID).
|
Str("X-TT-LOGID", logID).
|
||||||
Int("image_bytes", size).
|
Int("imageBufSize", size).
|
||||||
Int64("elapsed(ms)", elapsed.Milliseconds()).
|
Int("statusCode", statusCode).
|
||||||
Msg("request OCR service success")
|
Msgf("request veDEM OCR service failed, retry %d", i)
|
||||||
break
|
time.Sleep(1 * time.Second)
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
log.Error().Err(err).
|
|
||||||
|
log.Debug().
|
||||||
Str("X-TT-LOGID", logID).
|
Str("X-TT-LOGID", logID).
|
||||||
Int("imageBufSize", size).
|
Int("image_bytes", size).
|
||||||
Msgf("request veDEM OCR service failed, retry %d", i)
|
Int64("elapsed(ms)", elapsed.Milliseconds()).
|
||||||
time.Sleep(1 * time.Second)
|
Msg("request OCR service success")
|
||||||
|
break
|
||||||
}
|
}
|
||||||
if resp == nil {
|
if resp == nil {
|
||||||
err = code.CVServiceConnectionError
|
err = code.CVServiceConnectionError
|
||||||
@@ -286,7 +294,7 @@ func (s *veDEMImageService) GetImage(imageBuf *bytes.Buffer, options ...ActionOp
|
|||||||
|
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
results, err := ioutil.ReadAll(resp.Body)
|
results, err := io.ReadAll(resp.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Wrap(code.CVResponseError,
|
err = errors.Wrap(code.CVResponseError,
|
||||||
fmt.Sprintf("read response body error: %v", err))
|
fmt.Sprintf("read response body error: %v", err))
|
||||||
|
|||||||
Reference in New Issue
Block a user