change: request vedem service

This commit is contained in:
lilong.129
2023-08-29 17:25:41 +08:00
parent 84352926cf
commit 97430bf451
+23 -15
View File
@@ -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))