mirror of
https://github.com/tauri-apps/plugins-workspace.git
synced 2026-08-27 19:50:24 +02:00
refactor(store)!: more reworks (#1860)
* refactor(store): more reworks * Enable auto save by default * Store to resource table by default * Remove share store * Clean up * Add close store * Add store function * Add lazy store * Add init to lazy store * refresh cache in example * Add get-or-create-store * Revert "Add get-or-create-store" This reverts commit7ffd769240. * try get first * Docs * Use absolute path for store * more docs * Allow js to use pre-stored (de)serialize functions * Fix js get and close store * Show case how to use pretty json * Update readme * Use store instead of `store_builder` in example * Build * Fix example * More docs for StoreBuilder::build * Add default (de)serialize fn * Use pretty json by default * Use `undefined` for empty value in get * Change files * Differentiate json null from no value for events * Add create or existing * Build * Rename inner store's inset method to set * Update readme * Apply suggestions from code review * Use close instead * Update breaking changes * Return result in resolve_store_path * Change to close_resource and take &self * Clean up * Apply suggestions from code review Co-authored-by: Amr Bashir <amr.bashir2015@gmail.com> * Remove unused pub(crate) * Update change file * Expose resolve_store_path * Remove with_store * Remove StoreInner from pub and expose is_empty * Fix wrong jsdoc param * Update readme * rename createOrExistingStore to createOrLoad * make api consistent with the JS implementation, add examples * fmt * reintroduce "get existing store" behavior for create_or_load * rename createOrLoad to newOrExisting * keep store lock throughout whole new_or_existing_inner * Remove load * Missed if load * Don't make StoreState public * Remove R: Runtime from Builder * rename newOrExisting to load, load to reload * update docs * rename missing reload fn * rename builder fn to build() * fix default permission * Fix description and create_new logic * Clippy * Update docs * Update docs * remove create_store command * remove close_store command since we extend from Resource * Revert "remove close_store command since we extend from Resource" This reverts commit4a29fc8990. * Reapply "remove close_store command since we extend from Resource" This reverts commit70a1830e7d. --------- Co-authored-by: Amr Bashir <amr.bashir2015@gmail.com> Co-authored-by: Lucas Nogueira <lucas@tauri.app>
This commit is contained in:
Tony
co-authored by
Amr Bashir
Lucas Nogueira
Amr Bashir
Lucas Nogueira
parent
cfd48b3b2e
commit
8c67d44aef
+292
-153
@@ -12,234 +12,392 @@
|
||||
)]
|
||||
|
||||
pub use error::{Error, Result};
|
||||
use log::warn;
|
||||
use serde::Serialize;
|
||||
use serde::{Deserialize, Serialize};
|
||||
pub use serde_json::Value as JsonValue;
|
||||
use std::{
|
||||
collections::HashMap,
|
||||
path::{Path, PathBuf},
|
||||
sync::{Mutex, Weak},
|
||||
sync::{Arc, Mutex},
|
||||
time::Duration,
|
||||
};
|
||||
pub use store::{Store, StoreBuilder, StoreInner};
|
||||
pub use store::{resolve_store_path, DeserializeFn, SerializeFn, Store, StoreBuilder};
|
||||
use tauri::{
|
||||
plugin::{self, TauriPlugin},
|
||||
AppHandle, Manager, ResourceId, RunEvent, Runtime, Webview,
|
||||
AppHandle, Manager, ResourceId, RunEvent, Runtime, State,
|
||||
};
|
||||
|
||||
mod error;
|
||||
mod store;
|
||||
|
||||
#[derive(Serialize, Clone)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct ChangePayload<'a> {
|
||||
path: &'a Path,
|
||||
resource_id: Option<u32>,
|
||||
key: &'a str,
|
||||
value: &'a JsonValue,
|
||||
value: Option<&'a JsonValue>,
|
||||
exists: bool,
|
||||
}
|
||||
|
||||
pub struct StoreCollection<R: Runtime> {
|
||||
stores: Mutex<HashMap<PathBuf, Weak<Mutex<StoreInner<R>>>>>,
|
||||
// frozen: bool,
|
||||
#[derive(Debug)]
|
||||
struct StoreState {
|
||||
stores: Arc<Mutex<HashMap<PathBuf, ResourceId>>>,
|
||||
serialize_fns: HashMap<String, SerializeFn>,
|
||||
deserialize_fns: HashMap<String, DeserializeFn>,
|
||||
default_serialize: SerializeFn,
|
||||
default_deserialize: DeserializeFn,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
#[serde(untagged)]
|
||||
enum AutoSave {
|
||||
DebounceDuration(u64),
|
||||
Bool(bool),
|
||||
}
|
||||
|
||||
fn builder<R: Runtime>(
|
||||
app: AppHandle<R>,
|
||||
store_state: State<'_, StoreState>,
|
||||
path: PathBuf,
|
||||
auto_save: Option<AutoSave>,
|
||||
serialize_fn_name: Option<String>,
|
||||
deserialize_fn_name: Option<String>,
|
||||
create_new: bool,
|
||||
) -> Result<StoreBuilder<R>> {
|
||||
let mut builder = app.store_builder(path);
|
||||
if let Some(auto_save) = auto_save {
|
||||
match auto_save {
|
||||
AutoSave::DebounceDuration(duration) => {
|
||||
builder = builder.auto_save(Duration::from_millis(duration));
|
||||
}
|
||||
AutoSave::Bool(false) => {
|
||||
builder = builder.disable_auto_save();
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
|
||||
if let Some(serialize_fn_name) = serialize_fn_name {
|
||||
let serialize_fn = store_state
|
||||
.serialize_fns
|
||||
.get(&serialize_fn_name)
|
||||
.ok_or_else(|| crate::Error::SerializeFunctionNotFound(serialize_fn_name))?;
|
||||
builder = builder.serialize(*serialize_fn);
|
||||
}
|
||||
|
||||
if let Some(deserialize_fn_name) = deserialize_fn_name {
|
||||
let deserialize_fn = store_state
|
||||
.deserialize_fns
|
||||
.get(&deserialize_fn_name)
|
||||
.ok_or_else(|| crate::Error::DeserializeFunctionNotFound(deserialize_fn_name))?;
|
||||
builder = builder.deserialize(*deserialize_fn);
|
||||
}
|
||||
|
||||
if create_new {
|
||||
builder = builder.create_new();
|
||||
}
|
||||
|
||||
Ok(builder)
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn create_store<R: Runtime>(
|
||||
async fn load<R: Runtime>(
|
||||
app: AppHandle<R>,
|
||||
webview: Webview<R>,
|
||||
store_state: State<'_, StoreState>,
|
||||
path: PathBuf,
|
||||
auto_save: Option<u64>,
|
||||
auto_save: Option<AutoSave>,
|
||||
serialize_fn_name: Option<String>,
|
||||
deserialize_fn_name: Option<String>,
|
||||
create_new: Option<bool>,
|
||||
) -> Result<ResourceId> {
|
||||
let mut builder = app.store_builder(path);
|
||||
if let Some(auto_save) = auto_save {
|
||||
builder = builder.auto_save(Duration::from_millis(auto_save));
|
||||
}
|
||||
let store = builder.build();
|
||||
Ok(webview.resources_table().add(store))
|
||||
let builder = builder(
|
||||
app,
|
||||
store_state,
|
||||
path,
|
||||
auto_save,
|
||||
serialize_fn_name,
|
||||
deserialize_fn_name,
|
||||
create_new.unwrap_or_default(),
|
||||
)?;
|
||||
let (_, rid) = builder.build_inner()?;
|
||||
Ok(rid)
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn get_store<R: Runtime>(
|
||||
app: AppHandle<R>,
|
||||
store_state: State<'_, StoreState>,
|
||||
path: PathBuf,
|
||||
) -> Result<Option<ResourceId>> {
|
||||
let stores = store_state.stores.lock().unwrap();
|
||||
Ok(stores.get(&resolve_store_path(&app, path)?).copied())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn set<R: Runtime>(
|
||||
webview: Webview<R>,
|
||||
app: AppHandle<R>,
|
||||
rid: ResourceId,
|
||||
key: String,
|
||||
value: JsonValue,
|
||||
) -> Result<()> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
store.set(key, value);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn get<R: Runtime>(
|
||||
webview: Webview<R>,
|
||||
app: AppHandle<R>,
|
||||
rid: ResourceId,
|
||||
key: String,
|
||||
) -> Result<Option<JsonValue>> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.get(key))
|
||||
) -> Result<(Option<JsonValue>, bool)> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
let value = store.get(key);
|
||||
let exists = value.is_some();
|
||||
Ok((value, exists))
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn has<R: Runtime>(webview: Webview<R>, rid: ResourceId, key: String) -> Result<bool> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn has<R: Runtime>(app: AppHandle<R>, rid: ResourceId, key: String) -> Result<bool> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.has(key))
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn delete<R: Runtime>(webview: Webview<R>, rid: ResourceId, key: String) -> Result<bool> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn delete<R: Runtime>(app: AppHandle<R>, rid: ResourceId, key: String) -> Result<bool> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.delete(key))
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn clear<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn clear<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
store.clear();
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn reset<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn reset<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
store.reset();
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn keys<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<Vec<String>> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn keys<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<Vec<String>> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.keys())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn values<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<Vec<JsonValue>> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn values<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<Vec<JsonValue>> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.values())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn entries<R: Runtime>(
|
||||
webview: Webview<R>,
|
||||
app: AppHandle<R>,
|
||||
rid: ResourceId,
|
||||
) -> Result<Vec<(String, JsonValue)>> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.entries())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn length<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<usize> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn length<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<usize> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
Ok(store.length())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn load<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
store.load()
|
||||
async fn reload<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
store.reload()
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
async fn save<R: Runtime>(webview: Webview<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = webview.resources_table().get::<Store<R>>(rid)?;
|
||||
async fn save<R: Runtime>(app: AppHandle<R>, rid: ResourceId) -> Result<()> {
|
||||
let store = app.resources_table().get::<Store<R>>(rid)?;
|
||||
store.save()
|
||||
}
|
||||
|
||||
pub trait StoreExt<R: Runtime> {
|
||||
fn store(&self, path: impl AsRef<Path>) -> Store<R>;
|
||||
/// Create a store or load an existing store with default settings at the given path.
|
||||
///
|
||||
/// If the store is already loaded, its instance is automatically returned.
|
||||
///
|
||||
/// # Examples
|
||||
///
|
||||
/// ```
|
||||
/// use tauri_plugin_store::StoreExt;
|
||||
///
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(tauri_plugin_store::Builder::default().build())
|
||||
/// .setup(|app| {
|
||||
/// let store = app.store("my-store")?;
|
||||
/// Ok(())
|
||||
/// });
|
||||
/// ```
|
||||
fn store(&self, path: impl AsRef<Path>) -> Result<Arc<Store<R>>>;
|
||||
/// Get a store builder.
|
||||
///
|
||||
/// The builder can be used to configure the store.
|
||||
/// To use the default settings see [`Self::store`].
|
||||
///
|
||||
/// # Examples
|
||||
///
|
||||
/// ```
|
||||
/// use tauri_plugin_store::StoreExt;
|
||||
/// use std::time::Duration;
|
||||
///
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(tauri_plugin_store::Builder::default().build())
|
||||
/// .setup(|app| {
|
||||
/// let store = app.store_builder("users.json").auto_save(Duration::from_secs(1)).build()?;
|
||||
/// Ok(())
|
||||
/// });
|
||||
/// ```
|
||||
fn store_builder(&self, path: impl AsRef<Path>) -> StoreBuilder<R>;
|
||||
/// Get a handle of an already loaded store.
|
||||
///
|
||||
/// If the store is not loaded or does not exist, it returns `None`.
|
||||
///
|
||||
/// Note that using this function can cause race conditions if you fallback to creating or loading the store,
|
||||
/// so you should consider using [`Self::store`] if you are not sure if the store is loaded or not.
|
||||
///
|
||||
/// # Examples
|
||||
///
|
||||
/// ```
|
||||
/// use tauri_plugin_store::StoreExt;
|
||||
///
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(tauri_plugin_store::Builder::default().build())
|
||||
/// .setup(|app| {
|
||||
/// let store = if let Some(s) = app.get_store("store.json") {
|
||||
/// s
|
||||
/// } else {
|
||||
/// // this is not thread safe; if another thread is doing the same load/create,
|
||||
/// // there will be a race condition; in this case we could remove the get_store
|
||||
/// // and only run app.store() as it will return the existing store if it has been loaded
|
||||
/// app.store("store.json")?
|
||||
/// };
|
||||
/// Ok(())
|
||||
/// });
|
||||
/// ```
|
||||
fn get_store(&self, path: impl AsRef<Path>) -> Option<Arc<Store<R>>>;
|
||||
}
|
||||
|
||||
impl<R: Runtime, T: Manager<R>> StoreExt<R> for T {
|
||||
fn store(&self, path: impl AsRef<Path>) -> Store<R> {
|
||||
fn store(&self, path: impl AsRef<Path>) -> Result<Arc<Store<R>>> {
|
||||
StoreBuilder::new(self.app_handle(), path).build()
|
||||
}
|
||||
|
||||
fn store_builder(&self, path: impl AsRef<Path>) -> StoreBuilder<R> {
|
||||
StoreBuilder::new(self.app_handle(), path)
|
||||
}
|
||||
|
||||
fn get_store(&self, path: impl AsRef<Path>) -> Option<Arc<Store<R>>> {
|
||||
let collection = self.state::<StoreState>();
|
||||
let stores = collection.stores.lock().unwrap();
|
||||
stores
|
||||
.get(&resolve_store_path(self.app_handle(), path.as_ref()).ok()?)
|
||||
.and_then(|rid| self.resources_table().get(*rid).ok())
|
||||
}
|
||||
}
|
||||
|
||||
// #[derive(Default)]
|
||||
pub struct Builder<R: Runtime> {
|
||||
stores: HashMap<PathBuf, Store<R>>,
|
||||
// frozen: bool,
|
||||
fn default_serialize(
|
||||
cache: &HashMap<String, JsonValue>,
|
||||
) -> std::result::Result<Vec<u8>, Box<dyn std::error::Error + Send + Sync>> {
|
||||
Ok(serde_json::to_vec_pretty(&cache)?)
|
||||
}
|
||||
|
||||
impl<R: Runtime> Default for Builder<R> {
|
||||
fn default_deserialize(
|
||||
bytes: &[u8],
|
||||
) -> std::result::Result<HashMap<String, JsonValue>, Box<dyn std::error::Error + Send + Sync>> {
|
||||
serde_json::from_slice(bytes).map_err(Into::into)
|
||||
}
|
||||
|
||||
pub struct Builder {
|
||||
serialize_fns: HashMap<String, SerializeFn>,
|
||||
deserialize_fns: HashMap<String, DeserializeFn>,
|
||||
default_serialize: SerializeFn,
|
||||
default_deserialize: DeserializeFn,
|
||||
}
|
||||
|
||||
impl Default for Builder {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
stores: Default::default(),
|
||||
// frozen: false,
|
||||
serialize_fns: Default::default(),
|
||||
deserialize_fns: Default::default(),
|
||||
default_serialize,
|
||||
default_deserialize,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<R: Runtime> Builder<R> {
|
||||
impl Builder {
|
||||
pub fn new() -> Self {
|
||||
Self::default()
|
||||
}
|
||||
|
||||
// /// Registers a store with the plugin.
|
||||
// ///
|
||||
// /// # Examples
|
||||
// ///
|
||||
// /// ```
|
||||
// /// use tauri_plugin_store::{StoreBuilder, Builder};
|
||||
// ///
|
||||
// /// tauri::Builder::default()
|
||||
// /// .setup(|app| {
|
||||
// /// let store = StoreBuilder::new("store.bin").build(app.handle().clone());
|
||||
// /// let builder = Builder::default().store(store);
|
||||
// /// Ok(())
|
||||
// /// });
|
||||
// /// ```
|
||||
// pub fn store(mut self, store: Store<R>) -> Self {
|
||||
// self.stores.insert(store.path.clone(), store);
|
||||
// self
|
||||
// }
|
||||
/// Register a serialize function to access it from the JavaScript side
|
||||
///
|
||||
/// # Examples
|
||||
///
|
||||
/// ```
|
||||
/// fn no_pretty_json(
|
||||
/// cache: &std::collections::HashMap<String, serde_json::Value>,
|
||||
/// ) -> Result<Vec<u8>, Box<dyn std::error::Error + Send + Sync>> {
|
||||
/// Ok(serde_json::to_vec(&cache)?)
|
||||
/// }
|
||||
///
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(
|
||||
/// tauri_plugin_store::Builder::default()
|
||||
/// .register_serialize_fn("no-pretty-json".to_owned(), no_pretty_json)
|
||||
/// .build(),
|
||||
/// );
|
||||
/// ```
|
||||
pub fn register_serialize_fn(mut self, name: String, serialize_fn: SerializeFn) -> Self {
|
||||
self.serialize_fns.insert(name, serialize_fn);
|
||||
self
|
||||
}
|
||||
|
||||
// /// Registers multiple stores with the plugin.
|
||||
// ///
|
||||
// /// # Examples
|
||||
// ///
|
||||
// /// ```
|
||||
// /// use tauri_plugin_store::{StoreBuilder, Builder};
|
||||
// ///
|
||||
// /// tauri::Builder::default()
|
||||
// /// .setup(|app| {
|
||||
// /// let store = StoreBuilder::new("store.bin").build(app.handle().clone());
|
||||
// /// let builder = Builder::default().stores([store]);
|
||||
// /// Ok(())
|
||||
// /// });
|
||||
// /// ```
|
||||
// pub fn stores<T: IntoIterator<Item = Store<R>>>(mut self, stores: T) -> Self {
|
||||
// self.stores = stores
|
||||
// .into_iter()
|
||||
// .map(|store| (store.path.clone(), store))
|
||||
// .collect();
|
||||
// self
|
||||
// }
|
||||
/// Register a deserialize function to access it from the JavaScript side
|
||||
pub fn register_deserialize_fn(mut self, name: String, deserialize_fn: DeserializeFn) -> Self {
|
||||
self.deserialize_fns.insert(name, deserialize_fn);
|
||||
self
|
||||
}
|
||||
|
||||
// /// Freezes the collection.
|
||||
// ///
|
||||
// /// This causes requests for plugins that haven't been registered to fail
|
||||
// ///
|
||||
// /// # Examples
|
||||
// ///
|
||||
// /// ```
|
||||
// /// use tauri_plugin_store::{StoreBuilder, Builder};
|
||||
// ///
|
||||
// /// tauri::Builder::default()
|
||||
// /// .setup(|app| {
|
||||
// /// let store = StoreBuilder::new("store.bin").build(app.handle().clone());
|
||||
// /// app.handle().plugin(Builder::default().freeze().build());
|
||||
// /// Ok(())
|
||||
// /// });
|
||||
// /// ```
|
||||
// pub fn freeze(mut self) -> Self {
|
||||
// self.frozen = true;
|
||||
// self
|
||||
// }
|
||||
/// Use this serialize function for stores by default
|
||||
///
|
||||
/// # Examples
|
||||
///
|
||||
/// ```
|
||||
/// fn no_pretty_json(
|
||||
/// cache: &std::collections::HashMap<String, serde_json::Value>,
|
||||
/// ) -> Result<Vec<u8>, Box<dyn std::error::Error + Send + Sync>> {
|
||||
/// Ok(serde_json::to_vec(&cache)?)
|
||||
/// }
|
||||
///
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(
|
||||
/// tauri_plugin_store::Builder::default()
|
||||
/// .default_serialize_fn(no_pretty_json)
|
||||
/// .build(),
|
||||
/// );
|
||||
/// ```
|
||||
pub fn default_serialize_fn(mut self, serialize_fn: SerializeFn) -> Self {
|
||||
self.default_serialize = serialize_fn;
|
||||
self
|
||||
}
|
||||
|
||||
/// Use this deserialize function for stores by default
|
||||
pub fn default_deserialize_fn(mut self, deserialize_fn: DeserializeFn) -> Self {
|
||||
self.default_deserialize = deserialize_fn;
|
||||
self
|
||||
}
|
||||
|
||||
/// Builds the plugin.
|
||||
///
|
||||
@@ -249,56 +407,37 @@ impl<R: Runtime> Builder<R> {
|
||||
/// tauri::Builder::default()
|
||||
/// .plugin(tauri_plugin_store::Builder::default().build())
|
||||
/// .setup(|app| {
|
||||
/// let store = tauri_plugin_store::StoreBuilder::new(app, "store.bin").build();
|
||||
/// let store = tauri_plugin_store::StoreBuilder::new(app, "store.bin").build()?;
|
||||
/// Ok(())
|
||||
/// });
|
||||
/// ```
|
||||
pub fn build(mut self) -> TauriPlugin<R> {
|
||||
pub fn build<R: Runtime>(self) -> TauriPlugin<R> {
|
||||
plugin::Builder::new("store")
|
||||
.invoke_handler(tauri::generate_handler![
|
||||
create_store,
|
||||
set,
|
||||
get,
|
||||
has,
|
||||
delete,
|
||||
clear,
|
||||
reset,
|
||||
keys,
|
||||
values,
|
||||
length,
|
||||
entries,
|
||||
load,
|
||||
save
|
||||
load, get_store, set, get, has, delete, clear, reset, keys, values, length,
|
||||
entries, reload, save,
|
||||
])
|
||||
.setup(move |app_handle, _api| {
|
||||
for (path, store) in self.stores.iter_mut() {
|
||||
// ignore loading errors, just use the default
|
||||
if let Err(err) = store.load() {
|
||||
warn!(
|
||||
"Failed to load store {path:?} from disk: {err}. Falling back to default values."
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
app_handle.manage(StoreCollection::<R> {
|
||||
stores: Mutex::new(HashMap::new()),
|
||||
// frozen: self.frozen,
|
||||
app_handle.manage(StoreState {
|
||||
stores: Arc::new(Mutex::new(HashMap::new())),
|
||||
serialize_fns: self.serialize_fns,
|
||||
deserialize_fns: self.deserialize_fns,
|
||||
default_serialize: self.default_serialize,
|
||||
default_deserialize: self.default_deserialize,
|
||||
});
|
||||
|
||||
Ok(())
|
||||
})
|
||||
.on_event(|_app_handle, event| {
|
||||
.on_event(|app_handle, event| {
|
||||
if let RunEvent::Exit = event {
|
||||
// let collection = app_handle.state::<StoreCollection<R>>();
|
||||
|
||||
// for store in collection.stores.lock().expect("mutex poisoned").values_mut() {
|
||||
// if let Some(sender) = store.auto_save_debounce_sender.take() {
|
||||
// let _ = sender.send(AutoSaveMessage::Cancel);
|
||||
// }
|
||||
// if let Err(err) = store.save() {
|
||||
// eprintln!("failed to save store {:?} with error {:?}", store.path, err);
|
||||
// }
|
||||
// }
|
||||
let collection = app_handle.state::<StoreState>();
|
||||
let stores = collection.stores.lock().unwrap();
|
||||
for (path, rid) in stores.iter() {
|
||||
if let Ok(store) = app_handle.resources_table().get::<Store<R>>(*rid) {
|
||||
if let Err(err) = store.save() {
|
||||
log::error!("failed to save store {path:?} with error {err:?}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.build()
|
||||
|
||||
Reference in New Issue
Block a user