@514labs/moose-lib 0.6.256-ci-1-g6ca86038 → 0.6.256-ci-4-g0ca62054

package/dist/index.d.mts

@@ -44,6 +44,25 @@ declare const RETRY_INITIAL_TIME_MS = 100;
 declare const MAX_RETRIES_PRODUCER = 150;
 declare const RETRY_FACTOR_PRODUCER = 0.2;
 declare const ACKs = -1;
+/**
+ * Creates the base producer configuration for Kafka.
+ * Used by both the SDK stream publishing and streaming function workers.
+ *
+ * @param maxMessageBytes - Optional max message size in bytes (synced with topic config)
+ * @returns Producer configuration object for the Confluent Kafka client
+ */
+declare function createProducerConfig(maxMessageBytes?: number): {
+    "message.max.bytes"?: number | undefined;
+    kafkaJS: {
+        idempotent: boolean;
+        acks: number;
+        retry: {
+            retries: number;
+            maxRetryTime: number;
+        };
+    };
+    "linger.ms": number;
+};
 type KafkaClientConfig = {
     clientId: string;
     broker: string;
@@ -55,8 +74,12 @@ type KafkaClientConfig = {
 /**
  * Dynamically creates and connects a KafkaJS producer using the provided configuration.
  * Returns a connected producer instance.
+ *
+ * @param cfg - Kafka client configuration
+ * @param logger - Logger instance
+ * @param maxMessageBytes - Optional max message size in bytes (synced with topic config)
  */
-declare function getKafkaProducer(cfg: KafkaClientConfig, logger: Logger): Promise<Producer>;
+declare function getKafkaProducer(cfg: KafkaClientConfig, logger: Logger, maxMessageBytes?: number): Promise<Producer>;
 /**
  * Interface for logging functionality
  */
@@ -531,4 +554,4 @@ type DataModelConfig<T> = Partial<{
     parallelism?: number;
 }>;
 
-export { ACKs, ApiUtil, type CSVParsingConfig, CSV_DELIMITERS, type CliLogData, DEFAULT_CSV_CONFIG, DEFAULT_JSON_CONFIG, type DataModelConfig, DataSource, type DataSourceConfig, type ExpressRequestWithMoose, type ExtractionResult, type JSONParsingConfig, type KafkaClientConfig, type Logger, MAX_RETRIES, MAX_RETRIES_PRODUCER, MAX_RETRY_TIME_MS, MOOSE_RUNTIME_ENV_PREFIX, MooseCache, MooseClient, type Producer, RETRY_FACTOR_PRODUCER, RETRY_INITIAL_TIME_MS, type StripDateIntersection, type TaskConfig, type TaskDefinition, type TaskFunction, antiCachePath, cliLog, compilerLog, createApi, createConsumptionApi, expressMiddleware, getClickhouseClient, getFileName, getKafkaClient, getKafkaProducer, getMooseClients, getMooseUtils, isValidCSVDelimiter, logError, mapTstoJs, mooseEnvSecrets, mooseRuntimeEnv, parseCSV, parseJSON, parseJSONWithDates };
+export { ACKs, ApiUtil, type CSVParsingConfig, CSV_DELIMITERS, type CliLogData, DEFAULT_CSV_CONFIG, DEFAULT_JSON_CONFIG, type DataModelConfig, DataSource, type DataSourceConfig, type ExpressRequestWithMoose, type ExtractionResult, type JSONParsingConfig, type KafkaClientConfig, type Logger, MAX_RETRIES, MAX_RETRIES_PRODUCER, MAX_RETRY_TIME_MS, MOOSE_RUNTIME_ENV_PREFIX, MooseCache, MooseClient, type Producer, RETRY_FACTOR_PRODUCER, RETRY_INITIAL_TIME_MS, type StripDateIntersection, type TaskConfig, type TaskDefinition, type TaskFunction, antiCachePath, cliLog, compilerLog, createApi, createConsumptionApi, createProducerConfig, expressMiddleware, getClickhouseClient, getFileName, getKafkaClient, getKafkaProducer, getMooseClients, getMooseUtils, isValidCSVDelimiter, logError, mapTstoJs, mooseEnvSecrets, mooseRuntimeEnv, parseCSV, parseJSON, parseJSONWithDates };

package/dist/index.d.ts

@@ -44,6 +44,25 @@ declare const RETRY_INITIAL_TIME_MS = 100;
 declare const MAX_RETRIES_PRODUCER = 150;
 declare const RETRY_FACTOR_PRODUCER = 0.2;
 declare const ACKs = -1;
+/**
+ * Creates the base producer configuration for Kafka.
+ * Used by both the SDK stream publishing and streaming function workers.
+ *
+ * @param maxMessageBytes - Optional max message size in bytes (synced with topic config)
+ * @returns Producer configuration object for the Confluent Kafka client
+ */
+declare function createProducerConfig(maxMessageBytes?: number): {
+    "message.max.bytes"?: number | undefined;
+    kafkaJS: {
+        idempotent: boolean;
+        acks: number;
+        retry: {
+            retries: number;
+            maxRetryTime: number;
+        };
+    };
+    "linger.ms": number;
+};
 type KafkaClientConfig = {
     clientId: string;
     broker: string;
@@ -55,8 +74,12 @@ type KafkaClientConfig = {
 /**
  * Dynamically creates and connects a KafkaJS producer using the provided configuration.
  * Returns a connected producer instance.
+ *
+ * @param cfg - Kafka client configuration
+ * @param logger - Logger instance
+ * @param maxMessageBytes - Optional max message size in bytes (synced with topic config)
  */
-declare function getKafkaProducer(cfg: KafkaClientConfig, logger: Logger): Promise<Producer>;
+declare function getKafkaProducer(cfg: KafkaClientConfig, logger: Logger, maxMessageBytes?: number): Promise<Producer>;
 /**
  * Interface for logging functionality
  */
@@ -531,4 +554,4 @@ type DataModelConfig<T> = Partial<{
     parallelism?: number;
 }>;
 
-export { ACKs, ApiUtil, type CSVParsingConfig, CSV_DELIMITERS, type CliLogData, DEFAULT_CSV_CONFIG, DEFAULT_JSON_CONFIG, type DataModelConfig, DataSource, type DataSourceConfig, type ExpressRequestWithMoose, type ExtractionResult, type JSONParsingConfig, type KafkaClientConfig, type Logger, MAX_RETRIES, MAX_RETRIES_PRODUCER, MAX_RETRY_TIME_MS, MOOSE_RUNTIME_ENV_PREFIX, MooseCache, MooseClient, type Producer, RETRY_FACTOR_PRODUCER, RETRY_INITIAL_TIME_MS, type StripDateIntersection, type TaskConfig, type TaskDefinition, type TaskFunction, antiCachePath, cliLog, compilerLog, createApi, createConsumptionApi, expressMiddleware, getClickhouseClient, getFileName, getKafkaClient, getKafkaProducer, getMooseClients, getMooseUtils, isValidCSVDelimiter, logError, mapTstoJs, mooseEnvSecrets, mooseRuntimeEnv, parseCSV, parseJSON, parseJSONWithDates };
+export { ACKs, ApiUtil, type CSVParsingConfig, CSV_DELIMITERS, type CliLogData, DEFAULT_CSV_CONFIG, DEFAULT_JSON_CONFIG, type DataModelConfig, DataSource, type DataSourceConfig, type ExpressRequestWithMoose, type ExtractionResult, type JSONParsingConfig, type KafkaClientConfig, type Logger, MAX_RETRIES, MAX_RETRIES_PRODUCER, MAX_RETRY_TIME_MS, MOOSE_RUNTIME_ENV_PREFIX, MooseCache, MooseClient, type Producer, RETRY_FACTOR_PRODUCER, RETRY_INITIAL_TIME_MS, type StripDateIntersection, type TaskConfig, type TaskDefinition, type TaskFunction, antiCachePath, cliLog, compilerLog, createApi, createConsumptionApi, createProducerConfig, expressMiddleware, getClickhouseClient, getFileName, getKafkaClient, getKafkaProducer, getMooseClients, getMooseUtils, isValidCSVDelimiter, logError, mapTstoJs, mooseEnvSecrets, mooseRuntimeEnv, parseCSV, parseJSON, parseJSONWithDates };

package/dist/index.js

@@ -42,6 +42,7 @@ __export(commons_exports, {
   antiCachePath: () => antiCachePath,
   cliLog: () => cliLog,
   compilerLog: () => compilerLog,
+  createProducerConfig: () => createProducerConfig,
   getClickhouseClient: () => getClickhouseClient,
   getFileName: () => getFileName,
   getKafkaClient: () => getKafkaClient,
@@ -64,18 +65,25 @@ function isTruthy(value) {
 function mapTstoJs(filePath) {
   return filePath.replace(/\.ts$/, ".js").replace(/\.cts$/, ".cjs").replace(/\.mts$/, ".mjs");
 }
-async function getKafkaProducer(cfg, logger) {
-  const kafka = await getKafkaClient(cfg, logger);
-  const producer = kafka.producer({
+function createProducerConfig(maxMessageBytes) {
+  return {
     kafkaJS: {
-      idempotent: true,
+      idempotent: false,
+      // Not needed for at-least-once delivery
       acks: ACKs,
       retry: {
         retries: MAX_RETRIES_PRODUCER,
         maxRetryTime: MAX_RETRY_TIME_MS
       }
-    }
-  });
+    },
+    "linger.ms": 0,
+    // This is to make sure at least once delivery with immediate feedback on the send
+    ...maxMessageBytes && { "message.max.bytes": maxMessageBytes }
+  };
+}
+async function getKafkaProducer(cfg, logger, maxMessageBytes) {
+  const kafka = await getKafkaClient(cfg, logger);
+  const producer = kafka.producer(createProducerConfig(maxMessageBytes));
   await producer.connect();
   return producer;
 }
@@ -417,6 +425,7 @@ __export(index_exports, {
   createClickhouseParameter: () => createClickhouseParameter,
   createConsumptionApi: () => createConsumptionApi,
   createMaterializedView: () => createMaterializedView,
+  createProducerConfig: () => createProducerConfig,
   dropView: () => dropView,
   expressMiddleware: () => expressMiddleware,
   getApi: () => getApi,
@@ -3405,6 +3414,7 @@ var DataSource = class {
   createClickhouseParameter,
   createConsumptionApi,
   createMaterializedView,
+  createProducerConfig,
   dropView,
   expressMiddleware,
   getApi,