Generate AbortTransaction operation

GODRIVER-1043

Change-Id: Ic7a1501c06d3cc6a3aa4b30c1e45e43a5f1dbfa2

Author: connieyu

mongo/session.go

@@ -20,7 +20,6 @@ import (
 	"go.mongodb.org/mongo-driver/x/mongo/driver/operation"
 	"go.mongodb.org/mongo-driver/x/mongo/driver/session"
 	"go.mongodb.org/mongo-driver/x/mongo/driver/topology"
-	"go.mongodb.org/mongo-driver/x/mongo/driverlegacy"
 	"go.mongodb.org/mongo-driver/x/network/command"
 )
 
@@ -183,14 +182,21 @@ func (s *sessionImpl) AbortTransaction(ctx context.Context) error {
 		return s.clientSession.AbortTransaction()
 	}
 
-	cmd := command.AbortTransaction{
-		Session: s.clientSession,
+	var selector description.ServerSelectorFunc = func(t description.Topology, svrs []description.Server) ([]description.Server, error) {
+		if s.clientSession.PinnedServer != nil {
+			return s.clientSession.PinnedServer.SelectServer(t, svrs)
+		}
+		return description.WriteSelector().SelectServer(t, svrs)
 	}
 
 	s.clientSession.Aborting = true
-	_, err = driverlegacy.AbortTransaction(ctx, cmd, s.topo, description.WriteSelector())
+	err = operation.NewAbortTransaction().Session(s.clientSession).ClusterClock(s.client.clock).Database("admin").
+		Deployment(s.topo).WriteConcern(s.clientSession.CurrentWc).ServerSelector(selector).
+		Retry(driver.RetryOncePerCommand).CommandMonitor(s.client.monitor).RecoveryToken(bsoncore.Document(s.clientSession.RecoveryToken)).Execute(ctx)
 
+	s.clientSession.Aborting = false
 	_ = s.clientSession.AbortTransaction()
+
 	return replaceErrors(err)
 }
 

x/mongo/driver/operation.go

@@ -269,7 +269,7 @@ func (op Operation) Execute(ctx context.Context, scratch []byte) error {
 		op.Client.RetryWrite = false
 		if *op.RetryMode > RetryNone {
 			op.Client.RetryWrite = true
-			if !op.Client.Committing {
+			if !op.Client.Committing && !op.Client.Aborting {
 				op.Client.IncrementTxnNumber()
 			}
 		}
@@ -460,7 +460,7 @@ func (op Operation) Execute(ctx context.Context, scratch []byte) error {
 func (op Operation) retryable(desc description.Server) RetryType {
 	switch op.RetryType {
 	case RetryWrite:
-		if op.Client != nil && op.Client.Committing {
+		if op.Client != nil && (op.Client.Committing || op.Client.Aborting) {
 			return RetryWrite
 		}
 		if op.Deployment.SupportsRetry() &&

x/mongo/driver/operation/abort_transaction.go

@@ -0,0 +1,17 @@
+version = 0
+name = "AbortTransaction"
+documentation = "AbortTransaction performs an abortTransaction operation."
+
+[properties]
+enabled = ["write concern"]
+retryable = {mode = "once per command", type = "writes"}
+
+[command]
+name = "abortTransaction"
+parameter = "database"
+
+[request.recoveryToken]
+type = "document"
+documentation = """
+RecoveryToken sets the recovery token to use when committing or aborting a sharded transaction.\
+"""

x/mongo/driver/operation/abort_transaction.toml

@@ -10,3 +10,4 @@ package operation
 //go:generate operationgen drop_indexes.toml operation drop_indexes.go
 //go:generate operationgen drop_database.toml operation drop_database.go
 //go:generate operationgen commit_transaction.toml operation commit_transaction.go
+//go:generate operationgen abort_transaction.toml operation abort_transaction.go