elsanctum/tenquestionmarks
2
0
Fork 0
Code Issues 8 Pull Requests Releases Wiki Activity

implement a simple cache for autolink module

Browse Source
This commit is contained in:
Adrian Malacoda 2018-02-22 03:07:27 -06:00
parent ca6f3391d4
commit bcbc14ed17
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/modules/autolink.rs
View File

@ -15,6 +15,7 @@ use stc::searchers::mtg::{MtgCard, MtgSearcher};
use stc::searchers::mediawiki::MediawikiSearcher; use stc::searchers::mediawiki::MediawikiSearcher;
use regex::Regex; use regex::Regex;
use std::collections::BTreeMap;
pub struct AutolinkModule {} pub struct AutolinkModule {}
@ -89,14 +90,17 @@ impl EventLoop for AutolinkModule {
searchers.add_searcher("gcl", Box::new(MediawikiSearcher::new(String::from("https://glitchcity.info/wiki/")))); searchers.add_searcher("gcl", Box::new(MediawikiSearcher::new(String::from("https://glitchcity.info/wiki/"))));
searchers.add_searcher("wp", Box::new(MediawikiSearcher::new(String::from("https://en.wikipedia.org/wiki/")))); searchers.add_searcher("wp", Box::new(MediawikiSearcher::new(String::from("https://en.wikipedia.org/wiki/"))));
let mut searcher_cache = BTreeMap::new();
loop { loop {
match receiver.recv() { match receiver.recv() {
Ok(envelope) => { Ok(envelope) => {
if let Event::Message { ref message } = envelope.event { if let Event::Message { ref message } = envelope.event {
debug!("Received message from module {:?}... {:?}", envelope.from, message.content); debug!("Received message from module {:?}... {:?}", envelope.from, message.content);
for cap in link_regex.captures_iter(&message.content) { for cap in link_regex.captures_iter(&message.content) {
if let Some(item) = searchers.exact_search(&cap[1]) { let term = cap[1].to_owned();
print_any_link(&item, message); if let &mut Some(ref mut item) = searcher_cache.entry(term.to_owned()).or_insert_with(|| searchers.exact_search(&term)) {
print_any_link(item, message);
if let Some(card) = item.downcast_ref::<MtgCard>() { if let Some(card) = item.downcast_ref::<MtgCard>() {
print_mtg_card(card, message); print_mtg_card(card, message);
} else if let Some(card) = item.downcast_ref::<YugiohCard>() { } else if let Some(card) = item.downcast_ref::<YugiohCard>() {