Skip to content

Commit 7b1fd28

Browse files
committed
fix(rivetkit): ack ws messages
1 parent 9fbbebe commit 7b1fd28

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,8 +545,15 @@ export class EngineActorDriver implements ActorDriver {
545545
});
546546
}
547547

548-
websocket.addEventListener("message", (event) => {
548+
websocket.addEventListener("message", (event: RivetMessageEvent) => {
549549
wsHandlerPromise.then((x) => x.onMessage?.(event, wsContext));
550+
551+
invariant(event.rivetRequestId, "missing rivetRequestId");
552+
invariant(event.rivetMessageIndex, "missing rivetMessageIndex");
553+
this.#runner.sendWebsocketMessageAck(
554+
event.rivetRequestId,
555+
event.rivetMessageIndex,
556+
);
550557
});
551558

552559
websocket.addEventListener("close", (event) => {

0 commit comments

Comments
 (0)