Make singleton

Signed-off-by: Siddhant Deshmukh <deshsid@amazon.com>
deshsidd · Jul 18, 2024 · 2c3e40a · 2c3e40a
1 parent 49bcc31
commit 2c3e40a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/action/search/TransportSearchAction.java b/server/src/main/java/org/opensearch/action/search/TransportSearchAction.java
@@ -435,7 +435,7 @@ private void executeRequest(
                 originalSearchRequest,
                 logger,
                 TraceableSearchRequestOperationsListener.create(tracer, requestSpan),
-                new SearchTaskRequestOperationsListener(taskResourceTrackingService)
+                SearchTaskRequestOperationsListener.getInstance(taskResourceTrackingService)
             );
             SearchRequestContext searchRequestContext = new SearchRequestContext(
                 requestOperationsListeners,

diff --git a/server/src/main/java/org/opensearch/tasks/SearchTaskRequestOperationsListener.java b/server/src/main/java/org/opensearch/tasks/SearchTaskRequestOperationsListener.java
@@ -19,12 +19,20 @@
  *
  */
 public final class SearchTaskRequestOperationsListener extends SearchRequestOperationsListener {
+    private static SearchTaskRequestOperationsListener instance;
     private final TaskResourceTrackingService taskResourceTrackingService;
 
-    public SearchTaskRequestOperationsListener(TaskResourceTrackingService taskResourceTrackingService) {
+    private SearchTaskRequestOperationsListener(TaskResourceTrackingService taskResourceTrackingService) {
         this.taskResourceTrackingService = taskResourceTrackingService;
     }
 
+    public static synchronized SearchTaskRequestOperationsListener getInstance(TaskResourceTrackingService taskResourceTrackingService) {
+        if (instance == null) {
+            instance = new SearchTaskRequestOperationsListener(taskResourceTrackingService);
+        }
+        return instance;
+    }
+
     @Override
     public void onRequestEnd(SearchPhaseContext context, SearchRequestContext searchRequestContext) {
         taskResourceTrackingService.refreshResourceStats(context.getTask());