From ac1afe4c7b0f62160f62b848d7f747cb7be74bbd Mon Sep 17 00:00:00 2001
From: Jokler <jokler.contact@gmail.com>
Date: Sun, 25 Feb 2018 01:30:17 +0100
Subject: Return zero if count_factoids could not find the factoid

---
 src/plugins/factoids/database.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/plugins')

diff --git a/src/plugins/factoids/database.rs b/src/plugins/factoids/database.rs
index ccebfee..386d3f7 100644
--- a/src/plugins/factoids/database.rs
+++ b/src/plugins/factoids/database.rs
@@ -152,14 +152,17 @@ impl Database for Pool<ConnectionManager<MysqlConnection>> {
     }
 
     fn count_factoids(&self, name: &str) -> Result<i32, &'static str> {
+        use diesel;
+
         let conn = &*self.get().expect("Failed to get connection");
         let count: Result<i64, _> = factoids::table
             .filter(factoids::columns::name.eq(name))
             .count()
-            .first(conn);
+            .get_result(conn);
 
         match count {
             Ok(c) => Ok(c as i32),
+            Err(diesel::NotFound) => Ok(0),
             Err(e) => {
                 error!("DB Count Error: {}", e);
                 Err("Database Error")
-- 
cgit v1.2.3-70-g09d2