Merge pull request #74 from viamrobotics/connection-status-1

Only execute robot queries when connected

Author: micheal-parks

.changeset/itchy-sides-wink.md

@@ -0,0 +1,5 @@
+---
+'@viamrobotics/svelte-sdk': patch
+---
+
+Only execute robot queries when connected

src/lib/hooks/create-robot-query.svelte.ts

@@ -4,11 +4,12 @@ import {
   type QueryObserverResult,
 } from '@tanstack/svelte-query';
 
-import type { RobotClient } from '@viamrobotics/sdk';
+import { MachineConnectionEvent, type RobotClient } from '@viamrobotics/sdk';
 import { toStore, fromStore } from 'svelte/store';
 import { usePolling } from './use-polling.svelte';
 import { useQueryLogger } from '$lib/query-logger';
 import { useEnabledQueries } from './use-enabled-queries.svelte';
+import { useConnectionStatus } from './robot-clients.svelte';
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export type ArgumentsType<T> = T extends (...args: infer U) => any ? U : never;
@@ -37,6 +38,8 @@ export const createRobotQuery = <T extends RobotClient, K extends keyof T>(
       ]
     | [options?: (() => QueryOptions) | QueryOptions]
 ): { current: QueryObserverResult<ResolvedReturnType<T[K]>> } => {
+  const partID = $derived((client.current as T & { partID: string })?.partID);
+  const connectionStatus = useConnectionStatus(() => partID);
   const debug = useQueryLogger();
   const enabledQueries = useEnabledQueries();
   let [args, options] = additional;
@@ -52,7 +55,8 @@ export const createRobotQuery = <T extends RobotClient, K extends keyof T>(
   const _args = $derived(typeof args === 'function' ? args() : args);
   const methodName = $derived(String(method));
   const enabled = $derived(
-    client.current !== undefined &&
+    connectionStatus.current === MachineConnectionEvent.CONNECTED &&
+      client.current !== undefined &&
       _options?.enabled !== false &&
       enabledQueries.robotQueries
   );
@@ -62,7 +66,7 @@ export const createRobotQuery = <T extends RobotClient, K extends keyof T>(
       queryKey: [
         'viam-svelte-sdk',
         'partID',
-        (client.current as T & { partID: string })?.partID,
+        partID,
         'robotClient',
         methodName,
         ...(_args ? [_args] : []),

src/routes/components/parts.svelte

@@ -35,15 +35,15 @@ let streaming = true;
       </button>
 
       {#if id === partID.current}
+        <Version partID={id} />
+
         <Part partID={id} />
       {/if}
     {/each}
 
     {status.current}
   </div>
 
-  <Version {partID} />
-
   <h2 class="py-2">Resources</h2>
   {#if resources.error}
     Error fetching: {resources.error.message}