crdb: better max retry errors (especially with retries disabled)

authzed · Sep 18, 2023 · b40a0d0 · b40a0d0
1 parent 652d309
commit b40a0d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/internal/datastore/crdb/pool/pool.go b/internal/datastore/crdb/pool/pool.go
@@ -309,7 +309,7 @@ func (p *RetryPool) withRetries(ctx context.Context, fn func(conn *pgxpool.Conn)
 		log.Ctx(ctx).Warn().Err(err).Uint8("retries", retries).Msg("error is not resettable or retryable")
 		return err
 	}
-	return &MaxRetryError{MaxRetries: p.maxRetries, LastErr: err}
+	return &MaxRetryError{MaxRetries: maxRetries, LastErr: err}
 }
 
 // GC marks a connection for destruction on the next Acquire.

diff --git a/internal/datastore/crdb/pool/slqerrors.go b/internal/datastore/crdb/pool/slqerrors.go
@@ -27,8 +27,12 @@ type MaxRetryError struct {
 }
 
 func (e *MaxRetryError) Error() string {
-	return strconv.Itoa(int(e.MaxRetries)) + "max retries reached" + ": " + e.LastErr.Error()
+	if e.MaxRetries == 0 {
+		return "retries disabled: " + e.LastErr.Error()
+	}
+	return strconv.Itoa(int(e.MaxRetries)) + " max retries reached" + ": " + e.LastErr.Error()
 }
+
 func (e *MaxRetryError) Unwrap() error { return e.LastErr }
 
 // ResettableError is an error that we think may succeed if retried against a new connection.