Minor fix in dropping covering index

Signed-off-by: Vamsi Manohar <[email protected]>
opensearch-project · Oct 5, 2023 · c30aa2e · c30aa2e
1 parent a83482d
commit c30aa2e
Show file tree

Hide file tree

Showing 6 changed files with 20 additions and 20 deletions.
diff --git a/plugin/src/main/java/org/opensearch/sql/plugin/SQLPlugin.java b/plugin/src/main/java/org/opensearch/sql/plugin/SQLPlugin.java
@@ -94,7 +94,7 @@
 import org.opensearch.sql.spark.asyncquery.AsyncQueryJobMetadataStorageService;
 import org.opensearch.sql.spark.asyncquery.OpensearchAsyncQueryJobMetadataStorageService;
 import org.opensearch.sql.spark.client.EMRServerlessClient;
-import org.opensearch.sql.spark.client.EmrServerlessClientImplEMR;
+import org.opensearch.sql.spark.client.EmrServerlessClientImpl;
 import org.opensearch.sql.spark.config.SparkExecutionEngineConfig;
 import org.opensearch.sql.spark.dispatcher.SparkQueryDispatcher;
 import org.opensearch.sql.spark.flint.FlintIndexMetadataReaderImpl;
@@ -325,7 +325,7 @@ private EMRServerlessClient createEMRServerlessClient() {
                       .withRegion(sparkExecutionEngineConfig.getRegion())
                       .withCredentials(new DefaultAWSCredentialsProviderChain())
                       .build();
-              return new EmrServerlessClientImplEMR(awsemrServerless);
+              return new EmrServerlessClientImpl(awsemrServerless);
             });
   }
 }
