• DRY (不写重复代码)

    DRY (不写重复代码)

    通过提取函数或测试单元的公共部分，宏允许编写 DRY 代码（DRY 是 Don’t Repeat Yourself 的缩写，意思为“不要写重复代码”）。这里给出一个例子，实现并测试了关于 Vec<T> 的 +=、*= 和 -= 等运算符。

    1. use std::ops::{Add, Mul, Sub};
    3. macro_rules! assert_equal_len {
    4.     // `tt` （token tree，令牌树）指示符用于运算符和令牌。
    5.     // （原文：The `tt` (token tree) designator is used for
    6.     // operators and tokens.）
    7.     ($a:ident, $b: ident, $func:ident, $op:tt) => (
    8.         assert!($a.len() == $b.len(),
    9.                 "{:?}: dimension mismatch: {:?} {:?} {:?}",
    10.                 stringify!($func),
    11.                 ($a.len(),),
    12.                 stringify!($op),
    13.                 ($b.len(),));
    14.     )
    15. }
    17. macro_rules! op {
    18.     ($func:ident, $bound:ident, $op:tt, $method:ident) => (
    19.         fn $func<T: $bound<T, Output=T> + Copy>(xs: &mut Vec<T>, ys: &Vec<T>) {
    20.             assert_equal_len!(xs, ys, $func, $op);
    22.             for (x, y) in xs.iter_mut().zip(ys.iter()) {
    23.                 *x = $bound::$method(*x, *y);
    24.                 // *x = x.$method(*y);
    25.             }
    26.         }
    27.     )
    28. }
    30. // 实现 `add_assign`、`mul_assign` 和 `sub_assign` 等函数。
    31. op!(add_assign, Add, +=, add);
    32. op!(mul_assign, Mul, *=, mul);
    33. op!(sub_assign, Sub, -=, sub);
    35. mod test {
    36.     use std::iter;
    37.     macro_rules! test {
    38.         ($func: ident, $x:expr, $y:expr, $z:expr) => {
    39.             #[test]
    40.             fn $func() {
    41.                 for size in 0usize..10 {
    42.                     let mut x: Vec<_> = iter::repeat($x).take(size).collect();
    43.                     let y: Vec<_> = iter::repeat($y).take(size).collect();
    44.                     let z: Vec<_> = iter::repeat($z).take(size).collect();
    46.                     super::$func(&mut x, &y);
    48.                     assert_eq!(x, z);
    49.                 }
    50.             }
    51.         }
    52.     }
    54.     // 测试 `add_assign`、`mul_assign` 和 `sub_assign`
    55.     test!(add_assign, 1u32, 2u32, 3u32);
    56.     test!(mul_assign, 2u32, 3u32, 6u32);
    57.     test!(sub_assign, 3u32, 2u32, 1u32);
    58. }
    1. $ rustc --test dry.rs && ./dry
    2. running 3 tests
    3. test test::mul_assign ... ok
    4. test test::add_assign ... ok
    5. test test::sub_assign ... ok
    7. test result: ok. 3 passed; 0 failed; 0 ignored; 0 measured