Skip to content

Commit 80899dc

Browse files
committed
chore: replace all internal usage of pin_mut
Signed-off-by: tison <[email protected]>
1 parent 1d2089a commit 80899dc

File tree

9 files changed

+42
-44
lines changed

9 files changed

+42
-44
lines changed

futures-channel/tests/mpsc.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1+
use std::pin::pin;
12
use futures::channel::{mpsc, oneshot};
23
use futures::executor::{block_on, block_on_stream};
34
use futures::future::{poll_fn, FutureExt};
4-
use futures::pin_mut;
55
use futures::sink::{Sink, SinkExt};
66
use futures::stream::{Stream, StreamExt};
77
use futures::task::{Context, Poll};
@@ -30,7 +30,8 @@ fn send_recv_no_buffer() {
3030
// Run on a task context
3131
block_on(poll_fn(move |cx| {
3232
let (tx, rx) = mpsc::channel::<i32>(0);
33-
pin_mut!(tx, rx);
33+
let mut tx = pin!(tx);
34+
let mut rx = pin!(rx);
3435

3536
assert!(tx.as_mut().poll_flush(cx).is_ready());
3637
assert!(tx.as_mut().poll_ready(cx).is_ready());

futures-executor/src/local_pool.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@ use futures_core::stream::Stream;
44
use futures_core::task::{Context, Poll};
55
use futures_task::{waker_ref, ArcWake};
66
use futures_task::{FutureObj, LocalFutureObj, LocalSpawn, Spawn, SpawnError};
7-
use futures_util::pin_mut;
87
use futures_util::stream::FuturesUnordered;
98
use futures_util::stream::StreamExt;
109
use std::cell::RefCell;
1110
use std::ops::{Deref, DerefMut};
11+
use std::pin::pin;
1212
use std::rc::{Rc, Weak};
1313
use std::sync::{
1414
atomic::{AtomicBool, Ordering},
@@ -155,7 +155,7 @@ impl LocalPool {
155155
/// one of the pool's run or poll methods. While the function is running,
156156
/// however, all tasks in the pool will try to make progress.
157157
pub fn run_until<F: Future>(&mut self, future: F) -> F::Output {
158-
pin_mut!(future);
158+
let mut future = pin!(future);
159159

160160
run_executor(|cx| {
161161
{
@@ -312,7 +312,7 @@ impl Default for LocalPool {
312312
///
313313
/// Use a [`LocalPool`] if you need finer-grained control over spawned tasks.
314314
pub fn block_on<F: Future>(f: F) -> F::Output {
315-
pin_mut!(f);
315+
let mut f = pin!(f);
316316
run_executor(|cx| f.as_mut().poll(cx))
317317
}
318318

futures-util/src/future/future/mod.rs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,9 @@
77
use alloc::boxed::Box;
88
use core::convert::Infallible;
99
use core::pin::Pin;
10-
10+
use std::pin::pin;
1111
use crate::fns::{inspect_fn, into_fn, ok_fn, InspectFn, IntoFn, OkFn};
1212
use crate::future::{assert_future, Either};
13-
use crate::pin_mut;
1413
use crate::stream::assert_stream;
1514
#[cfg(feature = "alloc")]
1615
use futures_core::future::{BoxFuture, LocalBoxFuture};
@@ -592,8 +591,7 @@ pub trait FutureExt: Future {
592591
let noop_waker = crate::task::noop_waker();
593592
let mut cx = Context::from_waker(&noop_waker);
594593

595-
let this = self;
596-
pin_mut!(this);
594+
let this = pin!(self);
597595
match this.poll(&mut cx) {
598596
Poll::Ready(x) => Some(x),
599597
_ => None,

futures/tests/async_await_macros.rs

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,15 @@ use futures::sink::SinkExt;
55
use futures::stream::StreamExt;
66
use futures::task::{Context, Poll};
77
use futures::{
8-
join, pending, pin_mut, poll, select, select_biased, stream, stream_select, try_join,
8+
join, pending, poll, select, select_biased, stream, stream_select, try_join,
99
};
10-
use std::mem;
10+
use std::pin::pin;
1111

1212
#[test]
1313
fn poll_and_pending() {
1414
let pending_once = async { pending!() };
1515
block_on(async {
16-
pin_mut!(pending_once);
16+
let mut pending_once = pin!(pending_once);
1717
assert_eq!(Poll::Pending, poll!(&mut pending_once));
1818
assert_eq!(Poll::Ready(()), poll!(&mut pending_once));
1919
});
@@ -30,7 +30,7 @@ fn join() {
3030
};
3131

3232
block_on(async {
33-
pin_mut!(fut);
33+
let mut fut = pin!(fut);
3434
assert_eq!(Poll::Pending, poll!(&mut fut));
3535
tx1.send(1).unwrap();
3636
assert_eq!(Poll::Pending, poll!(&mut fut));
@@ -124,7 +124,7 @@ fn select_streams() {
124124
},
125125
// runs last
126126
complete => break,
127-
};
127+
}
128128
}
129129
});
130130
assert!(ran);
@@ -181,7 +181,7 @@ fn select_size() {
181181
_ = ready => {},
182182
}
183183
};
184-
assert_eq!(mem::size_of_val(&fut), 24);
184+
assert_eq!(size_of_val(&fut), 24);
185185

186186
let fut = async {
187187
let mut ready1 = future::ready(0i32);
@@ -191,7 +191,7 @@ fn select_size() {
191191
_ = ready2 => {},
192192
}
193193
};
194-
assert_eq!(mem::size_of_val(&fut), 40);
194+
assert_eq!(size_of_val(&fut), 40);
195195
}
196196

197197
#[test]
@@ -204,7 +204,7 @@ fn select_on_non_unpin_expressions() {
204204
select! {
205205
value_1 = make_non_unpin_fut().fuse() => select_res = value_1,
206206
value_2 = make_non_unpin_fut().fuse() => select_res = value_2,
207-
};
207+
}
208208
select_res
209209
});
210210
assert_eq!(res, 5);
@@ -221,7 +221,7 @@ fn select_on_non_unpin_expressions_with_default() {
221221
value_1 = make_non_unpin_fut().fuse() => select_res = value_1,
222222
value_2 = make_non_unpin_fut().fuse() => select_res = value_2,
223223
default => select_res = 7,
224-
};
224+
}
225225
select_res
226226
});
227227
assert_eq!(res, 5);
@@ -238,11 +238,11 @@ fn select_on_non_unpin_size() {
238238
select! {
239239
value_1 = make_non_unpin_fut().fuse() => select_res = value_1,
240240
value_2 = make_non_unpin_fut().fuse() => select_res = value_2,
241-
};
241+
}
242242
select_res
243243
};
244244

245-
assert_eq!(32, mem::size_of_val(&fut));
245+
assert_eq!(32, size_of_val(&fut));
246246
}
247247

248248
#[test]
@@ -364,14 +364,14 @@ fn join_size() {
364364
let ready = future::ready(0i32);
365365
join!(ready)
366366
};
367-
assert_eq!(mem::size_of_val(&fut), 24);
367+
assert_eq!(size_of_val(&fut), 24);
368368

369369
let fut = async {
370370
let ready1 = future::ready(0i32);
371371
let ready2 = future::ready(0i32);
372372
join!(ready1, ready2)
373373
};
374-
assert_eq!(mem::size_of_val(&fut), 40);
374+
assert_eq!(size_of_val(&fut), 40);
375375
}
376376

377377
#[cfg_attr(not(target_pointer_width = "64"), ignore)]
@@ -381,14 +381,14 @@ fn try_join_size() {
381381
let ready = future::ready(Ok::<i32, i32>(0));
382382
try_join!(ready)
383383
};
384-
assert_eq!(mem::size_of_val(&fut), 24);
384+
assert_eq!(size_of_val(&fut), 24);
385385

386386
let fut = async {
387387
let ready1 = future::ready(Ok::<i32, i32>(0));
388388
let ready2 = future::ready(Ok::<i32, i32>(0));
389389
try_join!(ready1, ready2)
390390
};
391-
assert_eq!(mem::size_of_val(&fut), 48);
391+
assert_eq!(size_of_val(&fut), 48);
392392
}
393393

