tcc grpc rollback_reson

dtmgrpc/dtmgimp/utils.go

@@ -37,6 +37,7 @@ func DtmGrpcCall(s *dtmimp.TransBase, operation string) error {
 			PassthroughHeaders: s.PassthroughHeaders,
 			BranchHeaders:      s.BranchHeaders,
 			RequestTimeout:     s.RequestTimeout,
+			RollbackReason:     s.RollbackReason,
 		},
 		QueryPrepared: s.QueryPrepared,
 		CustomedData:  s.CustomData,

dtmsvr/trans_class.go

@@ -94,12 +94,13 @@ func TransFromDtmRequest(ctx context.Context, c *dtmgpb.DtmRequest) *TransGlobal
 		o = c.TransOptions
 	}
 	r := TransGlobal{TransGlobalStore: storage.TransGlobalStore{
-		Gid:           c.Gid,
-		TransType:     c.TransType,
-		QueryPrepared: c.QueryPrepared,
-		Protocol:      "grpc",
-		BinPayloads:   c.BinPayloads,
-		CustomData:    c.CustomedData,
+		Gid:            c.Gid,
+		TransType:      c.TransType,
+		QueryPrepared:  c.QueryPrepared,
+		Protocol:       "grpc",
+		BinPayloads:    c.BinPayloads,
+		CustomData:     c.CustomedData,
+		RollbackReason: o.RollbackReason,
 		TransOptions: dtmcli.TransOptions{
 			WaitResult:         o.WaitResult,
 			TimeoutToFail:      o.TimeoutToFail,
@@ -107,6 +108,7 @@ func TransFromDtmRequest(ctx context.Context, c *dtmgpb.DtmRequest) *TransGlobal
 			PassthroughHeaders: o.PassthroughHeaders,
 			BranchHeaders:      o.BranchHeaders,
 			RequestTimeout:     o.RequestTimeout,
+			RollbackReason:     o.RollbackReason,
 		},
 	}}
 	if c.Steps != "" {

test/tcc_grpc_test.go

@@ -52,6 +52,7 @@ func TestTccGrpcRollback(t *testing.T) {
 	cronTransOnce(t, gid)
 	assert.Equal(t, StatusFailed, getTransStatus(gid))
 	assert.Equal(t, []string{StatusSucceed, StatusPrepared, StatusSucceed, StatusPrepared}, getBranchesStatus(gid))
+	assert.Equal(t, "rpc error: code = Aborted desc = reason:", getTrans(gid).RollbackReason)
 }
 
 func TestTccGrpcNested(t *testing.T) {