package com.doumee.service.business.snapshot;
|
|
import cn.hutool.http.HttpRequest;
|
import cn.hutool.http.HttpResponse;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
@Slf4j
|
@Component
|
public class SnapshotInferClient {
|
|
@Autowired
|
private SnapshotInferProperties properties;
|
|
public boolean healthCheck() {
|
if (StringUtils.isBlank(properties.getBaseUrl())) {
|
return false;
|
}
|
String url = normalizeBaseUrl() + "/health";
|
try {
|
HttpResponse response = HttpRequest.get(url)
|
.timeout(properties.getConnectTimeoutMs())
|
.execute();
|
if (response.getStatus() != 200) {
|
return false;
|
}
|
JSONObject json = JSON.parseObject(response.body());
|
return json != null && "ok".equalsIgnoreCase(json.getString("status"));
|
} catch (Exception e) {
|
log.warn("snapshot-infer health 失败: {}", e.getMessage());
|
return false;
|
}
|
}
|
|
public SnapshotAnalyzeResponse analyze(SnapshotAnalyzeRequest request) {
|
if (StringUtils.isBlank(properties.getBaseUrl())) {
|
return resolveFailure(request, "snapshot.infer.base-url 未配置");
|
}
|
String url = normalizeBaseUrl() + "/analyze";
|
String payload = JSON.toJSONString(request);
|
try {
|
HttpResponse response = HttpRequest.post(url)
|
.body(payload)
|
.contentType("application/json")
|
.setConnectionTimeout(properties.getConnectTimeoutMs())
|
.timeout(properties.getReadTimeoutMs())
|
.execute();
|
String body = response.body();
|
if (response.getStatus() != 200) {
|
log.warn("snapshot-infer analyze HTTP {} body={}", response.getStatus(), body);
|
return resolveFailure(request, "推理服务 HTTP " + response.getStatus());
|
}
|
SnapshotAnalyzeResponse parsed = JSON.parseObject(body, SnapshotAnalyzeResponse.class);
|
if (parsed == null) {
|
return resolveFailure(request, "推理响应为空");
|
}
|
if (Boolean.FALSE.equals(parsed.getSuccess())) {
|
String msg = StringUtils.defaultIfBlank(parsed.getMessage(), "推理未检测到有效时刻");
|
log.warn("snapshot-infer analyze 失败: {}", msg);
|
return resolveFailure(request, msg);
|
}
|
if (parsed.getStorefront() == null || parsed.getHandover() == null) {
|
return resolveFailure(request, "推理响应缺少门头或交付时刻");
|
}
|
if (parsed.getSuccess() == null) {
|
parsed.setSuccess(true);
|
}
|
if (parsed.getAsrHits() == null) {
|
parsed.setAsrHits(new java.util.ArrayList<>());
|
}
|
return parsed;
|
} catch (Exception e) {
|
log.warn("snapshot-infer analyze 异常: {}", e.getMessage());
|
return resolveFailure(request, e.getMessage());
|
}
|
}
|
|
private SnapshotAnalyzeResponse resolveFailure(SnapshotAnalyzeRequest request, String message) {
|
if (properties.isFailOpenMock()) {
|
log.warn("fail-open-mock 启用，使用本地 mock: {}", message);
|
return buildLocalMock(request);
|
}
|
SnapshotAnalyzeResponse fail = new SnapshotAnalyzeResponse();
|
fail.setSuccess(false);
|
fail.setMessage(message);
|
fail.setModelVersion("unavailable");
|
return fail;
|
}
|
|
private SnapshotAnalyzeResponse buildLocalMock(SnapshotAnalyzeRequest request) {
|
double duration = request.getDurationSec() != null && request.getDurationSec() > 0
|
? request.getDurationSec() : 1200.0;
|
double t1 = Math.round(duration * 0.25 * 100.0) / 100.0;
|
double t2 = Math.round(duration * 0.75 * 100.0) / 100.0;
|
if (t2 <= t1) {
|
t2 = t1 + 60.0;
|
}
|
SnapshotAnalyzeResponse response = new SnapshotAnalyzeResponse();
|
response.setSuccess(true);
|
response.setModelVersion("local-mock");
|
response.setDurationSec(duration);
|
SnapshotAnalyzeResponse.SnapshotHit storefront = new SnapshotAnalyzeResponse.SnapshotHit();
|
storefront.setTimeSec(t1);
|
storefront.setConfidence(0.5);
|
storefront.setSource("local-mock");
|
SnapshotAnalyzeResponse.SnapshotHit handover = new SnapshotAnalyzeResponse.SnapshotHit();
|
handover.setTimeSec(t2);
|
handover.setConfidence(0.5);
|
handover.setSource("local-mock");
|
response.setStorefront(storefront);
|
response.setHandover(handover);
|
return response;
|
}
|
|
private String normalizeBaseUrl() {
|
return properties.getBaseUrl().trim().replaceAll("/+$", "");
|
}
|
}
|
