drizzle-seed 0.3.2-905c951 → 0.3.2-b5a9650

package/index.d.cts

@@ -263,7 +263,7 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  * // 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: {
@@ -284,6 +284,17 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  *   },
  * }));
  *
+ * // 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.
+ *     },
+ *   },
+ * }));
+ *
  * ```
  */
 export declare function seed<DB extends PgDatabase<any, any> | MySqlDatabase<any, any, any, any> | BaseSQLiteDatabase<any, any> | LibSQLDatabase<any>, SCHEMA extends {

package/index.d.mts

@@ -263,7 +263,7 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  * // 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: {
@@ -284,6 +284,17 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  *   },
  * }));
  *
+ * // 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.
+ *     },
+ *   },
+ * }));
+ *
  * ```
  */
 export declare function seed<DB extends PgDatabase<any, any> | MySqlDatabase<any, any, any, any> | BaseSQLiteDatabase<any, any> | LibSQLDatabase<any>, SCHEMA extends {

package/index.d.ts

@@ -263,7 +263,7 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  * // 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: {
@@ -284,6 +284,17 @@ export declare function seedForDrizzleStudio({ sqlDialect, drizzleStudioObject,
  *   },
  * }));
  *
+ * // 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.
+ *     },
+ *   },
+ * }));
+ *
  * ```
  */
 export declare function seed<DB extends PgDatabase<any, any> | MySqlDatabase<any, any, any, any> | BaseSQLiteDatabase<any, any> | LibSQLDatabase<any>, SCHEMA extends {

package/index.mjs

@@ -1,6 +1,6 @@
-import { is, entityKind, eq, sql, Relations, getTableName, extractTablesRelationalConfig, createTableRelationsHelpers, One } from 'drizzle-orm';
+import { is, entityKind, sql, eq, Relations, getTableName, extractTablesRelationalConfig, createTableRelationsHelpers, One } from 'drizzle-orm';
 import { MySqlDatabase, MySqlTable, getTableConfig as getTableConfig$1 } from 'drizzle-orm/mysql-core';
-import { PgDatabase, getTableConfig, PgTable } from 'drizzle-orm/pg-core';
+import { PgSmallInt, PgInteger, PgBigInt53, PgBigInt64, PgSmallSerial, PgSerial, PgBigSerial53, PgBigSerial64, PgDatabase, getTableConfig, PgTable } from 'drizzle-orm/pg-core';
 import { BaseSQLiteDatabase, SQLiteTable, getTableConfig as getTableConfig$2 } from 'drizzle-orm/sqlite-core';
 import prand from 'pure-rand';
 
@@ -131217,6 +131217,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 is(column, PgSmallInt)
+        || is(column, PgInteger)
+        || is(column, PgBigInt53)
+        || is(column, PgBigInt64)
+        || is(column, PgSmallSerial)
+        || is(column, PgSerial)
+        || is(column, PgBigSerial53)
+        || is(column, PgBigSerial64);
+};
 
 /* eslint-disable drizzle-internal/require-entity-kind */
 class AbstractGenerator {
@@ -135222,7 +135233,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
@@ -135353,6 +135365,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 && is(db, PgDatabase) && schema !== undefined && tableName !== undefined
+            && schema[tableName] !== undefined) {
+            const tableConfig = 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 (is(db, (PgDatabase))) {
             // @ts-ignore
@@ -135384,6 +135415,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)) {
@@ -135430,9 +135467,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 (is(db, PgDatabase)) {
+            const fullTableName = schemaName ? `"${schemaName}"."${tableName}"` : `"${tableName}"`;
+            const rawQuery = `SELECT setval(pg_get_serial_sequence('${fullTableName}', '${columnName}'), ${(valueToUpdate ?? 'null').toString()}, true);`;
+            await db.execute(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 (is(db, (PgDatabase))) {
             const query = db.insert(schema[tableName]);
@@ -135571,7 +135627,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: {
@@ -135592,6 +135648,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) {