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
59
60
61
62
63
64
65
66
67
68
69
70
extern crate bytes;
#[macro_export]
macro_rules! templatify {
( $head_template:expr $(;$key:expr; $template:expr)* ) => {
{
let mut total_length = 0;
total_length += $head_template.len();
$(
total_length = total_length + $key.len() + $template.len();
)*
let mut output_string = String::with_capacity(total_length);
output_string.push_str($head_template);
$(
output_string.push_str($key);
output_string.push_str($template);
)*
output_string
}
}
}
#[macro_export]
macro_rules! templatify_buffer {
( $buffer:ident, $head_template:expr $(;$key:expr; $template:expr)* ) => {
{
let mut total_length = 0;
total_length += $head_template.len();
$(
total_length = total_length + $key.len() + $template.len();
)*
$buffer.reserve(total_length);
$buffer.put($head_template);
$(
$buffer.put($key);
$buffer.put($template);
)*
}
}
}
#[cfg(test)]
mod tests {
use bytes::{BytesMut, BufMut};
#[test]
fn templatify_should_work() {
let world = "world";
let results: String = templatify! { "hello, "; world ;"!" };
assert!(results == "hello, world!");
}
#[test]
fn templatify_buffer_should_work() {
let mut buf = BytesMut::new();
let world = "world";
templatify_buffer! { buf, "hello, "; world ;"!" };
assert!(buf == "hello, world!");
}
}