feat: 添加持久化 Rust Sync 上传队列与 HTTP 传输

2026-09-08 15:25:49 +08:00
parent e9df45b7ca
commit 2bc9ef84b9
8 changed files with 851 additions and 7 deletions
@@ -0,0 +1,139 @@
+#![cfg(feature = "desktop")]
+use notesagent_host::{sync_client::SyncClient, workspace::Workspace};
+use serde_json::{json, Value};
+use std::{
+    io::{BufRead, BufReader},
+    path::Path,
+    process::{Child, Command, Stdio},
+    sync::{Arc, Mutex},
+};
+use zeroize::Zeroizing;
+struct Server(Child);
+impl Drop for Server {
+    fn drop(&mut self) {
+        self.0.stdin.take();
+        let _ = self.0.kill();
+        let _ = self.0.wait();
+    }
+}
+
+#[tokio::test]
+async fn actual_service_accepts_ordered_push_and_repeat_commit_without_duplicates() {
+    let root = tempfile::tempdir().unwrap();
+    std::fs::write(root.path().join(".opennexus-test"), b"fixture").unwrap();
+    let service = Path::new(env!("CARGO_MANIFEST_DIR"))
+        .join("../../server sync")
+        .canonicalize()
+        .unwrap();
+    let python = service.join(if cfg!(windows) {
+        ".venv/Scripts/python.exe"
+    } else {
+        ".venv/bin/python"
+    });
+    let mut server = Server(
+        Command::new(python)
+            .args(["-m", "tests.host_fixture"])
+            .arg(root.path())
+            .current_dir(service)
+            .stdin(Stdio::piped())
+            .stdout(Stdio::piped())
+            .stderr(Stdio::null())
+            .spawn()
+            .unwrap(),
+    );
+    let mut line = String::new();
+    BufReader::new(server.0.stdout.take().unwrap())
+        .read_line(&mut line)
+        .unwrap();
+    let ready: Value = serde_json::from_str(&line).unwrap();
+    let endpoint = format!("http://127.0.0.1:{}", ready["port"]);
+    let public = SyncClient::new(&endpoint, Zeroizing::new(String::new()), true).unwrap();
+    public.handshake().await.unwrap();
+    let session = public
+        .login(
+            "rust-fixture",
+            Zeroizing::new("controlled-fixture-password".into()),
+            "Rust integration",
+        )
+        .await
+        .unwrap();
+    let client = SyncClient::new(
+        &endpoint,
+        Zeroizing::new(session.access_token.clone()),
+        true,
+    )
+    .unwrap();
+    let vault = client
+        .json(
+            reqwest::Method::POST,
+            "sync/v1/vaults",
+            Some(json!({"name":"Rust test"})),
+        )
+        .await
+        .unwrap();
+    let remote = vault["vault_id"].as_str().unwrap();
+    client.verify_empty(remote).await.unwrap();
+    let local = tempfile::tempdir().unwrap();
+    let workspace = Arc::new(Mutex::new(Workspace::open(local.path()).unwrap()));
+    let binding = {
+        let mut ws = workspace.lock().unwrap();
+        let mut digest = String::new();
+        for index in 0..20 {
+            digest = ws
+                .write(
+                    "note.md",
+                    &digest,
+                    format!("fixture-{index}").as_bytes(),
+                    "local",
+                )
+                .unwrap()
+                .hash;
+        }
+        ws.sync_bind_empty(&endpoint, remote, "rust-fixture")
+            .unwrap()
+    };
+    let first = workspace
+        .lock()
+        .unwrap()
+        .sync_next(&binding.id)
+        .unwrap()
+        .unwrap();
+    assert!(client.push_one(&workspace, &binding).await.unwrap());
+    let payload = workspace
+        .lock()
+        .unwrap()
+        .sync_commit_payload(&first)
+        .unwrap();
+    for _ in 0..100 {
+        let replay = client
+            .json(
+                reqwest::Method::POST,
+                &format!("sync/v1/vaults/{remote}/revisions"),
+                Some(payload.clone()),
+            )
+            .await
+            .unwrap();
+        assert_eq!(replay["sequence"], 1);
+    }
+    for _ in 1..20 {
+        assert!(client.push_one(&workspace, &binding).await.unwrap());
+    }
+    assert!(!client.push_one(&workspace, &binding).await.unwrap());
+    let changes = client
+        .json(
+            reqwest::Method::GET,
+            &format!("sync/v1/vaults/{remote}/changes"),
+            None,
+        )
+        .await
+        .unwrap();
+    assert_eq!(changes["items"].as_array().unwrap().len(), 20);
+    for (index, revision) in changes["items"].as_array().unwrap().iter().enumerate() {
+        assert_eq!(revision["base_revision"], index as i64);
+        assert_eq!(revision["sequence"], index as i64 + 1);
+    }
+    assert_eq!(
+        client.verify_empty(remote).await.unwrap_err().code,
+        "SYNC_RECONCILIATION_REQUIRED"
+    );
+}