fix: leak of memory in guardin by stoped policy (#4782)

Co-authored-by: Ihar <[email protected]>

Author: ihar-tsykala

common/src/mq/nats-service.ts

@@ -127,9 +127,11 @@ export abstract class NatsService {
      * @param subject
      * @param data
      * @param isResponseCallback
+     * @param externalMessageId
      */
-    public sendMessage<T>(subject: string, data?: unknown, isResponseCallback: boolean = true): Promise<T> {
-        const messageId = GenerateUUIDv4();
+    public sendMessage<T>(subject: string, data?: unknown, isResponseCallback: boolean = true, externalMessageId?: string): Promise<T> {
+        const messageId = externalMessageId ?? GenerateUUIDv4();
+
         return new Promise(async (resolve, reject) => {
             const head = headers();
             head.append('messageId', messageId);
@@ -159,14 +161,18 @@ export abstract class NatsService {
      * @param data
      */
     public sendMessageWithTimeout<T>(subject: string, timeout: number, data?: unknown): Promise<T> {
-        return Promise.race([
-            this.sendMessage<T>(subject, data),
-            new Promise<T>((_, reject) => {
-                setTimeout(() => {
-                    reject(new Error('Timeout exceed'))
-                }, timeout)
-            })
-        ])
+        const messageId = GenerateUUIDv4();
+
+        const messagePromise = this.sendMessage<T>(subject, data, true, messageId);
+
+        const timeoutPromise = new Promise<T>((_, reject) => {
+            setTimeout(() => {
+                this.responseCallbacksMap.delete(messageId);
+                reject(new Error('Timeout exceed'));
+            }, timeout);
+        });
+
+        return Promise.race([messagePromise, timeoutPromise]);
     }
 
     /**

guardian-service/src/helpers/guardians.ts

@@ -74,6 +74,7 @@ export class GuardiansService extends NatsService {
             }),
             new Promise<T>((resolve, reject) => {
                 setTimeout(() => {
+                    this.responseCallbacksMap.delete(messageId);
                     resolve(null);
                 }, awaitInterval)
             }),