src/inventory/adapters/output/db/postgres/category_view.rs

@@ -3,7 +3,6 @@
 // SPDX-License-Identifier: AGPL-3.0-or-later
 
 use async_trait::async_trait;
-use cqrs_es::persist::GenericQuery;
 use cqrs_es::persist::{PersistenceError, ViewContext, ViewRepository};
 use cqrs_es::{EventEnvelope, Query, View};
 use serde::{Deserialize, Serialize};
@@ -179,7 +178,22 @@ impl Query<Category> for SimpleLoggingQuery {
     }
 }
 
-// Our second query, this one will be handled with Postgres `GenericQuery`
-// which will serialize and persist our view after it is updated. It also
-// provides a `load` method to deserialize the view on request.
-pub type CategoryQuery = GenericQuery<InventoryDBPostgresAdapter, CategoryView, Category>;
+#[async_trait]
+impl Query<Category> for InventoryDBPostgresAdapter {
+    async fn dispatch(&self, category_id: &str, events: &[EventEnvelope<Category>]) {
+        let res = self
+            .load_with_context(&category_id)
+            .await
+            .unwrap_or_else(|_| {
+                Some((
+                    CategoryView::default(),
+                    ViewContext::new(category_id.into(), 0),
+                ))
+            });
+        let (mut view, view_context): (CategoryView, ViewContext) = res.unwrap();
+        for event in events {
+            view.update(event);
+        }
+        self.update_view(view, view_context).await.unwrap();
+    }
+}