Function replace 

pub fn replace(budget: usize) -> BudgetGuard

Replace the current budget returning a guard that will restore the one which was in effect when it is dropped.