Method executeInTransactionMode

backend/plugin/db/spanner/spanner.go:198–228

(ctx context.Context, stmts []string, opts db.ExecuteOptions)

}

func (d *Driver) executeInTransactionMode(ctx context.Context, stmts []string, opts db.ExecuteOptions) (int64, error) {
	var rowCount int64

	// Log transaction start
	opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_BEGIN, "")

	committed := false
	if _, err := d.client.ReadWriteTransaction(ctx, func(ctx context.Context, rwt *spanner.ReadWriteTransaction) error {
		spannerStmts := []spanner.Statement{}
		for _, stmt := range stmts {
			spannerStmts = append(spannerStmts, spanner.NewStatement(stmt))
		}
		counts, err := rwt.BatchUpdate(ctx, spannerStmts)
		if err != nil {
			return err
		}
		for _, count := range counts {
			rowCount += count
		}
		committed = true
		return nil
	}); err != nil {
		opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_ROLLBACK, "")
		return 0, err
	}

	if committed {
		opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_COMMIT, "")
	}
	return rowCount, nil
}

func (d *Driver) executeInAutoCommitMode(ctx context.Context, stmts []string, _ db.ExecuteOptions) (int64, error) {
	var rowCount int64