@hatchet-dev/typescript-sdk 1.9.8 → 1.10.1

package/protoc/workflows/workflows.js

@@ -1,7 +1,7 @@
 "use strict";
 // Code generated by protoc-gen-ts_proto. DO NOT EDIT.
 // versions:
-//   protoc-gen-ts_proto  v2.7.0
+//   protoc-gen-ts_proto  v2.7.7
 //   protoc               v3.19.1
 // source: workflows/workflows.proto
 Object.defineProperty(exports, "__esModule", { value: true });
@@ -267,7 +267,7 @@ exports.PutWorkflowRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBasePutWorkflowRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -377,7 +377,7 @@ exports.CreateWorkflowVersionOpts = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseCreateWorkflowVersionOpts();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -619,7 +619,7 @@ exports.WorkflowConcurrencyOpts = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseWorkflowConcurrencyOpts();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -717,7 +717,7 @@ exports.CreateWorkflowJobOpts = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseCreateWorkflowJobOpts();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -816,7 +816,7 @@ exports.DesiredWorkerLabels = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseDesiredWorkerLabels();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -962,7 +962,7 @@ exports.CreateWorkflowStepOpts = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseCreateWorkflowStepOpts();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1167,7 +1167,7 @@ exports.CreateWorkflowStepOpts_WorkerLabelsEntry = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseCreateWorkflowStepOpts_WorkerLabelsEntry();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1258,7 +1258,7 @@ exports.CreateStepRateLimit = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseCreateStepRateLimit();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1371,7 +1371,7 @@ exports.ListWorkflowsRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseListWorkflowsRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1445,7 +1445,7 @@ exports.ScheduleWorkflowRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseScheduleWorkflowRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1605,7 +1605,7 @@ exports.ScheduledWorkflow = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseScheduledWorkflow();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1697,7 +1697,7 @@ exports.WorkflowVersion = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseWorkflowVersion();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1830,7 +1830,7 @@ exports.WorkflowTriggerEventRef = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseWorkflowTriggerEventRef();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1899,7 +1899,7 @@ exports.WorkflowTriggerCronRef = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseWorkflowTriggerCronRef();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -1965,7 +1965,7 @@ exports.BulkTriggerWorkflowRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseBulkTriggerWorkflowRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -2022,7 +2022,7 @@ exports.BulkTriggerWorkflowResponse = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseBulkTriggerWorkflowResponse();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -2113,7 +2113,7 @@ exports.TriggerWorkflowRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseTriggerWorkflowRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -2269,7 +2269,7 @@ exports.TriggerWorkflowResponse = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBaseTriggerWorkflowResponse();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -2329,7 +2329,7 @@ exports.PutRateLimitRequest = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBasePutRateLimitRequest();
         while (reader.pos < end) {
             const tag = reader.uint32();
@@ -2404,7 +2404,7 @@ exports.PutRateLimitResponse = {
     },
     decode(input, length) {
         const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        let end = length === undefined ? reader.len : reader.pos + length;
+        const end = length === undefined ? reader.len : reader.pos + length;
         const message = createBasePutRateLimitResponse();
         while (reader.pos < end) {
             const tag = reader.uint32();

package/util/config-loader/config-loader.js

@@ -44,7 +44,7 @@ const token_1 = require("./token");
 const DEFAULT_CONFIG_FILE = '.hatchet.yaml';
 class ConfigLoader {
     static loadClientConfig(override, config) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6;
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8;
         const yaml = this.loadYamlConfig(config === null || config === void 0 ? void 0 : config.path);
         const tlsConfig = (_a = override === null || override === void 0 ? void 0 : override.tls_config) !== null && _a !== void 0 ? _a : {
             tls_strategy: (_d = (_c = (_b = yaml === null || yaml === void 0 ? void 0 : yaml.tls_config) === null || _b === void 0 ? void 0 : _b.tls_strategy) !== null && _c !== void 0 ? _c : this.env('HATCHET_CLIENT_TLS_STRATEGY')) !== null && _d !== void 0 ? _d : 'tls',
@@ -54,6 +54,10 @@ class ConfigLoader {
             server_name: (_m = (_l = yaml === null || yaml === void 0 ? void 0 : yaml.tls_config) === null || _l === void 0 ? void 0 : _l.server_name) !== null && _m !== void 0 ? _m : this.env('HATCHET_CLIENT_TLS_SERVER_NAME'),
         };
         const token = (_p = (_o = override === null || override === void 0 ? void 0 : override.token) !== null && _o !== void 0 ? _o : yaml === null || yaml === void 0 ? void 0 : yaml.token) !== null && _p !== void 0 ? _p : this.env('HATCHET_CLIENT_TOKEN');
+        const healthCheckConfig = (_r = (_q = override === null || override === void 0 ? void 0 : override.healthcheck) !== null && _q !== void 0 ? _q : yaml === null || yaml === void 0 ? void 0 : yaml.healthcheck) !== null && _r !== void 0 ? _r : {
+            enabled: this.env('HATCHET_CLIENT_WORKER_HEALTHCHECK_ENABLED') === 'true',
+            port: parseInt(this.env('HATCHET_CLIENT_WORKER_HEALTHCHECK_PORT') || '8001', 10),
+        };
         if (!token) {
             throw new Error('No token provided. Provide it by setting the HATCHET_CLIENT_TOKEN environment variable.');
         }
@@ -66,25 +70,26 @@ class ConfigLoader {
         try {
             const addresses = (0, token_1.getAddressesFromJWT)(token);
             grpcBroadcastAddress =
-                (_s = (_r = (_q = override === null || override === void 0 ? void 0 : override.host_port) !== null && _q !== void 0 ? _q : yaml === null || yaml === void 0 ? void 0 : yaml.host_port) !== null && _r !== void 0 ? _r : this.env('HATCHET_CLIENT_HOST_PORT')) !== null && _s !== void 0 ? _s : addresses.grpcBroadcastAddress;
+                (_u = (_t = (_s = override === null || override === void 0 ? void 0 : override.host_port) !== null && _s !== void 0 ? _s : yaml === null || yaml === void 0 ? void 0 : yaml.host_port) !== null && _t !== void 0 ? _t : this.env('HATCHET_CLIENT_HOST_PORT')) !== null && _u !== void 0 ? _u : addresses.grpcBroadcastAddress;
             apiUrl =
-                (_v = (_u = (_t = override === null || override === void 0 ? void 0 : override.api_url) !== null && _t !== void 0 ? _t : yaml === null || yaml === void 0 ? void 0 : yaml.api_url) !== null && _u !== void 0 ? _u : this.env('HATCHET_CLIENT_API_URL')) !== null && _v !== void 0 ? _v : addresses.serverUrl;
+                (_x = (_w = (_v = override === null || override === void 0 ? void 0 : override.api_url) !== null && _v !== void 0 ? _v : yaml === null || yaml === void 0 ? void 0 : yaml.api_url) !== null && _w !== void 0 ? _w : this.env('HATCHET_CLIENT_API_URL')) !== null && _x !== void 0 ? _x : addresses.serverUrl;
         }
         catch (e) {
             grpcBroadcastAddress =
-                (_x = (_w = override === null || override === void 0 ? void 0 : override.host_port) !== null && _w !== void 0 ? _w : yaml === null || yaml === void 0 ? void 0 : yaml.host_port) !== null && _x !== void 0 ? _x : this.env('HATCHET_CLIENT_HOST_PORT');
-            apiUrl = (_z = (_y = override === null || override === void 0 ? void 0 : override.api_url) !== null && _y !== void 0 ? _y : yaml === null || yaml === void 0 ? void 0 : yaml.api_url) !== null && _z !== void 0 ? _z : this.env('HATCHET_CLIENT_API_URL');
+                (_z = (_y = override === null || override === void 0 ? void 0 : override.host_port) !== null && _y !== void 0 ? _y : yaml === null || yaml === void 0 ? void 0 : yaml.host_port) !== null && _z !== void 0 ? _z : this.env('HATCHET_CLIENT_HOST_PORT');
+            apiUrl = (_1 = (_0 = override === null || override === void 0 ? void 0 : override.api_url) !== null && _0 !== void 0 ? _0 : yaml === null || yaml === void 0 ? void 0 : yaml.api_url) !== null && _1 !== void 0 ? _1 : this.env('HATCHET_CLIENT_API_URL');
         }
-        let namespace = (_1 = (_0 = override === null || override === void 0 ? void 0 : override.namespace) !== null && _0 !== void 0 ? _0 : yaml === null || yaml === void 0 ? void 0 : yaml.namespace) !== null && _1 !== void 0 ? _1 : this.env('HATCHET_CLIENT_NAMESPACE');
+        let namespace = (_3 = (_2 = override === null || override === void 0 ? void 0 : override.namespace) !== null && _2 !== void 0 ? _2 : yaml === null || yaml === void 0 ? void 0 : yaml.namespace) !== null && _3 !== void 0 ? _3 : this.env('HATCHET_CLIENT_NAMESPACE');
         if (namespace && !(namespace === null || namespace === void 0 ? void 0 : namespace.endsWith('_'))) {
             namespace = `${namespace}_`;
         }
         return {
-            token: (_3 = (_2 = override === null || override === void 0 ? void 0 : override.token) !== null && _2 !== void 0 ? _2 : yaml === null || yaml === void 0 ? void 0 : yaml.token) !== null && _3 !== void 0 ? _3 : this.env('HATCHET_CLIENT_TOKEN'),
+            token: (_5 = (_4 = override === null || override === void 0 ? void 0 : override.token) !== null && _4 !== void 0 ? _4 : yaml === null || yaml === void 0 ? void 0 : yaml.token) !== null && _5 !== void 0 ? _5 : this.env('HATCHET_CLIENT_TOKEN'),
             host_port: grpcBroadcastAddress,
             api_url: apiUrl,
             tls_config: tlsConfig,
-            log_level: (_6 = (_5 = (_4 = override === null || override === void 0 ? void 0 : override.log_level) !== null && _4 !== void 0 ? _4 : yaml === null || yaml === void 0 ? void 0 : yaml.log_level) !== null && _5 !== void 0 ? _5 : this.env('HATCHET_CLIENT_LOG_LEVEL')) !== null && _6 !== void 0 ? _6 : 'INFO',
+            healthcheck: healthCheckConfig,
+            log_level: (_8 = (_7 = (_6 = override === null || override === void 0 ? void 0 : override.log_level) !== null && _6 !== void 0 ? _6 : yaml === null || yaml === void 0 ? void 0 : yaml.log_level) !== null && _7 !== void 0 ? _7 : this.env('HATCHET_CLIENT_LOG_LEVEL')) !== null && _8 !== void 0 ? _8 : 'INFO',
             tenant_id: tenantId,
             namespace: namespace ? `${namespace}`.toLowerCase() : '',
         };

package/v1/client/worker/health-server.d.ts

@@ -0,0 +1,30 @@
+import { Logger } from '../../../util/logger';
+export declare const workerStatus: {
+    readonly INITIALIZED: "INITIALIZED";
+    readonly STARTING: "STARTING";
+    readonly HEALTHY: "HEALTHY";
+    readonly UNHEALTHY: "UNHEALTHY";
+};
+export type WorkerStatus = (typeof workerStatus)[keyof typeof workerStatus];
+export declare class HealthServer {
+    private port;
+    private getStatus;
+    private workerName;
+    private getSlots;
+    private getActions;
+    private getLabels;
+    private logger;
+    private server;
+    private register;
+    private workerStatusGauge;
+    private workerSlotsGauge;
+    private workerActionsGauge;
+    private metricsInitialized;
+    constructor(port: number, getStatus: () => WorkerStatus, workerName: string, getSlots: () => number, getActions: () => string[], getLabels: () => Record<string, string | number>, logger: Logger);
+    private handleRequest;
+    private handleHealth;
+    private initializeMetrics;
+    private handleMetrics;
+    start(): Promise<void>;
+    stop(): Promise<void>;
+}

package/v1/client/worker/health-server.js

@@ -0,0 +1,161 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HealthServer = exports.workerStatus = void 0;
+const node_http_1 = require("node:http");
+exports.workerStatus = {
+    INITIALIZED: 'INITIALIZED',
+    STARTING: 'STARTING',
+    HEALTHY: 'HEALTHY',
+    UNHEALTHY: 'UNHEALTHY',
+};
+class HealthServer {
+    constructor(port, getStatus, workerName, getSlots, getActions, getLabels, logger) {
+        this.port = port;
+        this.getStatus = getStatus;
+        this.workerName = workerName;
+        this.getSlots = getSlots;
+        this.getActions = getActions;
+        this.getLabels = getLabels;
+        this.logger = logger;
+        this.server = null;
+        this.register = null;
+        this.workerStatusGauge = null;
+        this.workerSlotsGauge = null;
+        this.workerActionsGauge = null;
+        this.metricsInitialized = false;
+        this.initializeMetrics();
+    }
+    handleRequest(req, res) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const url = req.url || '';
+            if (url === '/health' && req.method === 'GET') {
+                yield this.handleHealth(res);
+            }
+            else if (url === '/metrics' && req.method === 'GET') {
+                yield this.handleMetrics(res);
+            }
+            else {
+                res.writeHead(404, { 'Content-Type': 'text/plain' });
+                res.end('Not Found');
+            }
+        });
+    }
+    handleHealth(res) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const response = {
+                status: this.getStatus(),
+                name: this.workerName,
+                slots: this.getSlots(),
+                actions: this.getActions(),
+                labels: this.getLabels(),
+                nodeVersion: process.version,
+            };
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            yield res.end(JSON.stringify(response));
+        });
+    }
+    initializeMetrics() {
+        try {
+            // @ts-ignore - prom-client is an optional dependency
+            // eslint-disable-next-line
+            const { Registry, Gauge, collectDefaultMetrics } = require('prom-client');
+            this.register = new Registry();
+            collectDefaultMetrics({ register: this.register });
+            this.workerStatusGauge = new Gauge({
+                name: 'hatchet_worker_status',
+                help: 'Current status of the Hatchet worker',
+                registers: [this.register],
+                collect: () => {
+                    this.workerStatusGauge.set(this.getStatus() === exports.workerStatus.HEALTHY ? 1 : 0);
+                },
+            });
+            this.workerSlotsGauge = new Gauge({
+                name: 'hatchet_worker_slots',
+                help: 'Total slots available on the worker',
+                registers: [this.register],
+                collect: () => {
+                    this.workerSlotsGauge.set(this.getSlots());
+                },
+            });
+            this.workerActionsGauge = new Gauge({
+                name: 'hatchet_worker_actions',
+                help: 'Number of registered actions on the worker',
+                registers: [this.register],
+                collect: () => {
+                    this.workerActionsGauge.set(this.getActions().length);
+                },
+            });
+            this.metricsInitialized = true;
+        }
+        catch (error) {
+            this.metricsInitialized = false;
+            this.logger.error('Metrics initialization failed - prom-client dependency not installed');
+        }
+    }
+    handleMetrics(res) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.metricsInitialized || !this.register) {
+                this.logger.error('Metrics initialization failed - prom-client dependency not installed');
+                res.writeHead(503, { 'Content-Type': 'text/plain' });
+                res.end('Metrics initialization failed');
+                return;
+            }
+            try {
+                const metrics = yield this.register.metrics();
+                res.writeHead(200, { 'Content-Type': this.register.contentType });
+                res.end(metrics);
+            }
+            catch (error) {
+                this.logger.error(`Error generating metrics: ${error}`);
+                res.writeHead(500, { 'Content-Type': 'text/plain' });
+                res.end('Error generating metrics');
+            }
+        });
+    }
+    start() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return new Promise((resolve, reject) => {
+                try {
+                    this.server = (0, node_http_1.createServer)((req, res) => {
+                        this.handleRequest(req, res);
+                    });
+                    this.server.listen(this.port, '0.0.0.0', () => {
+                        this.logger.info(`Health check server running on port ${this.port}`);
+                        resolve();
+                    });
+                    this.server.on('error', (error) => {
+                        this.logger.error(`Failed to start health check server: ${error.message}`);
+                        reject(error);
+                    });
+                }
+                catch (error) {
+                    this.logger.error(`Failed to start health check server: ${error}`);
+                    reject(error);
+                }
+            });
+        });
+    }
+    stop() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.server) {
+                return new Promise((resolve) => {
+                    this.server.close(() => {
+                        this.logger.info('Health check server stopped!');
+                        resolve();
+                    });
+                });
+            }
+            return Promise.resolve();
+        });
+    }
+}
+exports.HealthServer = HealthServer;

