mpp: support returning regions that need retry (pingcap#751)

* mpp: support return of region errors * return region list Co-authored-by: Zhuhe Fang <fzhedu@gmail.com>
windtalker · May 7, 2021 · 76b14e4 · 76b14e4
1 parent 39b8846
commit 76b14e4
Show file tree

Hide file tree

Showing 2 changed files with 109 additions and 46 deletions.
diff --git a/pkg/mpp/mpp.pb.go b/pkg/mpp/mpp.pb.go
diff --git a/proto/mpp.proto b/proto/mpp.proto
@@ -3,6 +3,7 @@ package mpp;
 
 import "gogoproto/gogo.proto";
 import "coprocessor.proto";
+import "metapb.proto";
 
 option (gogoproto.marshaler_all) = true;
 option (gogoproto.sizer_all) = true;
@@ -31,6 +32,7 @@ message DispatchTaskRequest {
 // Get response of DispatchTaskRequest.
 message DispatchTaskResponse {
 	Error error = 1;
+	repeated metapb.Region retry_regions = 2;
 }
 
 // CancelTaskRequest closes the execution of a task.