New API to define handlers/shared handlers without wrapped functions. This supports proper contextual typing, and allows to evolve the Typed State API as well.

Author: slinkydeveloper

packages/restate-sdk-clients/src/api.ts

@@ -7,6 +7,7 @@ import type {
   WorkflowDefinitionFrom,
   Serde,
   Duration,
+  FlattenHandlersDefinition,
   JournalValueCodec,
 } from "@restatedev/restate-sdk-core";
 import { millisOrDurationToMillis } from "@restatedev/restate-sdk-core";
@@ -26,7 +27,9 @@ export interface Ingress {
   /**
    * Create a client from a {@link ServiceDefinition}.
    */
-  serviceClient<D>(opts: ServiceDefinitionFrom<D>): IngressClient<Service<D>>;
+  serviceClient<D>(
+    opts: ServiceDefinitionFrom<D>
+  ): IngressClient<FlattenHandlersDefinition<Service<D>>>;
 
   /**
    * Create a client from a {@link WorkflowDefinition}.
@@ -36,7 +39,7 @@ export interface Ingress {
   workflowClient<D>(
     opts: WorkflowDefinitionFrom<D>,
     key: string
-  ): IngressWorkflowClient<Workflow<D>>;
+  ): IngressWorkflowClient<FlattenHandlersDefinition<Workflow<D>>>;
 
   /**
    * Create a client from a {@link VirtualObjectDefinition}.
@@ -45,22 +48,22 @@ export interface Ingress {
   objectClient<D>(
     opts: VirtualObjectDefinitionFrom<D>,
     key: string
-  ): IngressClient<VirtualObject<D>>;
+  ): IngressClient<FlattenHandlersDefinition<VirtualObject<D>>>;
 
   /**
    * Create a client from a {@link ServiceDefinition}.
    */
   serviceSendClient<D>(
     opts: ServiceDefinitionFrom<D>
-  ): IngressSendClient<Service<D>>;
+  ): IngressSendClient<FlattenHandlersDefinition<Service<D>>>;
 
   /**
    * Create a client from a {@link VirtualObjectDefinition}.
    */
   objectSendClient<D>(
     opts: VirtualObjectDefinitionFrom<D>,
     key: string
-  ): IngressSendClient<VirtualObject<D>>;
+  ): IngressSendClient<FlattenHandlersDefinition<VirtualObject<D>>>;
 
   /**
    * Resolve an awakeable from the ingress client.

packages/restate-sdk-clients/src/ingress.ts

@@ -19,6 +19,7 @@ import {
   type Serde,
   serde,
   type JournalValueCodec,
+  type FlattenHandlersDefinition,
 } from "@restatedev/restate-sdk-core";
 import type {
   ConnectionOpts,
@@ -296,21 +297,27 @@ class HttpIngress implements Ingress {
     );
   }
 
-  serviceClient<D>(opts: ServiceDefinitionFrom<D>): IngressClient<Service<D>> {
-    return this.proxy(opts.name) as IngressClient<Service<D>>;
+  serviceClient<D>(
+    opts: ServiceDefinitionFrom<D>
+  ): IngressClient<FlattenHandlersDefinition<Service<D>>> {
+    return this.proxy(opts.name) as IngressClient<
+      FlattenHandlersDefinition<Service<D>>
+    >;
   }
 
   objectClient<D>(
     opts: VirtualObjectDefinitionFrom<D>,
     key: string
-  ): IngressClient<VirtualObject<D>> {
-    return this.proxy(opts.name, key) as IngressClient<VirtualObject<D>>;
+  ): IngressClient<FlattenHandlersDefinition<VirtualObject<D>>> {
+    return this.proxy(opts.name, key) as IngressClient<
+      FlattenHandlersDefinition<VirtualObject<D>>
+    >;
   }
 
   workflowClient<D>(
     opts: WorkflowDefinitionFrom<D>,
     key: string
-  ): IngressWorkflowClient<Workflow<D>> {
+  ): IngressWorkflowClient<FlattenHandlersDefinition<Workflow<D>>> {
     const component = opts.name;
     const conn = this.opts;
 
@@ -392,23 +399,23 @@ class HttpIngress implements Ingress {
           };
         },
       }
-    ) as IngressWorkflowClient<Workflow<D>>;
+    ) as IngressWorkflowClient<FlattenHandlersDefinition<Workflow<D>>>;
   }
 
   objectSendClient<D>(
     opts: VirtualObjectDefinitionFrom<D>,
     key: string
-  ): IngressSendClient<VirtualObject<D>> {
+  ): IngressSendClient<FlattenHandlersDefinition<VirtualObject<D>>> {
     return this.proxy(opts.name, key, true) as IngressSendClient<
-      VirtualObject<D>
+      FlattenHandlersDefinition<VirtualObject<D>>
     >;
   }
 
   serviceSendClient<D>(
     opts: ServiceDefinitionFrom<D>
-  ): IngressSendClient<Service<D>> {
+  ): IngressSendClient<FlattenHandlersDefinition<Service<D>>> {
     return this.proxy(opts.name, undefined, true) as IngressSendClient<
-      Service<D>
+      FlattenHandlersDefinition<Service<D>>
     >;
   }
 

packages/restate-sdk-clients/src/public_api.ts

@@ -22,6 +22,7 @@ export type {
   VirtualObject,
   Duration,
   JournalValueCodec,
+  FlattenHandlersDefinition,
 } from "@restatedev/restate-sdk-core";
 
 export { serde } from "@restatedev/restate-sdk-core";

packages/restate-sdk-core/src/core.ts

@@ -29,19 +29,28 @@ export interface RestateWorkflowContext
 
 // ----------- service -------------------------------------------------------
 
+/**
+ * @deprecated
+ */
 export type ArgType<T> = T extends (ctx: any) => any
   ? void
   : T extends (ctx: any, input: infer I) => any
   ? I
   : never;
 
