rt: simplify coop implementation (tokio-rs#2498)

Simplifies coop implementation. Prunes unused code, create a `Budget`
type to track the current budget.

Author: carllerche