@whaly/connector-sdk 0.3.5 → 0.3.6

package/dist/index.mjs

@@ -1133,6 +1133,7 @@ var Stream = class {
   // Private sync methods:
   async _syncRecords(parent) {
     let rowsSent = 0;
+    const dryRunLimit = isDryRun() ? getDryRunLimit() : void 0;
     let recordSyncedMetrics = [];
     if (this.rowsSyncedMetricsConf.isEnabled() === true) {
       recordSyncedMetrics = getCounterMetrics(
@@ -1174,6 +1175,10 @@ var Stream = class {
         metric.increment();
       });
       rowsSent += 1;
+      if (dryRunLimit !== void 0 && rowsSent >= dryRunLimit) {
+        logger.info(`\u{1F6D1} Stream: ${this.streamId} - DRY_RUN_LIMIT reached (${dryRunLimit} records), stopping.`);
+        break;
+      }
       if (rowsSent % 1e3 === 0) {
         const percentStr = this.totalRows ? ` (${Math.round(rowsSent / this.totalRows * 100)}%)` : "";
         logger.info(`\u23F3 Stream: ${this.streamId} - ${rowsSent} records synced so far...${percentStr}`);
@@ -2199,11 +2204,12 @@ var ITarget = class _ITarget {
       streamState.setSchema(schema);
       streamState.setCompiledValidateFn(this.ajv.compile({ type: "object", properties: schema }));
       const streamReplicationMethod = this.streams[streamId]?.getReplicationMethod();
-      if (message.keyProperties.length === 0 && streamReplicationMethod !== "APPEND" /* APPEND */) {
+      if (message.keyProperties.length === 0 && streamReplicationMethod === "INCREMENTAL" /* INCREMENTAL */) {
         throw new Error(`StreamId: ${message.stream} - \`key_properties\` field is required and can't be empty in SCHEMA message.
             Received SCHEMA message: ${JSON.stringify(message)}`);
       }
       streamState.setKeyProperties(message.keyProperties);
+      dbSync.renamedColumnStore.computeColumnNameForStream(streamId, schema);
       if (!isDryRun()) {
         await dbSync.updateSchemaInWarehouse(schema, message.keyProperties);
         const {
@@ -4082,6 +4088,12 @@ var BigQueryDBSync = class extends StreamWarehouseSyncService {
     });
   };
   getReplaceQueries = () => {
+    if (this.primaryKeys.length === 0) {
+      return [
+        `TRUNCATE TABLE ${this.sqlTableId};`,
+        ...this.getAppendQueries()
+      ];
+    }
     return [
       `TRUNCATE TABLE ${this.sqlTableId};`,
       ...this.getMergeQueries()