Pass testStatsObjectCreationViaStream

Signed-off-by: Bhumika Saini <[email protected]>
opensearch-project · Aug 14, 2023 · 4e8f6e8 · 4e8f6e8
1 parent 9adab8b
commit 4e8f6e8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/node/Node.java b/server/src/main/java/org/opensearch/node/Node.java
@@ -45,6 +45,7 @@
 import org.opensearch.index.IndexModule;
 import org.opensearch.index.IndexingPressureService;
 import org.opensearch.index.recovery.RemoteStoreRestoreService;
+import org.opensearch.index.remote.RemoteStorePressureService;
 import org.opensearch.index.store.remote.filecache.FileCache;
 import org.opensearch.index.store.remote.filecache.FileCacheCleaner;
 import org.opensearch.index.store.remote.filecache.FileCacheFactory;
@@ -260,6 +261,7 @@
 import java.util.stream.Stream;
 
 import static java.util.stream.Collectors.toList;
+import static org.opensearch.common.util.FeatureFlags.REMOTE_STORE;
 import static org.opensearch.common.util.FeatureFlags.TELEMETRY;
 import static org.opensearch.env.NodeEnvironment.collectFileCacheDataPath;
 import static org.opensearch.index.ShardIndexingPressureSettings.SHARD_INDEXING_PRESSURE_ENABLED_ATTRIBUTE_KEY;
@@ -1266,6 +1268,12 @@ public Node start() throws NodeValidationException {
         nodeService.getSearchBackpressureService().start();
         nodeService.getTaskCancellationMonitoringService().start();
 
+        if (FeatureFlags.isEnabled(REMOTE_STORE)) {
+            IndicesService indicesService = injector.getInstance(IndicesService.class);
+            RemoteStorePressureService pressureService = injector.getInstance(RemoteStorePressureService.class);
+            indicesService.setPressureService(pressureService);
+        }
+
         final ClusterService clusterService = injector.getInstance(ClusterService.class);
 
         final NodeConnectionsService nodeConnectionsService = injector.getInstance(NodeConnectionsService.class);

diff --git a/server/src/test/java/org/opensearch/index/remote/RemoteTranslogTrackerTests.java b/server/src/test/java/org/opensearch/index/remote/RemoteTranslogTrackerTests.java
@@ -422,15 +422,16 @@ private void populateUploadBytesStarted() {
 
     private void populateDummyStats() {
         int startedBytesUpload = randomIntBetween(10, 100);
+        int startedUploads = randomIntBetween(6, 10);
 
         tracker.addUploadBytesStarted(startedBytesUpload);
         tracker.addUploadBytesFailed(randomIntBetween(1, startedBytesUpload / 2));
         tracker.addUploadBytesSucceeded(randomIntBetween(1, startedBytesUpload / 2));
         tracker.addUploadTimeInMillis(randomIntBetween(10, 100));
         tracker.setLastSuccessfulUploadTimestamp(System.currentTimeMillis() + randomIntBetween(10, 100));
         tracker.addUploadsStarted(randomIntBetween(6, 10));
-        tracker.addUploadsFailed(randomIntBetween(1, 5));
-        tracker.addUploadsSucceeded(randomIntBetween(1, 5));
+        tracker.addUploadsFailed(randomIntBetween(1, startedUploads / 2));
+        tracker.addUploadsSucceeded(randomIntBetween(1, startedUploads / 2));
 
         tracker.addDownloadBytesSucceeded(randomIntBetween(10, 100));
         tracker.addDownloadTimeInMillis(randomIntBetween(10, 100));