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

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

use anyhow::Result;
pub use idx_sized::anyhow;

use idx_sized::{Found, IdxSized, Removed};
use various_data_file::VariousDataFile;

pub mod entity;
use entity::FieldEntity;

pub struct FieldData {
    pub(crate) index: IdxSized<FieldEntity>,
    data_file: VariousDataFile,
}
impl FieldData {
    pub fn new<P: AsRef<Path>>(path: P) -> io::Result<Self> {
        let path = path.as_ref();
        Ok(FieldData {
            index: IdxSized::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> {
        if let Some(org) = self.index.value(row) {
            if unsafe { self.data_file.bytes(org.data_address()) } == content {
                return Ok(row);
            }
            if let Removed::Last(data) = self.index.delete(row)? {
                self.data_file.remove(&data.data_address())?;
            }
        }
        let found = self.search(content);
        self.index.update_manually(
            row,
            || -> Result<FieldEntity> {
                let data_address = self.data_file.insert(content)?;
                Ok(FieldEntity::new(
                    data_address.address(),
                    unsafe { std::str::from_utf8_unchecked(content) }
                        .parse()
                        .unwrap_or(0.0),
                ))
            },
            found,
        )
    }
    pub fn delete(&mut self, row: u32) -> std::io::Result<()> {
        self.index.delete(row)?;
        Ok(())
    }

    pub(super) fn search(&self, content: &[u8]) -> Found {
        self.index.triee().search_cb(|data| -> Ordering {
            let bytes = unsafe { self.data_file.bytes(data.data_address()) };
            if content == bytes {
                Ordering::Equal
            } else {
                natord::compare(unsafe { std::str::from_utf8_unchecked(content) }, unsafe {
                    std::str::from_utf8_unchecked(bytes)
                })
            }
        })
    }
}