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
71
72
73
74
75
76
77
78
//! Request hook methods for `WireframeClientBuilder`.
use std::sync::Arc;
use super::WireframeClientBuilder;
use crate::serializer::Serializer;
impl<S, P, C> WireframeClientBuilder<S, P, C>
where
S: Serializer + Send + Sync,
{
/// Register a hook invoked after serialization, before each frame is sent.
///
/// Multiple hooks may be registered; they execute in registration order.
/// Each hook receives a mutable reference to the serialized bytes, allowing
/// inspection or modification (e.g., prepending an authentication token,
/// incrementing a metrics counter).
///
/// # Examples
///
/// ```
/// use std::sync::{
/// Arc,
/// atomic::{AtomicUsize, Ordering},
/// };
///
/// use wireframe::client::WireframeClientBuilder;
///
/// let counter = Arc::new(AtomicUsize::new(0));
/// let count = counter.clone();
/// let builder = WireframeClientBuilder::new().before_send(move |_bytes: &mut Vec<u8>| {
/// count.fetch_add(1, Ordering::Relaxed);
/// });
/// let _ = builder;
/// ```
#[must_use]
pub fn before_send<F>(mut self, f: F) -> Self
where
F: Fn(&mut Vec<u8>) + Send + Sync + 'static,
{
self.request_hooks.before_send.push(Arc::new(f));
self
}
/// Register a hook invoked after each frame is read, before
/// deserialization.
///
/// Multiple hooks may be registered; they execute in registration order.
/// Each hook receives a mutable reference to the raw frame bytes, allowing
/// inspection or modification before the deserializer processes them.
///
/// # Examples
///
/// ```
/// use std::sync::{
/// Arc,
/// atomic::{AtomicUsize, Ordering},
/// };
///
/// use wireframe::client::WireframeClientBuilder;
///
/// let counter = Arc::new(AtomicUsize::new(0));
/// let count = counter.clone();
/// let builder =
/// WireframeClientBuilder::new().after_receive(move |_bytes: &mut bytes::BytesMut| {
/// count.fetch_add(1, Ordering::Relaxed);
/// });
/// let _ = builder;
/// ```
#[must_use]
pub fn after_receive<F>(mut self, f: F) -> Self
where
F: Fn(&mut bytes::BytesMut) + Send + Sync + 'static,
{
self.request_hooks.after_receive.push(Arc::new(f));
self
}
}