diff --git a/client-auth/src/main/java/org/apache/atlas/auth/client/auth/AbstractAuthClient.java b/client-auth/src/main/java/org/apache/atlas/auth/client/auth/AbstractAuthClient.java index cb13431384..988a120bfd 100644 --- a/client-auth/src/main/java/org/apache/atlas/auth/client/auth/AbstractAuthClient.java +++ b/client-auth/src/main/java/org/apache/atlas/auth/client/auth/AbstractAuthClient.java @@ -38,7 +38,7 @@ public class AbstractAuthClient { private static final String BEARER = "Bearer "; private static final int TIMEOUT_IN_SEC = 60; private static final String INTEGRATION = "integration"; - private static final String KEYCLOAK = "keycloak"; + private static final String AUTH = "auth"; protected final AuthConfig authConfig; protected final RetrofitKeycloakClient retrofitKeycloakClient; @@ -84,10 +84,10 @@ public AbstractAuthClient(AuthConfig authConfig) { Interceptor responseLoggingInterceptor = chain -> { Request request = chain.request(); String rawPath = request.url().uri().getRawPath(); - Timer.Sample timerSample = this.metricUtils.start(rawPath); + Timer.Sample timerSample = this.metricUtils.start(rawPath, false); okhttp3.Response response = chain.proceed(request); this.metricUtils.recordHttpTimer(timerSample, request.method(), rawPath, response.code(), - INTEGRATION, KEYCLOAK); + INTEGRATION, AUTH); return response; }; diff --git a/common/src/main/java/org/apache/atlas/service/metrics/MetricUtils.java b/common/src/main/java/org/apache/atlas/service/metrics/MetricUtils.java index c309190391..523225ce7f 100644 --- a/common/src/main/java/org/apache/atlas/service/metrics/MetricUtils.java +++ b/common/src/main/java/org/apache/atlas/service/metrics/MetricUtils.java @@ -52,6 +52,10 @@ public Timer.Sample start(String uri) { return matchCanonicalPattern(uri).isPresent() ? Timer.start(getMeterRegistry()) : null; } + public Timer.Sample start(String uri, boolean checkCanonicalPattern) { + return checkCanonicalPattern ? start(uri) : Timer.start(getMeterRegistry()); + } + public void recordHttpTimer(Timer.Sample sample, String method, String rawPath, int code, String... additionalTags) { if (Objects.isNull(sample)) { return; diff --git a/webapp/src/main/java/org/apache/atlas/web/rest/DiscoveryREST.java b/webapp/src/main/java/org/apache/atlas/web/rest/DiscoveryREST.java index 79eb0a6c1f..7717baa49a 100644 --- a/webapp/src/main/java/org/apache/atlas/web/rest/DiscoveryREST.java +++ b/webapp/src/main/java/org/apache/atlas/web/rest/DiscoveryREST.java @@ -51,7 +51,6 @@ import javax.inject.Inject; import javax.inject.Singleton; import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; import javax.ws.rs.Consumes; import javax.ws.rs.DELETE; import javax.ws.rs.GET; @@ -388,8 +387,7 @@ public AtlasSearchResult searchWithParameters(SearchParameters parameters) throw @Path("indexsearch") @POST @Timed - public AtlasSearchResult indexSearch(@Context HttpServletRequest servletRequest, IndexSearchParams parameters, - @Context HttpServletResponse response) throws AtlasBaseException { + public AtlasSearchResult indexSearch(@Context HttpServletRequest servletRequest, IndexSearchParams parameters) throws AtlasBaseException { AtlasPerfTracer perf = null; long startTime = System.currentTimeMillis();