feat: autoscaling target

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "0.20.0",
+  "version": "0.21.0",
   "description": "Background task orchestration & visibility for developers",
   "types": "dist/index.d.ts",
   "files": [

src/clients/dispatcher/dispatcher-client.ts

@@ -49,10 +49,17 @@ export class DispatcherClient {
   }
 
   async getActionListener(options: GetActionListenerOptions) {
+    // merge preset labels with labels
+
+    const mergedLabels = {
+      ...this.config.preset_labels,
+      ...options.labels,
+    };
+
     // Register the worker
     const registration = await this.client.register({
       ...options,
-      labels: options.labels ? mapLabels(options.labels) : undefined,
+      labels: mergedLabels ? mapLabels(mergedLabels) : undefined,
       runtimeInfo: this.getRuntimeInfo(),
     });
 

src/clients/hatchet-client/client-config.ts

@@ -18,6 +18,8 @@ export const ClientConfigSchema = z.object({
   log_level: z.enum(['OFF', 'DEBUG', 'INFO', 'WARN', 'ERROR']).optional(),
   tenant_id: z.string(),
   namespace: z.string().optional(),
+  autoscaling_target: z.string().optional(),
+  preset_labels: z.record(z.string(), z.string()).optional(),
 });
 
 export type LogConstructor = (context: string, logLevel?: LogLevel) => Logger;

src/util/config-loader/config-loader.ts

@@ -17,7 +17,8 @@ type EnvVars =
   | 'HATCHET_CLIENT_TLS_ROOT_CA_FILE'
   | 'HATCHET_CLIENT_TLS_SERVER_NAME'
   | 'HATCHET_CLIENT_LOG_LEVEL'
-  | 'HATCHET_CLIENT_NAMESPACE';
+  | 'HATCHET_CLIENT_NAMESPACE'
+  | 'HATCHET_CLIENT_AUTOSCALING_TARGET';
 
 type TLSStrategy = 'tls' | 'mtls';
 
@@ -83,6 +84,16 @@ export class ConfigLoader {
     const namespace =
       override?.namespace ?? yaml?.namespace ?? this.env('HATCHET_CLIENT_NAMESPACE');
 
+    const autoscalingTarget =
+      override?.autoscaling_target ??
+      yaml?.autoscaling_target ??
+      this.env('HATCHET_CLIENT_AUTOSCALING_TARGET');
+
+    const presetLabels = {
+      ...(override?.preset_labels ?? yaml?.preset_labels ?? {}),
+      ...(autoscalingTarget ? { 'hatchet-autoscaling-target': autoscalingTarget } : {}),
+    };
+
     return {
       token: override?.token ?? yaml?.token ?? this.env('HATCHET_CLIENT_TOKEN'),
       host_port: grpcBroadcastAddress,
@@ -95,6 +106,8 @@ export class ConfigLoader {
         'INFO',
       tenant_id: tenantId,
       namespace: namespace ? `${namespace}_`.toLowerCase() : '',
+      autoscaling_target: autoscalingTarget,
+      preset_labels: presetLabels,
     };
   }