4-10 树域 Rust 家族化

wolai-frontend/src/components/sidebar/use-preferred-sidebar-snapshot.ts

@@ -1,9 +1,6 @@
 import { useMemo } from "react";
 import type { SidebarInitialData } from "@/components/sidebar/types";
-import {
-  buildSidebarDataSyncKey,
-  getSidebarDataFreshness,
-} from "@/components/sidebar/sidebar-sync";
+import { buildSidebarDataSyncKey } from "@/components/sidebar/sidebar-sync";
 
 export type PreferredSidebarSnapshotSource = "initial" | "query" | "tree_stream";
 
@@ -11,6 +8,7 @@ export function usePreferredSidebarSnapshot(input: {
   initialData: SidebarInitialData;
   sidebarQueryData: SidebarInitialData | null;
   treeStreamData: SidebarInitialData | null;
+  treeStreamStatus?: "idle" | "connecting" | "live" | "fallback";
 }) {
   const querySyncKey = useMemo(
     () => (input.sidebarQueryData ? buildSidebarDataSyncKey(input.sidebarQueryData) : null),
@@ -21,23 +19,10 @@ export function usePreferredSidebarSnapshot(input: {
     [input.treeStreamData],
   );
   const initialSyncKey = useMemo(() => buildSidebarDataSyncKey(input.initialData), [input.initialData]);
-  const queryFreshness = useMemo(
-    () => (input.sidebarQueryData ? getSidebarDataFreshness(input.sidebarQueryData) : Number.NEGATIVE_INFINITY),
-    [input.sidebarQueryData],
-  );
-  const treeStreamFreshness = useMemo(
-    () => (input.treeStreamData ? getSidebarDataFreshness(input.treeStreamData) : Number.NEGATIVE_INFINITY),
-    [input.treeStreamData],
-  );
+  const streamIsPreferred = input.treeStreamStatus !== "fallback";
 
   const source = useMemo<PreferredSidebarSnapshotSource>(() => {
-    if (input.treeStreamData && input.sidebarQueryData) {
-      if (treeStreamSyncKey === querySyncKey) {
-        return "tree_stream";
-      }
-      return queryFreshness > treeStreamFreshness ? "query" : "tree_stream";
-    }
-    if (input.treeStreamData) {
+    if (input.treeStreamData && streamIsPreferred) {
       return "tree_stream";
     }
     if (input.sidebarQueryData) {
@@ -47,10 +32,7 @@ export function usePreferredSidebarSnapshot(input: {
   }, [
     input.sidebarQueryData,
     input.treeStreamData,
-    queryFreshness,
-    querySyncKey,
-    treeStreamFreshness,
-    treeStreamSyncKey,
+    streamIsPreferred,
   ]);
 
   const data =
@@ -61,7 +43,7 @@ export function usePreferredSidebarSnapshot(input: {
         : input.treeStreamData ?? input.sidebarQueryData ?? input.initialData;
   const syncKey =
     source === "tree_stream"
-      ? treeStreamSyncKey ?? initialSyncKey
+      ? treeStreamSyncKey ?? querySyncKey ?? initialSyncKey
       : source === "query"
         ? querySyncKey ?? initialSyncKey
         : initialSyncKey;