sabiql 1.9.0

A fast, driver-less TUI for browsing and editing PostgreSQL databases
Documentation 
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
111
112
113
114
115

use std::collections::VecDeque;
use std::sync::Arc;

use crate::domain::QueryResult;

#[derive(Debug, Clone)]
pub struct ResultHistory {
    entries: VecDeque<Arc<QueryResult>>,
    capacity: usize,
}

impl Default for ResultHistory {
    fn default() -> Self {
        Self::new(20)
    }
}

impl ResultHistory {
    pub fn new(capacity: usize) -> Self {
        Self {
            entries: VecDeque::with_capacity(capacity),
            capacity,
        }
    }

    pub fn push(&mut self, result: Arc<QueryResult>) {
        if self.entries.len() >= self.capacity {
            self.entries.pop_front();
        }
        self.entries.push_back(result);
    }

    pub fn get(&self, index: usize) -> Option<&QueryResult> {
        self.entries.get(index).map(|arc| &**arc)
    }

    pub fn len(&self) -> usize {
        self.entries.len()
    }

    pub fn is_empty(&self) -> bool {
        self.entries.is_empty()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::domain::QuerySource;

    fn make_result(query: &str) -> QueryResult {
        QueryResult::success(
            query.to_string(),
            vec!["col1".to_string()],
            vec![vec!["val1".to_string()]],
            10,
            QuerySource::Adhoc,
        )
    }

    #[test]
    fn push_and_get_returns_entries_in_order() {
        let mut history = ResultHistory::new(3);

        history.push(Arc::new(make_result("SELECT 1")));
        history.push(Arc::new(make_result("SELECT 2")));

        assert_eq!(history.get(0).unwrap().query, "SELECT 1");
        assert_eq!(history.get(1).unwrap().query, "SELECT 2");
        assert!(history.get(2).is_none());
    }

    #[test]
    fn len_returns_entry_count() {
        let mut history = ResultHistory::new(5);

        assert_eq!(history.len(), 0);

        history.push(Arc::new(make_result("SELECT 1")));
        assert_eq!(history.len(), 1);

        history.push(Arc::new(make_result("SELECT 2")));
        assert_eq!(history.len(), 2);
    }

    #[test]
    fn is_empty_returns_true_when_no_entries() {
        let history = ResultHistory::new(5);

        assert!(history.is_empty());
    }

    #[test]
    fn is_empty_returns_false_after_push() {
        let mut history = ResultHistory::new(5);

        history.push(Arc::new(make_result("SELECT 1")));

        assert!(!history.is_empty());
    }

    #[test]
    fn push_evicts_oldest_when_at_capacity() {
        let mut history = ResultHistory::new(2);

        history.push(Arc::new(make_result("SELECT 1")));
        history.push(Arc::new(make_result("SELECT 2")));
        history.push(Arc::new(make_result("SELECT 3")));

        // SELECT 1 should be evicted
        assert_eq!(history.get(0).unwrap().query, "SELECT 2");
        assert_eq!(history.get(1).unwrap().query, "SELECT 3");
        assert!(history.get(2).is_none());
    }
}