@514labs/moose-lib 0.6.256-ci-4-g0ca62054 → 0.6.256-ci-3-gafce5840

package/dist/dmv2/index.mjs

@@ -20,7 +20,6 @@ __export(commons_exports, {
   antiCachePath: () => antiCachePath,
   cliLog: () => cliLog,
   compilerLog: () => compilerLog,
-  createProducerConfig: () => createProducerConfig,
   getClickhouseClient: () => getClickhouseClient,
   getFileName: () => getFileName,
   getKafkaClient: () => getKafkaClient,
@@ -46,25 +45,18 @@ function isTruthy(value) {
 function mapTstoJs(filePath) {
   return filePath.replace(/\.ts$/, ".js").replace(/\.cts$/, ".cjs").replace(/\.mts$/, ".mjs");
 }
-function createProducerConfig(maxMessageBytes) {
-  return {
+async function getKafkaProducer(cfg, logger) {
+  const kafka = await getKafkaClient(cfg, logger);
+  const producer = kafka.producer({
     kafkaJS: {
-      idempotent: false,
-      // Not needed for at-least-once delivery
+      idempotent: true,
       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;
 }
@@ -506,16 +498,6 @@ function emptyIfUndefined(value) {
   return value === void 0 ? "" : value;
 }
 
-// src/blocks/helpers.ts
-function dropView(name) {
-  return `DROP VIEW IF EXISTS ${quoteIdentifier(name)}`.trim();
-}
-function createMaterializedView(options) {
-  return `CREATE MATERIALIZED VIEW IF NOT EXISTS ${quoteIdentifier(options.name)}
-        TO ${quoteIdentifier(options.destinationTable)}
-        AS ${options.select}`.trim();
-}
-
 // src/dmv2/internal.ts
 import process2 from "process";
 
@@ -569,7 +551,9 @@ var moose_internal = {
   apis: /* @__PURE__ */ new Map(),
   sqlResources: /* @__PURE__ */ new Map(),
   workflows: /* @__PURE__ */ new Map(),
-  webApps: /* @__PURE__ */ new Map()
+  webApps: /* @__PURE__ */ new Map(),
+  materializedViews: /* @__PURE__ */ new Map(),
+  customViews: /* @__PURE__ */ new Map()
 };
 var defaultRetentionPeriod = 60 * 60 * 24 * 7;
 var getMooseInternal = () => globalThis.moose_internal;
@@ -2185,6 +2169,67 @@ var ETLPipeline = class {
   }
 };
 
+// src/dmv2/sdk/materializedView.ts
+var requireTargetTableName = (tableName) => {
+  if (typeof tableName === "string") {
+    return tableName;
+  } else {
+    throw new Error("Name of targetTable is not specified.");
+  }
+};
+var MaterializedView = class {
+  /** @internal */
+  kind = "MaterializedView";
+  /** The name of the materialized view */
+  name;
+  /** The target OlapTable instance where the materialized data is stored. */
+  targetTable;
+  /** The SELECT SQL statement */
+  selectSql;
+  /** Names of source tables that the SELECT reads from */
+  sourceTables;
+  /** @internal Source file path where this MV was defined */
+  sourceFile;
+  constructor(options, targetSchema, targetColumns) {
+    let selectStatement = options.selectStatement;
+    if (typeof selectStatement !== "string") {
+      selectStatement = toStaticQuery(selectStatement);
+    }
+    if (targetSchema === void 0 || targetColumns === void 0) {
+      throw new Error(
+        "Supply the type param T so that the schema is inserted by the compiler plugin."
+      );
+    }
+    const targetTable = options.targetTable instanceof OlapTable ? options.targetTable : new OlapTable(
+      requireTargetTableName(
+        options.targetTable?.name ?? options.tableName
+      ),
+      {
+        orderByFields: options.targetTable?.orderByFields ?? options.orderByFields,
+        engine: options.targetTable?.engine ?? options.engine ?? "MergeTree" /* MergeTree */
+      },
+      targetSchema,
+      targetColumns
+    );
+    if (targetTable.name === options.materializedViewName) {
+      throw new Error(
+        "Materialized view name cannot be the same as the target table name."
+      );
+    }
+    this.name = options.materializedViewName;
+    this.targetTable = targetTable;
+    this.selectSql = selectStatement;
+    this.sourceTables = options.selectTables.map((t) => t.name);
+    const stack = new Error().stack;
+    this.sourceFile = getSourceFileFromStack(stack);
+    const materializedViews = getMooseInternal().materializedViews;
+    if (!isClientOnlyMode() && materializedViews.has(this.name)) {
+      throw new Error(`MaterializedView with name ${this.name} already exists`);
+    }
+    materializedViews.set(this.name, this);
+  }
+};
+
 // src/dmv2/sdk/sqlResource.ts
 var SqlResource = class {
   /** @internal */
@@ -2230,66 +2275,18 @@ var SqlResource = class {
   }
 };
 
-// src/dmv2/sdk/materializedView.ts
-var requireTargetTableName = (tableName) => {
-  if (typeof tableName === "string") {
-    return tableName;
-  } else {
-    throw new Error("Name of targetTable is not specified.");
-  }
-};
-var MaterializedView = class extends SqlResource {
-  /** The target OlapTable instance where the materialized data is stored. */
-  targetTable;
-  constructor(options, targetSchema, targetColumns) {
-    let selectStatement = options.selectStatement;
-    if (typeof selectStatement !== "string") {
-      selectStatement = toStaticQuery(selectStatement);
-    }
-    if (targetSchema === void 0 || targetColumns === void 0) {
-      throw new Error(
-        "Supply the type param T so that the schema is inserted by the compiler plugin."
-      );
-    }
-    const targetTable = options.targetTable instanceof OlapTable ? options.targetTable : new OlapTable(
-      requireTargetTableName(
-        options.targetTable?.name ?? options.tableName
-      ),
-      {
-        orderByFields: options.targetTable?.orderByFields ?? options.orderByFields,
-        engine: options.targetTable?.engine ?? options.engine ?? "MergeTree" /* MergeTree */
-      },
-      targetSchema,
-      targetColumns
-    );
-    if (targetTable.name === options.materializedViewName) {
-      throw new Error(
-        "Materialized view name cannot be the same as the target table name."
-      );
-    }
-    super(
-      options.materializedViewName,
-      [
-        createMaterializedView({
-          name: options.materializedViewName,
-          destinationTable: targetTable.name,
-          select: selectStatement
-        })
-        // Population is now handled automatically by Rust infrastructure
-        // based on table engine type and whether this is a new or updated view
-      ],
-      [dropView(options.materializedViewName)],
-      {
-        pullsDataFrom: options.selectTables,
-        pushesDataTo: [targetTable]
-      }
-    );
-    this.targetTable = targetTable;
-  }
-};
-
 // src/dmv2/sdk/view.ts
-var View = class extends SqlResource {
+var View = class {
+  /** @internal */
+  kind = "CustomView";
+  /** The name of the view */
+  name;
+  /** The SELECT SQL statement that defines the view */
+  selectSql;
+  /** Names of source tables/views that the SELECT reads from */
+  sourceTables;
+  /** @internal Source file path where this view was defined */
+  sourceFile;
   /**
    * Creates a new View instance.
    * @param name The name of the view to be created.
@@ -2300,17 +2297,16 @@ var View = class extends SqlResource {
     if (typeof selectStatement !== "string") {
       selectStatement = toStaticQuery(selectStatement);
     }
-    super(
-      name,
-      [
-        `CREATE VIEW IF NOT EXISTS ${name} 
-          AS ${selectStatement}`.trim()
-      ],
-      [dropView(name)],
-      {
-        pullsDataFrom: baseTables
-      }
-    );
+    this.name = name;
+    this.selectSql = selectStatement;
+    this.sourceTables = baseTables.map((t) => t.name);
+    const stack = new Error().stack;
+    this.sourceFile = getSourceFileFromStack(stack);
+    const customViews = getMooseInternal().customViews;
+    if (!isClientOnlyMode() && customViews.has(this.name)) {
+      throw new Error(`View with name ${this.name} already exists`);
+    }
+    customViews.set(this.name, this);
   }
 };