diff --git a/modules/JiShe.CollectBus.Kafka/KafkaSubscribeExtensions.cs b/modules/JiShe.CollectBus.Kafka/KafkaSubscribeExtensions.cs
index 3c50aae..a0438db 100644
--- a/modules/JiShe.CollectBus.Kafka/KafkaSubscribeExtensions.cs
+++ b/modules/JiShe.CollectBus.Kafka/KafkaSubscribeExtensions.cs
@@ -137,13 +137,11 @@ namespace JiShe.CollectBus.Kafka
 
             foreach (var sub in subscribedMethods)
             {
-                int partitionCount = 3;// kafkaOptionConfig.NumPartitions;
-#if DEBUG
+                int partitionCount = sub.Attribute!.TaskCount==-1?3: sub.Attribute!.TaskCount;// kafkaOptionConfig.NumPartitions;
                 var adminClientService = provider.GetRequiredService<IAdminClientService>();
                 int topicCount = adminClientService.GetTopicPartitionsNum(sub.Attribute!.Topic);
                 partitionCount = partitionCount > topicCount ? topicCount : partitionCount;
-#endif
-                //int partitionCount = sub.Attribute!.TaskCount==-1?adminClientService.GetTopicPartitionsNum(sub.Attribute!.Topic) : sub.Attribute!.TaskCount;
+                //partitionCount = sub.Attribute!.TaskCount == -1 ? adminClientService.GetTopicPartitionsNum(sub.Attribute!.Topic) : sub.Attribute!.TaskCount;
                 if (partitionCount <= 0)
                     partitionCount = 1;
                 for (int i = 0; i < partitionCount; i++)