fix: fix trace agent

traas-stack · Aug 15, 2023 · 606643a · 606643a
1 parent abcb9d5
commit 606643a
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/...ome-web/src/main/java/io/holoinsight/server/home/web/controller/TraceAgentFacadeImpl.java b/...ome-web/src/main/java/io/holoinsight/server/home/web/controller/TraceAgentFacadeImpl.java
@@ -9,6 +9,7 @@
 import io.holoinsight.server.home.biz.common.MetaDictType;
 import io.holoinsight.server.home.biz.common.MetaDictUtil;
 import io.holoinsight.server.home.biz.service.ApiKeyService;
+import io.holoinsight.server.home.biz.service.TenantInitService;
 import io.holoinsight.server.home.common.util.scope.RequestContext;
 import io.holoinsight.server.home.dal.model.ApiKey;
 import io.holoinsight.server.home.web.common.AesUtil;
@@ -39,6 +40,10 @@ public class TraceAgentFacadeImpl extends BaseFacade {
   @Autowired
   private ApiKeyService apiKeyService;
 
+  @Autowired
+  private TenantInitService tenantInitService;
+
+
   @Autowired
   private TraceAuthEncryptConfiguration traceAuthEncryptConfiguration;
 
@@ -55,10 +60,7 @@ public JsonResult<Map<String, Object>> traceAgent(
     final JsonResult<Map<String, Object>> result = new JsonResult<>();
     try {
       String tenant = RequestContext.getContext().ms.getTenant();
-      Map<String, Object> conditions = new HashMap<>();
-      conditions.put("tenant", tenant);
-      conditions.put("status", true);
-      List<ApiKey> apiKeys = apiKeyService.listByMap(conditions);
+      List<ApiKey> apiKeys = apiKeyService.listByMap(apikeyConditions(tenant, extendInfo));
 
       Map<String, Object> sysMap = new HashMap<>();
       String apikey = "";
@@ -98,6 +100,13 @@ public JsonResult<Map<String, Object>> traceAgent(
     return result;
   }
 
+  public Map<String, Object> apikeyConditions(String tenant, Map<String, String> extendInfo) {
+    Map<String, Object> conditions = new HashMap<>();
+    conditions.put("tenant", tenantInitService.getTsdbTenant(tenant));
+    conditions.put("status", true);
+    return conditions;
+  }
+
   public String getCollectorAddress(Map<String, String> extendInfo) {
     return MetaDictUtil.getStringValue(MetaDictType.TRACE_AGENT_CONFIG, MetaDictKey.COLLECTOR_HOST);
   }