dynamically generate reload tasks based on debug versions of quickstart configs

Author: chakru-r

docker/build.gradle

@@ -80,6 +80,22 @@ ext {
                     preserveVolumes: true
             ]
     ]
+
+    moduleToContainer = [
+        ':metadata-service:war': 'datahub-gms-debug',
+        ':datahub-upgrade': 'system-update-debug',
+        ':datahub-frontend': 'frontend-debug',
+        ':metadata-jobs:mce-consumer-job': 'datahub-mce-consumer-job-debug',
+        ':metadata-jobs:mae-consumer-job': 'datahub-mae-consumer-job-debug',
+        ':docker:datahub-ingestion': 'datahub-ingestion-debug',
+        ':docker:mysql-setup': 'datahub-mysql-setup-debug',
+        ':docker:kafka-setup': 'datahub-kafka-setup-debug',
+        ':docker:elasticsearch-setup': 'datahub-elasticsearch-setup-debug',
+        ':docker:postgres-setup': 'datahub-postgres-setup-debug',
+        ':docker:kafka-setup': 'datahub-kafka-setup-debug',
+        ':docker:elasticsearch-setup': 'datahub-elasticsearch-setup-debug',
+        ':docker:postgres-setup': 'datahub-postgres-setup-debug', 
+    ]
 }
 
 // Register all quickstart tasks
@@ -186,39 +202,36 @@ tasks.withType(ComposeUp).configureEach {
     dependsOn tasks.named("minDockerCompose2.20")
 }
 
-task debugReload(type: Exec) {
-    //TODO: Generate *reloadTasks for all quickStart configs. Need to move the moduleToContainer to ext and define mappings for all containers.   
-    // Map of module paths to their corresponding container names
-    def moduleToContainer = [
-        ':metadata-service:war': 'datahub-gms-debug',
-        ':datahub-upgrade': 'system-update-debug',
-        ':datahub-frontend': 'frontend-debug'
-    ]
-    
-    doFirst {
-        // Get all tasks that were executed as part of this build
-        def executedTasks = project.gradle.taskGraph.allTasks.findAll { it.state.executed }
-        
-        // Find which modules actually had work done
-        def containersToRestart = []
-        moduleToContainer.each { modulePath, containerName ->
-            def moduleProject = project.project(modulePath)
-            def dockerPrepareTask = moduleProject.tasks.findByName('dockerPrepare')
-            
-            if (dockerPrepareTask && executedTasks.contains(dockerPrepareTask) && !dockerPrepareTask.state.upToDate) {
-                containersToRestart << containerName
+// Register all quickstart Reload tasks. For quickstartDebug, the reload task is DebugReload. (Taskname without quickstart prefix)
+quickstart_configs.each { taskName, config ->
+    if (config.isDebug) {
+        def reloadTaskName = taskName.replaceFirst(/^quickstart/, "")
+        tasks.register("${reloadTaskName}Reload", Exec) {
+            dependsOn tasks.named("${taskName}PrepareAll")
+            group = 'quickstartart'
+            description = "Build and reload only changed containers for the ${taskName} task"
+            doFirst {
+                def executedTasks = project.gradle.taskGraph.allTasks.findAll { it.state.executed }
+                def containersToRestart = []
+
+                moduleToContainer.each { modulePath, containerName ->
+                    def moduleProject = project.project(modulePath)
+                    def dockerPrepareTask = moduleProject.tasks.findByName('dockerPrepare')
+                    
+                    if (dockerPrepareTask && executedTasks.contains(dockerPrepareTask) && !dockerPrepareTask.state.upToDate) {
+                        containersToRestart << containerName
+                    }
+                }
+                
+                // Only restart containers that had their modules rebuilt
+                if (containersToRestart) {
+                    def cmd = ["docker compose -p datahub --profile ${config.profile}"] + ['-f', compose_base] + ['restart'] + containersToRestart
+                    commandLine 'bash', '-c', cmd.join(" ")
+                } else {
+                    // If no containers need restart, make this a no-op
+                    commandLine 'bash', '-c', 'echo "No containers need restarting - all modules are up to date"'
+                }    
             }
         }
-        
-        // Only restart containers that had their modules rebuilt
-        if (containersToRestart) {
-            def cmd = ['docker compose -p datahub --profile debug'] + ['-f', compose_base] + ['restart'] + containersToRestart
-            commandLine 'bash', '-c', cmd.join(" ")
-        } else {
-            // If no containers need restart, make this a no-op
-            commandLine 'bash', '-c', 'echo "No containers need restarting - all modules are up to date"'
-        }
     }
-    
-    dependsOn tasks.named("quickstartDebugPrepareAll")
-}
+}