wizardchen
33da010605
Added a new utility function, WithDeadlockRetry, to handle MySQL deadlock errors during chunk creation. This function retries the operation up to three times with exponential backoff. Updated the CreateChunks method to utilize this new function, ensuring robust handling of concurrent write scenarios and preventing potential deadlocks. Additionally, improved documentation for CreateChunks to clarify the use of Omit for avoiding auto-increment column issues.
46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/Tencent/WeKnora/internal/logger"
|
|
)
|
|
|
|
const defaultMaxDeadlockRetries = 3
|
|
|
|
// WithDeadlockRetry retries fn when MySQL returns Error 1213 (deadlock).
|
|
// Retries up to 3 times with exponential backoff (50ms, 100ms, 200ms).
|
|
// Safe for all DB drivers — non-deadlock errors are returned immediately.
|
|
func WithDeadlockRetry(ctx context.Context, fn func() error) error {
|
|
var err error
|
|
for attempt := 0; attempt <= defaultMaxDeadlockRetries; attempt++ {
|
|
err = fn()
|
|
if err == nil || !IsDeadlockError(err) {
|
|
return err
|
|
}
|
|
if attempt < defaultMaxDeadlockRetries {
|
|
wait := time.Duration(50<<attempt) * time.Millisecond
|
|
logger.Warnf(ctx, "[DeadlockRetry] attempt %d/%d failed with deadlock, retrying in %v: %v",
|
|
attempt+1, defaultMaxDeadlockRetries, wait, err)
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case <-time.After(wait):
|
|
}
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
// IsDeadlockError checks whether err is a MySQL deadlock (Error 1213).
|
|
func IsDeadlockError(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
msg := err.Error()
|
|
return strings.Contains(msg, "Error 1213") ||
|
|
strings.Contains(msg, "Deadlock found when trying to get lock")
|
|
}
|