From 0a148721de3e3be5e94213aad06e65bc4bb06c7c Mon Sep 17 00:00:00 2001
From: Kitaiti Makoto <KitaitiMakoto@gmail.com>
Date: Thu, 7 Jan 2021 04:46:35 +0900
Subject: [PATCH] Define SearchActor

---
 plume-models/src/search/actor.rs | 58 ++++++++++++++++++++++++++++++++
 plume-models/src/search/mod.rs   |  1 +
 2 files changed, 59 insertions(+)
 create mode 100644 plume-models/src/search/actor.rs

diff --git a/plume-models/src/search/actor.rs b/plume-models/src/search/actor.rs
new file mode 100644
index 00000000..9a939f93
--- /dev/null
+++ b/plume-models/src/search/actor.rs
@@ -0,0 +1,58 @@
+use super::Searcher;
+use crate::{db_conn::DbPool, posts::PostEvent, POST_CHAN};
+use riker::actors::{Actor, ActorFactoryArgs, Context, Sender, Subscribe, Tell};
+use std::sync::Arc;
+use tracing::error;
+
+pub struct SearchActor {
+    searcher: Arc<Searcher>,
+    conn: DbPool,
+}
+
+impl Actor for SearchActor {
+    type Msg = PostEvent;
+
+    fn pre_start(&mut self, ctx: &Context<Self::Msg>) {
+        &POST_CHAN.tell(
+            Subscribe {
+                actor: Box::new(ctx.myself()),
+                topic: "*".into(),
+            },
+            None,
+        );
+    }
+
+    fn recv(&mut self, _ctx: &Context<Self::Msg>, msg: Self::Msg, _sender: Sender) {
+        use PostEvent::*;
+
+        match msg {
+            PostPublished(post) => {
+                let conn = self.conn.get();
+                if conn.is_ok() {
+                    self.searcher
+                        .add_document(&conn.unwrap(), &post)
+                        .unwrap_or_else(|e| error!("{:?}", e));
+                } else {
+                    error!("Failed to get database connection");
+                }
+            }
+            PostUpdated(post) => {
+                let conn = self.conn.get();
+                if conn.is_ok() {
+                    self.searcher
+                        .update_document(&conn.unwrap(), &post)
+                        .unwrap_or_else(|e| error!("{:?}", e));
+                } else {
+                    error!("Failed to get database connection");
+                }
+            }
+            PostDeleted(post) => self.searcher.delete_document(&post),
+        }
+    }
+}
+
+impl ActorFactoryArgs<(Arc<Searcher>, DbPool)> for SearchActor {
+    fn create_args((searcher, conn): (Arc<Searcher>, DbPool)) -> Self {
+        Self { searcher, conn }
+    }
+}
diff --git a/plume-models/src/search/mod.rs b/plume-models/src/search/mod.rs
index 83b9bf62..33de8d49 100644
--- a/plume-models/src/search/mod.rs
+++ b/plume-models/src/search/mod.rs
@@ -1,3 +1,4 @@
+pub mod actor;
 mod query;
 mod searcher;
 mod tokenizer;