Merge branch 'main' into next

Rabatta-ApS · Sep 27, 2022 · 81c4b7f · 81c4b7f
1 parent 3904459
commit 81c4b7f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/internal/endpoint-runtime.ts b/src/internal/endpoint-runtime.ts
@@ -25,7 +25,7 @@ export interface EndpointRuntime {
   onMessage: <Data extends JsonValue, K extends DataTypeKey = DataTypeKey>(
     messageID: K,
     callback: OnMessageCallback<GetDataType<K, Data>, GetReturnType<K, any>>
-  ) => void
+  ) => (() => void)
   /**
    * @internal
    */
@@ -181,6 +181,7 @@ export const createEndpointRuntime = (
     },
     onMessage: (messageID, callback) => {
       onMessageListeners.set(messageID, callback)
+      return () => onMessageListeners.delete(messageID)
     },
   }
 }