394394
#[allow(clippy::let_underscore_future)]

futures/tests/future_join_all.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
use futures::executor::block_on;
22
use futures::future::{join_all, ready, Future, JoinAll};
3-
use futures::pin_mut;
43
use std::fmt::Debug;
4+
use std::pin::pin;
55

66
#[track_caller]
77
fn assert_done<T>(actual_fut: impl Future<Output = T>, expected: T)
88
where
99
T: PartialEq + Debug,
1010
{
11-
pin_mut!(actual_fut);
11+
let actual_fut = pin!(actual_fut);
1212
let output = block_on(actual_fut);
1313
assert_eq!(output, expected);
1414
}

futures/tests/future_try_join_all.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
use futures::executor::block_on;
22
use futures::future::{err, ok, try_join_all, Future, TryJoinAll};
3-
use futures::pin_mut;
43
use std::fmt::Debug;
4+
use std::pin::pin;
55

66
#[track_caller]
77
fn assert_done<T>(actual_fut: impl Future<Output = T>, expected: T)
88
where
99
T: PartialEq + Debug,
1010
{
11-
pin_mut!(actual_fut);
11+
let actual_fut = pin!(actual_fut);
1212
let output = block_on(actual_fut);
1313
assert_eq!(output, expected);
1414
}

futures/tests/io_buf_reader.rs

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,12 @@ use futures::io::{
44
AllowStdIo, AsyncBufRead, AsyncBufReadExt, AsyncRead, AsyncReadExt, AsyncSeek, AsyncSeekExt,
55
BufReader, SeekFrom,
66
};
7-
use futures::pin_mut;
87
use futures::task::{Context, Poll};
98
use futures_test::task::noop_context;
109
use pin_project::pin_project;
1110
use std::cmp;
1211
use std::io;
13-
use std::pin::Pin;
12+
use std::pin::{pin, Pin};
1413

