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
extern crate proc_macro; use proc_macro::TokenStream; use quote::quote; #[proc_macro] pub fn process_lock(_item: TokenStream) -> TokenStream { let result = quote! { use huber_common::model::config::Config; use std::fs::File; use fs2::FileExt; use log::{error, info}; let lock_path = Config::new().lock_file().unwrap(); let f = if !lock_path.exists() { File::create(&lock_path) } else { File::open(&lock_path) }.unwrap(); let r = f.try_lock_exclusive(); match r { Ok(_) => { info!("{}: {:?}", "Locking the operation", lock_path); }, Err(e) => { error!("{:?}: {:?}", lock_path, e); return Err(anyhow!("huber is already running by another process for the exclusion operation. Please try after the operation finished. {:?}", e)) } } }; result.into() }