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

use yew::prelude::*;

/// A hook that counts component renders.
///
/// # Example
///
/// ```rust
/// # use yew::prelude::*;
/// #
/// use yew_hooks::prelude::*;
///
/// #[function_component(Update)]
/// fn update() -> Html {
///     let count = use_renders_count();
///     let update = use_update();
///
///     let onclick = Callback::from(move |_| {
///         update();
///     });
///     
///     html! {
///         <>
///             <button {onclick}>{ "Update" }</button>
///             { count }
///         </>
///     }
/// }
/// ```
#[hook]
pub fn use_renders_count() -> i32 {
    let count = use_mut_ref(|| 0);

    let current = *count.borrow();
    *count.borrow_mut() = current + 1;
    let current = *count.borrow();

    current
}