feat: 建立可观测性、容量与恢复验证基线
This commit is contained in:
@@ -77,6 +77,7 @@ func NewAIClient(baseURL string) *AIClient {
|
||||
|
||||
// Detect 上传图片到 ai-service /detect,返回检测结果(无状态接口,天然幂等)
|
||||
func (c *AIClient) Detect(ctx context.Context, imageBytes []byte, filename string) (*AIDetectResponse, error) {
|
||||
start := time.Now()
|
||||
var body bytes.Buffer
|
||||
writer := multipart.NewWriter(&body)
|
||||
part, err := writer.CreateFormFile("file", filename)
|
||||
@@ -97,7 +98,9 @@ func (c *AIClient) Detect(ctx context.Context, imageBytes []byte, filename strin
|
||||
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
resp, err := c.client.Do(req)
|
||||
latency := time.Since(start)
|
||||
if err != nil {
|
||||
RecordDependency(DependencyAI, err, latency)
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
@@ -107,12 +110,16 @@ func (c *AIClient) Detect(ctx context.Context, imageBytes []byte, filename strin
|
||||
return nil, err
|
||||
}
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, fmt.Errorf("ai-service /detect 返回 %d: %s", resp.StatusCode, string(data))
|
||||
err := fmt.Errorf("ai-service /detect 返回 %d: %s", resp.StatusCode, string(data))
|
||||
RecordDependency(DependencyAI, err, latency)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var out AIDetectResponse
|
||||
if err := json.Unmarshal(data, &out); err != nil {
|
||||
RecordDependency(DependencyAI, err, latency)
|
||||
return nil, err
|
||||
}
|
||||
RecordDependency(DependencyAI, nil, latency)
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user