drop_code 1.0.1

A macro that allows you to implement code that will be automatically executed after the function code has finished, be it the end of the function or even a panic state.
Documentation 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use drop_code::drop_code;
use std::sync::{Mutex, OnceLock};

#[test]
fn easy_use_emptyargs() {
	static TEST_CINTERNAL: Mutex<u8> = Mutex::new(0);

	fn __tinternal(a: usize, b: usize) -> bool {
		drop_code! {
			let mut lock = TEST_CINTERNAL.lock().unwrap();
			*lock += 1;
		}
		if a == b {
			// autorun drop code
			return true;
		}

		// autorun drop code
		false
	}

	assert_eq!(*TEST_CINTERNAL.lock().unwrap(), 0);
	let rin = __tinternal(1, 1);
	assert!(rin);
	assert_eq!(*TEST_CINTERNAL.lock().unwrap(), 1);
}

#[test]
fn easy_use_twoargs() {
	static OLD_A: OnceLock<usize> = OnceLock::new();
	static OLD_B: OnceLock<usize> = OnceLock::new();

	fn __tinternal(a: usize, b: usize) -> bool {
		drop_code!((a: usize, b: usize) {
			OLD_A.set(*a).unwrap();
			OLD_B.set(*b).unwrap();
		});
		if a == b {
			// autorun drop code
			return true;
		}

		if a == 1 {
			// autorun drop code
			return true;
		}

		// autorun drop code
		false
	}

	assert_eq!(OLD_A.get(), None);
	assert_eq!(OLD_B.get(), None);
	let rin = __tinternal(1, 1);
	assert!(rin);
	assert_eq!(OLD_A.get(), Some(&1));
	assert_eq!(OLD_B.get(), Some(&1));
}