fix compile on rust 1.83.0

Author: loongs-zhang

core/src/co_pool/mod.rs

@@ -6,7 +6,7 @@ use crate::common::ordered_work_steal::{OrderedLocalQueue, OrderedWorkStealQueue
 use crate::common::{get_timeout_time, now, CondvarBlocker};
 use crate::coroutine::suspender::Suspender;
 use crate::scheduler::{SchedulableCoroutine, Scheduler};
-use crate::{impl_current_for, impl_display_by_debug, impl_for_named, trace};
+use crate::{error, impl_current_for, impl_display_by_debug, impl_for_named, trace};
 use dashmap::DashMap;
 use std::cell::Cell;
 use std::ffi::c_longlong;
@@ -70,10 +70,15 @@ impl Drop for CoroutinePool<'_> {
             self.get_running_size(),
             "There are still tasks in progress !"
         );
-        assert!(
-            self.task_queue.is_empty(),
-            "There are still tasks to be carried out !"
-        );
+        if !self.task_queue.is_empty() {
+            cfg_if::cfg_if!{
+                if #[cfg(debug_assertions)] {
+                    panic!("There are still tasks to be carried out !");
+                } else {
+                    error!("Forget some tasks when closing the pool");
+                }
+            }
+        }
     }
 }
 

open-coroutine/build.rs

@@ -1,5 +1,6 @@
 use cargo_metadata::MetadataCommand;
 use std::env::var;
+use std::fs::{copy, read_dir};
 use std::path::PathBuf;
 use tracing::{info, Level};
 use tracing_appender::rolling::{RollingFileAppender, Rotation};
@@ -74,7 +75,7 @@ fn main() {
             .join("registry")
             .join("src");
         let crates_parent_dirs = Vec::from_iter(
-            std::fs::read_dir(dep_src_dir.clone())
+            read_dir(dep_src_dir.clone())
                 .expect("Failed to read deps")
                 .flatten(),
         );
@@ -163,12 +164,56 @@ fn main() {
     {
         panic!("failed to build dylib {}", e);
     }
+    // correct dylib path
+    let hook_deps = out_dir
+        .join(target)
+        .join(if cfg!(debug_assertions) {
+            "debug"
+        } else {
+            "release"
+        })
+        .join("deps");
+    let deps = out_dir
+        .parent()
+        .expect("can not find deps dir")
+        .parent()
+        .expect("can not find deps dir")
+        .parent()
+        .expect("can not find deps dir")
+        .join("deps");
+    for entry in read_dir(hook_deps.clone())
+        .expect("can not find deps dir")
+        .flatten()
+    {
+        let file_name = entry.file_name().to_string_lossy().to_string();
+        if !file_name.contains("open_coroutine_hook") {
+            continue;
+        }
+        if cfg!(target_os = "linux") && file_name.ends_with(".so") {
+            let from = hook_deps.join(file_name);
+            let to = deps.join("libopen_coroutine_hook.so");
+            copy(from.clone(), to.clone()).expect("copy to libopen_coroutine_hook.so failed!");
+            info!("copy {:?} to {:?} success!", from, to);
+        } else if cfg!(target_os = "macos") && file_name.ends_with(".dylib") {
+            let from = hook_deps.join(file_name);
+            let to = deps.join("libopen_coroutine_hook.dylib");
+            copy(from.clone(), to.clone()).expect("copy to libopen_coroutine_hook.dylib failed!");
+            info!("copy {:?} to {:?} success!", from, to);
+        } else if cfg!(windows) {
+            if file_name.ends_with(".dll") {
+                let from = hook_deps.join(file_name);
+                let to = deps.join("open_coroutine_hook.dll");
+                copy(from.clone(), to.clone()).expect("copy to open_coroutine_hook.dll failed!");
+                info!("copy {:?} to {:?} success!", from, to);
+            } else if file_name.ends_with(".lib") {
+                let from = hook_deps.join(file_name);
+                let to = deps.join("open_coroutine_hook.lib");
+                copy(from.clone(), to.clone()).expect("copy to open_coroutine_hook.lib failed!");
+                info!("copy {:?} to {:?} success!", from, to);
+            }
+        }
+    }
     // link dylib
-    let hook_deps = out_dir.join(target).join(if cfg!(debug_assertions) {
-        "debug"
-    } else {
-        "release"
-    });
-    println!("cargo:rustc-link-search=native={}", hook_deps.display());
+    println!("cargo:rustc-link-search=native={}", deps.display());
     println!("cargo:rustc-link-lib=dylib=open_coroutine_hook");
 }