qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-11-22 05:24:39 -07:00
Code Releases Activity

sync: don't keep func alive after OnceFunc panics

Browse Source 
Change-Id: Iac9621280787c7c56964d985de8f46efe7de26cf
This commit is contained in:
Mateusz Poliwczak 2024-07-29 20:41:51 +02:00
parent 165bf241f2
commit 37646b6bbd
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/sync/oncefunc.go
View File

@ -17,6 +17,7 @@ func OnceFunc(f func()) func() {
// Construct the inner closure just once to reduce costs on the fast path.
g := func() {
defer func() {
f = nil // Do not keep f alive after invoking it.
p = recover()
if !valid {
// Re-panic immediately so on the first call the user gets a
@ -25,7 +26,6 @@ func OnceFunc(f func()) func() {
}
}()
f()
f = nil // Do not keep f alive after invoking it.
valid = true // Set only if f does not panic.
}
return func() {
@ -48,6 +48,7 @@ func OnceValue[T any](f func() T) func() T {
result T
)
g := func() {
f = nil // Do not keep f alive after invoking it.
defer func() {
p = recover()
if !valid {
@ -55,7 +56,6 @@ func OnceValue[T any](f func() T) func() T {
}
}()
result = f()
f = nil
valid = true
}
return func() T {
@ -80,6 +80,7 @@ func OnceValues[T1, T2 any](f func() (T1, T2)) func() (T1, T2) {
r2 T2
)
g := func() {
f = nil // Do not keep f alive after invoking it.
defer func() {
p = recover()
if !valid {
@ -87,7 +88,6 @@ func OnceValues[T1, T2 any](f func() (T1, T2)) func() (T1, T2) {
}
}()
r1, r2 = f()
f = nil
valid = true
}
return func() (T1, T2) {