Merge pull request #2472 from atlanhq/PLT-245-beta-apikey-counter

Add metric counter when authenticating a request from a service-account
atlanhq · Nov 14, 2023 · 2fd5091 · 2fd5091
2 parents 5c491e3 + ac5d5e1
commit 2fd5091
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/webapp/src/main/java/org/apache/atlas/web/security/AtlasKeycloakAuthenticationProvider.java b/webapp/src/main/java/org/apache/atlas/web/security/AtlasKeycloakAuthenticationProvider.java
@@ -16,6 +16,8 @@
  */
 package org.apache.atlas.web.security;
 
+import io.micrometer.core.instrument.Counter;
+import org.apache.atlas.service.metrics.MetricUtils;
 import org.apache.atlas.ApplicationProperties;
 import org.apache.commons.configuration.Configuration;
 import org.keycloak.adapters.springsecurity.authentication.KeycloakAuthenticationProvider;
@@ -66,6 +68,11 @@ public Authentication authenticate(Authentication authentication) {
       }
     }
 
+    if(authentication.getName().startsWith("service-account-apikey")) {
+      // Increment the counter when the authentication is for a service account.
+      Counter.builder("service_account_apikey_request_counter").register(MetricUtils.getMeterRegistry()).increment();
+    }
+
     return authentication;
   }