Shared transactions between drizzle and payload

[siimsams]

Hey!

It would be awesome to have non "hacky" way to share transactions between payload and drizzle.

Currently it's possible in this undocumented way.

const payloadTransactionID = await payload.db.beginTransaction();

try {
  if (!payloadTransactionID) {
    log.error("Failed to create transaction");
    throw new Error("Failed to create transaction");
  }

  // Get the drizzle transaction. This is the tx you would get by using `await payload.db.drizzle.transaction(async (tx) => {});`
  // This is a bit of an undocumented hack.
  const tx = payload.db.sessions![payloadTransactionID].db as PgTransaction<
    NodePgQueryResultHKT,
    Record<string, unknown>,
    ExtractTablesWithRelations<Record<string, unknown>>
  >;

  // Use tx as normal with drizzle
  const userCount = await tx.$count(
    users,
    and(
      eq(users.id, "id"),
    ),
  );

  // You can also keep using payloadTransactionID with Payload local api calls
  // ...

  // This will commit everything
  await payload.db.commitTransaction(payloadTransactionID);
} catch (error) {
  if (payloadTransactionID) {
    await payload.db.rollbackTransaction(payloadTransactionID);
  }
}