mirror
/
go
mirror of https://github.com/golang/go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

runtime: convert extram and extraMWaiters to internal atomic type 

Updates #53821

Change-Id: Id579b2f8e48dfbe9f37e02d2fa8c94354f9887a4
Reviewed-on: https://go-review.googlesource.com/c/go/+/425480
Reviewed-by: Michael Knyszek <mknyszek@google.com>
Reviewed-by: Michael Pratt <mpratt@google.com>
Run-TryBot: hopehook <hopehook@golangcn.org>
Auto-Submit: Michael Pratt <mpratt@google.com>
TryBot-Result: Gopher Robot <gobot@golang.org>
This commit is contained in:
Andy Pan 2022-08-25 21:08:02 +08:00 committed by Gopher Robot
parent 3486735bf2
commit 5634629f0b
2 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/runtime/cgocall.go
View File

@ -86,7 +86,6 @@ package runtime
import (
"internal/goarch"
"runtime/internal/atomic"
"runtime/internal/sys"
"unsafe"
)
@ -259,7 +258,7 @@ func cgocallbackg1(fn, frame unsafe.Pointer, ctxt uintptr) {
// We must still stay on the same m.
defer unlockOSThread()
if gp.m.needextram || atomic.Load(&extraMWaiters) > 0 {
if gp.m.needextram || extraMWaiters.Load() > 0 {
gp.m.needextram = false
systemstack(newextram)
}

14
src/runtime/proc.go
View File

@ -1878,7 +1878,7 @@ var earlycgocallback = []byte("fatal error: cgo callback before cgo call\n")
// It is called with a working local m, so that it can do things
// like call schedlock and allocate.
func newextram() {
c := atomic.Xchg(&extraMWaiters, 0)
c := extraMWaiters.Swap(0)
if c > 0 {
for i := uint32(0); i < c; i++ {
oneNewExtraM()
@ -1999,9 +1999,9 @@ func getm() uintptr {
return uintptr(unsafe.Pointer(getg().m))
}
var extram uintptr
var extram atomic.Uintptr
var extraMCount uint32 // Protected by lockextra
var extraMWaiters uint32
var extraMWaiters atomic.Uint32
// lockextra locks the extra list and returns the list head.
// The caller must unlock the list by storing a new list head
@ -2015,7 +2015,7 @@ func lockextra(nilokay bool) *m {
incr := false
for {
old := atomic.Loaduintptr(&extram)
old := extram.Load()
if old == locked {
osyield_no_g()
continue
@ -2025,13 +2025,13 @@ func lockextra(nilokay bool) *m {
// Add 1 to the number of threads
// waiting for an M.
// This is cleared by newextram.
atomic.Xadd(&extraMWaiters, 1)
extraMWaiters.Add(1)
incr = true
}
usleep_no_g(1)
continue
}
if atomic.Casuintptr(&extram, old, locked) {
if extram.CompareAndSwap(old, locked) {
return (*m)(unsafe.Pointer(old))
}
osyield_no_g()
@ -2041,7 +2041,7 @@ func lockextra(nilokay bool) *m {
//go:nosplit
func unlockextra(mp *m) {
atomic.Storeuintptr(&extram, uintptr(unsafe.Pointer(mp)))
extram.Store(uintptr(unsafe.Pointer(mp)))
}
var (