diff --git a/spark/src/main/java/org/opensearch/sql/spark/client/EMRServerlessClient.java b/spark/src/main/java/org/opensearch/sql/spark/client/EMRServerlessClient.java
@@ -13,7 +13,7 @@
 /**
  * Client Interface for spark Job Submissions. Can have multiple implementations based on the
  * underlying spark infrastructure. Currently, we have one for EMRServerless {@link
- * EmrServerlessClientImplEMR}
+ * EmrServerlessClientImpl}
  */
 public interface EMRServerlessClient {
 

diff --git a/...rk/client/EmrServerlessClientImplEMR.java → ...spark/client/EmrServerlessClientImpl.java b/...rk/client/EmrServerlessClientImplEMR.java → ...spark/client/EmrServerlessClientImpl.java
@@ -23,12 +23,12 @@
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
-public class EmrServerlessClientImplEMR implements EMRServerlessClient {
+public class EmrServerlessClientImpl implements EMRServerlessClient {
 
   private final AWSEMRServerless emrServerless;
-  private static final Logger logger = LogManager.getLogger(EmrServerlessClientImplEMR.class);
+  private static final Logger logger = LogManager.getLogger(EmrServerlessClientImpl.class);
 
-  public EmrServerlessClientImplEMR(AWSEMRServerless emrServerless) {
+  public EmrServerlessClientImpl(AWSEMRServerless emrServerless) {
     this.emrServerless = emrServerless;
   }
 

diff --git a/spark/src/main/java/org/opensearch/sql/spark/flint/FlintIndexMetadataReaderImpl.java b/spark/src/main/java/org/opensearch/sql/spark/flint/FlintIndexMetadataReaderImpl.java
@@ -42,7 +42,7 @@ private String getIndexName(IndexDetails indexDetails) {
           + "_"
           + fullyQualifiedTableName.getTableName()
           + "_"
-          + indexDetails.getIndexType().getName();
+          + indexDetails.getIndexType().getSuffix();
     } else if (FlintIndexType.COVERING.equals(indexDetails.getIndexType())) {
       return "flint"
           + "_"
@@ -54,7 +54,7 @@ private String getIndexName(IndexDetails indexDetails) {
           + "_"
           + indexDetails.getIndexName()
           + "_"
-          + indexDetails.getIndexType().getName();
+          + indexDetails.getIndexType().getSuffix();
     } else {
       throw new UnsupportedOperationException(
           String.format("Unsupported Index Type : %s", indexDetails.getIndexType()));

diff --git a/spark/src/main/java/org/opensearch/sql/spark/flint/FlintIndexType.java b/spark/src/main/java/org/opensearch/sql/spark/flint/FlintIndexType.java
@@ -12,24 +12,24 @@
 /** Enum for FlintIndex Type. */
 public enum FlintIndexType {
   SKIPPING("skipping_index"),
-  COVERING("covering_index"),
+  COVERING("index"),
   MATERIALIZED("materialized_view");
 
-  private final String name;
+  private final String suffix;
   private static final Map<String, FlintIndexType> ENUM_MAP;
 
-  FlintIndexType(String name) {
-    this.name = name;
+  FlintIndexType(String suffix) {
+    this.suffix = suffix;
   }
 
-  public String getName() {
-    return this.name;
+  public String getSuffix() {
+    return this.suffix;
   }
 
   static {
     Map<String, FlintIndexType> map = new HashMap<>();
     for (FlintIndexType instance : FlintIndexType.values()) {
-      map.put(instance.getName().toLowerCase(), instance);
+      map.put(instance.getSuffix().toLowerCase(), instance);
     }
     ENUM_MAP = Collections.unmodifiableMap(map);
   }

diff --git a/spark/src/test/java/org/opensearch/sql/spark/client/EmrServerlessClientImplTest.java b/spark/src/test/java/org/opensearch/sql/spark/client/EmrServerlessClientImplTest.java
@@ -36,7 +36,7 @@ void testStartJobRun() {
     StartJobRunResult response = new StartJobRunResult();
     when(emrServerless.startJobRun(any())).thenReturn(response);
 
-    EmrServerlessClientImplEMR emrServerlessClient = new EmrServerlessClientImplEMR(emrServerless);
+    EmrServerlessClientImpl emrServerlessClient = new EmrServerlessClientImpl(emrServerless);
     emrServerlessClient.startJobRun(
         new StartJobRequest(
             QUERY,
@@ -54,7 +54,7 @@ void testStartJobRunResultIndex() {
     StartJobRunResult response = new StartJobRunResult();
     when(emrServerless.startJobRun(any())).thenReturn(response);
 
-    EmrServerlessClientImplEMR emrServerlessClient = new EmrServerlessClientImplEMR(emrServerless);
+    EmrServerlessClientImpl emrServerlessClient = new EmrServerlessClientImpl(emrServerless);
     emrServerlessClient.startJobRun(
         new StartJobRequest(
             QUERY,
@@ -74,15 +74,15 @@ void testGetJobRunState() {
     GetJobRunResult response = new GetJobRunResult();
     response.setJobRun(jobRun);
     when(emrServerless.getJobRun(any())).thenReturn(response);
-    EmrServerlessClientImplEMR emrServerlessClient = new EmrServerlessClientImplEMR(emrServerless);
+    EmrServerlessClientImpl emrServerlessClient = new EmrServerlessClientImpl(emrServerless);
     emrServerlessClient.getJobRunResult(EMRS_APPLICATION_ID, "123");
   }
 
   @Test
   void testCancelJobRun() {
     when(emrServerless.cancelJobRun(any()))
         .thenReturn(new CancelJobRunResult().withJobRunId(EMR_JOB_ID));
-    EmrServerlessClientImplEMR emrServerlessClient = new EmrServerlessClientImplEMR(emrServerless);
+    EmrServerlessClientImpl emrServerlessClient = new EmrServerlessClientImpl(emrServerless);
     CancelJobRunResult cancelJobRunResult =
         emrServerlessClient.cancelJobRun(EMRS_APPLICATION_ID, EMR_JOB_ID);
     Assertions.assertEquals(EMR_JOB_ID, cancelJobRunResult.getJobRunId());
@@ -91,7 +91,7 @@ void testCancelJobRun() {
   @Test
   void testCancelJobRunWithValidationException() {
     doThrow(new ValidationException("Error")).when(emrServerless).cancelJobRun(any());
-    EmrServerlessClientImplEMR emrServerlessClient = new EmrServerlessClientImplEMR(emrServerless);
+    EmrServerlessClientImpl emrServerlessClient = new EmrServerlessClientImpl(emrServerless);
     IllegalArgumentException illegalArgumentException =
         Assertions.assertThrows(
             IllegalArgumentException.class,