Correct TypeScript type for tx when passing Drizzle ORM transactions to another function?

[riculum]

I’m using Drizzle ORM and need to pass the transaction object (tx) to another function. Currently, I have defined tx as any, but I’d like to replace it with the correct, type-safe version.

Here’s an example of my function:

const addSubscriptions = ({
  orderId,
  subscriptions,
  tx,
}: {
  orderId: string,
  subscriptions: z.infer<typeof orderValidation>['subscriptions'],
  tx: any, // <-- I'd like to replace `any` with the correct type
}) => {
  // Function implementation
}

[zemd]

I found that whether DatabaseType or TransactionType can be used

export type DatabaseType = typeof database;
export type TransactionType = Parameters<Parameters<DatabaseType['transaction']>[0]>[0];

[ogaken-1]

Hello!

I recently realized that drizzle actually already provides typed tx.

Are you defining a union database type?
I've noticed that when using a union database type, the transaction parameter seems to be resolved as any.

import type { drizzle as postgres } from "drizzle-orm/node-postgres";
import type { drizzle as pglite } from "drizzle-orm/pglite";
import type * as schema from "../schema/index";

export type Schema = typeof schema;
export type Postgres = ReturnType<typeof postgres<Schema>>;
export type PGlite = ReturnType<typeof pglite<Schema>>;
export type Database = Postgres | PGlite;
const any = (db: Database) => {
  // (parameter) tx: any
  db.transaction(tx => {})
}
const correct = (db: Postgres) => {
  // (parameter) tx: PgTransaction<NodePgQueryResultHKT, typeof schema, ExtractTablesWithRelations<typeof schema>>
  db.transaction(tx => {})
}