fix: multiple custom transport cleanup

open-rpc · Sep 3, 2021 · 78651fa · 78651fa
1 parent 5c9f2dc
commit 78651fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/containers/Inspector.tsx b/src/containers/Inspector.tsx
@@ -207,7 +207,7 @@ const Inspector: React.FC<IProps> = (props) => {
 
   useEffect(() => {
     if (selectedTransport !== undefined) {
-      setTransport(selectedTransport!);
+      setTransport(selectedTransport);
       const s: IWebTransport = selectedTransport as IWebTransport;
       if (s.schema !== undefined && s.schema !== true && s.schema !== false) {
         setTransportOptions((s.schema.examples as ExampleObject[])[0]);

diff --git a/src/hooks/useTransport.ts b/src/hooks/useTransport.ts
@@ -28,7 +28,7 @@ const getTransportFromType = async (
 ): Promise<Transport> => {
   let localTransport: any;
   const localTransportType = transports.find((value) => {
-    return value.type === transport.type;
+    return value.type === transport.type && value.name === transport.name;
   });
   if (localTransportType?.type === "websocket") {
     localTransport = new WebSocketTransport(uri);
@@ -74,6 +74,7 @@ const getTransportFromType = async (
       }
       public close() {
         intermediateTransport.unsubscribe();
+        intermediateTransport.close();
         return intermediateTransport.sendData({
           internalID: 0,
           request: {
@@ -119,6 +120,10 @@ const useTransport: TUseTransport = (transports, url, defaultTransportType, tran
       return localTransport.connect().then(() => {
         setTransportConnected(true);
         setTransport(localTransport);
+      }).catch((e) => {
+        localTransport.unsubscribe()
+        localTransport.close();
+        throw e;
       });
     };