When the main thread was exiting it was not flushing and properly clo…

…sing file descriptors

lib/wasix/src/fs/mod.rs

@@ -462,10 +462,6 @@ impl WasiFs {
     pub async fn close_all(&self) {
         // TODO: this should close all uniquely owned files instead of just flushing.
 
-        if let Ok(mut map) = self.fd_map.write() {
-            map.clear();
-        }
-
         let to_close = {
             if let Ok(map) = self.fd_map.read() {
                 map.keys().copied().collect::<Vec<_>>()
@@ -475,7 +471,7 @@ impl WasiFs {
         };
 
         let _ = tokio::join!(
-            // Make sure the STDOUT and STDERR are explicitely flushed
+            // Make sure the STDOUT and STDERR are explicitly flushed
             self.flush(__WASI_STDOUT_FILENO),
             self.flush(__WASI_STDERR_FILENO),
             async {
@@ -485,6 +481,10 @@ impl WasiFs {
                 }
             }
         );
+
+        if let Ok(mut map) = self.fd_map.write() {
+            map.clear();
+        }
     }
 
     /// Will conditionally union the binary file system with this one