cl
2026-04-16 f66b6a060100d8e117df8c01dc9416b37b474497
rsf-open-api/src/main/java/com/vincent/rsf/openApi/tv/TvRcsStationPollService.java
@@ -4,21 +4,32 @@
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.Collection;
import java.util.HashMap;
import java.util.Map;
/**
 * 调用 RCS 站点接口,结果写入电视机 Redis,由 zy-monitor-admin WebSocket 推送到电视机
 * 定时:GET RCS /cv/station/getTaskNo;POST RCS /cv/station/getError;任务号存在时调 WMS queryTask 合并料箱与物料明细进 Redis
 */
@Slf4j
@Service
public class TvRcsStationPollService {
    private static final int MAX_LOG_LEN = 2000;
    @Resource
    private RestTemplate restTemplate;
@@ -28,12 +39,18 @@
    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);
                applyTaskNoResponse(raw, pollProperties.getTaskNoStationId());
                String raw = pollTaskNoRaw();
                String taskNo = applyTaskNoResponse(raw, stationId);
                enrichSnapshotFromWms(stationId, taskNo);
            } catch (RestClientException e) {
                log.warn("RCS 任务号轮询 HTTP 失败: {}", e.getMessage());
            } catch (Exception e) {
@@ -42,7 +59,7 @@
        }
        if (StringUtils.hasText(pollProperties.getErrorPollUrl())) {
            try {
                String raw = restTemplate.getForObject(pollProperties.getErrorPollUrl(), String.class);
                String raw = pollErrorRaw();
                applyErrorResponse(raw, pollProperties.getErrorStationId());
            } catch (RestClientException e) {
                log.warn("RCS 异常轮询 HTTP 失败: {}", e.getMessage());
@@ -52,22 +69,166 @@
        }
    }
    private void applyTaskNoResponse(String raw, String staNo) {
        if (!StringUtils.hasText(staNo)) {
    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);
                wcsStationStatusService.clearStationSnapshot(stationId);
                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);
                wcsStationStatusService.clearStationSnapshot(stationId);
                return;
            }
            Object data = res.get("data");
            @SuppressWarnings("unchecked")
            Map<String, Object> dataMap = data instanceof Map ? (Map<String, Object>) data : null;
            if (dataMap == null || !hasWrkDetls(dataMap)) {
                // queryTask 空数据时清空站点缓存,避免电视机显示历史物料
                wcsStationStatusService.clearStationSnapshot(stationId);
                return;
            }
            wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, dataMap);
        } catch (Exception e) {
            log.warn("WMS queryTask 调用失败: {}", e.getMessage());
//            wcsStationStatusService.upsertFromRcsPoll(stationId, taskNo, null);
            wcsStationStatusService.clearStationSnapshot(stationId);
        }
    }
    private static boolean hasWrkDetls(Map<String, Object> dataMap) {
        if (dataMap == null) {
            return false;
        }
        Object wrkDetls = dataMap.get("wrkDetls");
        if (wrkDetls == null) {
            return false;
        }
        if (wrkDetls instanceof Collection) {
            return !((Collection<?>) wrkDetls).isEmpty();
        }
        if (wrkDetls.getClass().isArray()) {
            return java.lang.reflect.Array.getLength(wrkDetls) > 0;
        }
        return true;
    }
    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 {
        String url = pollProperties.getErrorPollUrl();
        if (pollProperties.isErrorPollUsePost()) {
            String reqBody = objectMapper.writeValueAsString(createStationBody(pollProperties.getErrorStationId()));
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_JSON);
            HttpEntity<String> entity = new HttpEntity<>(reqBody, headers);
            log.info("RCS站点轮询请求 POST /cv/station/getError url={} body={}", url, trimForLog(reqBody));
            ResponseEntity<String> resp = restTemplate.exchange(
                    url, HttpMethod.POST, entity, String.class);
            String respBody = resp.getBody();
            log.info("RCS站点轮询响应 POST /cv/station/getError body={}", trimForLog(respBody));
            return respBody;
        }
        log.info("RCS站点轮询请求 GET /cv/station/getError url={}", url);
        String raw = restTemplate.getForObject(url, String.class);
        log.info("RCS站点轮询响应 GET /cv/station/getError body={}", trimForLog(raw));
        return raw;
    }
    private String pollTaskNoRaw() throws Exception {
        String url = pollProperties.getTaskNoPollUrl();
        if (pollProperties.isTaskNoPollUsePost()) {
            String reqBody = objectMapper.writeValueAsString(createStationBody(pollProperties.getTaskNoStationId()));
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_JSON);
            HttpEntity<String> entity = new HttpEntity<>(reqBody, headers);
            // log.info("RCS站点轮询请求 GET /cv/station/getTaskNo url={}", url);
            log.info("RCS站点轮询请求 POST /cv/station/getTaskNo url={} body={}", url, trimForLog(reqBody));
            ResponseEntity<String> resp = restTemplate.exchange(
                    url, HttpMethod.POST, entity, String.class);
            String respBody = resp.getBody();
            // log.info("RCS站点轮询响应 GET /cv/station/getTaskNo body={}", trimForLog(raw));
            log.info("RCS站点轮询响应 POST /cv/station/getTaskNo body={}", trimForLog(respBody));
            return respBody;
        }
        log.info("RCS站点轮询请求 GET /cv/station/getTaskNo url={}", url);
        String raw = restTemplate.getForObject(url, String.class);
        log.info("RCS站点轮询响应 GET /cv/station/getTaskNo body={}", trimForLog(raw));
        return raw;
    }
    private ObjectNode createStationBody(String stationId) {
        ObjectNode body = objectMapper.createObjectNode();
        if (StringUtils.hasText(stationId)) {
            body.put("stationId", stationId);
            body.put("staNo", stationId);
        }
        return body;
    }
    /** 写入 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;
            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) {
@@ -122,15 +283,15 @@
        rcsTvCallbackService.handleStationError(body);
    }
    /**
     * 归一成 {@link com.vincent.rsf.openApi.service.RcsTvCallbackService#handleStationError} 可识别的 data 数组
     */
    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);
        }
@@ -194,4 +355,15 @@
        }
        return arr;
    }
    private static String trimForLog(String raw) {
        if (raw == null) {
            return "null";
        }
        String text = raw.trim();
        if (text.length() <= MAX_LOG_LEN) {
            return text;
        }
        return text.substring(0, MAX_LOG_LEN) + "...(truncated)";
    }
}