JChargePointProtocol/jcpp-protocol-api/src/main/java/sanbing/jcpp/protocol/adapter/DownlinkController.java

/**
 * 开源代码，仅供学习和交流研究使用，商用请联系三丙
 * 微信：mohan_88888
 * 抖音：程序员三丙
 * 付费课程知识星球：https://t.zsxq.com/aKtXo
 */
package sanbing.jcpp.protocol.adapter;

import jakarta.annotation.Resource;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.async.DeferredResult;
import sanbing.jcpp.proto.gen.ProtocolProto.DownlinkRequestMessage;
import sanbing.jcpp.protocol.domain.ProtocolSession;
import sanbing.jcpp.protocol.provider.ProtocolSessionRegistryProvider;

import java.util.UUID;

/**
 * @author baigod
 */
@RestController
@RequestMapping("/api")
@RequiredArgsConstructor
@Slf4j
public class DownlinkController {

    @Value("${api.timeout.onDownlink:3000}")
    public long onDownlinkTimeout;

    @Resource
    ProtocolSessionRegistryProvider protocolSessionRegistryProvider;

    @PostMapping(value = "/onDownlink", consumes = "application/x-protobuf", produces = "application/x-protobuf")
    public DeferredResult<ResponseEntity<String>> onDownlink(@RequestBody DownlinkRequestMessage downlinkMsg) {
        log.debug("收到REST下行请求 {}", downlinkMsg);

        final DeferredResult<ResponseEntity<String>> response = new DeferredResult<>(onDownlinkTimeout,
                ResponseEntity.status(HttpStatus.REQUEST_TIMEOUT).build());

        UUID protocolSessionId = new UUID(downlinkMsg.getSessionIdMSB(), downlinkMsg.getSessionIdLSB());

        ProtocolSession protocolSession = protocolSessionRegistryProvider.get(protocolSessionId);

        try {
            if (protocolSession != null) {

                protocolSession.onDownlink(downlinkMsg);

                response.setResult(ResponseEntity.status(HttpStatus.OK).build());
            } else {

                log.info("下发报文时Session未找到 sessionId: {}", protocolSessionId);

                response.setResult(ResponseEntity.status(HttpStatus.NOT_FOUND).body("Protocol Session not found for ID:" + protocolSessionId));
            }
        } catch (Exception e) {

            log.warn("下发报文时处理失败 sessionId: {}", protocolSessionId, e);

            if (!response.isSetOrExpired()) {

                response.setResult(ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(e.getMessage()));
            }
        }

        return response;
    }
}