twiq: Fix in_memory_user_repository

bouzuya · Oct 6, 2022 · 306ed98 · 306ed98
1 parent 01022a1
commit 306ed98
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/twiq/crates/db/src/in_memory_user_repository.rs b/twiq/crates/db/src/in_memory_user_repository.rs
@@ -11,14 +11,19 @@ use crate::in_memory_event_store::InMemoryEventStore;
 #[derive(Debug, Default)]
 pub struct InMemoryUserRepository {
     event_store: Arc<Mutex<InMemoryEventStore>>,
+    user_ids: Arc<Mutex<HashMap<UserId, EventStreamId>>>,
     index: Arc<Mutex<HashMap<TwitterUserId, UserId>>>,
 }
 
 #[async_trait]
 impl UserRepository for InMemoryUserRepository {
     async fn find(&self, id: UserId) -> Result<Option<User>> {
         let event_store = self.event_store.lock().await;
-        let event_stream_id = EventStreamId::from(id);
+        let user_ids = self.user_ids.lock().await;
+        let event_stream_id = match user_ids.get(&id) {
+            None => return Ok(None),
+            Some(event_stream_id) => *event_stream_id,
+        };
         let event_stream = event_store
             .find_event_stream(event_stream_id)
             .await
@@ -47,6 +52,7 @@ impl UserRepository for InMemoryUserRepository {
 
     async fn store(&self, before: Option<User>, after: User) -> Result<()> {
         let event_store = self.event_store.lock().await;
+        let mut user_ids = self.user_ids.lock().await;
         let mut index = self.index.lock().await;
         event_store
             .store(
@@ -55,6 +61,7 @@ impl UserRepository for InMemoryUserRepository {
             )
             .await
             .map_err(|e| Error::Unknown(e.to_string()))?;
+        user_ids.insert(after.id(), after.event_stream().id());
         index.insert(after.twitter_user_id().clone(), after.id());
         Ok(())
     }

diff --git a/twiq/crates/domain/src/aggregate/user.rs b/twiq/crates/domain/src/aggregate/user.rs
@@ -66,6 +66,10 @@ impl User {
         Ok(())
     }
 
+    pub fn event_stream(&self) -> &EventStream {
+        &self.event_stream
+    }
+
     pub fn twitter_user_id(&self) -> &TwitterUserId {
         &self.twitter_user_id
     }