Add to upload list method during individual file upload

Signed-off-by: Ashish Singh <[email protected]>
opensearch-project · Apr 18, 2023 · 71cf169 · 71cf169
1 parent 2348e3a
commit 71cf169
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/server/src/main/java/org/opensearch/index/RemoteSegmentUploadShardStatsTracker.java b/server/src/main/java/org/opensearch/index/RemoteSegmentUploadShardStatsTracker.java
@@ -247,10 +247,6 @@ public void updateLatestLocalFileNameLengthMap(Map<String, Long> latestLocalFile
         this.latestLocalFileNameLengthMap = latestLocalFileNameLengthMap;
     }
 
-    public Set<String> getLatestUploadFiles() {
-        return latestUploadFiles;
-    }
-
     public void updateLatestUploadFiles(Set<String> latestUploadFiles) {
         this.latestUploadFiles = latestUploadFiles;
     }
@@ -325,4 +321,8 @@ public void updateUploadBytesPerSecMovingAverageWindowSize(int updatedSize) {
     public void updateUploadTimeMovingAverageWindowSize(int updatedSize) {
         this.uploadTimeMovingAverageReference.set(new MovingAverage(updatedSize));
     }
+
+    public void addToLatestUploadFiles(String file) {
+        this.latestUploadFiles.add(file);
+    }
 }
diff --git a/server/src/main/java/org/opensearch/index/shard/RemoteStoreRefreshListener.java b/server/src/main/java/org/opensearch/index/shard/RemoteStoreRefreshListener.java
@@ -322,6 +322,7 @@ UploadStatus uploadNewSegments(
                 remoteDirectory.copyFrom(storeDirectory, file, file, IOContext.DEFAULT);
                 // Upload succeeded
                 statsTracker.incrementUploadBytesSucceeded(fileSize);
+                statsTracker.addToLatestUploadFiles(file);
                 success = true;
             } catch (IOException e) {
                 // ToDO: Handle transient and permanent un-availability of the remote store (GitHub #3397)