From 6dd3236069d5cbdd6cb946b408cd6efb7e91d0f9 Mon Sep 17 00:00:00 2001
From: "Carpenter, Adam (CORP)" <adam.carpenter@adp.com>
Date: Fri, 26 Nov 2021 15:58:13 -0500
Subject: feat: impl lazy-refreshing cache for quick access to be used for
 haystack

---
 angelsharkd/src/routes/extensions/mod.rs           |  4 +-
 angelsharkd/src/routes/extensions/simple_search.rs | 54 +++++++++++++++++++---
 angelsharkd/src/routes/mod.rs                      |  1 -
 3 files changed, 51 insertions(+), 8 deletions(-)

(limited to 'angelsharkd/src/routes')

diff --git a/angelsharkd/src/routes/extensions/mod.rs b/angelsharkd/src/routes/extensions/mod.rs
index 5dac1cf..defbca9 100644
--- a/angelsharkd/src/routes/extensions/mod.rs
+++ b/angelsharkd/src/routes/extensions/mod.rs
@@ -13,7 +13,9 @@ pub fn filter(config: &Config) -> impl Filter<Extract = impl Reply, Error = Reje
     let filters = default().or(default());
 
     #[cfg(feature = "simple_search")]
-    let filters = filters.or(simple_search::filter());
+    let filters = filters
+        .or(simple_search::search(config))
+        .or(simple_search::refresh());
 
     #[cfg(feature = "simple_deprov")]
     let filters = filters.or(simple_deprov::filter());
diff --git a/angelsharkd/src/routes/extensions/simple_search.rs b/angelsharkd/src/routes/extensions/simple_search.rs
index 454dba3..58f6e52 100644
--- a/angelsharkd/src/routes/extensions/simple_search.rs
+++ b/angelsharkd/src/routes/extensions/simple_search.rs
@@ -1,14 +1,56 @@
-use crate::config::Config;
+use crate::{config::Config, routes::with_runner};
+use cached::proc_macro::cached;
+use libangelshark::AcmRunner;
 use std::convert::Infallible;
-use warp::{path, post, Filter, Rejection, Reply};
+use warp::{
+    body::{content_length_limit, json},
+    get, path, post, Filter, Rejection, Reply,
+};
 
 const CMD_LIST_EXT: &str = "list extension-type";
 const CMD_LIST_STAT: &str = "list station";
 
-pub fn filter() -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
-    path("search").and(post()).and_then(handle_simple_search)
+type Terms = Vec<String>;
+
+pub fn search(config: &Config) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
+    let runner = config.runner.clone();
+    // TODO: anti-caching headers on resp?
+
+    path("search")
+        .and(post())
+        .and(content_length_limit(1024 * 16))
+        .and(json::<Terms>())
+        .and(with_runner(runner))
+        .and_then(handle_simple_search)
 }
 
-async fn handle_simple_search() -> Result<impl Reply, Infallible> {
-    Ok("Search!")
+pub fn refresh() -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
+    path!("search" / "refresh")
+        .and(get())
+        .and_then(handle_refresh)
+}
+
+async fn handle_refresh() -> Result<impl Reply, Infallible> {
+    fetch(String::from("test"), true);
+    Ok("")
+}
+
+async fn handle_simple_search(terms: Terms, runner: AcmRunner) -> Result<impl Reply, Infallible> {
+    Ok(fetch(terms[0].clone(), false))
+}
+
+#[cached]
+fn fetch(param: String, refresh: bool) -> String {
+    if refresh {
+        let param = param.clone();
+        tokio::spawn(async move {
+            std::thread::sleep_ms(10000);
+            let mut handle = FETCH.lock().unwrap();
+            handle.cache_clear();
+            handle.cache_set((param.clone(), false), format!("blargh {}", param.clone()));
+            eprintln!("evicted");
+        });
+    }
+
+    param
 }
diff --git a/angelsharkd/src/routes/mod.rs b/angelsharkd/src/routes/mod.rs
index 52373d6..cee0657 100644
--- a/angelsharkd/src/routes/mod.rs
+++ b/angelsharkd/src/routes/mod.rs
@@ -13,7 +13,6 @@ use warp::{
 };
 
 mod dtos;
-#[cfg(feature = "extensions")]
 pub mod extensions;
 
 /// GET / -> Name and version # of app.
-- 
cgit v1.2.3