feat(cluster): extend initial config (#131)

Byndyusoft · Oct 20, 2023 · ef93ee2 · ef93ee2
1 parent 5098e28
commit ef93ee2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -124,6 +124,7 @@ export class ConfigModule {
           password: process.env.KAFKA_PASSWORD,
           ssl: process.env.KAFKA_SSL,
           ca: process.env.KAFKA_CA,
+          connectionTimeout: process.env.KAFKA_CONNECTION_TIMEOUT, // default is 1 s.
         },
         consumer: {
           groupId: process.env.KAFKA_CONSUMER_GROUP_ID as string,

diff --git a/src/config/kafkaClusterConfigDto.ts b/src/config/kafkaClusterConfigDto.ts
@@ -14,13 +14,17 @@
  * limitations under the License.
  */
 
-import { TransformToBoolean } from "@byndyusoft/class-validator-extended";
+import {
+  TransformToBoolean,
+  TransformToNumber,
+} from "@byndyusoft/class-validator-extended";
 import { Transform } from "class-transformer";
 import {
   ArrayMinSize,
   IsArray,
   IsBoolean,
   IsIn,
+  IsInt,
   IsOptional,
   IsString,
 } from "class-validator";
@@ -46,6 +50,7 @@ export class KafkaClusterConfigDto {
 
     return {
       brokers,
+      connectionTimeout: config.connectionTimeout,
       ...this.getKafkaSslConfig(transformedConfig),
       ...this.getKafkaSaslConfig(transformedConfig),
     };
@@ -122,4 +127,9 @@ export class KafkaClusterConfigDto {
   )
   @IsOptional()
   public readonly ca?: string | Buffer;
+
+  @IsInt()
+  @IsOptional()
+  @TransformToNumber()
+  public readonly connectionTimeout?: number;
 }