From f00c8fe6e13d3171905cb601c7c1f3580b27347d Mon Sep 17 00:00:00 2001
From: Pol Pinol Castuera <pol.pinol@mongodb.com>
Date: Thu, 29 May 2025 09:47:03 +0200
Subject: [PATCH] Read with the correct doc from mongo

---
 .../infrastructure/listeners/PubSubUsersSubscriber.java     | 2 +-
 .../read_models/MongoUsersWalletReadModel.java              | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/main/java/io/autoinvestor/infrastructure/listeners/PubSubUsersSubscriber.java b/src/main/java/io/autoinvestor/infrastructure/listeners/PubSubUsersSubscriber.java
index 777a642..f6851bd 100644
--- a/src/main/java/io/autoinvestor/infrastructure/listeners/PubSubUsersSubscriber.java
+++ b/src/main/java/io/autoinvestor/infrastructure/listeners/PubSubUsersSubscriber.java
@@ -85,7 +85,7 @@ private void processMessage(PubsubMessage message, AckReplyConsumer consumer) {
                         event.getAggregateId()
                 );
                 this.commandHandler.handle(command);
-                log.info("Decision registered for userId={} msgId={}", command.userId(), msgId);
+                log.info("User registered for userId={} msgId={}", command.userId(), msgId);
             } else {
                 log.debug("Ignored unsupported event type={} msgId={}", event.getType(), msgId);
             }
diff --git a/src/main/java/io/autoinvestor/infrastructure/read_models/MongoUsersWalletReadModel.java b/src/main/java/io/autoinvestor/infrastructure/read_models/MongoUsersWalletReadModel.java
index 02dd7b3..9cd61bb 100644
--- a/src/main/java/io/autoinvestor/infrastructure/read_models/MongoUsersWalletReadModel.java
+++ b/src/main/java/io/autoinvestor/infrastructure/read_models/MongoUsersWalletReadModel.java
@@ -24,10 +24,10 @@ public void add(UsersWalletReadModelDTO dto) {
 
     @Override
     public String getWalletId(String userId) {
-        UsersWalletReadModelDTO dto = template.findById(userId, UsersWalletReadModelDTO.class);
+        MongoUsersWalletReadModelDocument doc = template.findById(userId, MongoUsersWalletReadModelDocument.class);
 
-        if (dto != null) {
-            return dto.walletId();
+        if (doc != null) {
+            return doc.walletId();
         }
 
         return null;