aboutsummaryrefslogtreecommitdiff
path: root/database/src/sort/none.rs
blob: 92859982ef7fe57e7635f53f97b40f07866ea773 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2026 metamuffin <metamuffin.org>
*/

use crate::{
    backends::WriteTransaction,
    filter::binning::Binning,
    sort::Index,
    table::{RowNum, TableNum},
};
use anyhow::Result;
use jellyobject::Object;

pub struct UnsortedIndex {
    id: TableNum,
    binning: Binning,
}

impl UnsortedIndex {
    pub fn new(id: TableNum, binning: Binning) -> Self {
        Self { id, binning }
    }
    fn keys(&self, id: RowNum, ob: Object) -> Vec<Vec<u8>> {
        let mut keys = vec![self.id.to_be_bytes().to_vec()];
        self.binning.apply(ob, &mut keys);
        for k in &mut keys {
            k.extend(id.to_ne_bytes());
        }
        keys
    }
}
impl Index for UnsortedIndex {
    fn add(&self, db: &mut dyn WriteTransaction, id: RowNum, ob: Object) -> Result<()> {
        for key in self.keys(id, ob) {
            db.set(&key, &[])?;
        }
        Ok(())
    }
    fn remove(&self, db: &mut dyn WriteTransaction, id: RowNum, ob: Object) -> Result<()> {
        for key in self.keys(id, ob) {
            db.del(&key)?;
        }
        Ok(())
    }
    fn compare(&self, before: Object, after: Object) -> bool {
        self.keys(0, before) == self.keys(0, after)
    }
}