fix: panics

Author: abelanger5

internal/services/controllers/olap/process_dag_status_updates.go

@@ -53,11 +53,15 @@ func (o *OLAPControllerImpl) runDAGStatusUpdates(ctx context.Context) func() {
 				return
 			}
 
-			err = o.notifyDAGsUpdated(ctx, rows)
+			migrationDisabled := os.Getenv("MIGRATION_DISABLE_EXTRA") == "true"
 
-			if err != nil {
-				o.l.Error().Err(err).Msg("failed to notify updated DAG statuses")
-				return
+			if !migrationDisabled {
+				err = o.notifyDAGsUpdated(ctx, rows)
+
+				if err != nil {
+					o.l.Error().Err(err).Msg("failed to notify updated DAG statuses")
+					return
+				}
 			}
 		}
 

internal/services/controllers/olap/process_task_status_updates.go

@@ -81,11 +81,15 @@ func (o *OLAPControllerImpl) runTaskStatusUpdates(ctx context.Context) func() {
 				return
 			}
 
-			err = o.notifyTasksUpdated(ctx, rows)
+			migrationDisabled := os.Getenv("MIGRATION_DISABLE_EXTRA") == "true"
 
-			if err != nil {
-				o.l.Error().Err(err).Msg("failed to notify updated task statuses")
-				return
+			if !migrationDisabled {
+				err = o.notifyTasksUpdated(ctx, rows)
+
+				if err != nil {
+					o.l.Error().Err(err).Msg("failed to notify updated task statuses")
+					return
+				}
 			}
 		}
 	}