feat(workers): add decorator to catch unhandled exceptions and report them to Hawk with worker type context

Author: akulistus

lib/utils/catchAndReport.ts

@@ -0,0 +1,29 @@
+import HawkCatcher from "@hawk.so/nodejs";
+import { Worker } from "../worker";
+
+/**
+* Catches unhandled exceptions from the decorated method, sends them to HawkCatcher
+* with the current worker’s type, and re‑throws.
+*/
+export function catchAndReport(): MethodDecorator {
+  return function (_target, propertyKey, descriptor) {
+    const original = descriptor.value;
+
+    if (typeof original !== "function") {
+      throw new Error(
+        `@catchAndReport can only be applied to methods (${String(propertyKey)})`
+      )
+    }
+
+    descriptor.value = async function (...args: any[]) {
+      try {
+        return await original.apply(this, args);
+      } catch (error) {
+        HawkCatcher.send(error, {
+          workerType: (this as Worker).type
+        });
+        throw error;
+      }
+    } as typeof original;
+  }
+};

runner.ts

@@ -1,3 +1,4 @@
+import { EventContext } from '@hawk.so/types';
 import * as utils from './lib/utils';
 
 /* Prometheus client for pushing metrics to the pushgateway */
@@ -192,7 +193,9 @@ class WorkerRunner {
 
           utils.sendReport(worker.constructor.name + ' started');
         } catch (startingError) {
-          this.exceptionHandler(startingError);
+          this.exceptionHandler(startingError, {
+            workerType: worker.type
+          });
 
           utils.sendReport(worker.constructor.name + ' failed to start');
 
@@ -209,8 +212,8 @@ class WorkerRunner {
    *
    * @param error - error to handle
    */
-  private exceptionHandler(error: Error): void {
-    HawkCatcher.send(error);
+  private exceptionHandler(error: Error, context?: EventContext): void {
+    HawkCatcher.send(error, context);
 
     console.log(
       '\x1b[41m%s\x1b[0m',
@@ -285,7 +288,9 @@ class WorkerRunner {
         `\n\n Worker ${worker.constructor.name} stopped \n`
       );
     } catch (finishingError) {
-      HawkCatcher.send(finishingError);
+      HawkCatcher.send(finishingError, {
+        workerType: worker.type
+      });
       console.error('Error while finishing Worker: ', finishingError);
     }
   }

workers/accountant/src/index.ts

@@ -5,6 +5,7 @@ import { Worker } from '../../../lib/worker';
 import * as pkg from '../package.json';
 import { AccountantEvent, EventType, IncomeTransactionPayload, TransactionEvent, TransactionType } from '../types/accountant-worker-events';
 import { WorkspaceDBScheme } from '@hawk.so/types';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 /**
  * Worker for managing workspaces balance
@@ -63,6 +64,7 @@ export default class AccountantWorker extends Worker {
    *
    * @param event
    */
+  @catchAndReport()
   public async handle(event: AccountantEvent): Promise<void> {
     switch (event.type) {
       case EventType.Transaction:

workers/archiver/src/index.ts

@@ -9,6 +9,7 @@ import * as dotenv from 'dotenv';
 import prettysize from 'prettysize';
 import { ProjectDBScheme } from '@hawk.so/types';
 import { HOURS_IN_DAY, MINUTES_IN_HOUR, MS_IN_SEC, SECONDS_IN_MINUTE } from '../../../lib/utils/consts';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 dotenv.config({ path: path.resolve(__dirname, '../.env') });
 
@@ -77,6 +78,7 @@ export default class ArchiverWorker extends Worker {
   /**
    * Task handling function
    */
+  @catchAndReport()
   public async handle(): Promise<void> {
     const dbSizeOnStart = (await this.eventsDbConnection.stats()).dataSize;
 

workers/default/src/index.ts

@@ -2,6 +2,7 @@ import { ErrorsCatcherType } from '@hawk.so/types';
 import { EventWorker } from '../../../lib/event-worker';
 import * as pkg from '../package.json';
 import { DefaultEventWorkerTask } from '../types/default-event-worker-task';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 /**
  * Worker for handling Default events
@@ -17,6 +18,7 @@ export default class DefaultEventWorker extends EventWorker {
    *
    * @param event - event to handle
    */
+  @catchAndReport()
   public async handle(event: DefaultEventWorkerTask): Promise<void> {
     /**
      * Define  event type

workers/grouper/src/index.ts

@@ -31,6 +31,7 @@ import { hasValue } from '../../../lib/utils/hasValue';
  */
 /* eslint-disable-next-line no-unused-vars */
 import { memoize } from '../../../lib/memoize';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 /**
  * eslint does not count decorators as a variable usage
@@ -136,6 +137,7 @@ export default class GrouperWorker extends Worker {
    *
    * @param task - event to handle
    */
+  @catchAndReport()
   public async handle(task: GroupWorkerTask<ErrorsCatcherType>): Promise<void> {
     let uniqueEventHash = await this.getUniqueEventHash(task);
 

workers/javascript/src/index.ts

@@ -14,6 +14,7 @@ import { beautifyUserAgent, getFunctionContext } from './utils';
 import { Collection } from 'mongodb';
 /* eslint-disable-next-line no-unused-vars */
 import { memoize } from '../../../lib/memoize';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 /**
  * eslint does not count decorators as a variable usage
@@ -70,6 +71,7 @@ export default class JavascriptEventWorker extends EventWorker {
    *
    * @param event - event to handle
    */
+  @catchAndReport()
   public async handle(event: JavaScriptEventWorkerTask): Promise<void> {
     if (event.payload.release && event.payload.backtrace) {
       this.logger.info('beautifyBacktrace called');

workers/limiter/src/index.ts

@@ -13,6 +13,7 @@ import { WorkspaceWithTariffPlan } from '../types';
 import * as WorkerNames from '../../../lib/workerNames';
 import { DbHelper } from './dbHelper';
 import * as telegram from '../../../lib/utils/telegram';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 dotenv.config({ path: path.resolve(__dirname, '../.env') });
 
@@ -91,6 +92,7 @@ export default class LimiterWorker extends Worker {
    *
    * @param event - worker event to handle
    */
+  @catchAndReport()
   public async handle(event: LimiterEvent): Promise<void> {
     switch (event.type) {
       case 'regular-workspaces-check':

workers/notifier/src/index.ts

@@ -11,6 +11,7 @@ import { SenderWorkerTask } from 'hawk-worker-sender/types/sender-task';
 import RuleValidator from './validator';
 import TimeMs from '../../../lib/utils/time';
 import RedisHelper from './redisHelper';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 /**
  * Worker to buffer events before sending notifications about them
@@ -56,6 +57,7 @@ export default class NotifierWorker extends Worker {
    *
    * @param task — notifier task to handle
    */
+  @catchAndReport()
   public async handle(task: NotifierWorkerTask): Promise<void> {
     try {
       const { projectId, event } = task;

workers/paymaster/src/index.ts

@@ -10,6 +10,7 @@ import axios from 'axios';
 import * as WorkerNames from '../../../lib/workerNames';
 import HawkCatcher from '@hawk.so/nodejs';
 import { countDaysBeforePayday, countDaysAfterPayday, countDaysAfterBlock } from '../../../lib/utils/payday';
+import { catchAndReport } from '../../../lib/utils/catchAndReport';
 
 dotenv.config({
   path: path.resolve(__dirname, '../.env'),
@@ -136,6 +137,7 @@ export default class PaymasterWorker extends Worker {
    *
    * @param event - event to handle
    */
+  @catchAndReport()
   public async handle(event: PaymasterEvent): Promise<void> {
     switch (event.type) {
       case EventType.WorkspaceSubscriptionCheck: