drizzle-seed 0.3.2-905c951 → 0.3.2-b5a9650

package/index.cjs

@@ -131219,6 +131219,17 @@ const equalSets = (set1, set2) => {
 const isValidDate = (date) => {
     return !Number.isNaN(date.getTime());
 };
+const intMax = (args) => args.reduce((m, e) => e > m ? e : m);
+const isPostgresColumnIntLike = (column) => {
+    return drizzleOrm.is(column, pgCore.PgSmallInt)
+        || drizzleOrm.is(column, pgCore.PgInteger)
+        || drizzleOrm.is(column, pgCore.PgBigInt53)
+        || drizzleOrm.is(column, pgCore.PgBigInt64)
+        || drizzleOrm.is(column, pgCore.PgSmallSerial)
+        || drizzleOrm.is(column, pgCore.PgSerial)
+        || drizzleOrm.is(column, pgCore.PgBigSerial53)
+        || drizzleOrm.is(column, pgCore.PgBigSerial64);
+};
 
 /* eslint-disable drizzle-internal/require-entity-kind */
 class AbstractGenerator {
@@ -135224,7 +135235,8 @@ class SeedService {
                 };
             }
             // get values to generate columns with foreign key
-            // if table posts contains foreign key to table users, then rel.table === 'posts' and rel.refTable === 'users', because table posts has reference to table users.
+            // if table posts contains foreign key to table users, then rel.table === 'posts' and rel.refTable === 'users',
+            // because table posts has reference to table users.
             if (filteredRelations.length !== 0) {
                 for (const rel of filteredRelations) {
                     if (table.withFromTable[rel.refTable] !== undefined
@@ -135355,6 +135367,25 @@ class SeedService {
             // 	columnsGenerators[columnName] = uniqueGen;
             // }
         }
+        // sequence updates will only be performed for PostgreSQL, since MySQL and SQLite already update their sequences correctly on their own.
+        const columnsToUpdateSeq = new Map();
+        if (count > 0 && drizzleOrm.is(db, pgCore.PgDatabase) && schema !== undefined && tableName !== undefined
+            && schema[tableName] !== undefined) {
+            const tableConfig = pgCore.getTableConfig(schema[tableName]);
+            for (const column of tableConfig.columns) {
+                // TODO should I filter only primary key columns?
+                // should I filter column by dataType or by column drizzle type?
+                // column.dataType === 'number' || column.dataType === 'bigint'
+                if (isPostgresColumnIntLike(column)) {
+                    columnsToUpdateSeq.set(column.name, {
+                        schemaName: tableConfig.schema,
+                        tableName: tableConfig.name,
+                        columnName: column.name,
+                        valueToUpdate: undefined,
+                    });
+                }
+            }
+        }
         let maxParametersNumber;
         if (drizzleOrm.is(db, (pgCore.PgDatabase))) {
             // @ts-ignore
@@ -135386,6 +135417,12 @@ class SeedService {
                 //   | boolean;
                 generatedValue = columnsGenerators[columnName].generate({ i });
                 row[columnName] = generatedValue;
+                const colToUpdateSeq = columnsToUpdateSeq.get(columnName);
+                if (columnsToUpdateSeq.size !== 0 && colToUpdateSeq !== undefined) {
+                    colToUpdateSeq.valueToUpdate = colToUpdateSeq?.valueToUpdate === undefined
+                        ? generatedValue
+                        : intMax([colToUpdateSeq.valueToUpdate, generatedValue]);
+                }
             }
             if ((insertDataInDb === true || updateDataInDb === true)
                 && ((i + 1) % batchSize === 0 || i === count - 1)) {
@@ -135432,9 +135469,28 @@ class SeedService {
                     }
                 }
             }
+            const columnsToUpdateSeqFiltered = [...columnsToUpdateSeq.values()].filter((col) => col.valueToUpdate !== undefined);
+            if (i === count - 1
+                && columnsToUpdateSeqFiltered.length !== 0 && db !== undefined) {
+                for (const columnConfig of columnsToUpdateSeq.values()) {
+                    if (columnConfig) {
+                        await this.updateColumnSequence({ db, columnConfig });
+                    }
+                }
+            }
         }
         return preserveData === true ? generatedValues : [];
     };
+    updateColumnSequence = async ({ db, columnConfig: { schemaName, tableName, columnName, valueToUpdate } }) => {
+        if (drizzleOrm.is(db, pgCore.PgDatabase)) {
+            const fullTableName = schemaName ? `"${schemaName}"."${tableName}"` : `"${tableName}"`;
+            const rawQuery = `SELECT setval(pg_get_serial_sequence('${fullTableName}', '${columnName}'), ${(valueToUpdate ?? 'null').toString()}, true);`;
+            await db.execute(drizzleOrm.sql.raw(rawQuery));
+        }
+        // mysql updates auto_increment or serial column by itself
+        // sqlite updates autoincrement  column by itself
+        return;
+    };
     insertInDb = async ({ generatedValues, db, schema, tableName, override, }) => {
         if (drizzleOrm.is(db, (pgCore.PgDatabase))) {
             const query = db.insert(schema[tableName]);
@@ -135573,7 +135629,7 @@ async function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject, drizzleSt
  * // seeding with count and seed specified
  * await seed(db, schema, { count: 100000, seed: 1 });
  *
- * //seeding using refine
+ * // seeding using refine
  * await seed(db, schema, { count: 1000 }).refine((funcs) => ({
  *   users: {
  *     columns: {
@@ -135594,6 +135650,17 @@ async function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject, drizzleSt
  *   },
  * }));
  *
+ * // seeding while ignoring column
+ * await seed(db, schema).refine((funcs) => ({
+ *   users: {
+ *     count: 5,
+ *     columns: {
+ *       name: funcs.fullName(),
+ *       photo: false, // the photo column will not be seeded, allowing the database to use its default value.
+ *     },
+ *   },
+ * }));
+ *
  * ```
  */
 function seed(db, schema, options) {