Revert datasource state when delete fails

Signed-off-by: Heemin Kim <[email protected]>
heemin32 · Aug 15, 2023 · cf344ac · cf344ac
1 parent d8718ad
commit cf344ac
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/geospatial/ip2geo/action/DeleteDatasourceTransportAction.java b/src/main/java/org/opensearch/geospatial/ip2geo/action/DeleteDatasourceTransportAction.java
@@ -111,9 +111,18 @@ protected void deleteDatasource(final String datasourceName) throws IOException
         if (datasource == null) {
             throw new ResourceNotFoundException("no such datasource exist");
         }
-
+        DatasourceState previousState = datasource.getState();
         setDatasourceStateAsDeleting(datasource);
-        geoIpDataDao.deleteIp2GeoDataIndex(datasource.getIndices());
+
+        try {
+            geoIpDataDao.deleteIp2GeoDataIndex(datasource.getIndices());
+        } catch (Exception e) {
+            if (previousState.equals(datasource.getState()) == false) {
+                datasource.setState(previousState);
+                datasourceDao.updateDatasource(datasource);
+            }
+            throw e;
+        }
         datasourceDao.deleteDatasource(datasource);
     }
 

diff --git a/...st/java/org/opensearch/geospatial/ip2geo/action/DeleteDatasourceTransportActionTests.java b/...st/java/org/opensearch/geospatial/ip2geo/action/DeleteDatasourceTransportActionTests.java
@@ -8,6 +8,7 @@
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyLong;
 import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.times;
@@ -162,4 +163,19 @@ public void testDeleteDatasource_whenProcessorIsCreatedDuringDeletion_thenThrowE
         verify(geoIpDataDao, never()).deleteIp2GeoDataIndex(datasource.getIndices());
         verify(datasourceDao, never()).deleteDatasource(datasource);
     }
+
+    @SneakyThrows
+    public void testDeleteDatasource_whenDeleteFailsAfterStateIsChanged_thenRevertState() {
+        Datasource datasource = randomDatasource();
+        datasource.setState(DatasourceState.AVAILABLE);
+        when(datasourceDao.getDatasource(datasource.getName())).thenReturn(datasource);
+        doThrow(new RuntimeException()).when(geoIpDataDao).deleteIp2GeoDataIndex(datasource.getIndices());
+
+        // Run
+        expectThrows(RuntimeException.class, () -> action.deleteDatasource(datasource.getName()));
+
+        // Verify
+        verify(datasourceDao, times(2)).updateDatasource(datasource);
+        assertEquals(DatasourceState.AVAILABLE, datasource.getState());
+    }
 }