From 39d1cd64bc5ffe250dc2a30d05d0b6312e7abbc1 Mon Sep 17 00:00:00 2001
From: Vincent Ambo <tazjin@gmail.com>
Date: Sat, 14 Apr 2018 22:05:06 +0200
Subject: feat(main/db): Schedule regular search view refresh

Schedules refreshes of the database view used for search at one-minute
intervals.
---
 src/db.rs     | 21 +++++++++++++++++++++
 src/errors.rs | 10 ++++++++++
 src/main.rs   | 19 +++++++++++++++++++
 3 files changed, 50 insertions(+)

(limited to 'src')

diff --git a/src/db.rs b/src/db.rs
index 416e3fdd0f52..a4f86977c5f4 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -177,3 +177,24 @@ impl Handler<SearchPosts> for DbExecutor {
         Ok(search_results)
     }
 }
+
+/// Message that triggers a refresh of the view used for full-text
+/// searching.
+pub struct RefreshSearchView;
+
+impl Message for RefreshSearchView {
+    type Result = Result<()>;
+}
+
+const REFRESH_QUERY: &'static str = "REFRESH MATERIALIZED VIEW search_index";
+
+impl Handler<RefreshSearchView> for DbExecutor {
+    type Result = Result<()>;
+
+    fn handle(&mut self, _: RefreshSearchView, _: &mut Self::Context) -> Self::Result {
+        let conn = self.0.get()?;
+        debug!("Refreshing search_index view in DB");
+        sql_query(REFRESH_QUERY).execute(&conn)?;
+        Ok(())
+    }
+}
diff --git a/src/errors.rs b/src/errors.rs
index c7c1e14f4312..9de907430d6e 100644
--- a/src/errors.rs
+++ b/src/errors.rs
@@ -30,6 +30,7 @@ use diesel;
 use r2d2;
 use reqwest;
 use tera;
+use tokio_timer;
 
 pub type Result<T> = result::Result<T, ConverseError>;
 
@@ -50,6 +51,9 @@ pub enum ConverseError {
     #[fail(display = "error occured during request handling: {}", error)]
     ActixWeb { error: actix_web::Error },
 
+    #[fail(display = "error occured running timer: {}", error)]
+    Timer { error: tokio_timer::Error },
+
     // This variant is used as a catch-all for wrapping
     // actix-web-compatible response errors, such as the errors it
     // throws itself.
@@ -99,6 +103,12 @@ impl From<reqwest::Error> for ConverseError {
     }
 }
 
+impl From<tokio_timer::Error> for ConverseError {
+    fn from(error: tokio_timer::Error) -> ConverseError {
+        ConverseError::Timer { error }
+    }
+}
+
 // Support conversion of error type into HTTP error responses:
 
 impl ResponseError for ConverseError {
diff --git a/src/main.rs b/src/main.rs
index 61969bcff92a..9f4eea03fa31 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -42,6 +42,8 @@ extern crate rand;
 extern crate reqwest;
 extern crate serde;
 extern crate serde_json;
+extern crate tokio;
+extern crate tokio_timer;
 extern crate url;
 extern crate url_serde;
 
@@ -86,6 +88,8 @@ fn main() {
 
     let db_addr = SyncArbiter::start(2, move || DbExecutor(pool.clone()));
 
+    schedule_search_refresh(db_addr.clone());
+
     info!("Initialising OIDC integration ...");
     let oidc_url = config("OIDC_DISCOVERY_URL");
     let oidc_config = oidc::load_oidc(&oidc_url)
@@ -162,3 +166,18 @@ fn main() {
 
     let _ = sys.run();
 }
+
+fn schedule_search_refresh(db: Addr<Syn, DbExecutor>) {
+    use tokio::prelude::*;
+    use tokio::timer::Interval;
+    use std::time::{Duration, Instant};
+    use std::thread;
+
+    let task = Interval::new(Instant::now(), Duration::from_secs(60))
+        .from_err()
+        .for_each(move |_| db.send(db::RefreshSearchView).flatten())
+        .map_err(|err| error!("Error while updating search view: {}", err));
+        //.and_then(|_| debug!("Refreshed search view in DB"));
+
+    thread::spawn(|| tokio::run(task));
+}
-- 
cgit 1.4.1