Get batch arg, when looking for EndTransaction, just look at the last entry

Author: veteranlu

roachpb/batch.go

@@ -62,8 +62,13 @@ func (ba *BatchRequest) IsRange() bool {
 
 // GetArg returns the first request of the given type, if possible.
 func (ba *BatchRequest) GetArg(method Method) (Request, bool) {
-	// TODO(tschottdorf): when looking for EndTransaction, just look at the
-	// last entry.
+	// when looking for EndTransaction, just look at the last entry.
+	if length := len(ba.Requests); length > 0 && method == EndTransaction {
+		if req := ba.Requests[length-1].GetInner(); req.Method() == EndTransaction {
+			return req, true
+		}
+	}
+
 	for _, arg := range ba.Requests {
 		if req := arg.GetInner(); req.Method() == method {
 			return req, true