reorganize bookmarks memory model

This commit is contained in:
yggverse 2025-03-08 18:48:21 +02:00
parent 6f91efbc9c
commit 6526ca85d8
6 changed files with 93 additions and 99 deletions

View File

@ -207,13 +207,13 @@ impl Menu for MenuButton {
move |_| { move |_| {
// Bookmarks // Bookmarks
main_bookmarks.remove_all(); main_bookmarks.remove_all();
for request in profile.bookmark.recent() { for bookmark in profile.bookmark.recent() {
let menu_item = gio::MenuItem::new(Some(&ellipsize(&request, LABEL_MAX_LENGTH)), None); let menu_item = gio::MenuItem::new(Some(&ellipsize(&bookmark.request, LABEL_MAX_LENGTH)), None);
menu_item.set_action_and_target_value(Some(&format!( menu_item.set_action_and_target_value(Some(&format!(
"{}.{}", "{}.{}",
window_action.id, window_action.id,
window_action.load.simple_action.name() window_action.load.simple_action.name()
)), Some(&request.to_variant())); )), Some(&bookmark.request.to_variant()));
main_bookmarks.append_item(&menu_item); main_bookmarks.append_item(&menu_item);
} // @TODO `menu_item` } // @TODO `menu_item`

View File

@ -48,7 +48,7 @@ impl Bookmark for Button {
} }
fn update(&self, profile: &Profile, request: &Entry) { fn update(&self, profile: &Profile, request: &Entry) {
let has_bookmark = profile.bookmark.get(&request.text()).is_some(); let has_bookmark = profile.bookmark.contains_request(&request.text());
self.set_icon_name(icon_name(has_bookmark)); self.set_icon_name(icon_name(has_bookmark));
self.set_tooltip_text(Some(tooltip_text(has_bookmark))); self.set_tooltip_text(Some(tooltip_text(has_bookmark)));
} }

View File

@ -1,16 +1,18 @@
mod database; mod database;
mod item;
mod memory; mod memory;
use anyhow::Result; use anyhow::Result;
use database::Database; use database::Database;
use gtk::glib::DateTime; use gtk::glib::DateTime;
use item::Item;
use memory::Memory; use memory::Memory;
use sqlite::{Connection, Transaction}; use sqlite::{Connection, Transaction};
use std::{rc::Rc, sync::RwLock}; use std::{cell::RefCell, rc::Rc, sync::RwLock};
pub struct Bookmark { pub struct Bookmark {
database: Database, // permanent storage database: Database, // permanent storage
memory: Memory, // fast search index memory: RefCell<Memory>, // fast search index
} }
impl Bookmark { impl Bookmark {
@ -20,11 +22,17 @@ impl Bookmark {
pub fn build(connection: &Rc<RwLock<Connection>>, profile_id: &Rc<i64>) -> Result<Self> { pub fn build(connection: &Rc<RwLock<Connection>>, profile_id: &Rc<i64>) -> Result<Self> {
// Init children components // Init children components
let database = Database::new(connection, profile_id); let database = Database::new(connection, profile_id);
let memory = Memory::new(); let memory = RefCell::new(Memory::new());
// Build initial index // Build initial index
for record in database.records(None)? { {
memory.add(record.request, record.id)?; let mut memory = memory.borrow_mut();
for record in database.records(None)? {
memory.add(Item {
id: record.id,
request: record.request,
});
}
} }
// Return new `Self` // Return new `Self`
@ -33,25 +41,20 @@ impl Bookmark {
// Actions // Actions
/// Get record `id` by `request` from memory index
pub fn get(&self, request: &str) -> Option<i64> {
self.memory.get(request)
}
/// Toggle bookmark in `database` and `memory` index /// Toggle bookmark in `database` and `memory` index
/// * return `true` on bookmark create, `false` on delete /// * return `true` on bookmark create, `false` on delete
pub fn toggle(&self, request: &str) -> Result<bool> { pub fn toggle(&self, request: &str) -> Result<bool> {
Ok(match self.get(request) { let mut memory = self.memory.borrow_mut();
Some(id) => { Ok(match memory.delete_by_request(request) {
self.database.delete(id)?; Some(record) => {
self.memory.delete(request)?; self.database.delete(record.id)?;
false false
} }
None => { None => {
self.memory.add( memory.add(Item {
request.into(), id: self.database.add(DateTime::now_local()?, request)?,
self.database.add(DateTime::now_local()?, request)?, request: request.into(),
)?; });
true true
} }
}) })
@ -59,9 +62,14 @@ impl Bookmark {
// Getters // Getters
/// Get recent requests vector from `memory`, sorted by `ID` DESC /// Check `request` exists in the memory index
pub fn recent(&self) -> Vec<String> { pub fn contains_request(&self, request: &str) -> bool {
self.memory.recent() self.memory.borrow_mut().contains_request(request)
}
/// Get recent Items vector from `memory`, sorted by `ID` DESC
pub fn recent(&self) -> Vec<Item> {
self.memory.borrow().recent()
} }
} }

View File

@ -1,15 +1,9 @@
use super::Item;
use anyhow::Result; use anyhow::Result;
use gtk::glib::DateTime; use gtk::glib::DateTime;
use sqlite::{Connection, Transaction}; use sqlite::{Connection, Transaction};
use std::{rc::Rc, sync::RwLock}; use std::{rc::Rc, sync::RwLock};
pub struct Table {
pub id: i64,
//pub profile_id: i64,
//pub time: DateTime,
pub request: String,
}
pub struct Database { pub struct Database {
connection: Rc<RwLock<Connection>>, connection: Rc<RwLock<Connection>>,
profile_id: Rc<i64>, // multi-profile relationship profile_id: Rc<i64>, // multi-profile relationship
@ -29,7 +23,7 @@ impl Database {
// Getters // Getters
/// Get bookmark records from database with optional filter by `request` /// Get bookmark records from database with optional filter by `request`
pub fn records(&self, request: Option<&str>) -> Result<Vec<Table>> { pub fn records(&self, request: Option<&str>) -> Result<Vec<Item>> {
let readable = self.connection.read().unwrap(); // @TODO let readable = self.connection.read().unwrap(); // @TODO
let tx = readable.unchecked_transaction()?; let tx = readable.unchecked_transaction()?;
select(&tx, *self.profile_id, request) select(&tx, *self.profile_id, request)
@ -86,7 +80,7 @@ pub fn insert(tx: &Transaction, profile_id: i64, time: DateTime, request: &str)
Ok(tx.last_insert_rowid()) Ok(tx.last_insert_rowid())
} }
pub fn select(tx: &Transaction, profile_id: i64, request: Option<&str>) -> Result<Vec<Table>> { pub fn select(tx: &Transaction, profile_id: i64, request: Option<&str>) -> Result<Vec<Item>> {
let mut stmt = tx.prepare( let mut stmt = tx.prepare(
"SELECT `id`, `profile_id`, `time`, `request` "SELECT `id`, `profile_id`, `time`, `request`
FROM `profile_bookmark` FROM `profile_bookmark`
@ -94,7 +88,7 @@ pub fn select(tx: &Transaction, profile_id: i64, request: Option<&str>) -> Resul
)?; )?;
let result = stmt.query_map((profile_id, request.unwrap_or("%")), |row| { let result = stmt.query_map((profile_id, request.unwrap_or("%")), |row| {
Ok(Table { Ok(Item {
id: row.get(0)?, id: row.get(0)?,
//profile_id: row.get(1)?, //profile_id: row.get(1)?,
//time: DateTime::from_unix_local(row.get(2)?).unwrap(), //time: DateTime::from_unix_local(row.get(2)?).unwrap(),

View File

@ -0,0 +1,5 @@
#[derive(Clone)]
pub struct Item {
pub id: i64,
pub request: String,
}

View File

@ -1,10 +1,56 @@
use anyhow::Result; use super::Item;
use itertools::Itertools; use itertools::Itertools;
use std::{cell::RefCell, collections::HashMap};
/// Reduce disk usage by cache Bookmarks index in memory /// Reduce disk usage by cache Bookmarks index in memory
pub struct Memory { pub struct Memory(Vec<Item>);
index: RefCell<HashMap<String, i64>>,
impl Memory {
// Constructors
/// Create new `Self`
pub fn new() -> Self {
Self(Vec::new())
}
// Actions
/// Add new item
pub fn add(&mut self, item: Item) {
self.0.push(item)
}
/// Delete record from index by `request`
pub fn delete_by_request(&mut self, request: &str) -> Option<Item> {
for (i, item) in self.0.iter().enumerate() {
if item.request == request {
return Some(self.0.remove(i));
}
}
None
}
/// Check `request` exists in the memory index
pub fn contains_request(&self, request: &str) -> bool {
for item in self.0.iter() {
if item.request == request {
return true;
}
}
false
}
/// Get recent Items vector sorted by `ID` DESC
pub fn recent(&self) -> Vec<Item> {
let mut recent: Vec<Item> = Vec::new();
for item in self
.0
.iter()
.sorted_by(|a, b| Ord::cmp(&b.request, &a.request))
{
recent.push(item.clone())
}
recent
}
} }
impl Default for Memory { impl Default for Memory {
@ -12,62 +58,3 @@ impl Default for Memory {
Self::new() Self::new()
} }
} }
impl Memory {
// Constructors
/// Create new `Self`
pub fn new() -> Self {
Self {
index: RefCell::new(HashMap::new()),
}
}
// Actions
/// Add new record with `request` as key and `id` as value
/// * validate record with same key does not exist yet
pub fn add(&self, request: String, id: i64) -> Result<()> {
// Borrow shared index access
let mut index = self.index.borrow_mut();
// Prevent existing key overwrite
if index.contains_key(&request) {
panic!() // unexpected
}
// Slot should be free, let check it twice
match index.insert(request, id) {
Some(_) => panic!(), // unexpected
None => Ok(()),
}
}
/// Delete record from index by `request`
/// * validate record key is exist
pub fn delete(&self, request: &str) -> Result<()> {
match self.index.borrow_mut().remove(request) {
Some(_) => Ok(()),
None => panic!(), // unexpected
}
}
/// Get `id` by `request` from memory index
pub fn get(&self, request: &str) -> Option<i64> {
self.index.borrow().get(request).copied()
}
/// Get recent requests vector sorted by `ID` DESC
pub fn recent(&self) -> Vec<String> {
let mut recent: Vec<String> = Vec::new();
for (request, _) in self
.index
.borrow()
.iter()
.sorted_by(|a, b| Ord::cmp(&b.1, &a.1))
{
recent.push(request.to_string())
}
recent
}
}