Use budgeted poll macro

[emhane]

Describe the feature

Use existing pattern for budgeted poll of sub streams + remove magic number for budget in #11647

For example as has been done in the transactions manager

reth/crates/net/network/src/transactions/mod.rs

Lines 1257 to 1264 in 1ba631b

	let maybe_more_pool_imports = metered_poll_nested_stream_with_budget!(
	poll_durations.acc_pending_imports,
	"net::tx",
	"Batched pool imports stream",
	DEFAULT_BUDGET_TRY_DRAIN_PENDING_POOL_IMPORTS,
	this.pool_imports.poll_next_unpin(cx),
	|batch_results| this.on_batch_import_result(batch_results)
	);

A macro without use of metrics also exists

reth/crates/net/network/src/budget.rs

Lines 45 to 71 in 1ba631b

	/// Polls the given stream. Breaks with `true` if there maybe is more work.
	#[macro_export]
	macro_rules! poll_nested_stream_with_budget {
	($target:literal, $label:literal, $budget:ident, $poll_stream:expr, $on_ready_some:expr $(, $on_ready_none:expr;)? $(,)?) => {{
	let mut budget: u32 = $budget;
	loop {
	match $poll_stream {
	Poll::Ready(Some(item)) => {
	#[allow(unused_mut)]
	let mut f = $on_ready_some;
	f(item);
	budget = budget.saturating_sub(1);
	if budget == 0 {
	break true
	}
	}
	Poll::Ready(None) => {
	$($on_ready_none;)? // todo: handle error case with $target and $label
	break false
	}
	Poll::Pending => break false,
	}
	}
	}};
	}

Additional context

No response

[Rjected]

rather than a macro, could we extract the macro functionality to a struct somehow? macros can be hard to debug / make it harder to read and understand what's actually going on

[ayoubed]

I can take this.

[mattsse]

I think we have this everywhere where we can/want to use this