+/**
+ * @deprecated
+ */
 export type HandlerReturnType<T> = T extends (
   ctx: any,
   input: any
 ) => Promise<infer R>
   ? R
   : never;
 
+/**
+ * @deprecated
+ */
 export type ServiceHandler<F, C = RestateContext> = F extends (
   ctx: C
 ) => Promise<any>
@@ -65,6 +74,9 @@ export type ServiceDefinitionFrom<M> = M extends ServiceDefinition<
 
 // ----------- object -------------------------------------------------------
 
+/**
+ * @deprecated
+ */
 export type ObjectSharedHandler<
   F,
   SC = RestateObjectSharedContext
@@ -74,6 +86,9 @@ export type ObjectSharedHandler<
   ? F
   : (ctx: SC, param?: any) => Promise<any>;
 
+/**
+ * @deprecated
+ */
 export type ObjectHandler<F, C = RestateObjectContext> = F extends (
   ctx: C,
   param: any
@@ -104,6 +119,9 @@ export type VirtualObjectDefinitionFrom<M> = M extends VirtualObjectDefinition<
 
 // ----------- workflow -------------------------------------------------------
 
+/**
+ * @deprecated
+ */
 export type WorkflowSharedHandler<
   F,
   SC = RestateWorkflowSharedContext
@@ -113,6 +131,9 @@ export type WorkflowSharedHandler<
   ? F
   : (ctx: SC, param?: any) => Promise<any>;
 
+/**
+ * @deprecated
+ */
 export type WorkflowHandler<F, C = RestateWorkflowContext> = F extends (
   ctx: C,
   param: any
@@ -135,3 +156,28 @@ export type WorkflowDefinitionFrom<M> = M extends WorkflowDefinition<
 >
   ? M
   : WorkflowDefinition<string, M>;
+
+// -------- Type manipulation for clients
+
+export type FlattenHandlersDefinition<M> = {
+  [K in keyof M]: M[K] extends {
+    handler:
+      | ((ctx: any, param: any) => Promise<any>)
+      | ((ctx: any) => Promise<any>)
+      | ((ctx: any, param?: any) => Promise<any>);
+  }
+    ? M[K]["handler"]
+    : M[K] extends {
+        sharedHandler:
+          | ((ctx: any, param: any) => Promise<any>)
+          | ((ctx: any) => Promise<any>)
+          | ((ctx: any, param?: any) => Promise<any>);
+      }
+    ? M[K]["sharedHandler"]
+    : M[K] extends
+        | ((ctx: any, param: any) => Promise<any>)
+        | ((ctx: any) => Promise<any>)
+        | ((ctx: any, param?: any) => Promise<any>)
+    ? M[K]
+    : never;
+};

packages/restate-sdk-core/src/public_api.ts

@@ -31,6 +31,7 @@ export type {
   WorkflowDefinitionFrom,
   ArgType,
   HandlerReturnType,
+  FlattenHandlersDefinition,
 } from "./core.js";
 
 export type { Serde } from "./serde_api.js";

packages/restate-sdk-examples/src/object.ts

@@ -22,46 +22,25 @@ import {
 export const counter = object({
   name: "counter",
   handlers: {
-    /**
-     * Add amount to the currently stored count
-     */
-    add: async (ctx: ObjectContext, amount: number) => {
+    add: async (ctx, amount: number) => {
       const current = await ctx.get<number>("count");
       const updated = (current ?? 0) + amount;
       ctx.set("count", updated);
       return updated;
     },
-
-    /**
-     * Get the current amount.
-     *
-     * Notice that VirtualObjects can have "shared" handlers.
-     * These handlers can be executed concurrently to the exclusive handlers (i.e. add)
-     * But they can not modify the state (set is missing from the ctx).
-     */
-    current: createObjectSharedHandler(
-      async (ctx: ObjectSharedContext): Promise<number> => {
-        return (await ctx.get("count")) ?? 0;
-      }
-    ),
-
-    /**
-     * Handlers (shared or exclusive) can be configured to bypass JSON serialization,
-     * by specifying the input (accept) and output (contentType) content types.
-     *
-     * to call that handler with binary data, you can use the following curl command:
-     * curl -X POST -H "Content-Type: application/octet-stream" --data-binary 'hello' ${RESTATE_INGRESS_URL}/counter/mykey/binary
-     */
-    binary: createObjectHandler(
-      {
-        input: serde.binary,
-        output: serde.binary,
+    current: {
+      sharedHandler: async (ctx) => {
+        return (await ctx.get<number>("count")) ?? 0;
+      },
+    },
+    clear: {
+      handler: async (ctx) => {
+        ctx.clearAll();
+      },
+      options: {
+        enableLazyState: true,
       },
-      async (ctx: ObjectContext, data: Uint8Array) => {
-        // console.log("Received binary data", data);
-        return data;
-      }
-    ),
+    },
   },
 });
 

packages/restate-sdk/src/common_api.ts

@@ -83,6 +83,14 @@ export type {
   ServiceOptions,
   ObjectOptions,
   WorkflowOptions,
+  ServiceHandlers,
+  ObjectHandlers,
+  WorkflowHandlers,
+  ServiceHandlerDefinition,
+  ObjectHandlerDefinition,
+  WorkflowHandlerDefinition,
+  WorkflowSharedHandlerDefinition,
+  Handler,
 } from "./types/rpc.js";
 export {
   service,
@@ -99,6 +107,7 @@ export type {
   ServiceDefinition,
   VirtualObjectDefinition,
   WorkflowDefinition,
+  FlattenHandlersDefinition,
 } from "@restatedev/restate-sdk-core";
 
 export type {
@@ -131,8 +140,23 @@ export type { EndpointOptions } from "./endpoint/types.js";
 
 import { handlers } from "./types/rpc.js";
 
+/**
+ * @deprecated
+ */
 export const createServiceHandler = handlers.handler;
+/**
+ * @deprecated
+ */
 export const createObjectHandler = handlers.object.exclusive;
+/**
+ * @deprecated
+ */
 export const createObjectSharedHandler = handlers.object.shared;
+/**
+ * @deprecated
+ */
 export const createWorkflowHandler = handlers.workflow.workflow;
+/**
+ * @deprecated
+ */
 export const createWorkflowSharedHandler = handlers.workflow.shared;