feat: add SchedStatus

joway · Sep 7, 2023 · 569f849 · 569f849
1 parent 92184d0
commit 569f849
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lang/runtimex/sched.go b/lang/runtimex/sched.go
@@ -25,25 +25,29 @@ var (
 	//prunqField       = pType.FieldByName("runq")
 )
 
-func MPreemptOff() {
+func MPreemptOff() bool {
 	g := getg()
 	m := *((**_m)(unsafe.Pointer(g + mField.Offset())))
 	preemptoff := (*string)(unsafe.Pointer((uintptr)(unsafe.Pointer(m)) + mpreemptoffField.Offset()))
 	if *preemptoff == "" {
 		*preemptoff = "holding"
+		return true
 	}
+	return false
 }
 
-func MPreemptOn() {
+func MPreemptOn() bool {
 	g := getg()
 	m := *((**_m)(unsafe.Pointer(g + mField.Offset())))
 	preemptoff := (*string)(unsafe.Pointer((uintptr)(unsafe.Pointer(m)) + mpreemptoffField.Offset()))
 	if *preemptoff != "" {
 		*preemptoff = ""
+		return true
 	}
+	return false
 }
 
-func schedlog() {
+func SchedStatus() {
 	g := getg()
 	m := *((**_m)(unsafe.Pointer(g + mField.Offset())))
 	p := *((**_p)(unsafe.Pointer(uintptr(unsafe.Pointer(m)) + pField.Offset())))