1514
// helper for maybe_pending_* tests
1615
fn run<F: Future + Unpin>(mut f: F) -> F::Output {
@@ -157,7 +156,7 @@ fn test_buffered_reader_seek() {
157156
block_on(async {
158157
let inner: &[u8] = &[5, 6, 7, 0, 1, 2, 3, 4];
159158
let reader = BufReader::with_capacity(2, Cursor::new(inner));
160-
pin_mut!(reader);
159+
let mut reader = pin!(reader);
161160

162161
assert_eq!(reader.seek(SeekFrom::Start(3)).await.unwrap(), 3);
163162
assert_eq!(reader.as_mut().fill_buf().await.unwrap(), &[0, 1][..]);
@@ -175,7 +174,7 @@ fn test_buffered_reader_seek_relative() {
175174
block_on(async {
176175
let inner: &[u8] = &[5, 6, 7, 0, 1, 2, 3, 4];
177176
let reader = BufReader::with_capacity(2, Cursor::new(inner));
178-
pin_mut!(reader);
177+
let mut reader = pin!(reader);
179178

180179
assert!(reader.as_mut().seek_relative(3).await.is_ok());
181180
assert_eq!(reader.as_mut().fill_buf().await.unwrap(), &[0, 1][..]);
@@ -195,7 +194,7 @@ fn test_buffered_reader_invalidated_after_read() {
195194
block_on(async {
196195
let inner: &[u8] = &[5, 6, 7, 0, 1, 2, 3, 4];
197196
let reader = BufReader::with_capacity(3, Cursor::new(inner));
198-
pin_mut!(reader);
197+
let mut reader = pin!(reader);
199198

200199
assert_eq!(reader.as_mut().fill_buf().await.unwrap(), &[5, 6, 7][..]);
201200
reader.as_mut().consume(3);
@@ -216,7 +215,7 @@ fn test_buffered_reader_invalidated_after_seek() {
216215
block_on(async {
217216
let inner: &[u8] = &[5, 6, 7, 0, 1, 2, 3, 4];
218217
let reader = BufReader::with_capacity(3, Cursor::new(inner));
219-
pin_mut!(reader);
218+
let mut reader = pin!(reader);
220219

221220
assert_eq!(reader.as_mut().fill_buf().await.unwrap(), &[5, 6, 7][..]);
222221
reader.as_mut().consume(3);
@@ -265,7 +264,7 @@ fn test_buffered_reader_seek_underflow() {
265264

266265
block_on(async {
267266
let reader = BufReader::with_capacity(5, AllowStdIo::new(PositionReader { pos: 0 }));
268-
pin_mut!(reader);
267+
let mut reader = pin!(reader);
269268
assert_eq!(reader.as_mut().fill_buf().await.unwrap(), &[0, 1, 2, 3, 4][..]);
270269
assert_eq!(reader.seek(SeekFrom::End(-5)).await.unwrap(), u64::MAX - 5);
271270
assert_eq!(reader.as_mut().fill_buf().await.unwrap().len(), 5);
@@ -419,7 +418,7 @@ fn maybe_pending_seek() {
419418

420419
let inner: &[u8] = &[5, 6, 7, 0, 1, 2, 3, 4];
421420
let reader = BufReader::with_capacity(2, MaybePendingSeek::new(inner));
422-
pin_mut!(reader);
421+
let mut reader = pin!(reader);
423422

424423
assert_eq!(run(reader.seek(SeekFrom::Current(3))).ok(), Some(3));
425424
assert_eq!(run(reader.as_mut().fill_buf()).ok(), Some(&[0, 1][..]));

futures/tests/sink.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ use std::collections::VecDeque;
1111
use std::convert::Infallible;
1212
use std::fmt;
1313
use std::mem;
14-
use std::pin::Pin;
14+
use std::pin::{pin, Pin};
1515
use std::rc::Rc;
1616
use std::sync::atomic::{AtomicBool, Ordering};
1717
use std::sync::Arc;
@@ -509,7 +509,7 @@ fn sink_unfold() {
509509
Ok::<_, String>(())
510510
}
511511
});
512-
futures::pin_mut!(unfold);
512+
let mut unfold = pin!(unfold);
513513
assert_eq!(unfold.as_mut().start_send(1), Ok(()));
514514
assert_eq!(unfold.as_mut().poll_flush(cx), Poll::Ready(Ok(())));
515515
assert_eq!(rx.try_next().unwrap(), Some(1));

futures/tests/stream_peekable.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
1+
use std::pin::pin;
12
use futures::executor::block_on;
2-
use futures::pin_mut;
33
use futures::stream::{self, Peekable, StreamExt};
44

55
#[test]
66
fn peekable() {
77
block_on(async {
88
let peekable: Peekable<_> = stream::iter(vec![1u8, 2, 3]).peekable();
9-
pin_mut!(peekable);
9+
let mut peekable = pin!(peekable);
1010
assert_eq!(peekable.as_mut().peek().await, Some(&1u8));
1111
assert_eq!(peekable.collect::<Vec<u8>>().await, vec![1, 2, 3]);
1212

1313
let s = stream::once(async { 1 }).peekable();
14-
pin_mut!(s);
14+
let mut s = pin!(s);
1515
assert_eq!(s.as_mut().peek().await, Some(&1u8));
1616
assert_eq!(s.collect::<Vec<u8>>().await, vec![1]);
1717
});
@@ -21,7 +21,7 @@ fn peekable() {
2121
fn peekable_mut() {
2222
block_on(async {
2323
let s = stream::iter(vec![1u8, 2, 3]).peekable();
24-
pin_mut!(s);
24+
let mut s = pin!(s);
2525
if let Some(p) = s.as_mut().peek_mut().await {
2626
if *p == 1 {
2727
*p = 5;
@@ -36,7 +36,7 @@ fn peekable_next_if_eq() {
3636
block_on(async {
3737
// first, try on references
3838
let s = stream::iter(vec!["Heart", "of", "Gold"]).peekable();
39-
pin_mut!(s);
39+
let mut s = pin!(s);
4040
// try before `peek()`
4141
assert_eq!(s.as_mut().next_if_eq(&"trillian").await, None);
4242
assert_eq!(s.as_mut().next_if_eq(&"Heart").await, Some("Heart"));
@@ -49,7 +49,7 @@ fn peekable_next_if_eq() {
4949

5050
// make sure comparison works for owned values
5151
let s = stream::iter(vec![String::from("Ludicrous"), "speed".into()]).peekable();
52-
pin_mut!(s);
52+
let mut s = pin!(s);
5353
// make sure basic functionality works
5454
assert_eq!(s.as_mut().next_if_eq("Ludicrous").await, Some("Ludicrous".into()));
5555
assert_eq!(s.as_mut().next_if_eq("speed").await, Some("speed".into()));

0 commit comments

Comments
 (0)