feat: set metadata on state transaction

Author: fengmk2

src/implementation/Client/GRPCClient/state.ts

@@ -6,6 +6,7 @@ import { OperationType } from '../../../types/Operation.type';
 import { IRequestMetadata } from '../../../types/RequestMetadata.type';
 import IClientState from '../../../interfaces/Client/IClientState';
 import { KeyValueType } from '../../../types/KeyValue.type';
+import { merge } from '../../../utils/Map.util';
 
 // https://docs.dapr.io/reference/api/state_api/
 export default class GRPCClientState implements IClientState {
@@ -16,7 +17,7 @@ export default class GRPCClientState implements IClientState {
   }
 
   async save(storeName: string, stateObjects: KeyValuePairType[]): Promise<void> {
-    const stateList = [];
+    const stateList: StateItem[] = [];
 
     for (const stateObject of stateObjects) {
       const si = new StateItem();
@@ -163,6 +164,9 @@ export default class GRPCClientState implements IClientState {
     const msgService = new ExecuteStateTransactionRequest();
     msgService.setStorename(storeName);
     msgService.setOperationsList(transactionItems);
+    if (metadata) {
+      merge(msgService.getMetadataMap(), metadata);
+    }
 
     return new Promise((resolve, reject) => {
       const client = this.client.getClient();

src/utils/Map.util.ts

@@ -0,0 +1,8 @@
+import { Map } from 'google-protobuf';
+import { IRequestMetadata } from '../types/RequestMetadata.type';
+
+export function merge(metadataMap: Map<string, string>, metadata: IRequestMetadata): void {
+  for (const key of Object.keys(metadata)) {
+    metadataMap.set(key, metadata[key]);
+  }
+}

test/e2e/main.grpc.test.ts

@@ -301,6 +301,31 @@ describe('grpc/main', () => {
         expect(resTransactionDelete).toEqual('');
         expect(resTransactionUpsert).toEqual('my-new-data-1');
       });
+
+      it('should be able to perform a transaction with metadata', async () => {
+        await client.state.transaction('state-redis', [
+          {
+            operation: 'upsert',
+            request: {
+              key: 'key-with-metadata-1',
+              value: 'my-new-data-with-metadata-1',
+            },
+          },
+          {
+            operation: 'delete',
+            request: {
+              key: 'key-with-metadata-2',
+            },
+          },
+        ], {
+          trace_id: 'mock trace id here',
+        });
+
+        const resTransactionDelete = await client.state.get('state-redis', 'key-with-metadata-2');
+        const resTransactionUpsert = await client.state.get('state-redis', 'key-with-metadata-1');
+        expect(resTransactionDelete).toEqual('');
+        expect(resTransactionUpsert).toEqual('my-new-data-with-metadata-1');
+      });
     });
   });