implement RR feedback

Author: stephmarie17

source/includes/write/transaction.kt

@@ -1,32 +1,36 @@
 import com.mongodb.*
 import com.mongodb.client.ClientSession
 import com.mongodb.client.MongoClients
-import com.mongodb.client.MongoCollection
 import org.bson.Document
 
+// start-data-class
+data class Restaurant(val name: String, val cuisine: String)
+// end-data-class
+
 fun main() {
 // start-transaction
     // Creates a new MongoClient with a connection string
     val client = MongoClients.create("<connection string>")
 
     // Gets the database and collection
-    val restaurantsDb = client.getDatabase("sample_restaurants")
-    val restaurantsCollection: MongoCollection<Document> = restaurantsDb.getCollection("restaurants")
+    val database = client.getDatabase("sample_restaurants")
+    val collection = database.getCollection<Restaurant>("restaurants")
 
-    fun insertDocuments(session: ClientSession) {
+    // Defines options and inserts restaurants into the collection
+    fun insertRestaurantsInTransaction(session: ClientSession) {
         // Sets options for the collection operation within the transaction
-        val restaurantsCollectionWithOptions = restaurantsCollection
+        val restaurantsCollectionWithOptions = collection
             .withWriteConcern(WriteConcern("majority"))
             .withReadConcern(ReadConcern.LOCAL)
 
-        // Inserts documents within the transaction
+        // Inserts restaurants within the transaction
         restaurantsCollectionWithOptions.insertOne(
             session,
-            Document("name", "Kotlin Sync Pizza").append("cuisine", "Pizza")
+            Restaurant("name", "Kotlin Sync Pizza").append("cuisine", "Pizza")
         )
         restaurantsCollectionWithOptions.insertOne(
             session,
-            Document("name", "Kotlin Sync Burger").append("cuisine", "Burger")
+            Restaurant("name", "Kotlin Sync Burger").append("cuisine", "Burger")
         )
     }
 
@@ -40,7 +44,7 @@ fun main() {
 
             // Uses the withTransaction method to start a transaction and run the given function
             session.withTransaction({
-                insertDocuments(session)
+                insertRestaurantsInTransaction(session)
                 println("Transaction succeeded")
             }, txnOptions)
         } catch (e: Exception) {

source/write/transactions.txt

@@ -55,6 +55,15 @@ from the :atlas:`Atlas sample datasets </sample-data>`. To learn how to create a
 free MongoDB Atlas cluster and load the sample datasets, see the
 :atlas:`Get Started with Atlas </getting-started>` guide.
 
+The documents in this collection are modeled by the following {+language+} data class:
+
+.. literalinclude:: /includes/write/transaction.kt
+   :start-after: start-data-class
+   :end-before: end-data-class
+   :language: kotlin
+   :copyable:
+   :dedent:
+
 Methods
 -------
 
@@ -77,7 +86,7 @@ manage your transaction:
          this method, see the :manual:`startTransaction() page
          </reference/method/Session.startTransaction/>`  in the Server manual.
        |
-       | **Parameter**: ``transactionOptions``
+       | **Parameter**: ``TransactionOptions``
 
    * - ``abortTransaction()``
      - | Ends the active transaction for this session. Returns an
@@ -98,7 +107,7 @@ manage your transaction:
      - | Starts a transaction on this session and runs the given function within 
          a transaction.
        |
-       | **Parameters**: ``transactionBody, options``
+       | **Parameters**: transaction body function, ``TransactionOptions``
 
 Example
 -------