0.13.1 - Cache Indices

Author: ShayBox

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "vrc-log"
-version = "0.13.0"
+version = "0.13.1"
 authors = ["Shayne Hartford <shaybox@shaybox.com>"]
 edition = "2021"
 description = "VRChat Local Avatar ID Logger"

Cargo.toml

@@ -3,7 +3,7 @@ use std::{time::Duration, vec};
 use anyhow::{bail, Result};
 use async_trait::async_trait;
 use reqwest::{Client, StatusCode, Url};
-use serde::{Deserialize, Serialize};
+use serde::Deserialize;
 use serde_json::{json, Value};
 
 use crate::{

src/provider/avtrdb.rs

@@ -56,19 +56,25 @@ impl Cache {
                             ADD COLUMN updated_at DATETIME
                         ", [])?;
                     }
+                }
 
-                    #[rustfmt::skip] // Prevent a large burst after updating
-                    connection.execute("
-                        UPDATE avatars
-                        SET updated_at = datetime('now', '-31 days')
-                        WHERE updated_at IS NULL
-                    ", [])?;
-
-                    // Print cache statistics
-                    if let Ok(mut statement) = connection.prepare("SELECT COUNT(*) FROM avatars") {
-                        if let Ok(count) = statement.query_row([], |row| row.get::<_, i64>(0)) {
-                            info!("{} Cached Avatars", count);
-                        }
+                #[rustfmt::skip] // Prevent a large burst after updating
+                connection.execute("
+                    UPDATE avatars
+                    SET updated_at = datetime('now', '-31 days')
+                    WHERE updated_at IS NULL
+                ", [])?;
+
+                #[rustfmt::skip] // Speed up queries on large databases
+                connection.execute("
+                    CREATE INDEX IF NOT EXISTS idx_avatars_updated_at
+                    ON avatars(updated_at)
+                ", [])?;
+
+                // Print cache statistics
+                if let Ok(mut statement) = connection.prepare("SELECT COUNT(*) FROM avatars") {
+                    if let Ok(count) = statement.query_row([], |row| row.get::<_, i64>(0)) {
+                        info!("{} Cached Avatars", count);
                     }
                 }
 
@@ -89,9 +95,10 @@ impl Provider for Cache {
     async fn check_avatar_id(&self, avatar_id: &str) -> Result<bool> {
         let id = avatar_id.to_string();
         let query = "
-            SELECT 1 FROM avatars
-            WHERE id = (?) AND updated_at >= datetime('now', '-30 days')
-            LIMIT 1
+            SELECT EXISTS(
+                SELECT 1 FROM avatars
+                WHERE id = ? AND updated_at >= datetime('now', '-30 days')
+            )
         ";
 
         let is_ok = self

src/provider/cache.rs

@@ -1,12 +1,12 @@
-use std::{collections::HashMap, time::Duration};
+use std::time::Duration;
 
 use anyhow::{bail, Result};
 use async_trait::async_trait;
 use reqwest::{Client, StatusCode};
 use serde_json::json;
 
 use crate::{
-    provider::{avtrdb::AvtrDB, Provider, ProviderKind},
+    provider::{Provider, ProviderKind},
     settings::Settings,
     USER_AGENT,
 };