Merge pull request #86 from Together42/83-transaction_interceptor

83 transaction interceptor

Author: seo-wo

Diff for: src/common/interceptor/transaction.interceptor.ts

@@ -0,0 +1,48 @@
+import {
+  CallHandler,
+  ExecutionContext,
+  Injectable,
+  NestInterceptor,
+} from '@nestjs/common';
+import {
+  catchError,
+  finalize,
+  Observable,
+  concatMap,
+} from 'rxjs';
+import { QueryRunner, DataSource } from 'typeorm';
+
+@Injectable()
+export class TransactionInterceptor implements NestInterceptor {
+  constructor(private readonly dataSource: DataSource) {}
+
+  async intercept(
+    context: ExecutionContext,
+    next: CallHandler<any>,
+  ) : Promise<Observable<any>> {
+    const request = context.switchToHttp().getRequest();
+    const queryRunner: QueryRunner = await this.initRunner();
+    request.queryRunnerManager = queryRunner.manager;
+
+    return next.handle().pipe(
+      concatMap(async (data) => {
+        await queryRunner.commitTransaction();
+        return data;
+      }),
+      catchError(async (error) => {
+        await queryRunner.rollbackTransaction();
+        throw error;
+      }),
+      finalize(async () => {
+        await queryRunner.release();
+      }),
+    );
+  }
+
+  private async initRunner(): Promise<QueryRunner> {
+    const queryRunner: QueryRunner = this.dataSource.createQueryRunner();
+    await queryRunner.connect();
+    await queryRunner.startTransaction();
+    return queryRunner;
+  }
+}

Diff for: src/decorator/transaction.decorator.ts

@@ -0,0 +1,8 @@
+import { ExecutionContext, createParamDecorator } from "@nestjs/common";
+
+export const TransactionManager = createParamDecorator(
+  (data: unknown, ctx: ExecutionContext) => {
+    const request = ctx.switchToHttp().getRequest();
+    return request.queryRunnerManager;
+  }
+)