Use xstate function to bind machines to a restate endpoint

restatedev · Oct 8, 2024 · d210a66 · d210a66
1 parent 6c7f4f1
commit d210a66
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 9 deletions.
diff --git a/examples/auth/app.ts b/examples/auth/app.ts
@@ -10,7 +10,7 @@
  */
 
 import * as restate from "@restatedev/restate-sdk";
-import { bindXStateRouter, fromPromise } from "@restatedev/xstate";
+import { xstate, fromPromise } from "@restatedev/xstate";
 import { createMachine, sendTo } from "xstate";
 
 const authServerMachine = createMachine(
@@ -77,4 +77,4 @@ const authClientMachine = createMachine({
   },
 });
 
-await bindXStateRouter(restate.endpoint(), "auth", authClientMachine).listen();
+await restate.endpoint().bind(xstate("auth", authClientMachine)).listen();
diff --git a/examples/payment/app.ts b/examples/payment/app.ts
@@ -1,6 +1,6 @@
 import { log, setup } from "xstate";
 import * as restate from "@restatedev/restate-sdk";
-import { fromPromise, bindXStateRouter } from "@restatedev/xstate";
+import { fromPromise, xstate } from "@restatedev/xstate";
 
 export const machine = setup({
   types: {
@@ -130,4 +130,4 @@ export const machine = setup({
   },
 });
 
-await bindXStateRouter(restate.endpoint(), "payment", machine).listen();
+await restate.endpoint().bind(xstate("payment", machine)).listen();
diff --git a/src/lib.ts b/src/lib.ts
@@ -581,12 +581,11 @@ const actorObject = <TLogic extends AnyStateMachine>(
   });
 };
 
-export const bindXStateRouter = <TLogic extends AnyStateMachine>(
-  server: restate.RestateEndpoint,
+export const xstate = <TLogic extends AnyStateMachine>(
   path: string,
   logic: TLogic
-): restate.RestateEndpoint => {
-  return server.bind(actorObject(path, logic));
+) => {
+  return actorObject(path, logic);
 };
 
 export const xStateApi = <TLogic extends AnyStateMachine>(

diff --git a/src/public_api.ts b/src/public_api.ts
@@ -1,2 +1,2 @@
-export { bindXStateRouter } from "./lib.js";
+export { xstate } from "./lib.js";
 export { fromPromise } from "./promise.js";