CSOT

Author: rustagir

examples/src/test/kotlin/CompoundTest.kt

@@ -81,7 +81,6 @@ internal class CompoundOperationsTest {
  val update = Updates.set(FoodOrder::food.name, "pizza")
  val options = FindOneAndUpdateOptions()
  .upsert(true)
- .maxTime(5, TimeUnit.SECONDS)
  /* The result variable contains your document in the
  state before your update operation is performed
  or null if the document was inserted due to upsert
@@ -183,4 +182,4 @@ internal class CompoundOperationsTest {
  assertEquals("joe", roomAfterSafe.first().guest)
  assertTrue(roomAfterSafe.first().reserved)
  }
-}
+}

examples/src/test/kotlin/CsotTest.kt

@@ -0,0 +1,144 @@
+
+import com.mongodb.ClientSessionOptions
+import com.mongodb.ConnectionString
+import com.mongodb.MongoClientSettings
+import com.mongodb.TransactionOptions
+import com.mongodb.client.cursor.TimeoutMode
+import com.mongodb.client.model.Filters
+import com.mongodb.kotlin.client.coroutine.MongoClient
+import config.getConfig
+import kotlinx.coroutines.runBlocking
+import org.bson.Document
+import org.junit.jupiter.api.AfterAll
+import org.junit.jupiter.api.BeforeAll
+import org.junit.jupiter.api.Test
+import java.util.concurrent.TimeUnit
+import kotlin.test.Ignore
+
+class CsotTest {
+
+ companion object {
+ val config = getConfig()
+ val CONNECTION_URI_PLACEHOLDER = config.connectionUri
+ val client = MongoClient.create(CONNECTION_URI_PLACEHOLDER)
+ val database = client.getDatabase("db")
+ val collection = database.getCollection<Document>("people")
+
+ @BeforeAll
+ @JvmStatic
+ fun beforeAll() {
+ runBlocking {
+ val people = listOf(
+ Document("name", "Shelley Price").append("age", 56),
+ Document("name", "Garrett John").append("age", 39),
+ Document("name", "Kalima Sheik").append("age", 26)
+ )
+ collection.insertMany(people)
+ }
+ }
+
+ @AfterAll
+ @JvmStatic
+ fun afterAll() {
+ runBlocking {
+ database.drop()
+ client.close()
+ }
+ }
+ }
+
+ @Ignore
+ fun connectionStringTest() = runBlocking {
+ // :snippet-start: connection-string-timeout
+ val uri = "<connection string>/?timeoutMS=200"
+ val client = MongoClient.create(uri)
+ // :snippet-end:
+ }
+
+ // :replace-start: {
+ // "terms": {
+ // "uri": "\"<connection string>\""
+ // }
+ // }
+
+ @Test
+ fun mongoClientSettingsTest() = runBlocking {
+ val uri = CONNECTION_URI_PLACEHOLDER
+ // :snippet-start: mongoclientsettings-timeout
+ val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString(uri))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+ val client = MongoClient.create(settings)
+ // :snippet-end:
+ }
+
+ @Test
+ fun operationTimeoutTest() = runBlocking {
+ val uri = CONNECTION_URI_PLACEHOLDER
+ // :snippet-start: operation
+ val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString(uri))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+ val client = MongoClient.create(settings)
+ val collection = client
+ .getDatabase("db")
+ .getCollection<Document>("people")
+
+ collection.insertOne(Document("name", "Francine Loews"))
+ // :snippet-end:
+ }
+
+ @Test
+ fun overrideTimeoutTest() = runBlocking {
+ val uri = CONNECTION_URI_PLACEHOLDER
+ // :snippet-start: override
+ val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString(uri))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+ val client = MongoClient.create(settings)
+ val database = client.getDatabase("db")
+ val collection = database
+ .getCollection<Document>("people")
+ .withTimeout(300L, TimeUnit.MILLISECONDS)
+ // :snippet-end:
+ }
+
+ @Test
+ fun transactionTimeoutTest() = runBlocking {
+ // :snippet-start: session
+ val opts = ClientSessionOptions.builder()
+ .defaultTimeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+ val session = client.startSession(opts)
+ // ... perform operations on ClientSession
+ // :snippet-end:
+
+ // :snippet-start: transaction
+ val transactionOptions = TransactionOptions.builder()
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+ // :snippet-end:
+ }
+
+ @Test
+ fun cursorTimeoutTest() = runBlocking {
+ val collection = database
+ .getCollection<Document>("people")
+ .withTimeout(200L, TimeUnit.MILLISECONDS)
+
+ // :snippet-start: cursor-lifetime
+ val flowWithLifetimeTimeout = collection
+ .find(Filters.gte("age", 40))
+ .timeoutMode(TimeoutMode.CURSOR_LIFETIME)
+ // :snippet-end:
+ }
+}
+
+// :replace-end:

source/examples/generated/CompoundTest.snippet.find-one-update.kt

@@ -3,7 +3,6 @@ val filter = Filters.eq(FoodOrder::color.name, "green")
 val update = Updates.set(FoodOrder::food.name, "pizza")
 val options = FindOneAndUpdateOptions()
  .upsert(true)
- .maxTime(5, TimeUnit.SECONDS)
 /* The result variable contains your document in the
  state before your update operation is performed
  or null if the document was inserted due to upsert

source/examples/generated/CsotTest.snippet.connection-string-timeout.kt

@@ -0,0 +1,2 @@
+val uri = "<connection string>/?timeoutMS=200"
+val client = MongoClient.create(uri)

source/examples/generated/CsotTest.snippet.cursor-lifetime.kt

@@ -0,0 +1,3 @@
+val flowWithLifetimeTimeout = collection
+ .find(Filters.gte("age", 40))
+ .timeoutMode(TimeoutMode.CURSOR_LIFETIME)

source/examples/generated/CsotTest.snippet.mongoclientsettings-timeout.kt

@@ -0,0 +1,6 @@
+val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString("<connection string>"))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+val client = MongoClient.create(settings)

source/examples/generated/CsotTest.snippet.operation.kt

@@ -0,0 +1,11 @@
+val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString("<connection string>"))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+val client = MongoClient.create(settings)
+val collection = client
+ .getDatabase("db")
+ .getCollection<Document>("people")
+
+collection.insertOne(Document("name", "Francine Loews"))

source/examples/generated/CsotTest.snippet.override.kt

@@ -0,0 +1,10 @@
+val settings = MongoClientSettings.builder()
+ .applyConnectionString(ConnectionString("<connection string>"))
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+val client = MongoClient.create(settings)
+val database = client.getDatabase("db")
+val collection = database
+ .getCollection<Document>("people")
+ .withTimeout(300L, TimeUnit.MILLISECONDS)

source/examples/generated/CsotTest.snippet.session.kt

@@ -0,0 +1,6 @@
+val opts = ClientSessionOptions.builder()
+ .defaultTimeout(200L, TimeUnit.MILLISECONDS)
+ .build()
+
+val session = client.startSession(opts)
+// ... perform operations on ClientSession

source/examples/generated/CsotTest.snippet.transaction.kt

@@ -0,0 +1,3 @@
+val transactionOptions = TransactionOptions.builder()
+ .timeout(200L, TimeUnit.MILLISECONDS)
+ .build()