package com.zy.ai.service.impl;
|
|
import com.alibaba.fastjson.JSON;
|
import com.zy.ai.entity.AiDataAnalysisReport;
|
import com.zy.ai.entity.AiDataAnalysisUploadLog;
|
import com.zy.ai.service.AiDataAnalysisUploadLogService;
|
import com.zy.ai.service.DataAnalysisUploadService;
|
import com.zy.common.utils.HttpHandler;
|
import com.zy.system.service.ConfigService;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Date;
|
import java.util.LinkedHashMap;
|
import java.util.Map;
|
|
@Slf4j
|
@Service("dataAnalysisUploadService")
|
@RequiredArgsConstructor
|
public class DataAnalysisUploadServiceImpl implements DataAnalysisUploadService {
|
|
private static final String CONFIG_UPLOAD_ENABLED = "aiDataAnalysisUploadEnabled";
|
private static final String CONFIG_UPLOAD_URL = "aiDataAnalysisUploadUrl";
|
|
private final ConfigService configService;
|
private final AiDataAnalysisUploadLogService aiDataAnalysisUploadLogService;
|
|
@Override
|
public UploadResult upload(AiDataAnalysisReport report) {
|
if (!isUploadEnabled()) {
|
return UploadResult.skipped();
|
}
|
|
String url = configService.getConfigValue(CONFIG_UPLOAD_URL, "");
|
if (url == null || url.trim().isEmpty()) {
|
return UploadResult.skipped();
|
}
|
|
Map<String, Object> payload = new LinkedHashMap<>();
|
payload.put("reportId", report.getId());
|
payload.put("periodType", report.getPeriodType());
|
payload.put("periodStart", report.getPeriodStart());
|
payload.put("periodEnd", report.getPeriodEnd());
|
payload.put("triggerType", report.getTriggerType());
|
payload.put("status", report.getStatus());
|
payload.put("summary", report.getSummary());
|
payload.put("structuredData", report.getStructuredData());
|
payload.put("totalTokens", report.getTotalTokens());
|
payload.put("createTime", report.getCreateTime());
|
|
String jsonBody = JSON.toJSONString(payload);
|
AiDataAnalysisUploadLog uploadLog = new AiDataAnalysisUploadLog();
|
uploadLog.setReportId(report.getId());
|
uploadLog.setUploadUrl(url);
|
uploadLog.setRequestBody(jsonBody);
|
uploadLog.setCreateTime(new Date());
|
|
try {
|
HttpHandler httpHandler = new HttpHandler.Builder()
|
.setUri(url)
|
.setJson(jsonBody)
|
.setTimeout(30, java.util.concurrent.TimeUnit.SECONDS)
|
.build();
|
|
String response = httpHandler.doPost();
|
uploadLog.setHttpStatus(200);
|
uploadLog.setResponseBody(response);
|
uploadLog.setResult("success");
|
aiDataAnalysisUploadLogService.save(uploadLog);
|
log.info("Data analysis report uploaded, reportId={}, url={}", report.getId(), url);
|
return UploadResult.success(200, response);
|
} catch (Exception e) {
|
log.warn("Failed to upload data analysis report, reportId={}, url={}", report.getId(), url, e);
|
uploadLog.setResult("failed");
|
uploadLog.setErrorMessage(e.getMessage());
|
aiDataAnalysisUploadLogService.save(uploadLog);
|
return UploadResult.failed(null, e.getMessage());
|
}
|
}
|
|
private boolean isUploadEnabled() {
|
String value = configService.getConfigValue(CONFIG_UPLOAD_ENABLED, "0");
|
return "1".equals(value.trim());
|
}
|
}
|
