poll impl: fixes for self.table

crates/wasi/src/preview2/poll.rs

@@ -68,14 +68,12 @@ impl poll::Host for WasiView<'_> {
     async fn poll(&mut self, pollables: Vec<Resource<Pollable>>) -> Result<Vec<u32>> {
         type ReadylistIndex = u32;
 
-        let table = self.table;
-
         let mut table_futures: HashMap<u32, (MakeFuture, Vec<ReadylistIndex>)> = HashMap::new();
 
         for (ix, p) in pollables.iter().enumerate() {
             let ix: u32 = ix.try_into()?;
 
-            let pollable = table.get(p)?;
+            let pollable = self.table.get(p)?;
             let (_, list) = table_futures
                 .entry(pollable.index)
                 .or_insert((pollable.make_future, Vec::new()));
@@ -83,7 +81,7 @@ impl poll::Host for WasiView<'_> {
         }
 
         let mut futures: Vec<(PollableFuture<'_>, Vec<ReadylistIndex>)> = Vec::new();
-        for (entry, (make_future, readylist_indices)) in table.iter_entries(table_futures) {
+        for (entry, (make_future, readylist_indices)) in self.table.iter_entries(table_futures) {
             let entry = entry?;
             futures.push((make_future(entry), readylist_indices));
         }
@@ -121,16 +119,14 @@ impl poll::Host for WasiView<'_> {
 #[async_trait::async_trait]
 impl crate::preview2::bindings::io::poll::HostPollable for WasiView<'_> {
     async fn block(&mut self, pollable: Resource<Pollable>) -> Result<()> {
-        let table = self.table;
+        let pollable = self.table.get(&pollable)?;
-        let pollable = table.get(&pollable)?;
+        let ready = (pollable.make_future)(self.table.get_any_mut(pollable.index)?);
-        let ready = (pollable.make_future)(table.get_any_mut(pollable.index)?);
         ready.await;
         Ok(())
     }
     async fn ready(&mut self, pollable: Resource<Pollable>) -> Result<bool> {
-        let table = self.table;
+        let pollable = self.table.get(&pollable)?;
-        let pollable = table.get(&pollable)?;
+        let ready = (pollable.make_future)(self.table.get_any_mut(pollable.index)?);
-        let ready = (pollable.make_future)(table.get_any_mut(pollable.index)?);
         futures::pin_mut!(ready);
         Ok(matches!(
             futures::future::poll_immediate(ready).await,