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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use std::{cmp::Ordering, io, path::Path};

use anyhow::Result;
pub use idx_file::anyhow;

use idx_file::{Avltriee, Found, IdxFile, UOrd};
use various_data_file::VariousDataFile;

pub mod entity;
use entity::FieldEntity;

pub struct FieldData {
    pub(crate) index: IdxFile<FieldEntity>,
    pub(crate) data_file: VariousDataFile,
}

impl idx_file::RefIdxFile<FieldEntity> for FieldData {
    fn idx(&mut self) -> &mut IdxFile<FieldEntity> {
        &mut self.index
    }
}
impl UOrd<FieldEntity, &[u8]> for FieldData {
    fn triee(&self) -> &Avltriee<FieldEntity> {
        self.index.triee()
    }
    fn triee_mut(&mut self) -> &mut Avltriee<FieldEntity> {
        self.index.triee_mut()
    }
    fn cmp(&self, left: &FieldEntity, right: &&[u8]) -> Ordering {
        self.search(left, right)
    }

    fn search(&self, input: &&[u8]) -> Found {
        self.index
            .triee()
            .search_uord(|data| self.search(data, input))
    }

    fn value(&mut self, input: &&[u8]) -> Result<FieldEntity> {
        let data_address = self.data_file.insert(input)?;
        Ok(FieldEntity::new(
            data_address.address(),
            unsafe { std::str::from_utf8_unchecked(input) }
                .parse()
                .unwrap_or(0.0),
        ))
    }

    fn delete(&mut self, row: u32, delete_node: &FieldEntity) -> Result<()> {
        if !unsafe { self.index.triee().has_same(row) } {
            self.data_file.delete(&delete_node.data_address()).unwrap();
        }
        self.index.delete(row)?;
        Ok(())
    }
}

impl FieldData {
    pub fn new<P: AsRef<Path>>(path: P) -> io::Result<Self> {
        let path = path.as_ref();
        Ok(FieldData {
            index: IdxFile::new({
                let mut path = path.to_path_buf();
                path.push(".i");
                path
            })?,
            data_file: VariousDataFile::new({
                let mut path = path.to_path_buf();
                path.push(".d");
                path
            })?,
        })
    }
    pub fn get(&self, row: u32) -> Option<&'static [u8]> {
        if let Some(e) = self.index.value(row) {
            Some(unsafe { self.data_file.bytes(e.data_address()) })
        } else {
            None
        }
    }

    pub fn num(&self, row: u32) -> Option<f64> {
        if let Some(e) = self.index.value(row) {
            Some(e.num())
        } else {
            None
        }
    }
    pub fn update(&mut self, row: u32, content: &[u8]) -> Result<u32> {
        IdxFile::update_uord(self, row, content)
    }
    pub fn delete(&mut self, row: u32) -> std::io::Result<()> {
        self.index.delete(row)?;
        Ok(())
    }

    pub(crate) fn search(&self, data: &FieldEntity, content: &[u8]) -> Ordering {
        let left = unsafe { self.data_file.bytes(data.data_address()) };
        if left == content {
            Ordering::Equal
        } else {
            unsafe {
                natord::compare(
                    std::str::from_utf8_unchecked(left),
                    std::str::from_utf8_unchecked(content),
                )
            }
        }
    }
}