From 4f184a439056d2b5dff7aa2fa1b1a73a1cbe9582 Mon Sep 17 00:00:00 2001 From: 魏曹先生 <1992414357@qq.com> Date: Sat, 7 Feb 2026 18:08:42 +0800 Subject: Add asset system with file locking and atomic writes --- systems/asset/macros/Cargo.toml | 13 +++++++++++++ systems/asset/macros/src/lib.rs | 43 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 56 insertions(+) create mode 100644 systems/asset/macros/Cargo.toml create mode 100644 systems/asset/macros/src/lib.rs (limited to 'systems/asset/macros') diff --git a/systems/asset/macros/Cargo.toml b/systems/asset/macros/Cargo.toml new file mode 100644 index 0000000..d4142f2 --- /dev/null +++ b/systems/asset/macros/Cargo.toml @@ -0,0 +1,13 @@ +[package] +name = "asset_macros" +version.workspace = true +edition = "2024" + +[lib] +proc-macro = true + +[dependencies] +string_proc = { path = "../../../utils/string_proc" } +syn = { version = "2.0", features = ["full", "extra-traits"] } +quote = "1.0" +proc-macro2 = "1.0" diff --git a/systems/asset/macros/src/lib.rs b/systems/asset/macros/src/lib.rs new file mode 100644 index 0000000..f186633 --- /dev/null +++ b/systems/asset/macros/src/lib.rs @@ -0,0 +1,43 @@ +use proc_macro::TokenStream; +use quote::quote; +use string_proc::snake_case; +use syn::parse_macro_input; + +#[proc_macro_derive(RWDataTest)] +pub fn rw_data_test_derive(input: TokenStream) -> TokenStream { + let input = parse_macro_input!(input as syn::DeriveInput); + let name = &input.ident; + let test_mod_name = syn::Ident::new( + &format!("{}_rw_test", snake_case!(name.to_string())), + name.span(), + ); + + let expanded = quote! { + #[cfg(test)] + mod #test_mod_name { + use super::*; + use asset::rw::RWData; + + #[tokio::test] + async fn test() { + let temp_dir = std::env::current_dir().unwrap().join(".temp/"); + tokio::fs::create_dir_all(&temp_dir) + .await + .expect("Create dir failed!"); + let temp_file = temp_dir.join(stringify!(#name)); + + #name::write(#name::test_data(), &temp_file) + .await + .expect("Write test data failed!"); + + let read_data = #name::read(&temp_file) + .await + .expect("Re-read data failed!"); + + assert!(#name::verify_data(#name::test_data(), read_data)); + } + } + }; + + TokenStream::from(expanded) +} -- cgit