From ad54ae673789e6f1c00eba0c3a203163084f2394 Mon Sep 17 00:00:00 2001
From: Ben Darnell <ben@bendarnell.com>
Date: Sun, 11 Nov 2018 18:53:07 -0500
Subject: [PATCH] batcheval: Work around time-bound iterator bug in
 RefreshRange

Updates #28358
Fixes #31823

Release note (bug fix): Fix long stalls that can occur in contended
transactions.
---
 pkg/storage/batcheval/cmd_refresh_range.go | 27 ++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/pkg/storage/batcheval/cmd_refresh_range.go b/pkg/storage/batcheval/cmd_refresh_range.go
index e0fc9a24b862..b25b6a0100da 100644
--- a/pkg/storage/batcheval/cmd_refresh_range.go
+++ b/pkg/storage/batcheval/cmd_refresh_range.go
@@ -86,6 +86,33 @@ func RefreshRange(
 		if i.Txn.ID == h.Txn.ID {
 			continue
 		}
+
+		if i.Span.EndKey != nil {
+			return result.Result{}, errors.Errorf("unexpected range intent from MVCC storage")
+		}
+
+		// HACK(bdarnell): Time-bound iterators can return intents that
+		// shouldn't be there
+		// (https://github.com/cockroachdb/cockroach/issues/28358), and
+		// this can result in stalled traffic when it occurs in this
+		// method (https://github.com/cockroachdb/cockroach/issues/31823).
+		// When we get an intent, check with a regular iterator to ensure
+		// that it's really there.
+		_, realIntents, err := engine.MVCCGetWithTombstone(
+			ctx,
+			batch,
+			i.Span.Key,
+			h.Txn.Timestamp,
+			false, /* consistent */
+			nil,   /* txn */
+		)
+		if err != nil {
+			return result.Result{}, err
+		}
+		if len(realIntents) == 0 {
+			continue
+		}
+
 		// Return an error if an intent was written to the span.
 		return result.Result{}, errors.Errorf("encountered recently written intent %s @%s",
 			i.Span.Key, i.Txn.Timestamp)