summaryrefslogtreecommitdiffstats
path: root/third_party/rust/writeable/examples/writeable_message.rs
blob: b342fdb261816acd54a3298e789c502be2b4c88d (plain)
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
// This file is part of ICU4X. For terms of use, please see the file
// called LICENSE at the top level of the ICU4X source tree
// (online at: https://github.com/unicode-org/icu4x/blob/main/LICENSE ).

// This example illustrates a very simple type implementing Writeable.
icu_benchmark_macros::static_setup!();

use std::fmt;
use writeable::*;

struct WriteableMessage<W: Writeable>(W);

const GREETING: Part = Part {
    category: "meaning",
    value: "greeting",
};

const EMOJI: Part = Part {
    category: "meaning",
    value: "emoji",
};

impl<V: Writeable> Writeable for WriteableMessage<V> {
    fn write_to_parts<W: PartsWrite + ?Sized>(&self, sink: &mut W) -> fmt::Result {
        use fmt::Write;
        sink.with_part(GREETING, |g| {
            g.write_str("Hello")?;
            g.write_str(" ")?;
            self.0.write_to(g)
        })?;
        sink.write_char(' ')?;
        sink.with_part(EMOJI, |e| e.write_char('😅'))
    }

    fn writeable_length_hint(&self) -> LengthHint {
        LengthHint::exact(11) + self.0.writeable_length_hint()
    }
}

impl<V: Writeable> fmt::Display for WriteableMessage<V> {
    #[inline]
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.write_to(f)
    }
}

fn main() {
    icu_benchmark_macros::main_setup!();

    let (string, parts) = writeable_to_parts_for_test(&WriteableMessage("world")).unwrap();

    assert_eq!(string, "Hello world 😅");

    // Print the greeting only
    let (start, end, _) = parts
        .into_iter()
        .find(|(_, _, part)| part == &GREETING)
        .unwrap();
    println!("{}", &string[start..end]);
}