Add os.sched_yield, os.sched_get_priority_min, os.sched_get_priority_max

zetwhite · zetwhite · commit 7f38e33595ba · 2021-09-02T17:12:04.000+09:00
diff --git a/vm/src/stdlib/os.rs b/vm/src/stdlib/os.rs
@@ -2087,6 +2087,39 @@ mod posix {
     #[pyattr]
     const EX_CONFIG: i8 = exitcode::CONFIG as i8;
 
+    #[cfg(any(
+        target_os = "linux",
+        target_os = "android",
+        target_os = "freebsd",
+        target_os = "dragonfly",
+        target_os = "netbsd"
+    ))]
+    #[pyattr]
+    const SCHED_RR: i32 = libc::SCHED_RR;
+    #[cfg(any(
+        target_os = "linux",
+        target_os = "android",
+        target_os = "freebsd",
+        target_os = "dragonfly",
+        target_os = "netbsd"
+    ))]
+    #[pyattr]
+    const SCHED_FIFO: i32 = libc::SCHED_FIFO;
+    #[cfg(any(
+        target_os = "linux",
+        target_os = "freebsd",
+        target_os = "dragonfly",
+        target_os = "netbsd"
+    ))]
+    #[pyattr]
+    const SCHED_OTHER: i32 = libc::SCHED_OTHER;
+    #[cfg(any(target_os = "linux", target_os = "android"))]
+    #[pyattr]
+    const SCHED_IDLE: i32 = libc::SCHED_IDLE;
+    #[cfg(any(target_os = "linux", target_os = "android"))]
+    #[pyattr]
+    const SCHED_BATCH: i32 = libc::SCHED_BATCH;
+
     #[cfg(any(target_os = "linux", target_os = "freebsd", target_os = "macos"))]
     #[pyattr]
     const POSIX_SPAWN_OPEN: i32 = PosixSpawnFileActionIdentifier::Open as i32;
@@ -2369,6 +2402,32 @@ mod posix {
         }
     }
 
+    #[pyfunction]
+    fn sched_get_priority_max(policy: i32, vm: &VirtualMachine) -> PyResult<i32> {
+        let max = unsafe { libc::sched_get_priority_max(policy) };
+        if max == -1 {
+            Err(errno_err(vm))
+        } else {
+            Ok(max)
+        }
+    }
+
+    #[pyfunction]
+    fn sched_get_priority_min(policy: i32, vm: &VirtualMachine) -> PyResult<i32> {
+        let min = unsafe { libc::sched_get_priority_min(policy) };
+        if min == -1 {
+            Err(errno_err(vm))
+        } else {
+            Ok(min)
+        }
+    }
+
+    #[pyfunction]
+    fn sched_yield(vm: &VirtualMachine) -> PyResult<()> {
+        let _ = nix::sched::sched_yield().map_err(|e| e.into_pyexception(vm))?;
+        Ok(())
+    }
+
     #[pyfunction]
     fn get_inheritable(fd: RawFd, vm: &VirtualMachine) -> PyResult<bool> {
         use nix::fcntl::fcntl;
