Software-Mac/spacedrive
Software-Mac/spacedrive
1
0
Fork 0
mirror of https://github.com/spacedriveapp/spacedrive synced 2024-07-05 08:03:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove TryInto usage in sync manager

Browse source
This commit is contained in:
Brendan Allan 2023-02-01 11:42:43 +08:00
parent 95ba2e1246
commit 58da1822e5
1 changed files with 31 additions and 52 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
core/src/sync/manager.rs
View file

@ -6,7 +6,7 @@ use crate::{
use sd_sync::*;
use futures::future::join_all;
use serde_json::{from_slice, from_value, json, to_vec, Value};
use serde_json::{from_value, json, to_vec, Value};
use std::{
collections::{HashMap, HashSet},
sync::Arc,
@ -147,69 +147,48 @@ impl SyncManager {
}
pub async fn get_ops(&self) -> prisma_client_rust::Result<Vec<CRDTOperation>> {
owned_operation::include!(owned_op_with_node { node });
let db = &self.db;
impl TryInto<CRDTOperation> for owned_op_with_node::Data {
type Error = ();
fn try_into(self) -> Result<CRDTOperation, Self::Error> {
let id = Uuid::from_slice(&self.id).map_err(|_| ())?;
let node = Uuid::from_slice(&self.node.pub_id).map_err(|_| ())?;
Ok(CRDTOperation {
id,
node,
timestamp: NTP64(self.timestamp as u64),
typ: CRDTOperationType::Owned(OwnedOperation {
model: self.model,
items: serde_json::from_slice(&self.data).unwrap(),
}),
})
}
}
shared_operation::include!(shared_op_with_node { node });
impl TryInto<CRDTOperation> for shared_op_with_node::Data {
type Error = ();
fn try_into(self) -> Result<CRDTOperation, Self::Error> {
let id = Uuid::from_slice(&self.id).map_err(|_| ())?;
let node = Uuid::from_slice(&self.node.pub_id).map_err(|_| ())?;
Ok(CRDTOperation {
id,
node,
timestamp: NTP64(self.timestamp as u64),
typ: CRDTOperationType::Shared(SharedOperation {
record_id: serde_json::from_slice(&self.record_id).unwrap(),
model: self.model,
data: from_slice(&self.data).unwrap(),
}),
})
}
}
let owned = self
.db
let owned = db
.owned_operation()
.find_many(vec![])
.include(owned_op_with_node::include())
.include(owned_operation::include!({ node }))
.exec()
.await?
.into_iter()
.map(TryInto::try_into);
let shared = self
.db
.flat_map(|op| {
Some(CRDTOperation {
id: Uuid::from_slice(&op.id).ok()?,
node: Uuid::from_slice(&op.node.pub_id).ok()?,
timestamp: NTP64(op.timestamp as u64),
typ: CRDTOperationType::Owned(OwnedOperation {
model: op.model,
items: serde_json::from_slice(&op.data).ok()?,
}),
})
});
let shared = db
.shared_operation()
.find_many(vec![])
.include(shared_op_with_node::include())
.include(shared_operation::include!({ node }))
.exec()
.await?
.into_iter()
.map(TryInto::try_into);
.flat_map(|op| {
Some(CRDTOperation {
id: Uuid::from_slice(&op.id).ok()?,
node: Uuid::from_slice(&op.node.pub_id).ok()?,
timestamp: NTP64(op.timestamp as u64),
typ: CRDTOperationType::Shared(SharedOperation {
record_id: serde_json::from_slice(&op.record_id).ok()?,
model: op.model,
data: serde_json::from_slice(&op.data).ok()?,
}),
})
});
let mut result: Vec<CRDTOperation> = owned.chain(shared).flatten().collect();
let mut result: Vec<CRDTOperation> = owned.chain(shared).collect();
result.sort_by(|a, b| a.timestamp.cmp(&b.timestamp));