MongoDB 與@NestJs/mongoose 的交易不工作

Question

我真的需要你的幫助。 我與@NestJs/mongoose 的 MongoDB 交易無法正常工作...當我的條帶支付失敗時回滾無法正常工作...不過，我的訂單收集保存了數據...我該如何解決這個問題..？

  async create(orderData: CreateOrderServiceDto): Promise<any> {
    const session = await this.connection.startSession();
    session.startTransaction();
    try {
      const createOrder = new this.orderModel(orderData);
      const order = await createOrder.save();

      await this.stripeService.charge(
        orderData.amount,
        orderData.paymentMethodId,
        orderData.stripeCustomerId,
      );
      await session.commitTransaction();
      return order;
    } catch (error) {
      await session.abortTransaction();
      throw error;
    } finally {
      await session.endSession();
    }
  }

Answer 1

我遇到了同樣的問題，我發現在 github 上： Mongo DB Transactions With Mongoose & Nestjs

所以我認為，根據這個問題，你必須像這樣調用 model 的create方法：

const order = await this.orderModel.create(orderData, { session });

如您所見， Model.create方法有一個以SaveOptions作為參數的重載：

create(docs: (AnyKeys<T> | AnyObject)[], options?: SaveOptions): Promise<HydratedDocument<T, TMethodsAndOverrides, TVirtuals>[]>;

它采用可選的SaveOptions參數，該參數可以包含 session：

interface SaveOptions {
  checkKeys?: boolean;
  j?: boolean;
  safe?: boolean | WriteConcern;
  session?: ClientSession | null;
  timestamps?: boolean;
  validateBeforeSave?: boolean;
  validateModifiedOnly?: boolean;
  w?: number | string;
  wtimeout?: number;
}

請注意Model.save()也可以采用SaveOptions參數。 所以你也可以像你那樣做：

const createOrder = new this.orderModel(orderData);
const order = await createOrder.save({ session });

再遠一點...

當我做很多需要交易的事情時，我想出了這個助手來避免許多代碼重復：

import { InternalServerErrorException } from "@nestjs/common"
import { Connection, ClientSession } from "mongoose"

export const mongooseTransactionHandler = async <T = any>(
  method: (session: ClientSession) => Promise<T>,
  onError: (error: any) => any,
  connection: Connection, session?: ClientSession
): Promise<T> => {
  const isSessionFurnished = session === undefined ? false : true
  if (isSessionFurnished === false) {
    session = await connection.startSession()
    session.startTransaction()
  }

  let error
  let result: T
  try {
    result = await method(session)

    if (isSessionFurnished === false) {
      await session.commitTransaction()
    }
  } catch (err) {
    error = err
    if (isSessionFurnished === false) {
      await session.abortTransaction()
    }
  } finally {
    if (isSessionFurnished === false) {
      await session.endSession()
    }

    if (error) {
      onError(error)
    }

    return result
  }
}

細節

可選參數session是為了防止您正在執行嵌套嵌套事務。 這就是為什么我檢查是否提供了 session。 如果是，則意味着我們處於嵌套事務中。 因此，我們將讓主事務提交、中止並結束 session。

例子

例如：你刪除了一個User model，然后這個用戶的頭像是一個File model。

/** UserService **/
async deleteById(id: string): Promise<void> {
  const transactionHandlerMethod = async (session: ClientSession): Promise<void> => {
    const user = await this.userModel.findOneAndDelete(id, { session })
    await this.fileService.deleteById(user.avatar._id.toString(), session)
  }

  const onError = (error: any) => {
    throw error
  }

  await mongooseTransactionHandler<void>(
    transactionHandlerMethod,
    onError,
    this.connection
  )
}

/** FileService **/
async deleteById(id: string, session?: ClientSession): Promise<void> {
  const transactionHandlerMethod = async (session: ClientSession): Promise<void> => {
    await this.fileModel.findOneAndRemove(id, { session })
  }

  const onError = (error: any) => {
    throw error
  }

  await mongooseTransactionHandler<void>(
    transactionHandlerMethod,
    onError,
    this.connection,
    session
  )
}

所以，簡而言之：

你可以像這樣使用它：

async create(orderData: CreateOrderServiceDto): Promise<any> {
  const transactionHandlerMethod = async (session: ClientSession): Promise<Order> => {
    const createOrder = new this.orderModel(orderData);
    const order = await createOrder.save({ session });

    await this.stripeService.charge(
      orderData.amount,
      orderData.paymentMethodId,
      orderData.stripeCustomerId,
    );

    return order
  }

  const onError = (error: any): void => {
    throw error
  }

  const order = await mongooseTransactionHandler<Order>(
    transactionHandlerMethod,
    onError,
    this.connection
  )

  return order
}

希望它會有所幫助。