Skip to content

Fix sort order in crates api #398

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Aug 9, 2016
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions src/krate.rs
Original file line number Diff line number Diff line change
Expand Up @@ -469,12 +469,17 @@ pub fn index(req: &mut Request) -> CargoResult<Response> {
let mut args = vec![&limit as &ToSql, &offset];
let (q, cnt) = query.get("q").map(|query| {
args.insert(0, query);
let rank_sort_sql = match sort {
"downloads" => format!("{}, rank DESC", sort_sql),
_ => format!("rank DESC, {}", sort_sql),
};
format!("{},", sort_sql); // Append Comma
(format!("SELECT crates.* FROM crates,
plainto_tsquery($1) q,
ts_rank_cd(textsearchable_index_col, q) rank
WHERE q @@ textsearchable_index_col
ORDER BY name = $1 DESC, rank DESC, {}
LIMIT $2 OFFSET $3", sort_sql),
ORDER BY name = $1 DESC, {}
LIMIT $2 OFFSET $3", rank_sort_sql),
"SELECT COUNT(crates.*) FROM crates,
plainto_tsquery($1) q
WHERE q @@ textsearchable_index_col".to_string())
Expand Down
66 changes: 66 additions & 0 deletions src/tests/krate.rs
Original file line number Diff line number Diff line change
Expand Up @@ -168,6 +168,72 @@ fn exact_match_first_on_queries() {
assert_eq!(json.crates[2].name, "foo");
}

#[test]
fn exact_match_on_queries_with_sort() {
let (_b, app, middle) = ::app();

let mut req = ::req(app, Method::Get, "/api/v1/crates");
let _ = ::mock_user(&mut req, ::user("foo"));
let mut krate = ::krate("foo");
krate.description = Some("bar baz const".to_string());
krate.downloads = 50;
let (k, _) = ::mock_crate(&mut req, krate.clone());
let mut krate2 = ::krate("bar");
krate2.description = Some("foo baz foo baz const".to_string());
krate2.downloads = 3333;
let (k2, _) = ::mock_crate(&mut req, krate2.clone());
let mut krate3 = ::krate("baz");
krate3.description = Some("foo bar foo bar foo bar const".to_string());
krate3.downloads = 100000;
let (k3, _) = ::mock_crate(&mut req, krate3.clone());
let mut krate4 = ::krate("other");
krate4.description = Some("other const".to_string());
krate4.downloads = 999999;
let (k4, _) = ::mock_crate(&mut req, krate4.clone());

{
let req2: &mut Request = &mut req;
let tx = req2.tx().unwrap();
tx.execute("UPDATE crates set downloads = $1
WHERE id = $2", &[&krate.downloads, &k.id]).unwrap();
tx.execute("UPDATE crates set downloads = $1
WHERE id = $2", &[&krate2.downloads, &k2.id]).unwrap();
tx.execute("UPDATE crates set downloads = $1
WHERE id = $2", &[&krate3.downloads, &k3.id]).unwrap();
tx.execute("UPDATE crates set downloads = $1
WHERE id = $2", &[&krate4.downloads, &k4.id]).unwrap();
}

let mut response = ok_resp!(middle.call(req.with_query("q=foo&sort=downloads")));
let json: CrateList = ::json(&mut response);
assert_eq!(json.meta.total, 3);
assert_eq!(json.crates[0].name, "foo");
assert_eq!(json.crates[1].name, "baz");
assert_eq!(json.crates[2].name, "bar");

let mut response = ok_resp!(middle.call(req.with_query("q=bar&sort=downloads")));
let json: CrateList = ::json(&mut response);
assert_eq!(json.meta.total, 3);
assert_eq!(json.crates[0].name, "bar");
assert_eq!(json.crates[1].name, "baz");
assert_eq!(json.crates[2].name, "foo");

let mut response = ok_resp!(middle.call(req.with_query("q=baz&sort=downloads")));
let json: CrateList = ::json(&mut response);
assert_eq!(json.meta.total, 3);
assert_eq!(json.crates[0].name, "baz");
assert_eq!(json.crates[1].name, "bar");
assert_eq!(json.crates[2].name, "foo");

let mut response = ok_resp!(middle.call(req.with_query("q=const&sort=downloads")));
let json: CrateList = ::json(&mut response);
assert_eq!(json.meta.total, 4);
assert_eq!(json.crates[0].name, "other");
assert_eq!(json.crates[1].name, "baz");
assert_eq!(json.crates[2].name, "bar");
assert_eq!(json.crates[3].name, "foo");
}

#[test]
fn show() {
let (_b, app, middle) = ::app();
Expand Down