Address PR comments

Signed-off-by: Shivansh Arora <[email protected]>

server/src/main/java/org/opensearch/cluster/routing/remote/InternalRemoteRoutingTableService.java

@@ -274,7 +274,7 @@ private void uploadIndex(
     }
 
     @Override
-    public CheckedRunnable<IOException> getAsyncIndexRoutingReadAction(
+    public void getAsyncIndexRoutingReadAction(
         String uploadedFilename,
         Index index,
         LatchedActionListener<IndexRoutingTable> latchedActionListener
@@ -284,7 +284,7 @@ public CheckedRunnable<IOException> getAsyncIndexRoutingReadAction(
         BlobContainer blobContainer = blobStoreRepository.blobStore()
             .blobContainer(BlobPath.cleanPath().add(uploadedFilename.substring(0, idx)));
 
-        return () -> readAsync(
+        readAsync(
             blobContainer,
             blobFileName,
             index,

server/src/main/java/org/opensearch/cluster/routing/remote/NoopRemoteRoutingTableService.java

@@ -63,13 +63,12 @@ public List<ClusterMetadataManifest.UploadedIndexMetadata> getAllUploadedIndices
     }
 
     @Override
-    public CheckedRunnable<IOException> getAsyncIndexRoutingReadAction(
+    public void getAsyncIndexRoutingReadAction(
         String uploadedFilename,
         Index index,
         LatchedActionListener<IndexRoutingTable> latchedActionListener
     ) {
         // noop
-        return () -> {};
     }
 
     @Override

server/src/main/java/org/opensearch/cluster/routing/remote/RemoteRoutingTableService.java

@@ -46,7 +46,7 @@ public IndexRoutingTable read(StreamInput in, String key) throws IOException {
 
     List<IndexRoutingTable> getIndicesRouting(RoutingTable routingTable);
 
-    CheckedRunnable<IOException> getAsyncIndexRoutingReadAction(
+    void getAsyncIndexRoutingReadAction(
         String uploadedFilename,
         Index index,
         LatchedActionListener<IndexRoutingTable> latchedActionListener

server/src/main/java/org/opensearch/common/remote/AbstractRemoteWritableEntityManager.java

@@ -8,12 +8,10 @@
 
 package org.opensearch.common.remote;
 
-import org.opensearch.common.CheckedRunnable;
 import org.opensearch.core.action.ActionListener;
 import org.opensearch.gateway.remote.ClusterMetadataManifest;
 import org.opensearch.gateway.remote.model.RemoteReadResult;
 
-import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -71,20 +69,16 @@ protected abstract ActionListener<Object> getReadActionListener(
     );
 
     @Override
-    public CheckedRunnable<IOException> asyncWrite(
+    public void writeAsync(
         String component,
         AbstractRemoteWritableBlobEntity entity,
         ActionListener<ClusterMetadataManifest.UploadedMetadata> listener
     ) {
-        return () -> getStore(entity).writeAsync(entity, getWriteActionListener(component, entity, listener));
+        getStore(entity).writeAsync(entity, getWriteActionListener(component, entity, listener));
     }
 
     @Override
-    public CheckedRunnable<IOException> asyncRead(
-        String component,
-        AbstractRemoteWritableBlobEntity entity,
-        ActionListener<RemoteReadResult> listener
-    ) {
-        return () -> getStore(entity).readAsync(entity, getReadActionListener(component, entity, listener));
+    public void readAsync(String component, AbstractRemoteWritableBlobEntity entity, ActionListener<RemoteReadResult> listener) {
+        getStore(entity).readAsync(entity, getReadActionListener(component, entity, listener));
     }
 }

server/src/main/java/org/opensearch/common/remote/RemoteWritableEntityManager.java

@@ -8,43 +8,40 @@
 
 package org.opensearch.common.remote;
 
-import org.opensearch.common.CheckedRunnable;
 import org.opensearch.core.action.ActionListener;
 import org.opensearch.gateway.remote.ClusterMetadataManifest.UploadedMetadata;
 import org.opensearch.gateway.remote.model.RemoteReadResult;
 
-import java.io.IOException;
-
 /**
  * The RemoteWritableEntityManager interface provides async read and write methods for managing remote entities in the remote store
  */
 public interface RemoteWritableEntityManager {
 
     /**
-     * Returns a CheckedRunnable that performs an asynchronous read operation for the specified component and entity.
+     * Performs an asynchronous read operation for the specified component and entity.
      *
      * @param component the component for which the read operation is performed
      * @param entity the entity to be read
-     * @param listener the listener to be notified when the read operation completes
-     * @return a CheckedRunnable that performs the asynchronous read operation
+     * @param listener the listener to be notified when the read operation completes.
+     *                 The listener's {@link ActionListener#onResponse(Object)} method
+     *                 is called with a {@link RemoteReadResult} object containing the
+     *                 read data on successful read. The
+     *                 {@link ActionListener#onFailure(Exception)} method is called with
+     *                 an exception if the read operation fails.
      */
-    CheckedRunnable<IOException> asyncRead(
-        String component,
-        AbstractRemoteWritableBlobEntity entity,
-        ActionListener<RemoteReadResult> listener
-    );
+    void readAsync(String component, AbstractRemoteWritableBlobEntity entity, ActionListener<RemoteReadResult> listener);
 
     /**
-     * Returns a CheckedRunnable that performs an asynchronous write operation for the specified component and entity.
+     * Performs an asynchronous write operation for the specified component and entity.
      *
      * @param component the component for which the write operation is performed
      * @param entity the entity to be written
-     * @param listener the listener to be notified when the write operation completes
-     * @return a CheckedRunnable that performs the asynchronous write operation
+     * @param listener the listener to be notified when the write operation completes.
+     *                 The listener's {@link ActionListener#onResponse(Object)} method
+     *                 is called with a {@link UploadedMetadata} object containing the
+     *                 uploaded metadata on successful write. The
+     *                 {@link ActionListener#onFailure(Exception)} method is called with
+     *                 an exception if the write operation fails.
      */
-    CheckedRunnable<IOException> asyncWrite(
-        String component,
-        AbstractRemoteWritableBlobEntity entity,
-        ActionListener<UploadedMetadata> listener
-    );
+    void writeAsync(String component, AbstractRemoteWritableBlobEntity entity, ActionListener<UploadedMetadata> listener);
 }