grpc 实现

jcpp-app/src/main/java/sanbing/jcpp/app/service/DownlinkCallService.java

@@ -4,12 +4,68 @@
  */
 package sanbing.jcpp.app.service;
 
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import sanbing.jcpp.app.data.PileSession;
+import sanbing.jcpp.app.service.cache.session.PileSessionCacheKey;
+import sanbing.jcpp.infrastructure.cache.CacheValueWrapper;
+import sanbing.jcpp.infrastructure.cache.TransactionalCache;
+import sanbing.jcpp.infrastructure.queue.discovery.ServiceInfoProvider;
+import sanbing.jcpp.infrastructure.util.trace.Tracer;
+import sanbing.jcpp.infrastructure.util.trace.TracerContextUtil;
+import sanbing.jcpp.proto.gen.ProtocolProto;
 import sanbing.jcpp.proto.gen.ProtocolProto.DownlinkRequestMessage;
+import sanbing.jcpp.protocol.adapter.DownlinkController;
 
 /**
  * @author baigod
  */
-public interface DownlinkCallService {
+@Slf4j
+public abstract class DownlinkCallService {
 
-    void sendDownlinkMessage(DownlinkRequestMessage.Builder downlinkMessageBuilder, String pileCode);
+    @Resource
+    protected ServiceInfoProvider serviceInfoProvider;
+
+    @Resource
+    protected DownlinkController downlinkController;
+
+    @Resource
+    protected TransactionalCache<PileSessionCacheKey, PileSession> pileSessionCache;
+
+    @Value("${cache.type}")
+    protected String cacheType;
+
+    public void sendDownlinkMessage(DownlinkRequestMessage.Builder downlinkMessageBuilder, String pileCode) {
+        CacheValueWrapper<PileSession> pileSessionCacheValueWrapper = pileSessionCache.get(new PileSessionCacheKey(pileCode));
+
+        if (pileSessionCacheValueWrapper == null) {
+            log.warn("充电桩会话不存在 {}", pileCode);
+            return;
+        }
+
+        PileSession pileSession = pileSessionCacheValueWrapper.get();
+
+        if (serviceInfoProvider.isMonolith() &&
+                ("caffeine".equalsIgnoreCase(cacheType)) || serviceInfoProvider.getServiceId().equalsIgnoreCase(pileSession.getNodeId())) {
+
+            downlinkController.onDownlink(downlinkMessageBuilder.build())
+                    .setResultHandler(result -> log.debug("下行消息发送完成"));
+
+        } else {
+            Tracer currentTracer = TracerContextUtil.getCurrentTracer();
+
+            downlinkMessageBuilder.setTracer(ProtocolProto.TracerProto.newBuilder()
+                    .setId(currentTracer.getTraceId())
+                    .setOrigin(currentTracer.getOrigin())
+                    .setTs(currentTracer.getTracerTs())
+                    .build());
+
+
+            _sendDownlinkMessage(downlinkMessageBuilder.build(), pileSession);
+        }
+    }
+
+
+    protected abstract void _sendDownlinkMessage(DownlinkRequestMessage downlinkMessage, PileSession pileSession);
 }