云快充1.5.0 初始化

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

@@ -0,0 +1,78 @@
+/**
+ * 抖音关注：程序员三丙
+ * 知识星球：https://t.zsxq.com/j9b21
+ */
+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.DownlinkRestMessage;
+import sanbing.jcpp.protocol.domain.ProtocolSession;
+import sanbing.jcpp.protocol.provider.ProtocolSessionRegistryProvider;
+
+import java.util.UUID;
+import java.util.concurrent.CompletableFuture;
+
+/**
+ * @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 DownlinkRestMessage downlinkMsg) {
+        log.info("收到REST下行请求 {}", downlinkMsg);
+
+        final DeferredResult<ResponseEntity<String>> response = new DeferredResult<>(onDownlinkTimeout,
+                ResponseEntity.status(HttpStatus.REQUEST_TIMEOUT).build());
+
+        UUID protocolSessionId = new UUID(downlinkMsg.getSessionIdMSB(),downlinkMsg.getSessionIdLSB()) ;
+
+        CompletableFuture<ProtocolSession> protocolSessionCompletableFuture = protocolSessionRegistryProvider.get(protocolSessionId);
+
+        protocolSessionCompletableFuture.thenAccept(session -> {
+            if (session != null) {
+
+                session.onDownlink(downlinkMsg);
+
+                response.setResult(ResponseEntity.status(HttpStatus.OK).build());
+            } else {
+
+                log.warn("下发报文时Session未找到 sessionId: {}", protocolSessionId);
+
+                response.setResult(ResponseEntity.status(HttpStatus.NOT_FOUND).body("Protocol Session not found for ID:" + protocolSessionId));
+            }
+        }).whenComplete((unused, throwable) -> {
+            if (throwable != null) {
+
+                log.warn("下发报文时处理失败 sessionId: {}", protocolSessionId, throwable);
+
+                if (!response.isSetOrExpired()) {
+
+                    response.setResult(ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(throwable.getMessage()));
+                }
+            }
+        });
+
+        return response;
+    }
+}

jcpp-protocol-api/src/main/java/sanbing/jcpp/protocol/adapter/config/TracerInterceptor.java

@@ -0,0 +1,41 @@
+/**
+ * 抖音关注：程序员三丙
+ * 知识星球：https://t.zsxq.com/j9b21
+ */
+package sanbing.jcpp.protocol.adapter.config;
+
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import org.springframework.stereotype.Component;
+import org.springframework.web.servlet.HandlerInterceptor;
+import sanbing.jcpp.infrastructure.util.mdc.MDCUtils;
+import sanbing.jcpp.infrastructure.util.trace.TracerContextUtil;
+
+import static sanbing.jcpp.infrastructure.util.trace.TracerContextUtil.*;
+
+@Component
+public class TracerInterceptor implements HandlerInterceptor {
+
+    @Override
+    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
+        String tracerId = request.getHeader(JCPP_TRACER_ID);
+        String tracerOrigin = request.getHeader(JCPP_TRACER_ORIGIN);
+        String tracerTsStr = request.getHeader(JCPP_TRACER_TS);
+
+        long tracerTs;
+        if (tracerTsStr != null) {
+            try {
+                tracerTs = Long.parseLong(tracerTsStr);
+            } catch (NumberFormatException e) {
+                tracerTs = System.currentTimeMillis();
+            }
+        } else {
+            tracerTs = System.currentTimeMillis();
+        }
+
+        TracerContextUtil.newTracer(tracerId, tracerOrigin, tracerTs);
+        MDCUtils.recordTracer();
+
+        return true;
+    }
+}

jcpp-protocol-api/src/main/java/sanbing/jcpp/protocol/adapter/config/UndertowServletWebServerCustomizer.java

@@ -0,0 +1,24 @@
+/**
+ * 抖音关注：程序员三丙
+ * 知识星球：https://t.zsxq.com/j9b21
+ */
+package sanbing.jcpp.protocol.adapter.config;
+
+import io.undertow.server.DefaultByteBufferPool;
+import io.undertow.websockets.jsr.WebSocketDeploymentInfo;
+import org.springframework.boot.web.embedded.undertow.UndertowServletWebServerFactory;
+import org.springframework.boot.web.server.WebServerFactoryCustomizer;
+import org.springframework.stereotype.Component;
+
+
+@Component
+public class UndertowServletWebServerCustomizer implements WebServerFactoryCustomizer<UndertowServletWebServerFactory> {
+    @Override
+    public void customize(UndertowServletWebServerFactory factory) {
+        factory.addDeploymentInfoCustomizers(deploymentInfo -> {
+            WebSocketDeploymentInfo webSocketDeploymentInfo = new WebSocketDeploymentInfo();
+            webSocketDeploymentInfo.setBuffers(new DefaultByteBufferPool(true, 128 * 1024 * 1024));
+            deploymentInfo.addServletContextAttribute("io.undertow.websockets.jsr.WebSocketDeploymentInfo", webSocketDeploymentInfo);
+        });
+    }
+}

jcpp-protocol-api/src/main/java/sanbing/jcpp/protocol/adapter/config/WebMvcConfiguration.java

@@ -0,0 +1,45 @@
+/**
+ * 抖音关注：程序员三丙
+ * 知识星球：https://t.zsxq.com/j9b21
+ */
+package sanbing.jcpp.protocol.adapter.config;
+
+import jakarta.annotation.Resource;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.converter.HttpMessageConverter;
+import org.springframework.http.converter.StringHttpMessageConverter;
+import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;
+import org.springframework.http.converter.protobuf.ProtobufHttpMessageConverter;
+import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+
+@Configuration
+public class WebMvcConfiguration implements WebMvcConfigurer {
+
+    @Resource
+    private TracerInterceptor tracerInterceptor;
+
+    @Override
+    public void configureMessageConverters(List<HttpMessageConverter<?>> converters) {
+        for (HttpMessageConverter<?> converter : converters) {
+            if (converter instanceof StringHttpMessageConverter) {
+                ((StringHttpMessageConverter) converter).setDefaultCharset(StandardCharsets.UTF_8);
+            }
+
+            if (converter instanceof MappingJackson2HttpMessageConverter) {
+                ((MappingJackson2HttpMessageConverter) converter).setDefaultCharset(StandardCharsets.UTF_8);
+            }
+        }
+
+        // protobuf 序列化
+        converters.add( new ProtobufHttpMessageConverter());
+    }
+
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {
+        registry.addInterceptor(tracerInterceptor).addPathPatterns("/**");
+    }
+}