package com.zy.ai.service;
|
|
import org.springframework.stereotype.Service;
|
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
|
@Service
|
public class WcsQueryService {
|
|
// 这里注入你现有的各类 Service/Mapper：deviceService/taskService/logService/configService...
|
|
public Map<String, Object> buildDiagnosisSnapshot(Map<String, Object> args) {
|
Map<String, Object> focus = (Map<String, Object>) args.get("focus");
|
List<String> deviceIds = focus == null ? Collections.emptyList() : (List<String>) focus.get("deviceIds");
|
List<String> taskIds = focus == null ? Collections.emptyList() : (List<String>) focus.get("taskIds");
|
|
int timeWindowSec = getInt(args, "timeWindowSec", 600);
|
int logLimit = getInt(args, "logLimit", 400);
|
boolean includeConfigs = getBool(args, "includeConfigs", true);
|
boolean includeDerived = getBool(args, "includeDerivedSignals", true);
|
|
Date now = new Date();
|
Date from = new Date(now.getTime() - timeWindowSec * 1000L);
|
|
// 1) devices
|
List<Object> devices = new ArrayList<>();
|
for (String deviceId : safeList(deviceIds)) {
|
Map<String, Object> a = new HashMap<>();
|
a.put("deviceId", deviceId);
|
devices.add(getDeviceState(a));
|
}
|
|
// 2) tasks
|
Map<String, Object> taskArgs = new HashMap<>();
|
if (!safeList(taskIds).isEmpty()) {
|
// 你可以让 listTasks 支持 taskIds 精确过滤
|
taskArgs.put("taskIds", taskIds);
|
} else if (!safeList(deviceIds).isEmpty()) {
|
taskArgs.put("deviceId", deviceIds.get(0)); // 或改成支持 deviceIds 数组
|
taskArgs.put("statuses", Arrays.asList("WAITING","RUNNING","SUSPENDED"));
|
taskArgs.put("limit", 200);
|
}
|
Map<String, Object> taskResult = (Map<String, Object>) listTasks(taskArgs);
|
List<Object> tasks = taskResult == null ? new ArrayList<>() : (List<Object>) taskResult.get("tasks");
|
|
// 3) logs
|
Map<String, Object> logArgs = new HashMap<>();
|
logArgs.put("timeFrom", iso(from));
|
logArgs.put("timeTo", iso(now));
|
logArgs.put("limit", logLimit);
|
if (!safeList(deviceIds).isEmpty()) logArgs.put("deviceId", deviceIds.get(0));
|
if (!safeList(taskIds).isEmpty()) logArgs.put("taskId", taskIds.get(0));
|
Map<String, Object> logResult = (Map<String, Object>) queryLogs(logArgs);
|
List<Object> logs = logResult == null ? new ArrayList<>() : (List<Object>) logResult.get("logs");
|
|
// 4) configs
|
Object configs = null;
|
if (includeConfigs) {
|
configs = getSystemConfig(new HashMap<String, Object>());
|
}
|
|
// 5) derived
|
Object derived = null;
|
if (includeDerived) {
|
derived = deriveSignals(devices, tasks, logs);
|
}
|
|
// output
|
Map<String, Object> out = new LinkedHashMap<>();
|
Map<String, Object> meta = new LinkedHashMap<>();
|
meta.put("snapshotId", UUID.randomUUID().toString());
|
meta.put("generatedAt", iso(now));
|
meta.put("timeFrom", iso(from));
|
meta.put("timeTo", iso(now));
|
|
out.put("meta", meta);
|
out.put("devices", devices);
|
out.put("tasks", tasks);
|
out.put("logs", logs);
|
out.put("configs", configs);
|
out.put("derivedSignals", derived);
|
return out;
|
}
|
|
private Map<String, Object> deriveSignals(List<Object> devices, List<Object> tasks, List<Object> logs) {
|
Map<String, Object> d = new LinkedHashMap<>();
|
List<String> flags = new ArrayList<>();
|
List<String> causes = new ArrayList<>();
|
List<String> bottlenecks = new ArrayList<>();
|
|
// TODO：把你 WCS 领域规则塞这里（心跳超时、等待确认、命令无ACK、站台满、下游阻塞等）
|
// 先给个示例：
|
if (tasks != null && !tasks.isEmpty() && (devices == null || devices.isEmpty())) {
|
flags.add("HAS_TASKS_BUT_NO_DEVICE_SNAPSHOT");
|
}
|
|
d.put("anomalyFlags", flags);
|
d.put("suspectedRootCauses", causes);
|
d.put("suspectedBottlenecks", bottlenecks);
|
return d;
|
}
|
|
// ======= 下面这些函数，你用现有 service 实现即可 =======
|
public Map<String, Object> getDeviceState(Map<String, Object> args) { /* ... */ return new HashMap<>(); }
|
public Map<String, Object> listTasks(Map<String, Object> args) { /* ... */ return new HashMap<>(); }
|
public Map<String, Object> queryLogs(Map<String, Object> args) { /* ... */ return new HashMap<>(); }
|
public Object getSystemConfig(Map<String, Object> args) { /* ... */ return new HashMap<>(); }
|
|
// helpers
|
private int getInt(Map<String, Object> m, String k, int def) {
|
Object v = m.get(k);
|
if (v == null) return def;
|
if (v instanceof Number) return ((Number) v).intValue();
|
return Integer.parseInt(String.valueOf(v));
|
}
|
private boolean getBool(Map<String, Object> m, String k, boolean def) {
|
Object v = m.get(k);
|
if (v == null) return def;
|
if (v instanceof Boolean) return (Boolean) v;
|
return Boolean.parseBoolean(String.valueOf(v));
|
}
|
private List<String> safeList(List<String> l) { return l == null ? Collections.emptyList() : l; }
|
private String iso(Date d) {
|
SimpleDateFormat f = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSXXX");
|
f.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai"));
|
return f.format(d);
|
}
|
}
|
