package com.vincent.rsf.openApi.tv;
|
|
import com.fasterxml.jackson.databind.JsonNode;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import com.fasterxml.jackson.databind.node.ArrayNode;
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
import com.vincent.rsf.openApi.feign.wms.WmsServerFeignClient;
|
import com.vincent.rsf.openApi.service.RcsTvCallbackService;
|
import com.vincent.rsf.openApi.service.WcsStationStatusService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.http.HttpEntity;
|
import org.springframework.http.HttpHeaders;
|
import org.springframework.http.HttpMethod;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.client.RestClientException;
|
import org.springframework.web.client.RestTemplate;
|
|
import javax.annotation.Resource;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* 定时：GET RCS /station/getTaskNo；POST RCS /station/getError；任务号存在时调 WMS queryTask 合并料箱与物料明细进 Redis
|
*/
|
@Slf4j
|
@Service
|
public class TvRcsStationPollService {
|
|
@Resource
|
private RestTemplate restTemplate;
|
@Resource
|
private ObjectMapper objectMapper;
|
@Resource
|
private RcsTvCallbackService rcsTvCallbackService;
|
@Resource
|
private TvRcsStationPollProperties pollProperties;
|
@Resource
|
private WcsStationStatusService wcsStationStatusService;
|
@Resource
|
private WmsServerFeignClient wmsServerFeignClient;
|
|
public void pollOnce() {
|
String stationId = pollProperties.getTaskNoStationId();
|
if (StringUtils.hasText(pollProperties.getTaskNoPollUrl())) {
|
try {
|
String raw = restTemplate.getForObject(pollProperties.getTaskNoPollUrl(), String.class);
|
String taskNo = applyTaskNoResponse(raw, stationId);
|
enrichSnapshotFromWms(stationId, taskNo);
|
} catch (RestClientException e) {
|
log.warn("RCS 任务号轮询 HTTP 失败: {}", e.getMessage());
|
} catch (Exception e) {
|
log.warn("RCS 任务号轮询处理失败", e);
|
}
|
}
|
if (StringUtils.hasText(pollProperties.getErrorPollUrl())) {
|
try {
|
String raw = pollErrorRaw();
|
applyErrorResponse(raw, pollProperties.getErrorStationId());
|
} catch (RestClientException e) {
|
log.warn("RCS 异常轮询 HTTP 失败: {}", e.getMessage());
|
} catch (Exception e) {
|
log.warn("RCS 异常轮询处理失败", e);
|
}
|
}
|
}
|
|
private void enrichSnapshotFromWms(String stationId, String taskNo) {
|
if (!pollProperties.isEnrichTaskFromWms()) {
|
return;
|
}
|
if (!StringUtils.hasText(stationId)) {
|
return;
|
}
|
if (!StringUtils.hasText(taskNo) || "0".equals(taskNo.trim())) {
|
wcsStationStatusService.clearStationSnapshot(stationId);
|
return;
|
}
|
Map<String, Object> req = new HashMap<>();
|
req.put("taskNo", taskNo);
|
try {
|
Map<String, Object> res = wmsServerFeignClient.openAsrsQueryTask(req);
|
if (res == null) {
|
wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, null);
|
return;
|
}
|
Object codeObj = res.get("code");
|
int code = parseCode(codeObj);
|
if (code != 200) {
|
log.debug("WMS queryTask 非成功 code={} msg={}", code, res.get("msg"));
|
wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, null);
|
return;
|
}
|
Object data = res.get("data");
|
@SuppressWarnings("unchecked")
|
Map<String, Object> dataMap = data instanceof Map ? (Map<String, Object>) data : null;
|
wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, dataMap);
|
} catch (Exception e) {
|
log.warn("WMS queryTask 调用失败: {}", e.getMessage());
|
wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, null);
|
}
|
}
|
|
private static int parseCode(Object codeObj) {
|
if (codeObj == null) {
|
return 0;
|
}
|
if (codeObj instanceof Number) {
|
return ((Number) codeObj).intValue();
|
}
|
try {
|
return Integer.parseInt(String.valueOf(codeObj));
|
} catch (Exception e) {
|
return 0;
|
}
|
}
|
|
private String pollErrorRaw() throws Exception {
|
if (pollProperties.isErrorPollUsePost()) {
|
ObjectNode body = objectMapper.createObjectNode();
|
body.put("staNo", pollProperties.getErrorStationId());
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
HttpEntity<String> entity = new HttpEntity<>(objectMapper.writeValueAsString(body), headers);
|
ResponseEntity<String> resp = restTemplate.exchange(
|
pollProperties.getErrorPollUrl(), HttpMethod.POST, entity, String.class);
|
return resp.getBody();
|
}
|
return restTemplate.getForObject(pollProperties.getErrorPollUrl(), String.class);
|
}
|
|
/** 写入 tvRcs 任务号 Hash；返回解析出的 taskNo（用于 WMS 合并） */
|
private String applyTaskNoResponse(String raw, String staNo) {
|
if (!StringUtils.hasText(staNo)) {
|
return null;
|
}
|
if (!StringUtils.hasText(raw)) {
|
rcsTvCallbackService.writeStationTaskNo(staNo, null);
|
return null;
|
}
|
String trimmed = raw.trim();
|
try {
|
JsonNode root = objectMapper.readTree(trimmed);
|
if (!httpSuccess(root)) {
|
rcsTvCallbackService.writeStationTaskNo(staNo, null);
|
return null;
|
}
|
String taskNo = extractTaskNo(root);
|
rcsTvCallbackService.writeStationTaskNo(staNo, taskNo);
|
return taskNo;
|
} catch (Exception e) {
|
rcsTvCallbackService.writeStationTaskNo(staNo, trimmed);
|
return trimmed;
|
}
|
}
|
|
private static boolean httpSuccess(JsonNode root) {
|
if (root == null || !root.isObject()) {
|
return true;
|
}
|
JsonNode c = root.get("code");
|
if (c == null || c.isNull()) {
|
return true;
|
}
|
return c.asInt(200) == 200;
|
}
|
|
private static String extractTaskNo(JsonNode root) {
|
if (root == null || root.isNull()) {
|
return null;
|
}
|
if (root.isValueNode()) {
|
return textNode(root);
|
}
|
String[] paths = {"taskNo", "task_no", "taskCode", "seqNum"};
|
for (String p : paths) {
|
JsonNode n = root.get(p);
|
if (n != null && !n.isNull() && StringUtils.hasText(textNode(n))) {
|
return textNode(n);
|
}
|
}
|
JsonNode data = root.get("data");
|
if (data != null && !data.isNull()) {
|
if (data.isObject()) {
|
for (String p : paths) {
|
JsonNode n = data.get(p);
|
if (n != null && !n.isNull() && StringUtils.hasText(textNode(n))) {
|
return textNode(n);
|
}
|
}
|
} else if (data.isValueNode()) {
|
return textNode(data);
|
}
|
}
|
JsonNode result = root.get("result");
|
if (result != null && result.isObject()) {
|
for (String p : paths) {
|
JsonNode n = result.get(p);
|
if (n != null && !n.isNull() && StringUtils.hasText(textNode(n))) {
|
return textNode(n);
|
}
|
}
|
}
|
return null;
|
}
|
|
private static String textNode(JsonNode n) {
|
if (n == null || n.isNull()) {
|
return "";
|
}
|
String s = n.asText("");
|
return s == null ? "" : s.trim();
|
}
|
|
private void applyErrorResponse(String raw, String defaultStaNo) throws Exception {
|
JsonNode body = buildErrorCallbackBody(raw, defaultStaNo);
|
rcsTvCallbackService.handleStationError(body);
|
}
|
|
private JsonNode buildErrorCallbackBody(String raw, String defaultStaNo) throws Exception {
|
if (!StringUtils.hasText(raw)) {
|
return objectMapper.createArrayNode();
|
}
|
String trimmed = raw.trim();
|
JsonNode root = objectMapper.readTree(trimmed);
|
if (!httpSuccess(root)) {
|
return objectMapper.createArrayNode();
|
}
|
if (root.isArray()) {
|
return fillStaNo((ArrayNode) root, defaultStaNo);
|
}
|
if (root.isObject()) {
|
if (root.has("data") && root.get("data").isArray()) {
|
ObjectNode o = objectMapper.createObjectNode();
|
o.set("data", fillStaNo((ArrayNode) root.get("data"), defaultStaNo));
|
return o;
|
}
|
if (root.has("errors") && root.get("errors").isArray()) {
|
ObjectNode o = objectMapper.createObjectNode();
|
o.set("data", fillStaNo((ArrayNode) root.get("errors"), defaultStaNo));
|
return o;
|
}
|
String msg = firstMessageField(root);
|
if (StringUtils.hasText(msg)) {
|
return wrapSingleError(defaultStaNo, msg);
|
}
|
JsonNode data = root.get("data");
|
if (data != null && data.isTextual()) {
|
return wrapSingleError(defaultStaNo, data.asText());
|
}
|
}
|
return wrapSingleError(defaultStaNo, trimmed);
|
}
|
|
private JsonNode wrapSingleError(String staNo, String msg) {
|
ArrayNode arr = objectMapper.createArrayNode();
|
ObjectNode item = objectMapper.createObjectNode();
|
if (StringUtils.hasText(staNo)) {
|
item.put("staNo", staNo);
|
}
|
item.put("error", msg);
|
arr.add(item);
|
ObjectNode body = objectMapper.createObjectNode();
|
body.set("data", arr);
|
return body;
|
}
|
|
private static String firstMessageField(JsonNode o) {
|
String[] keys = {"errorMsg", "message", "msg", "error", "plcDesc", "desc"};
|
for (String k : keys) {
|
JsonNode n = o.get(k);
|
if (n != null && !n.isNull() && StringUtils.hasText(n.asText("").trim())) {
|
return n.asText("").trim();
|
}
|
}
|
return "";
|
}
|
|
private ArrayNode fillStaNo(ArrayNode arr, String defaultStaNo) {
|
for (int i = 0; i < arr.size(); i++) {
|
JsonNode el = arr.get(i);
|
if (!el.isObject()) {
|
continue;
|
}
|
ObjectNode obj = (ObjectNode) el;
|
if (!StringUtils.hasText(textNode(obj.get("staNo"))) && StringUtils.hasText(defaultStaNo)) {
|
obj.put("staNo", defaultStaNo);
|
}
|
}
|
return arr;
|
}
|
}
|
