mirror of
https://github.com/YGGverse/Yoda.git
synced 2025-09-03 10:22:21 +00:00
reorganize bookmarks memory model
This commit is contained in:
parent
6f91efbc9c
commit
6526ca85d8
@ -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`
|
||||||
|
@ -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)));
|
||||||
}
|
}
|
||||||
|
@ -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()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -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(),
|
||||||
|
5
src/profile/bookmark/item.rs
Normal file
5
src/profile/bookmark/item.rs
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Item {
|
||||||
|
pub id: i64,
|
||||||
|
pub request: String,
|
||||||
|
}
|
@ -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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user