fix: usestate modify panic
@@ -39,6 +39,7 @@ pub fn use_state<'a, T: 'static>(
let setter = Rc::new({
crate::to_owned![update_callback, slot];
move |new| {
+ {
let mut slot = slot.borrow_mut();
// if there's only one reference (weak or otherwise), we can just swap the values
@@ -48,7 +49,7 @@ pub fn use_state<'a, T: 'static>(
} else {
*slot = Rc::new(new);
}
-
+ }
update_callback();
}
});
@@ -153,8 +154,10 @@ impl<T: 'static> UseState<T> {
/// }
/// ```
pub fn modify(&self, f: impl FnOnce(&T) -> T) {
+ let new_val = {
let current = self.slot.borrow();
- let new_val = f(current.as_ref());
+ f(current.as_ref())
+ };
(self.setter)(new_val);
}
