Disabling database deletion as part of resource management (Azure#19374)

Co-authored-by: Amar Athavale <aathaval@linkedin.com>

Author: amarathavale

sdk/cosmos/azure-cosmos-benchmark/ctl/linkedin/run_benchmark.sh

@@ -98,9 +98,9 @@ jvm_opt=""
 
 if [ -z "$ctl_graphite_endpoint" ]
 then
-java -Xmx8g -Xms8g  $jvm_opt -Dcosmos.directModeProtocol=$protocol -Dazure.cosmos.directModeProtocol=$protocol -jar "$jar_file" -serviceEndpoint "$service_endpoint" -masterKey "$master_key" -databaseId "$database_name"  -collectionId "$collection_name" -numberOfCollectionForCtl "$number_Of_collection" -throughput $throughput  -consistencyLevel $consistency_level -concurrency $concurrency -numberOfOperations $number_of_operations -operation $operation -connectionMode $connection_mode -maxRunningTimeDuration $max_running_time_duration -numberOfPreCreatedDocuments $number_of_precreated_documents -printingInterval $printing_interval 2>&1 | tee -a "$log_filename"
+java -Xmx8g -Xms8g  $jvm_opt -Dcosmos.directModeProtocol=$protocol -Dazure.cosmos.directModeProtocol=$protocol -jar "$jar_file" -serviceEndpoint "$service_endpoint" -masterKey "$master_key" -databaseId "$database_name"  -collectionId "$collection_name" -numberOfCollectionForCtl "$number_Of_collection" -throughput $throughput  -consistencyLevel $consistency_level -concurrency $concurrency -numberOfOperations $number_of_operations -operation $operation -connectionMode $connection_mode -maxRunningTimeDuration $max_running_time_duration -numberOfPreCreatedDocuments $number_of_precreated_documents -printingInterval $printing_interval -manageResources 2>&1 | tee -a "$log_filename"
 else
-java -Xmx8g -Xms8g  $jvm_opt -Dcosmos.directModeProtocol=$protocol -Dazure.cosmos.directModeProtocol=$protocol -jar "$jar_file" -serviceEndpoint "$service_endpoint" -masterKey "$master_key" -databaseId "$database_name"  -collectionId "$collection_name" -numberOfCollectionForCtl "$number_Of_collection" -throughput $throughput  -consistencyLevel $consistency_level -concurrency $concurrency -numberOfOperations $number_of_operations -operation $operation -connectionMode $connection_mode -maxRunningTimeDuration $max_running_time_duration -graphiteEndpoint $ctl_graphite_endpoint -numberOfPreCreatedDocuments $number_of_precreated_documents -printingInterval $printing_interval 2>&1 | tee -a "$log_filename"
+java -Xmx8g -Xms8g  $jvm_opt -Dcosmos.directModeProtocol=$protocol -Dazure.cosmos.directModeProtocol=$protocol -jar "$jar_file" -serviceEndpoint "$service_endpoint" -masterKey "$master_key" -databaseId "$database_name"  -collectionId "$collection_name" -numberOfCollectionForCtl "$number_Of_collection" -throughput $throughput  -consistencyLevel $consistency_level -concurrency $concurrency -numberOfOperations $number_of_operations -operation $operation -connectionMode $connection_mode -maxRunningTimeDuration $max_running_time_duration -graphiteEndpoint $ctl_graphite_endpoint -numberOfPreCreatedDocuments $number_of_precreated_documents -printingInterval $printing_interval -manageResources 2>&1 | tee -a "$log_filename"
 fi
 
 end=`date +%s`

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/linkedin/ResourceManager.java

@@ -27,7 +27,7 @@ public interface ResourceManager {
     void createContainer() throws CosmosException;
 
     /**
-     * Delete all managed resources i.e. databases and containers
+     * Delete all managed resources e.g. account, databases and/or containers etc
      */
     void deleteResources();
 }

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/linkedin/ResourceManagerImpl.java

@@ -40,7 +40,7 @@ public ResourceManagerImpl(final Configuration configuration, final CosmosAsyncC
     @Override
     public void createDatabase() throws CosmosException {
         try {
-            LOGGER.info("Creating database {} for the ctl workload", _configuration.getDatabaseId());
+            LOGGER.info("Creating database {} for the ctl workload if one doesn't exist", _configuration.getDatabaseId());
             final ThroughputProperties throughputProperties = createManualThroughput(_configuration.getThroughput());
             _client.createDatabaseIfNotExists(_configuration.getDatabaseId(), throughputProperties)
                 .block(RESOURCE_CRUD_WAIT_TIME);
@@ -73,16 +73,6 @@ public void deleteResources() {
         // Delete all the containers in the database
         deleteExistingContainers();
 
-        // Followed by the main database used for testing
-        final CosmosAsyncDatabase database = _client.getDatabase(_configuration.getDatabaseId());
-        try {
-            LOGGER.info("Deleting the main database {} used in this test", _configuration.getDatabaseId());
-            database.delete()
-                .block(RESOURCE_CRUD_WAIT_TIME);
-        } catch (CosmosException e) {
-            LOGGER.error("Exception while deleting the database {}", _configuration.getDatabaseId(), e);
-            throw e;
-        }
         LOGGER.info("Resource cleanup completed");
     }