| 1 | /// Set if the task is scheduled for running. |
| 2 | /// |
| 3 | /// A task is considered to be scheduled whenever its `Runnable` exists. |
| 4 | /// |
| 5 | /// This flag can't be set when the task is completed. However, it can be set while the task is |
| 6 | /// running, in which case it will be rescheduled as soon as polling finishes. |
| 7 | pub(crate) const SCHEDULED: usize = 1 << 0; |
| 8 | |
| 9 | /// Set if the task is running. |
| 10 | /// |
| 11 | /// A task is in running state while its future is being polled. |
| 12 | /// |
| 13 | /// This flag can't be set when the task is completed. However, it can be in scheduled state while |
| 14 | /// it is running, in which case it will be rescheduled as soon as polling finishes. |
| 15 | pub(crate) const RUNNING: usize = 1 << 1; |
| 16 | |
| 17 | /// Set if the task has been completed. |
| 18 | /// |
| 19 | /// This flag is set when polling returns `Poll::Ready`. The output of the future is then stored |
| 20 | /// inside the task until it becomes closed. In fact, `Task` picks up the output by marking |
| 21 | /// the task as closed. |
| 22 | /// |
| 23 | /// This flag can't be set when the task is scheduled or running. |
| 24 | pub(crate) const COMPLETED: usize = 1 << 2; |
| 25 | |
| 26 | /// Set if the task is closed. |
| 27 | /// |
| 28 | /// If a task is closed, that means it's either canceled or its output has been consumed by the |
| 29 | /// `Task`. A task becomes closed in the following cases: |
| 30 | /// |
| 31 | /// 1. It gets canceled by `Runnable::drop()`, `Task::drop()`, or `Task::cancel()`. |
| 32 | /// 2. Its output gets awaited by the `Task`. |
| 33 | /// 3. It panics while polling the future. |
| 34 | /// 4. It is completed and the `Task` gets dropped. |
| 35 | pub(crate) const CLOSED: usize = 1 << 3; |
| 36 | |
| 37 | /// Set if the `Task` still exists. |
| 38 | /// |
| 39 | /// The `Task` is a special case in that it is only tracked by this flag, while all other |
| 40 | /// task references (`Runnable` and `Waker`s) are tracked by the reference count. |
| 41 | pub(crate) const TASK: usize = 1 << 4; |
| 42 | |
| 43 | /// Set if the `Task` is awaiting the output. |
| 44 | /// |
| 45 | /// This flag is set while there is a registered awaiter of type `Waker` inside the task. When the |
| 46 | /// task gets closed or completed, we need to wake the awaiter. This flag can be used as a fast |
| 47 | /// check that tells us if we need to wake anyone. |
| 48 | pub(crate) const AWAITER: usize = 1 << 5; |
| 49 | |
| 50 | /// Set if an awaiter is being registered. |
| 51 | /// |
| 52 | /// This flag is set when `Task` is polled and we are registering a new awaiter. |
| 53 | pub(crate) const REGISTERING: usize = 1 << 6; |
| 54 | |
| 55 | /// Set if the awaiter is being notified. |
| 56 | /// |
| 57 | /// This flag is set when notifying the awaiter. If an awaiter is concurrently registered and |
| 58 | /// notified, whichever side came first will take over the reposibility of resolving the race. |
| 59 | pub(crate) const NOTIFYING: usize = 1 << 7; |
| 60 | |
| 61 | /// A single reference. |
| 62 | /// |
| 63 | /// The lower bits in the state contain various flags representing the task state, while the upper |
| 64 | /// bits contain the reference count. The value of `REFERENCE` represents a single reference in the |
| 65 | /// total reference count. |
| 66 | /// |
| 67 | /// Note that the reference counter only tracks the `Runnable` and `Waker`s. The `Task` is |
| 68 | /// tracked separately by the `TASK` flag. |
| 69 | pub(crate) const REFERENCE: usize = 1 << 8; |
| 70 | |