Add waker functionality

- Forgot to add task wakers in futures

Author: StaticallyTypedAnxiety

src/engine.rs

@@ -75,6 +75,11 @@ impl WasmRuntimeAsyncEngine {
     }
 }
 
+
+pub fn next_id()
+{
+    
+}
 struct FutureWaker(Sender<()>);
 
 impl FutureWaker {
@@ -91,7 +96,7 @@ impl Wake for FutureWaker {
 
 #[cfg(test)]
 mod test {
-   
+
     use super::*;
     use std::future::Future;
 

src/io/net.rs

@@ -155,7 +155,7 @@ impl<'a> Future for ConnectionFuture<'a> {
     type Output = IOResult<()>;
     fn poll(
         self: std::pin::Pin<&mut Self>,
-        _: &mut std::task::Context<'_>,
+        cx: &mut std::task::Context<'_>,
     ) -> std::task::Poll<Self::Output> {
         let this = self.get_mut();
         if !REACTOR.lock().unwrap().is_pollable(&this.async_key) {
@@ -172,6 +172,7 @@ impl<'a> Future for ConnectionFuture<'a> {
             this.stream.finish_connecting()?;
             Poll::Ready(Ok(()))
         } else {
+            cx.waker().wake_by_ref();
             Poll::Pending
         }
     }

src/io/timer.rs

@@ -55,7 +55,7 @@ impl Future for TimeoutFuture {
     type Output = ();
     fn poll(
         self: std::pin::Pin<&mut Self>,
-        _: &mut std::task::Context<'_>,
+        cx: &mut std::task::Context<'_>,
     ) -> std::task::Poll<Self::Output> {
         let has_elapsed = TIMERS
             .get(&self.id)
@@ -64,6 +64,7 @@ impl Future for TimeoutFuture {
         if has_elapsed {
             return std::task::Poll::Ready(());
         }
+        cx.waker().wake_by_ref();
         std::task::Poll::Pending
     }
 }

src/poll_tasks.rs

@@ -1,7 +1,4 @@
-use std::{
-    collections::HashMap,
-    sync::Arc,
-};
+use std::{collections::HashMap, sync::Arc};
 
 use crate::bindings::wasi::io::poll::{poll, Pollable};
 
@@ -49,4 +46,3 @@ impl PollTasks {
         }
     }
 }
-