package/v1/client/worker/worker-internal.d.ts

@@ -14,6 +14,8 @@ export interface WorkerOpts {
     handleKill?: boolean;
     maxRuns?: number;
     labels?: WorkerLabels;
+    healthPort?: number;
+    enableHealthServer?: boolean;
 }
 export declare class V1Worker {
     client: HatchetClient;
@@ -30,12 +32,21 @@ export declare class V1Worker {
     logger: Logger;
     registeredWorkflowPromises: Array<Promise<any>>;
     labels: WorkerLabels;
+    healthPort: number;
+    enableHealthServer: boolean;
+    private healthServer;
+    private status;
     constructor(client: HatchetClient, options: {
         name: string;
         handleKill?: boolean;
         maxRuns?: number;
         labels?: WorkerLabels;
     });
+    private initializeHealthServer;
+    private getAvailableSlots;
+    private getRegisteredActions;
+    private getFilteredLabels;
+    private setStatus;
     private registerActions;
     getHandler(workflows: Workflow[]): void;
     registerWebhook(webhook: WebhookWorkerCreateRequest): Promise<import("axios").AxiosResponse<import("../../../clients/rest/generated/data-contracts").WebhookWorkerCreated, any, {}>>;

package/v1/client/worker/worker-internal.js

@@ -32,23 +32,61 @@ const step_1 = require("../../../step");
 const apply_namespace_1 = require("../../../util/apply-namespace");
 const context_1 = require("./context");
 const parent_run_context_vars_1 = require("../../parent-run-context-vars");
+const health_server_1 = require("./health-server");
 class V1Worker {
     constructor(client, options) {
+        var _a, _b, _c, _d;
         this.workflow_registry = [];
         this.futures = {};
         this.contexts = {};
         this.registeredWorkflowPromises = [];
         this.labels = {};
+        this.status = health_server_1.workerStatus.INITIALIZED;
         this.client = client;
         this.name = (0, apply_namespace_1.applyNamespace)(options.name, this.client.config.namespace);
         this.action_registry = {};
         this.maxRuns = options.maxRuns;
         this.labels = options.labels || {};
+        this.enableHealthServer = (_b = (_a = client.config.healthcheck) === null || _a === void 0 ? void 0 : _a.enabled) !== null && _b !== void 0 ? _b : false;
+        this.healthPort = (_d = (_c = client.config.healthcheck) === null || _c === void 0 ? void 0 : _c.port) !== null && _d !== void 0 ? _d : 8001;
         process.on('SIGTERM', () => this.exitGracefully(true));
         process.on('SIGINT', () => this.exitGracefully(true));
         this.killing = false;
         this.handle_kill = options.handleKill === undefined ? true : options.handleKill;
         this.logger = client.config.logger(`Worker/${this.name}`, this.client.config.log_level);
+        if (this.enableHealthServer && this.healthPort) {
+            this.initializeHealthServer();
+        }
+    }
+    initializeHealthServer() {
+        if (!this.healthPort) {
+            this.logger.warn('Health server enabled but no port specified');
+            return;
+        }
+        this.healthServer = new health_server_1.HealthServer(this.healthPort, () => this.status, this.name, () => this.getAvailableSlots(), () => this.getRegisteredActions(), () => this.getFilteredLabels(), this.logger);
+    }
+    getAvailableSlots() {
+        if (!this.maxRuns) {
+            return 0;
+        }
+        const currentRuns = Object.keys(this.futures).length;
+        return Math.max(0, this.maxRuns - currentRuns);
+    }
+    getRegisteredActions() {
+        return Object.keys(this.action_registry);
+    }
+    getFilteredLabels() {
+        const filtered = {};
+        for (const [key, value] of Object.entries(this.labels)) {
+            if (value !== undefined) {
+                filtered[key] = value;
+            }
+        }
+        return filtered;
+    }
+    setStatus(status) {
+        this.status = status;
+        this.logger.debug(`Worker status changed to: ${status}`);
     }
     registerActions(workflow) {
         var _a;
@@ -195,8 +233,10 @@ class V1Worker {
                     this.logger.warn(`\`on\` for event and cron triggers is deprecated and will be removed soon, use \`onEvents\` and \`onCrons\` instead for ${workflow.name}`);
                 }
                 const eventTriggers = [
-                    ...(workflow.onEvents || []),
-                    ...(((_u = workflow.on) === null || _u === void 0 ? void 0 : _u.event) ? [workflow.on.event] : []),
+                    ...(workflow.onEvents || []).map((event) => (0, apply_namespace_1.applyNamespace)(event, this.client.config.namespace)),
+                    ...(((_u = workflow.on) === null || _u === void 0 ? void 0 : _u.event)
+                        ? [(0, apply_namespace_1.applyNamespace)(workflow.on.event, this.client.config.namespace)]
+                        : []),
                 ];
                 const cronTriggers = [
                     ...(workflow.onCrons || []),
@@ -582,6 +622,7 @@ class V1Worker {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
             this.killing = true;
+            this.setStatus(health_server_1.workerStatus.UNHEALTHY);
             this.logger.info('Starting to exit...');
             try {
                 yield ((_a = this.listener) === null || _a === void 0 ? void 0 : _a.unregister());
@@ -593,6 +634,14 @@ class V1Worker {
             // attempt to wait for futures to finish
             yield Promise.all(Object.values(this.futures).map(({ promise }) => promise));
             this.logger.info('Successfully finished pending tasks.');
+            if (this.healthServer) {
+                try {
+                    yield this.healthServer.stop();
+                }
+                catch (e) {
+                    this.logger.error(`Could not stop health server: ${e.message}`);
+                }
+            }
             if (handleKill) {
                 this.logger.info('Exiting hatchet worker...');
                 process.exit(0);
@@ -602,6 +651,17 @@ class V1Worker {
     start() {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, e_1, _b, _c;
+            this.setStatus(health_server_1.workerStatus.STARTING);
+            if (this.healthServer) {
+                try {
+                    yield this.healthServer.start();
+                }
+                catch (e) {
+                    this.logger.error(`Could not start health server: ${e.message}`);
+                    this.setStatus(health_server_1.workerStatus.UNHEALTHY);
+                    return;
+                }
+            }
             // ensure all workflows are registered
             yield Promise.all(this.registeredWorkflowPromises);
             if (Object.keys(this.action_registry).length === 0) {
@@ -616,6 +676,7 @@ class V1Worker {
                     labels: this.labels,
                 });
                 this.workerId = this.listener.workerId;
+                this.setStatus(health_server_1.workerStatus.HEALTHY);
                 const generator = this.listener.actions();
                 this.logger.info(`Worker ${this.name} listening for actions`);
                 try {
@@ -636,6 +697,7 @@ class V1Worker {
                 }
             }
             catch (e) {
+                this.setStatus(health_server_1.workerStatus.UNHEALTHY);
                 if (this.killing) {
                     this.logger.info(`Exiting worker, ignoring error: ${e.message}`);
                     return;

package/version.d.ts

@@ -1 +1 @@
-export declare const HATCHET_VERSION = "1.9.8";
+export declare const HATCHET_VERSION = "1.10.1";

package/version.js

@@ -1,4 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.HATCHET_VERSION = void 0;
-exports.HATCHET_VERSION = '1.9.8';
+exports.HATCHET_VERSION = '1.10.1';