59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
|
package daos
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/pocketbase/dbx"
|
||
|
|
)
|
||
|
|
|
||
|
|
const defaultQueryTimeout time.Duration = 2 * time.Minute
|
||
|
|
|
||
|
|
const defaultMaxRetries int = 10
|
||
|
|
|
||
|
|
var defaultRetryIntervals = []int{100, 250, 350, 500, 700, 1000, 1200, 1500}
|
||
|
|
|
||
|
|
func onLockErrorRetry(s *dbx.SelectQuery, op func() error) error {
|
||
|
|
return baseLockRetry(func(attempt int) error {
|
||
|
|
// load a default timeout context if not set explicitly
|
||
|
|
if s.Context() == nil {
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), defaultQueryTimeout)
|
||
|
|
defer func() {
|
||
|
|
cancel()
|
||
|
|
s.WithContext(nil) // reset
|
||
|
|
}()
|
||
|
|
s.WithContext(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
return op()
|
||
|
|
}, defaultMaxRetries)
|
||
|
|
}
|
||
|
|
|
||
|
|
func baseLockRetry(op func(attempt int) error, maxRetries int) error {
|
||
|
|
attempt := 1
|
||
|
|
|
||
|
|
Retry:
|
||
|
|
err := op(attempt)
|
||
|
|
|
||
|
|
if err != nil &&
|
||
|
|
attempt <= maxRetries &&
|
||
|
|
// we are checking the err message to handle both the cgo and noncgo errors
|
||
|
|
strings.Contains(err.Error(), "database is locked") {
|
||
|
|
// wait and retry
|
||
|
|
time.Sleep(getDefaultRetryInterval(attempt))
|
||
|
|
attempt++
|
||
|
|
goto Retry
|
||
|
|
}
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func getDefaultRetryInterval(attempt int) time.Duration {
|
||
|
|
if attempt < 0 || attempt > len(defaultRetryIntervals)-1 {
|
||
|
|
return time.Duration(defaultRetryIntervals[len(defaultRetryIntervals)-1]) * time.Millisecond
|
||
|
|
}
|
||
|
|
|
||
|
|
return time.Duration(defaultRetryIntervals[attempt]) * time.Millisecond
|
||
|
|
}
|