Go from per document to per vault version ids to allow "since" queries

This commit is contained in:
Andras Schmelczer 2024-12-14 17:02:44 +00:00
parent fe8d236948
commit 70471f5142
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
6 changed files with 127 additions and 53 deletions

View file

@ -1,4 +1,4 @@
use anyhow::{anyhow, Context};
use anyhow::Context;
use axum::{
extract::{Path, State},
Json,
@ -14,7 +14,7 @@ use super::{auth::auth, requests::DeleteDocumentVersion};
use crate::{
app_state::AppState,
database::models::{StoredDocumentVersion, VaultId},
errors::{not_found_error, server_error, SyncServerError},
errors::{server_error, SyncServerError},
};
// This is required for aide to infer the path parameter types and names
@ -42,23 +42,16 @@ pub async fn delete_document(
.await
.map_err(server_error)?;
let latest_version = state
let last_update_id = state
.database
.get_latest_document(&vault_id, &relative_path, Some(&mut transaction))
.get_max_update_id_in_vault(&vault_id, Some(&mut transaction))
.await
.map_err(server_error)?
.map(Ok)
.unwrap_or_else(|| {
Err(not_found_error(anyhow!(
"Latest document version of document `{}` not found",
relative_path
)))
})?;
.map_err(server_error)?;
let new_version = StoredDocumentVersion {
vault_id,
vault_update_id: last_update_id + 1,
relative_path,
version_id: latest_version.version_id + 1,
content: vec![],
created_date: request.created_date,
updated_date: chrono::Utc::now(),

View file

@ -2,12 +2,12 @@ use chrono::{DateTime, Utc};
use schemars::JsonSchema;
use serde::{self, Deserialize};
use crate::database::models::DocumentVersionId;
use crate::database::models::VaultUpdateId;
#[derive(Debug, Deserialize, JsonSchema)]
#[serde(rename_all = "camelCase")]
pub struct UpdateDocumentVersion {
pub parent_version_id: Option<DocumentVersionId>,
pub parent_version_id: Option<VaultUpdateId>,
pub created_date: DateTime<Utc>,
pub content_base64: String,
}

View file

@ -7,6 +7,7 @@ use axum_extra::{
headers::{authorization::Bearer, Authorization},
TypedHeader,
};
use log::info;
use schemars::JsonSchema;
use serde::Deserialize;
use sync_lib::{base64_to_bytes, merge};
@ -36,10 +37,11 @@ pub async fn update_document(
Json(request): Json<UpdateDocumentVersion>,
) -> Result<Json<DocumentVersion>, SyncServerError> {
auth(&state, auth_header.token())?;
let parent_content = if let Some(parent_version_id) = request.parent_version_id {
state
.database
.get_document_version(&vault_id, &relative_path, &parent_version_id, None)
.get_document_version(&vault_id, parent_version_id, None)
.await
.map_err(server_error)?
.map(Ok)
@ -55,26 +57,59 @@ pub async fn update_document(
Ok(Vec::default())
}?;
let content_bytes = base64_to_bytes(&request.content_base64)
.context("Failed to decode base64 content in request")
.map_err(client_error)?;
let mut transaction = state
.database
.create_transaction()
.await
.map_err(server_error)?;
let mut last_update_id = state
.database
.get_max_update_id_in_vault(&vault_id, Some(&mut transaction))
.await
.map_err(server_error)?;
let latest_version = state
.database
.get_latest_document(&vault_id, &relative_path, Some(&mut transaction))
.await
.map_err(server_error)?;
let content_bytes = base64_to_bytes(&request.content_base64)
.context("Failed to decode base64 content in request")
.map_err(client_error)?;
if let Some(latest) = latest_version.as_ref() {
if content_bytes == latest.content && relative_path == latest.relative_path {
info!("Document content is the same as the latest version, skipping update");
transaction
.rollback()
.await
.context("Failed to rollback transaction")
.map_err(server_error)?;
return Ok(Json(latest.clone().into()));
} else if relative_path != latest.relative_path {
let delete_at_previous_path = StoredDocumentVersion {
vault_id: vault_id.clone(),
vault_update_id: last_update_id + 1,
relative_path: latest.relative_path.clone(),
content: vec![],
created_date: request.created_date,
updated_date: chrono::Utc::now(),
is_deleted: true,
};
last_update_id += 1;
state
.database
.insert_document_version(&delete_at_previous_path, Some(&mut transaction))
.await
.map_err(server_error)?;
}
}
let next_version = latest_version
.as_ref()
.map(|v| v.version_id + 1)
.unwrap_or(0);
let latest_version_content = latest_version
.map(|v| v.content)
.unwrap_or_else(Vec::default);
@ -85,8 +120,8 @@ pub async fn update_document(
let new_version = StoredDocumentVersion {
vault_id,
vault_update_id: last_update_id + 1,
relative_path,
version_id: next_version,
content: merged_content,
created_date: request.created_date,
updated_date: chrono::Utc::now(),