feat: support force sync by click (#4089)

Co-authored-by: JimmFly <yangjinfei001@gmail.com>
2026-07-23 13:29:02 +08:00 · 2023-09-01 01:15:07 -05:00
parent 83e7e9db8d
commit 92f0b31196
13 changed files with 341 additions and 117 deletions
@@ -0,0 +1,88 @@
+import { pushNotificationAtom } from '@affine/component/notification-center';
+import type {
+  AffineSocketIOProvider,
+  LocalIndexedDBBackgroundProvider,
+  SQLiteProvider,
+} from '@affine/env/workspace';
+import { type Status, syncDataSource } from '@affine/y-provider';
+import { assertExists } from '@blocksuite/global/utils';
+import type { Workspace } from '@blocksuite/store';
+import { useSetAtom } from 'jotai';
+import { startTransition, useCallback, useMemo, useState } from 'react';
+
+export function useDatasourceSync(workspace: Workspace) {
+  const [status, setStatus] = useState<Status>({
+    type: 'idle',
+  });
+  const pushNotification = useSetAtom(pushNotificationAtom);
+  const providers = workspace.providers;
+  const remoteProvider: AffineSocketIOProvider | undefined = useMemo(() => {
+    return providers.find(
+      (provider): provider is AffineSocketIOProvider =>
+        provider.flavour === 'affine-socket-io'
+    );
+  }, [providers]);
+  const localProvider = useMemo(() => {
+    const sqliteProvider = providers.find(
+      (provider): provider is SQLiteProvider => provider.flavour === 'sqlite'
+    );
+    const indexedDbProvider = providers.find(
+      (provider): provider is LocalIndexedDBBackgroundProvider =>
+        provider.flavour === 'local-indexeddb-background'
+    );
+    const provider = sqliteProvider || indexedDbProvider;
+    assertExists(provider, 'no local provider');
+    return provider;
+  }, [providers]);
+  return [
+    status,
+    useCallback(() => {
+      if (!remoteProvider) {
+        return;
+      }
+      startTransition(() => {
+        setStatus({
+          type: 'syncing',
+        });
+      });
+      syncDataSource(
+        () => [
+          workspace.doc.guid,
+          ...[...workspace.doc.subdocs].map(doc => doc.guid),
+        ],
+        remoteProvider.datasource,
+        localProvider.datasource
+      )
+        .then(() => {
+          startTransition(() => {
+            setStatus({
+              type: 'synced',
+            });
+            pushNotification({
+              title: 'Synced successfully',
+              type: 'success',
+            });
+          });
+        })
+        .catch(error => {
+          startTransition(() => {
+            setStatus({
+              type: 'error',
+              error,
+            });
+            pushNotification({
+              title: 'Unable to Sync',
+              message: 'Server error, please try again later.',
+              type: 'error',
+            });
+          });
+        });
+    }, [
+      remoteProvider,
+      localProvider.datasource,
+      workspace.doc.guid,
+      workspace.doc.subdocs,
+      pushNotification,
+    ]),
+  ] as const;
+}