Merge pull request #774 from ydb-platform/tli

* Added `yd.IsTransactionLocksInvalidated(err)` helper for checks `TL…

Author: asmyasnikov

CHANGELOG.md

@@ -1,3 +1,5 @@
+* Added `ydb.IsOperationErrorTransactionLocksInvalidated(err)` helper for checks `TLI` flag in err
+
 ## v3.48.3
 * Added `table/types.IsOptional()` helper
 

errors.go

@@ -82,6 +82,11 @@ func IsOperationErrorSchemeError(err error) bool {
 	return IsOperationError(err, Ydb.StatusIds_SCHEME_ERROR)
 }
 
+// IsOperationErrorTransactionLocksInvalidated checks does err a TLI issue
+func IsOperationErrorTransactionLocksInvalidated(err error) (isTLI bool) {
+	return xerrors.IsOperationErrorTransactionLocksInvalidated(err)
+}
+
 // IsRatelimiterAcquireError checks whether given err is an ratelimiter acquire error
 func IsRatelimiterAcquireError(err error) bool {
 	return ratelimiterErrors.IsAcquireError(err)

internal/xerrors/operation.go

@@ -118,6 +118,17 @@ func IsOperationError(err error, codes ...Ydb.StatusIds_StatusCode) bool {
 	return false
 }
 
+const issueCodeTransactionLocksInvalidated = 2001
+
+func IsOperationErrorTransactionLocksInvalidated(err error) (isTLI bool) {
+	if IsOperationError(err, Ydb.StatusIds_ABORTED) {
+		IterateByIssues(err, func(_ string, code Ydb.StatusIds_StatusCode, severity uint32) {
+			isTLI = isTLI || (code == issueCodeTransactionLocksInvalidated)
+		})
+	}
+	return isTLI
+}
+
 func (e *operationError) Type() Type {
 	switch e.code {
 	case

internal/xerrors/operation_test.go

@@ -28,6 +28,53 @@ func TestIsOperationError(t *testing.T) {
 	}
 }
 
+func TestIsOperationErrorTransactionLocksInvalidated(t *testing.T) {
+	for _, tt := range [...]struct {
+		err   error
+		isTLI bool
+	}{
+		{
+			err: Operation(
+				WithStatusCode(Ydb.StatusIds_ABORTED),
+				WithIssues([]*Ydb_Issue.IssueMessage{{
+					IssueCode: issueCodeTransactionLocksInvalidated,
+				}}),
+			),
+			isTLI: true,
+		},
+		{
+			err: Operation(
+				WithStatusCode(Ydb.StatusIds_OVERLOADED),
+				WithIssues([]*Ydb_Issue.IssueMessage{{
+					IssueCode: issueCodeTransactionLocksInvalidated,
+				}}),
+			),
+			isTLI: false,
+		},
+		{
+			err: Operation(
+				WithStatusCode(Ydb.StatusIds_ABORTED),
+			),
+			isTLI: false,
+		},
+		{
+			err: Operation(
+				WithStatusCode(Ydb.StatusIds_ABORTED),
+				WithIssues([]*Ydb_Issue.IssueMessage{{
+					Issues: []*Ydb_Issue.IssueMessage{{
+						IssueCode: issueCodeTransactionLocksInvalidated,
+					}},
+				}}),
+			),
+			isTLI: true,
+		},
+	} {
+		t.Run("", func(t *testing.T) {
+			require.Equal(t, tt.isTLI, IsOperationErrorTransactionLocksInvalidated(tt.err))
+		})
+	}
+}
+
 func Test_operationError_Error(t *testing.T) {
 	for _, tt := range []struct {
 		err  error