no need to fetch partition number if no need to broadcast

pingcap · Jan 15, 2025 · af18d34 · af18d34
1 parent d04d3b6
commit af18d34
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/downstreamadapter/worker/kafka_ddl_worker.go b/downstreamadapter/worker/kafka_ddl_worker.go
@@ -114,12 +114,12 @@ func (w *KafkaDDLWorker) WriteBlockEvent(ctx context.Context, event *event.DDLEv
 			return errors.Trace(err)
 		}
 		topic := w.eventRouter.GetTopicForDDL(e)
-		partitionNum, err := w.topicManager.GetPartitionNum(ctx, topic)
-		if err != nil {
-			return errors.Trace(err)
-		}
 
 		if w.partitionRule == PartitionAll {
+			partitionNum, err := w.topicManager.GetPartitionNum(ctx, topic)
+			if err != nil {
+				return errors.Trace(err)
+			}
 			err = w.statistics.RecordDDLExecution(func() error {
 				return w.producer.SyncBroadcastMessage(ctx, topic, partitionNum, message)
 			})