Skip to main content

aedb/migration/
mod.rs

1use crate::commit::validation::Mutation;
2use crate::error::AedbError;
3use serde::{Deserialize, Serialize};
4use sha2::{Digest, Sha256};
5
6#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
7pub struct MigrationRecord {
8    pub version: u64,
9    pub name: String,
10    pub project_id: String,
11    pub scope_id: String,
12    pub applied_at_micros: u64,
13    pub applied_seq: u64,
14    pub checksum_hex: String,
15}
16
17#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
18pub struct Migration {
19    pub version: u64,
20    pub name: String,
21    pub project_id: String,
22    pub scope_id: String,
23    pub mutations: Vec<Mutation>,
24    #[serde(default)]
25    pub down_mutations: Option<Vec<Mutation>>,
26}
27
28pub fn migration_key(version: u64) -> Vec<u8> {
29    format!("__migrations/{version:020}").into_bytes()
30}
31
32pub fn encode_record(record: &MigrationRecord) -> Result<Vec<u8>, AedbError> {
33    serde_json::to_vec(record).map_err(|e| AedbError::Encode(e.to_string()))
34}
35
36pub fn decode_record(bytes: &[u8]) -> Result<MigrationRecord, AedbError> {
37    serde_json::from_slice(bytes).map_err(|e| AedbError::Decode(e.to_string()))
38}
39
40pub fn checksum_hex(migration: &Migration) -> Result<String, AedbError> {
41    let bytes = rmp_serde::to_vec(migration).map_err(|e| AedbError::Encode(e.to_string()))?;
42    let mut hasher = Sha256::new();
43    hasher.update(&bytes);
44    Ok(hex::encode(hasher.finalize()))
45}