capture number of queries that timedout

atlanhq · Nov 14, 2024 · eac1e55 · eac1e55
1 parent ce06a59
commit eac1e55
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...anus/src/main/java/org/apache/atlas/repository/graphdb/janus/AtlasElasticsearchQuery.java b/...anus/src/main/java/org/apache/atlas/repository/graphdb/janus/AtlasElasticsearchQuery.java
@@ -176,6 +176,7 @@ private Map<String, Object> runQueryWithLowLevelClient(String query) throws Atla
 
     private DirectIndexQueryResult performAsyncDirectIndexQuery(SearchParams searchParams) throws AtlasBaseException, IOException {
         AtlasPerfMetrics.MetricRecorder metric = RequestContext.get().startMetricRecord("performAsyncDirectIndexQuery");
+        AtlasPerfMetrics.MetricRecorder metricSearchTimeout = RequestContext.get().startMetricRecord("asyncDirectIndexQueryTimeout");
         DirectIndexQueryResult result = null;
         boolean contextIdExists = StringUtils.isNotEmpty(searchParams.getSearchContextId()) && searchParams.getSearchContextSequenceNo() != null;
         try {
@@ -207,7 +208,8 @@ private DirectIndexQueryResult performAsyncDirectIndexQuery(SearchParams searchP
                 if (response ==  null) {
                     // Rather than null (if the response is null wil help returning @204 HTTP_NO_CONTENT to the user)
                     // return timeout exception to user
-                    LOG.error("query timeout exceeded {}:", searchParams.getQuery());
+                    LOG.error("timeout exceeded for query {}:", searchParams.getQuery());
+                    RequestContext.get().endMetricRecord(metricSearchTimeout);
                     throw new AtlasBaseException(AtlasErrorCode.INDEX_SEARCH_FAILED_DUE_TO_TIMEOUT, KeepAliveTime);
                 }
                 result = getResultFromResponse(response.getFullResponse(), true);