[dagit] tweak websocket disable (#7860)

dagster-io · May 12, 2022 · 61afd9a · 61afd9a
1 parent b71f150
commit 61afd9a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/js_modules/dagit/packages/core/src/runs/LogsProvider.tsx b/js_modules/dagit/packages/core/src/runs/LogsProvider.tsx
@@ -266,14 +266,15 @@ export const LogsProvider: React.FC<LogsProviderProps> = (props) => {
   const {children, runId} = props;
   const {availability, disabled} = React.useContext(WebSocketContext);
 
-  if (availability === 'attempting-to-connect') {
-    return <>{children({allNodes: [], loading: true})}</>;
-  }
-
+  // if disabled, drop to query variant immediately
   if (availability === 'unavailable' || disabled) {
     return <LogsProviderWithQuery runId={runId}>{children}</LogsProviderWithQuery>;
   }
 
+  if (availability === 'attempting-to-connect') {
+    return <>{children({allNodes: [], loading: true})}</>;
+  }
+
   return <LogsProviderWithSubscription runId={runId}>{children}</LogsProviderWithSubscription>;
 };