feat: 事务注解增加数据源选项

core/transaction-decorator/README.md

@@ -3,6 +3,7 @@
 事务注解
 
 ## Usage
+### 传播机制
 ```ts
 export class Foo {
 
@@ -16,6 +17,18 @@ export class Foo {
     console.log('has msg: ', msg);
   }
 
+}
+```
+
+### 数据源
+```ts
+export class Bar {
+
+  @Transactional({ dataSourceName: 'xx' })
+  async bar() {
+    await this.foo();
+  }
+
 }
 
 ```

core/transaction-decorator/src/Common.ts

@@ -8,4 +8,10 @@ export enum PropagationType {
 export interface TransactionalParams {
   /** 事务传播方式，默认 REQUIRED */
   propagation?: PropagationType;
+  /**
+   * 数据源，默认使用 module 的数据源，非 module 时将使用 default 数据源
+   * 需要注意的是数据源之间的连接是隔离的，回滚也是独立的
+   * 比如函数 B 在函数 A 中执行，A 执行异常时，不会回滚 B 中执行的 sql
+   * */
+  datasourceName?: string;
 }

core/transaction-decorator/src/decorator/Transactional.ts

@@ -7,13 +7,15 @@ export function Transactional(params?: TransactionalParams) {
   if (!Object.values(PropagationType).includes(propagation)) {
     throw new Error(`unknown propagation type ${propagation}`);
   }
+  const datasourceName = params?.datasourceName;
 
   return function(target: any, propertyKey: PropertyKey) {
     const constructor: EggProtoImplClass = target.constructor;
     TransactionMetadataUtil.setIsTransactionClazz(constructor);
     TransactionMetadataUtil.addTransactionMetadata(constructor, {
       propagation,
       method: propertyKey,
+      datasourceName,
     });
   };
 }

core/transaction-decorator/src/model/TransactionMetadata.ts

@@ -3,4 +3,5 @@ import { PropagationType } from '../Common';
 export interface TransactionMetadata {
   propagation: PropagationType;
   method: PropertyKey;
+  datasourceName?: string;
 }

core/transaction-decorator/test/builder/TransactionMetaBuilder.test.ts

@@ -14,29 +14,35 @@ describe('test/builder/TransactionMetaBuilder.test.ts', () => {
     assert.deepStrictEqual(fooBuilder.build(), [{
       propagation: PropagationType.REQUIRED,
       method: 'defaultPropagation',
+      datasourceName: undefined,
     }, {
       propagation: PropagationType.REQUIRED,
       method: 'requiredPropagation',
+      datasourceName: 'testDatasourceName1',
     }, {
       propagation: PropagationType.ALWAYS_NEW,
       method: 'alwaysNewPropagation',
+      datasourceName: undefined,
     }]);
 
     assert.ok(TransactionMetadataUtil.isTransactionClazz(Bar));
     const barBuilder = new TransactionMetaBuilder(Bar);
     assert.deepStrictEqual(barBuilder.build(), [{
       propagation: PropagationType.REQUIRED,
       method: 'foo',
+      datasourceName: 'datasourceName2',
     }, {
       propagation: PropagationType.ALWAYS_NEW,
       method: 'bar',
+      datasourceName: undefined,
     }]);
 
     assert.ok(TransactionMetadataUtil.isTransactionClazz(FooBar));
     const fooBarBuilder = new TransactionMetaBuilder(FooBar);
     assert.deepStrictEqual(fooBarBuilder.build(), [{
       propagation: PropagationType.ALWAYS_NEW,
       method: 'foo',
+      datasourceName: undefined,
     }]);
 
     const barFooBuilder = new TransactionMetaBuilder(BarFoo);

core/transaction-decorator/test/fixtures/transaction.ts

@@ -8,7 +8,9 @@ export class Foo {
     console.log('msg: ', msg);
   }
 
-  @Transactional({})
+  @Transactional({
+    datasourceName: 'testDatasourceName1',
+  })
   async requiredPropagation(msg) {
     console.log('msg: ', msg);
   }
@@ -22,7 +24,7 @@ export class Foo {
 
 export class Bar {
 
-  @Transactional()
+  @Transactional({ datasourceName: 'datasourceName2' })
   async foo(msg) {
     console.log('msg: ', msg);
   }