Edit

sendinblue · Mar 29, 2024 · 169168d · 169168d
1 parent d87518d
commit 169168d
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 157 deletions.
diff --git a/airbyte-integrations/connectors/source-mongodb-v2/build.gradle b/airbyte-integrations/connectors/source-mongodb-v2/build.gradle
@@ -51,7 +51,7 @@ dependencies {
     dataGeneratorImplementation ('com.github.javafaker:javafaker:1.0.2') { exclude module: 'snakeyaml' }
     dataGeneratorImplementation 'io.github.oshai:kotlin-logging-jvm:5.1.0'
     dataGeneratorImplementation 'org.jetbrains.kotlinx:kotlinx-cli-jvm:0.3.5'
-    dataGeneratorImplementation 'org.mongodb:mongodb-driver-sync:3.12.10'
+    dataGeneratorImplementation 'org.mongodb:mongodb-driver-sync:4.10.2'
 
     debeziumTestImplementation 'io.debezium:debezium-embedded:2.4.0.Final'
     debeziumTestImplementation 'io.debezium:debezium-connector-mongodb:2.4.0.Final'

diff --git a/...ngodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/InitialSnapshotHandler.java b/...ngodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/InitialSnapshotHandler.java
@@ -5,6 +5,7 @@
 package io.airbyte.integrations.source.mongodb;
 
 import com.mongodb.client.MongoCollection;
+import com.mongodb.client.MongoDatabase;
 import com.mongodb.client.model.*;
 import io.airbyte.cdk.integrations.source.relationaldb.state.SourceStateIterator;
 import io.airbyte.cdk.integrations.source.relationaldb.state.StateEmitFrequency;

diff --git a/...source-mongodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/MongoDatabase.java b/...source-mongodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/MongoDatabase.java
diff --git a/...ors/source-mongodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/MongoUtil.java b/...ors/source-mongodb-v2/src/main/java/io/airbyte/integrations/source/mongodb/MongoUtil.java
@@ -17,6 +17,7 @@
 import com.mongodb.client.MongoClient;
 import com.mongodb.client.MongoCollection;
 import com.mongodb.client.MongoCursor;
+import com.mongodb.client.MongoDatabase;
 import com.mongodb.client.MongoIterable;
 import com.mongodb.client.model.Aggregates;
 import com.mongodb.client.model.Projections;
@@ -176,7 +177,7 @@ public static int getDebeziumEventQueueSize(final MongoDbSourceConfig config) {
   public static Optional<CollectionStatistics> getCollectionStatistics(final MongoClient mongoClient, final ConfiguredAirbyteStream stream) {
     try {
       final Map<String, Object> collStats = Map.of(MongoConstants.STORAGE_STATS_KEY, Map.of(), MongoConstants.COUNT_KEY, Map.of());
-      final MongoDatabase mongoDatabase = (MongoDatabase) mongoClient.getDatabase(stream.getStream().getNamespace());
+      final MongoDatabase mongoDatabase = mongoClient.getDatabase(stream.getStream().getNamespace());
       final MongoCollection<Document> collection = mongoDatabase.getCollection(stream.getStream().getName());
       final AggregateIterable<Document> output = collection.aggregate(List.of(new Document("$collStats", collStats)));
 

diff --git a/...ce-mongodb-v2/src/test/java/io/airbyte/integrations/source/mongodb/MongoDbSourceTest.java b/...ce-mongodb-v2/src/test/java/io/airbyte/integrations/source/mongodb/MongoDbSourceTest.java
@@ -15,7 +15,6 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import com.mongodb.MongoCredential;
 import com.mongodb.MongoSecurityException;
-
 import com.mongodb.client.*;
 import com.mongodb.connection.ClusterDescription;
 import com.mongodb.connection.ClusterType;