初始化

2025-09-10 15:36:49 +08:00
commit 4330ffdff1
75 changed files with 6674 additions and 0 deletions
@@ -0,0 +1,48 @@
+package com.tongran.agent.client.netty.basics;
+
+import com.tongran.agent.client.netty.annotation.AgentDispatcher;
+import org.springframework.beans.BeansException;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
+import org.springframework.core.annotation.Order;
+import org.springframework.stereotype.Component;
+import org.springframework.util.CollectionUtils;
+
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Component
+@Order(1)
+public class AgentDispatcherManager implements ApplicationContextAware {
+
+    /**
+     * 所有实现的包处理器
+     */
+    private static Map<String, AgentHandler> MSG_HANDLER_MAP;
+
+    /**
+     * 唤醒时 初始化 packHandlerMap
+     */
+    @Override
+    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+        // 仅一次性初始化完成
+        if (MSG_HANDLER_MAP == null) {
+            MSG_HANDLER_MAP = new ConcurrentHashMap<>();
+            Map<String, Object> handlers = applicationContext.getBeansWithAnnotation(AgentDispatcher.class);
+            if (!CollectionUtils.isEmpty(handlers)) {
+                handlers.values().forEach(tempHandler -> {
+                    boolean result = tempHandler.getClass().isAnnotationPresent(AgentDispatcher.class);
+                    if (result) {
+                        AgentDispatcher annotation = tempHandler.getClass().getAnnotation(AgentDispatcher.class);
+                        MSG_HANDLER_MAP.put(annotation.msgId().getValue() + "&" + annotation.version().value, (AgentHandler) tempHandler);
+                    }
+                });
+            }
+        }
+    }
+
+    public AgentHandler getHandler(String msgIdVersion) {
+        return MSG_HANDLER_MAP == null ? null : MSG_HANDLER_MAP.get(msgIdVersion);
+    }
+
+}
@@ -0,0 +1,21 @@
+package com.tongran.agent.client.netty.basics;
+
+
+import com.tongran.agent.client.netty.model.UpMsgResponse;
+
+public interface AgentHandler {
+
+    /**
+     * 处理终端传入的消息, 然后进行返回
+     *
+     * @return 需要发送给终端的消息
+     */
+
+    default UpMsgResponse upHandle(String data, String clientId) {
+        return null;
+    }
+
+    default String downHandle(String data) {
+        return "";
+    }
+}