Extracted Config

Author: timotheekelly

src/main/java/com/mongodb/frauddetector/config/MongoDBConfig.java

@@ -0,0 +1,25 @@
+package com.mongodb.frauddetector.config;
+
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoCollection;
+import com.mongodb.client.MongoDatabase;
+import org.bson.Document;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class MongoDBConfig {
+
+ private static final String DATABASE_NAME = "fraud";
+ private static final String TRANSACTIONS_COLLECTION = "transactions";
+
+ @Bean
+ public MongoDatabase fraudDatabase(MongoClient mongoClient) {
+ return mongoClient.getDatabase(DATABASE_NAME);
+ }
+
+ @Bean
+ public MongoCollection<Document> transactionsCollection(MongoDatabase fraudDatabase) {
+ return fraudDatabase.getCollection(TRANSACTIONS_COLLECTION);
+ }
+}

src/main/java/com/mongodb/frauddetector/service/TransactionChangeStreamListener.java

@@ -18,24 +18,21 @@
 public class TransactionChangeStreamListener {
  private static final Logger logger = LoggerFactory.getLogger(TransactionChangeStreamListener.class);
 
- private final MongoClient mongoClient;
  private final TransactionVectorSearchService vectorSearchService;
  private final ExecutorService executorService = Executors.newSingleThreadExecutor(); // Keeps it synchronous
+ private final MongoCollection<Document> transactionsCollection;
 
- public TransactionChangeStreamListener(MongoClient mongoClient, TransactionVectorSearchService vectorSearchService) {
- this.mongoClient = mongoClient;
+ public TransactionChangeStreamListener(TransactionVectorSearchService vectorSearchService, MongoCollection<Document> transactionsCollection) {
+ this.transactionsCollection = transactionsCollection;
  this.vectorSearchService = vectorSearchService;
  }
 
  public void startListening() {
  executorService.submit(() -> {
- MongoDatabase database = mongoClient.getDatabase("fraud");
- MongoCollection<Document> collection = database.getCollection("transactions");
-
  // Filter to only listen for INSERT operations
  List<Bson> pipeline = List.of(Aggregates.match(Filters.eq("operationType", "insert")));
 
- try (MongoCursor<ChangeStreamDocument<Document>> cursor = collection.watch(pipeline).iterator()) {
+ try (MongoCursor<ChangeStreamDocument<Document>> cursor = transactionsCollection.watch(pipeline).iterator()) {
  while (cursor.hasNext()) {
  ChangeStreamDocument<Document> change = cursor.next();
  Document transactionDoc = change.getFullDocument();

src/main/java/com/mongodb/frauddetector/service/TransactionSeeder.java

@@ -1,8 +1,6 @@
 package com.mongodb.frauddetector.service;
 
-import com.mongodb.client.MongoClient;
 import com.mongodb.client.MongoCollection;
-import com.mongodb.client.MongoDatabase;
 import com.mongodb.frauddetector.model.Customer;
 import com.mongodb.frauddetector.model.Transaction;
 import com.mongodb.frauddetector.repository.CustomerRepository;
@@ -19,17 +17,16 @@
 @Service
 public class TransactionSeeder {
  private static final Logger logger = LoggerFactory.getLogger(TransactionSeeder.class);
- private final MongoCollection<Document> transactionCollection;
  private final CustomerRepository customerRepository;
  private final EmbeddingGenerator embeddingGenerator;
  private final TransactionRepository transactionRepository;
  private final TransactionChangeStreamListener transactionChangeStreamListener;
+ private final MongoCollection<Document> transactionsCollection;
 
- public TransactionSeeder(MongoClient mongoClient, CustomerRepository customerRepository,
+ public TransactionSeeder(CustomerRepository customerRepository,
  EmbeddingGenerator embeddingGenerator, TransactionRepository transactionRepository,
- TransactionChangeStreamListener transactionChangeStreamListener) {
- MongoDatabase database = mongoClient.getDatabase("fraud");
- this.transactionCollection = database.getCollection("transactions");
+ TransactionChangeStreamListener transactionChangeStreamListener, MongoCollection<Document> transactionsCollection) {
+ this.transactionsCollection = transactionsCollection;
  this.transactionRepository = transactionRepository;
  this.customerRepository = customerRepository;
  this.embeddingGenerator = embeddingGenerator;
@@ -38,7 +35,7 @@ public TransactionSeeder(MongoClient mongoClient, CustomerRepository customerRep
 
  @PostConstruct
  public void seedTransactions() {
- if (transactionCollection.countDocuments() > 0) {
+ if (transactionsCollection.countDocuments() > 0) {
  logger.info("Transactions already seeded.");
  return;
  }

src/main/java/com/mongodb/frauddetector/service/TransactionVectorSearchService.java

@@ -1,8 +1,6 @@
 package com.mongodb.frauddetector.service;
 
-import com.mongodb.client.MongoClient;
 import com.mongodb.client.MongoCollection;
-import com.mongodb.client.MongoDatabase;
 import com.mongodb.client.model.Aggregates;
 import com.mongodb.client.model.Filters;
 import com.mongodb.client.model.Updates;
@@ -28,9 +26,8 @@ public class TransactionVectorSearchService {
  private static final int SEARCH_LIMIT = 5; // Number of similar transactions to retrieve
  private static final int NUM_CANDIDATES = 50; // Number of approximate neighbors to consider
 
- public TransactionVectorSearchService(MongoClient mongoClient) {
- MongoDatabase database = mongoClient.getDatabase("fraud");
- this.transactionCollection = database.getCollection("transactions");
+ public TransactionVectorSearchService(MongoCollection<Document> transactionCollection) {
+ this.transactionCollection = transactionCollection;
  }
 
  public void evaluateTransactionFraud(Document transactionDoc) {