From c55a35402ded49b2b20c0e73123777042788ddf4 Mon Sep 17 00:00:00 2001
From: Gethyl Kurian <gethyl.kurian@kindredgroup.com>
Date: Tue, 1 Oct 2024 15:42:03 +1000
Subject: [PATCH] feat: allow creating topic with partitions using
 KAFKA_CREATE_TOPIC_PARTITIONS

---
 packages/talos_certifier_adapters/src/kafka/kafka_deploy.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/talos_certifier_adapters/src/kafka/kafka_deploy.rs b/packages/talos_certifier_adapters/src/kafka/kafka_deploy.rs
index d0b8c408..50a5358c 100644
--- a/packages/talos_certifier_adapters/src/kafka/kafka_deploy.rs
+++ b/packages/talos_certifier_adapters/src/kafka/kafka_deploy.rs
@@ -7,6 +7,7 @@ use rdkafka::{
     error::KafkaError,
     types::RDKafkaErrorCode,
 };
+use talos_common_utils::env_var_with_defaults;
 use talos_rdkafka_utils::kafka_config::KafkaConfig;
 use thiserror::Error as ThisError;
 
@@ -40,7 +41,7 @@ pub async fn create_topic() -> Result<KafkaDeployStatus, KafkaDeployError> {
         println!("Topic does not exist, creating...");
         let topic = NewTopic {
             name: &kafka_certification_topic,
-            num_partitions: 1,
+            num_partitions: env_var_with_defaults!("KAFKA_CREATE_TOPIC_PARTITIONS", i32, 1),
             replication: TopicReplication::Fixed(1),
             config: vec![("message.timestamp.type", "LogAppendTime")],
         };