database/models/download: Implement Selectable trait

Author: Turbo87

crates/crates_io_database/src/models/download.rs

@@ -4,18 +4,16 @@ use chrono::NaiveDate;
 use crates_io_diesel_helpers::SemverVersion;
 use diesel::prelude::*;
 
-#[derive(Queryable, Identifiable, Associations, Debug, Clone, Copy)]
+#[derive(Queryable, Identifiable, Selectable, Associations, Debug, Clone, Copy)]
 #[diesel(
     primary_key(version_id, date),
     belongs_to(FullVersion, foreign_key=version_id),
     belongs_to(Version),
 )]
 pub struct VersionDownload {
     pub version_id: i32,
-    pub downloads: i32,
-    pub counted: i32,
     pub date: NaiveDate,
-    pub processed: bool,
+    pub downloads: i32,
 }
 
 /// A subset of the columns of the `versions` table.

src/controllers/krate/downloads.rs

@@ -111,6 +111,7 @@ pub async fn get_crate_downloads(
     let (downloads, extra_downloads, versions_and_publishers, actions) = tokio::try_join!(
         VersionDownload::belonging_to(latest_five)
             .filter(version_downloads::date.gt(date(now - 90.days())))
+            .select(VersionDownload::as_select())
             .order((
                 version_downloads::date.asc(),
                 version_downloads::version_id.desc(),

src/controllers/version/downloads.rs

@@ -93,6 +93,7 @@ pub async fn get_version_downloads(
 
     let version_downloads = VersionDownload::belonging_to(&version)
         .filter(version_downloads::date.between(cutoff_start_date, cutoff_end_date))
+        .select(VersionDownload::as_select())
         .order(version_downloads::date)
         .load(&mut conn)
         .await?