1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
// This file is part of Gear.
// Copyright (C) 2021-2024 Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
macro_rules! impl_futures {
($f:ident, $r:ty, |$fut:ident, $cx:ident| => { $p:expr }) => {
impl_futures!($f, $r, ($fut, $cx), $p, );
};
($f:ident, $g: tt, $r:ty, |$fut:ident, $cx:ident| => { $p:expr }) => {
impl_futures!($f, $r, ($fut, $cx), $p, $g);
};
($f:ident, $r:ty, ($fut:ident, $cx:ident), $p:expr, $($g:tt)?) => {
impl $( <$g: Decode> )? FusedFuture for $f $( < $g > )? {
fn is_terminated(&self) -> bool {
!signals().waits_for(self.waiting_reply_to)
}
}
impl $( <$g: Decode> )? Future for $f $( < $g > )? {
type Output = Result<$r>;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
let $fut = &mut self;
let $cx = cx;
$p
}
}
impl $( <$g: Decode> )? $f $( < $g > )? {
/// Postpone handling for a maximum amount of blocks that could be paid, that
/// doesn't exceed a given duration.
pub fn up_to(self, duration: Option<u32>) -> Result<Self> {
async_runtime::locks().lock(
crate::msg::id(),
self.waiting_reply_to,
Lock::up_to(duration.unwrap_or(Config::wait_up_to()))?,
);
Ok(self)
}
/// Postpone handling for a given specific amount of blocks.
pub fn exactly(self, duration: Option<u32>) -> Result<Self> {
async_runtime::locks().lock(
crate::msg::id(),
self.waiting_reply_to,
Lock::exactly(duration.unwrap_or(Config::wait_for()))?,
);
Ok(self)
}
}
};
}
pub(super) use impl_futures;