Skip to content

Commit

Permalink
Update grpc client (#29)
Browse files Browse the repository at this point in the history
* update grpc client

decouple orchestrator name from orchestrator

* rollback split name

* more updates

* fix TaskHubGrpcWorker options

* add docs

* add human-interaction example

* update example

* minor update on example

* await terminate orchestration in test

* lint code style
  • Loading branch information
kaibocai authored Jan 8, 2024
1 parent 9dfa855 commit 5c07288
Show file tree
Hide file tree
Showing 17 changed files with 6,324 additions and 2,905 deletions.
10 changes: 9 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -144,4 +144,12 @@ temp/
build/

# version file is auto-generated
src/version.ts
src/version.ts

### vscode ###
.vscode
!.vscode/settings.json
!.vscode/tasks.json
!.vscode/launch.json
!.vscode/extensions.json
*.code-workspace
4 changes: 2 additions & 2 deletions examples/activity-sequence.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { TaskHubGrpcClient } from "../src/client";
import { TaskHubGrpcClient } from "../src/client/client";
import { ActivityContext } from "../src/task/context/activity-context";
import { OrchestrationContext } from "../src/task/context/orchestration-context";
import { TOrchestrator } from "../src/types/orchestrator.type";
Expand Down Expand Up @@ -41,7 +41,7 @@ import { TaskHubGrpcWorker } from "../src/worker/task-hub-grpc-worker";

// Schedule a new orchestration
try {
const id = await taskHubClient.scheduleNewOrchestration(sequence, 1);
const id = await taskHubClient.scheduleNewOrchestration(sequence);
console.log(`Orchestration scheduled with ID: ${id}`);

// Wait for orchestration completion
Expand Down
2 changes: 1 addition & 1 deletion examples/fanout-fanin.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { TaskHubGrpcClient } from "../src/client";
import { TaskHubGrpcClient } from "../src/client/client";
import { whenAll } from "../src/task";
import { ActivityContext } from "../src/task/context/activity-context";
import { OrchestrationContext } from "../src/task/context/orchestration-context";
Expand Down
110 changes: 110 additions & 0 deletions examples/human_interaction.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,110 @@
import { TaskHubGrpcClient } from "../src/client/client";
import { whenAny } from "../src/task";
import { ActivityContext } from "../src/task/context/activity-context";
import { OrchestrationContext } from "../src/task/context/orchestration-context";
import { Task } from "../src/task/task";
import { TOrchestrator } from "../src/types/orchestrator.type";
import { TaskHubGrpcWorker } from "../src/worker/task-hub-grpc-worker";
import * as readlineSync from "readline-sync";

// Wrap the entire code in an immediately-invoked async function
(async () => {
class Order {
cost: number;
product: string;
quantity: number;
constructor(cost: number, product: string, quantity: number) {
this.cost = cost;
this.product = product;
this.quantity = quantity;
}
}

function sleep(ms: number): Promise<void> {
return new Promise((resolve) => setTimeout(resolve, ms));
}

// Update the gRPC client and worker to use a local address and port
const grpcServerAddress = "localhost:4001";
let taskHubClient: TaskHubGrpcClient = new TaskHubGrpcClient(grpcServerAddress);
let taskHubWorker: TaskHubGrpcWorker = new TaskHubGrpcWorker(grpcServerAddress);

//Activity function that sends an approval request to the manager
const sendApprovalRequest = async (_: ActivityContext, order: Order) => {
// Simulate some work that takes an amount of time
await sleep(3000);
console.log(`Sending approval request for order: ${order.product}`);
};

// Activity function that places an order
const placeOrder = async (_: ActivityContext, order: Order) => {
console.log(`Placing order: ${order.product}`);
};

// Orchestrator function that represents a purchase order workflow
const purchaseOrderWorkflow: TOrchestrator = async function* (ctx: OrchestrationContext, order: Order): any {
// Orders under $1000 are auto-approved
if (order.cost < 1000) {
return "Auto-approved";
}

// Orders of $1000 or more require manager approval
yield ctx.callActivity(sendApprovalRequest, order);

// Approvals must be received within 24 hours or they will be cancled.
const tasks: Task<any>[] = [];
const approvalEvent = ctx.waitForExternalEvent("approval_received");
const timeoutEvent = ctx.createTimer(24 * 60 * 60);
tasks.push(approvalEvent);
tasks.push(timeoutEvent);
const winner = whenAny(tasks);

if (winner == timeoutEvent) {
return "Cancelled";
}

yield ctx.callActivity(placeOrder, order);
const approvalDetails = approvalEvent.getResult();
return `Approved by ${approvalDetails.approver}`;
};

taskHubWorker.addOrchestrator(purchaseOrderWorkflow);
taskHubWorker.addActivity(sendApprovalRequest);
taskHubWorker.addActivity(placeOrder);

// Wrap the worker startup in a try-catch block to handle any errors during startup
try {
await taskHubWorker.start();
console.log("Worker started successfully");
} catch (error) {
console.error("Error starting worker:", error);
}

// Schedule a new orchestration
try {
const cost = readlineSync.question("Cost of your order:");
const approver = readlineSync.question("Approver of your order:");
const timeout = readlineSync.question("Timeout for your order in seconds:");
const order = new Order(cost, "MyProduct", 1);
const id = await taskHubClient.scheduleNewOrchestration(purchaseOrderWorkflow, order);
console.log(`Orchestration scheduled with ID: ${id}`);

if (readlineSync.keyInYN("Press [Y] to approve the order... Y/yes, N/no")) {
const approvalEvent = { approver: approver };
await taskHubClient.raiseOrchestrationEvent(id, "approval_received", approvalEvent);
} else {
return "Order rejected";
}

// Wait for orchestration completion
const state = await taskHubClient.waitForOrchestrationCompletion(id, undefined, timeout + 2);

console.log(`Orchestration completed! Result: ${state?.serializedOutput}`);
} catch (error) {
console.error("Error scheduling or waiting for orchestration:", error);
}

// stop worker and client
await taskHubWorker.stop();
await taskHubClient.stop();
})();
Loading

0 comments on commit 5c07288

Please sign in to comment.