Skip to content

Commit dcf9045

Browse files
authored
typosquat/database: Sort by downloads from crate_downloads table (#8251)
1 parent c5d4a0f commit dcf9045

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

src/typosquat/database.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ impl TopCrates {
2020
pub fn new(conn: &mut PgConnection, num: i64) -> QueryResult<Self> {
2121
use crate::{
2222
models,
23-
schema::{crate_owners, crates},
23+
schema::{crate_downloads, crate_owners},
2424
};
2525
use diesel::prelude::*;
2626

@@ -43,7 +43,8 @@ impl TopCrates {
4343

4444
let mut crates: BTreeMap<i32, (String, Crate)> = BTreeMap::new();
4545
for result in models::Crate::all()
46-
.order(crates::downloads.desc())
46+
.inner_join(crate_downloads::table)
47+
.order(crate_downloads::downloads.desc())
4748
.limit(num)
4849
.load_iter::<models::Crate, DefaultLoadingMode>(conn)?
4950
{

src/typosquat/test_util.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ use crate::{
66
models::{
77
Crate, CrateOwner, NewCrate, NewTeam, NewUser, NewVersion, Owner, OwnerKind, User, Version,
88
},
9-
schema::{crate_owners, crates},
9+
schema::{crate_downloads, crate_owners},
1010
Emails,
1111
};
1212

@@ -60,9 +60,9 @@ impl Faker {
6060
}
6161
.create(conn, user.id)?;
6262

63-
diesel::update(crates::table)
64-
.filter(crates::id.eq(krate.id))
65-
.set(crates::downloads.eq(downloads))
63+
diesel::update(crate_downloads::table)
64+
.filter(crate_downloads::crate_id.eq(krate.id))
65+
.set(crate_downloads::downloads.eq(downloads as i64))
6666
.execute(conn)?;
6767

6868
let version = NewVersion::new(

0 commit comments

Comments
 (0)