25 lines
485 B
Go
25 lines
485 B
Go
|
package stm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"runtime/pprof"
|
||
|
|
)
|
||
|
|
|
||
|
|
var retries = pprof.NewProfile("stmRetries")
|
||
|
|
|
||
|
|
// retry is a sentinel value. When thrown via panic, it indicates that a
|
||
|
|
// transaction should be retried.
|
||
|
|
var retry = &struct{}{}
|
||
|
|
|
||
|
|
// catchRetry returns true if fn calls tx.Retry.
|
||
|
|
func catchRetry(fn Operation, tx *Tx) (result interface{}, gotRetry bool) {
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r == retry {
|
||
|
|
gotRetry = true
|
||
|
|
} else if r != nil {
|
||
|
|
panic(r)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
result = fn(tx)
|
||
|
|
return
|
||
|
|
}
|