use crate::{ errors::{DocError, DocResult}, services::doc::revision::{RevisionCache, RevisionUploadStream}, }; use flowy_database::ConnectionPool; use flowy_document_infra::{entities::doc::Doc, util::RevIdCounter}; use lib_infra::future::ResultFuture; use lib_ot::{ core::OperationTransformable, revision::{RevId, RevType, Revision, RevisionRange}, rich_text::RichTextDelta, }; use std::sync::Arc; use tokio::sync::mpsc; pub trait RevisionServer: Send + Sync { fn fetch_document(&self, doc_id: &str) -> ResultFuture; } pub struct RevisionManager { doc_id: String, rev_id_counter: RevIdCounter, cache: Arc, } impl RevisionManager { pub fn new( doc_id: &str, pool: Arc, server: Arc, ws_sender: mpsc::UnboundedSender, ) -> Self { let cache = Arc::new(RevisionCache::new(doc_id, pool, server)); spawn_upload_stream(cache.clone(), ws_sender); let rev_id_counter = RevIdCounter::new(0); Self { doc_id: doc_id.to_string(), rev_id_counter, cache, } } pub async fn load_document(&mut self) -> DocResult { let doc = self.cache.load_document().await?; self.update_rev_id_counter_value(doc.rev_id); Ok(doc.delta()?) } pub async fn add_revision(&self, revision: &Revision) -> Result<(), DocError> { let _ = self.cache.add_revision(revision.clone()).await?; Ok(()) } pub async fn ack_revision(&self, rev_id: RevId) -> Result<(), DocError> { self.cache.ack_revision(rev_id).await; Ok(()) } pub fn rev_id(&self) -> i64 { self.rev_id_counter.value() } pub fn next_rev_id(&self) -> (i64, i64) { let cur = self.rev_id_counter.value(); let next = self.rev_id_counter.next(); (cur, next) } pub fn update_rev_id_counter_value(&self, rev_id: i64) { self.rev_id_counter.set(rev_id); } pub async fn mk_revisions(&self, range: RevisionRange) -> Result { debug_assert!(range.doc_id == self.doc_id); let revisions = self.cache.revisions_in_range(range.clone()).await?; let mut new_delta = RichTextDelta::new(); for revision in revisions { match RichTextDelta::from_bytes(revision.delta_data) { Ok(delta) => { new_delta = new_delta.compose(&delta)?; }, Err(e) => log::error!("{}", e), } } let delta_data = new_delta.to_bytes(); let revision = Revision::new( range.start, range.end, delta_data.to_vec(), &self.doc_id, RevType::Remote, ); Ok(revision) } } fn spawn_upload_stream(cache: Arc, ws_sender: mpsc::UnboundedSender) { tokio::spawn(RevisionUploadStream::new(cache, ws_sender).run()); }