클라이언트 유량제어 eapim-online 이동

2026-05-26 16:33:15 +09:00
parent 912a282b54
commit 4f54ee5259
11 changed files with 465 additions and 12 deletions
@@ -0,0 +1,114 @@
+package com.eactive.eai.custom.inflow;
+
+import java.util.Collections;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+import org.springframework.stereotype.Component;
+
+import com.eactive.eai.common.inflow.Bucket;
+import com.eactive.eai.common.inflow.InflowTargetVO;
+import com.eactive.eai.common.inflow.InflowType;
+import com.eactive.eai.common.inflow.dual.DualCustomBucket;
+import com.eactive.eai.common.inflow.dual.DualInflowControlManager;
+import com.eactive.eai.common.lifecycle.LifecycleException;
+import com.eactive.eai.common.util.ApplicationContextProvider;
+import com.eactive.eai.common.util.Logger;
+
+/**
+ * 어댑터/인터페이스 이중 버킷(DualInflowControlManager)에 클라이언트 유량제어를 추가한 관리자.
+ *
+ * <p>적용 방법: INFLOW.properties에 아래 한 줄 추가.
+ * <pre>inflow.control.bucket.className=com.eactive.eai.common.inflow.dual.ClientDualInflowControlManager</pre>
+ */
+@Component
+public class ClientDualInflowControlManager extends DualInflowControlManager implements ClientDualBucket {
+
+    private static final Logger logger = Logger.getLogger(Logger.LOGGER_DEFAULT);
+
+    public static synchronized Bucket getBucket() {
+        return ApplicationContextProvider.getContext().getBean(ClientDualInflowControlManager.class);
+    }
+
+    private volatile Map<String, DualCustomBucket> clientBucketMap = new ConcurrentHashMap<>();
+    private final ConcurrentHashMap<String, TargetMetrics> clientMetricsMap = new ConcurrentHashMap<>();
+
+    // -------------------------------------------------------------------------
+    // Lifecycle
+    // -------------------------------------------------------------------------
+
+    @Override
+    public void start() throws LifecycleException {
+        super.start();
+        try {
+            initClients();
+        } catch (Exception e) {
+            throw new LifecycleException("ClientDualInflowControlManager init failed: " + e.getMessage());
+        }
+    }
+
+    // -------------------------------------------------------------------------
+    // 초기화 / 리로드
+    // -------------------------------------------------------------------------
+
+    private void initClients() throws Exception {
+        this.clientBucketMap = loadBucketMap(InflowType.CLIENT);
+        if (logger.isInfo()) logger.info("ClientDualInflowControlManager] initClients: " + clientBucketMap.size() + " buckets");
+    }
+
+    public synchronized void reloadClient() throws Exception {
+        initClients();
+        clientMetricsMap.clear();
+    }
+
+    public synchronized void reloadClient(String clientId) throws Exception {
+        reloadBucketMap(clientBucketMap, InflowType.CLIENT, clientId);
+        TargetMetrics m = clientMetricsMap.get(clientId);
+        if (m != null) m.reset();
+    }
+
+    public synchronized void removeClient(String clientId) {
+        clientBucketMap.remove(clientId);
+        clientMetricsMap.remove(clientId);
+    }
+
+    // -------------------------------------------------------------------------
+    // 클라이언트 메트릭 접근자
+    // -------------------------------------------------------------------------
+
+    public TargetMetrics getClientMetrics(String clientId)    { return clientMetricsMap.get(clientId); }
+    public Map<String, TargetMetrics> getAllClientMetrics()    { return Collections.unmodifiableMap(clientMetricsMap); }
+    public void resetClientMetrics(String clientId)           { TargetMetrics m = clientMetricsMap.get(clientId); if (m != null) m.reset(); }
+    public void resetAllClientMetrics()                       { clientMetricsMap.values().forEach(TargetMetrics::reset); }
+
+    // -------------------------------------------------------------------------
+    // ClientDualBucket — 클라이언트 유량제어
+    // -------------------------------------------------------------------------
+
+    @Override
+    public String isClientPassDetail(String clientId) {
+        DualCustomBucket b = clientBucketMap.get(clientId);
+        if (b == null) return DualCustomBucket.RESULT_PASS;
+        String result = b.tryConsume();
+        recordMetrics(clientMetricsMap, clientId, result);
+        return result;
+    }
+
+    @Override
+    public InflowTargetVO getClientInflowThreshold(String clientId) {
+        DualCustomBucket b = clientBucketMap.get(clientId);
+        return (b == null) ? null : b.getInflowTargetVo();
+    }
+
+    // -------------------------------------------------------------------------
+    // 모니터링용 접근자
+    // -------------------------------------------------------------------------
+
+    public DualCustomBucket getClientBucket(String clientId) {
+        return clientBucketMap.get(clientId);
+    }
+
+    public Map<String, DualCustomBucket> getClientBucketMap() {
+        return clientBucketMap;
+    }
+}