Skip to content

Commit 1c88de4

Browse files
committed
Downgrade more return types from AppResult to QueryResult
1 parent 16cd88a commit 1c88de4

File tree

4 files changed

+23
-20
lines changed

4 files changed

+23
-20
lines changed

src/models/krate.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ impl Crate {
387387

388388
/// Return both the newest (most recently updated) and
389389
/// highest version (in semver order) for the current crate.
390-
pub fn top_versions(&self, conn: &PgConnection) -> AppResult<TopVersions> {
390+
pub fn top_versions(&self, conn: &PgConnection) -> QueryResult<TopVersions> {
391391
use crate::schema::versions::dsl::*;
392392

393393
Ok(Version::top(
@@ -397,7 +397,7 @@ impl Crate {
397397
))
398398
}
399399

400-
pub fn owners(&self, conn: &PgConnection) -> AppResult<Vec<Owner>> {
400+
pub fn owners(&self, conn: &PgConnection) -> QueryResult<Vec<Owner>> {
401401
let users = CrateOwner::by_owner_kind(OwnerKind::User)
402402
.filter(crate_owners::crate_id.eq(self.id))
403403
.inner_join(users::table)

src/models/team.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ impl Team {
173173
team_with_gh_id_contains_user(app, self.github_id, user)
174174
}
175175

176-
pub fn owning(krate: &Crate, conn: &PgConnection) -> AppResult<Vec<Owner>> {
176+
pub fn owning(krate: &Crate, conn: &PgConnection) -> QueryResult<Vec<Owner>> {
177177
let base_query = CrateOwner::belonging_to(krate).filter(crate_owners::deleted.eq(false));
178178
let teams = base_query
179179
.inner_join(teams::table)

src/models/user.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ impl User {
116116
Self::find(conn, api_token.user_id)
117117
}
118118

119-
pub fn owning(krate: &Crate, conn: &PgConnection) -> AppResult<Vec<Owner>> {
119+
pub fn owning(krate: &Crate, conn: &PgConnection) -> QueryResult<Vec<Owner>> {
120120
let users = CrateOwner::by_owner_kind(OwnerKind::User)
121121
.inner_join(users::table)
122122
.select(users::all_columns)
@@ -157,7 +157,7 @@ impl User {
157157

158158
/// Queries the database for the verified emails
159159
/// belonging to a given user
160-
pub fn verified_email(&self, conn: &PgConnection) -> AppResult<Option<String>> {
160+
pub fn verified_email(&self, conn: &PgConnection) -> QueryResult<Option<String>> {
161161
Ok(Email::belonging_to(self)
162162
.select(emails::email)
163163
.filter(emails::verified.eq(true))

src/publish_rate_limit.rs

Lines changed: 18 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ impl PublishRateLimit {
5555
uploader: i32,
5656
now: NaiveDateTime,
5757
conn: &PgConnection,
58-
) -> AppResult<Bucket> {
58+
) -> QueryResult<Bucket> {
5959
use self::publish_limit_buckets::dsl::*;
6060
use diesel::sql_types::{Double, Interval, Text, Timestamp};
6161

@@ -93,7 +93,6 @@ impl PublishRateLimit {
9393
.eq(last_refill + self.refill_rate().into_sql::<Interval>() * tokens_to_add),
9494
))
9595
.get_result(conn)
96-
.map_err(Into::into)
9796
}
9897

9998
fn refill_rate(&self) -> PgInterval {
@@ -108,7 +107,7 @@ mod tests {
108107
use crate::test_util::*;
109108

110109
#[test]
111-
fn take_token_with_no_bucket_creates_new_one() -> AppResult<()> {
110+
fn take_token_with_no_bucket_creates_new_one() -> QueryResult<()> {
112111
let conn = pg_connection();
113112
let now = now();
114113

@@ -139,7 +138,7 @@ mod tests {
139138
}
140139

141140
#[test]
142-
fn take_token_with_existing_bucket_modifies_existing_bucket() -> AppResult<()> {
141+
fn take_token_with_existing_bucket_modifies_existing_bucket() -> QueryResult<()> {
143142
let conn = pg_connection();
144143
let now = now();
145144

@@ -159,7 +158,7 @@ mod tests {
159158
}
160159

161160
#[test]
162-
fn take_token_after_delay_refills() -> AppResult<()> {
161+
fn take_token_after_delay_refills() -> QueryResult<()> {
163162
let conn = pg_connection();
164163
let now = now();
165164

@@ -180,12 +179,13 @@ mod tests {
180179
}
181180

182181
#[test]
183-
fn refill_subsecond_rate() -> AppResult<()> {
182+
fn refill_subsecond_rate() -> QueryResult<()> {
184183
let conn = pg_connection();
185184
// Subsecond rates have floating point rounding issues, so use a known
186185
// timestamp that rounds fine
187186
let now =
188-
NaiveDateTime::parse_from_str("2019-03-19T21:11:24.620401", "%Y-%m-%dT%H:%M:%S%.f")?;
187+
NaiveDateTime::parse_from_str("2019-03-19T21:11:24.620401", "%Y-%m-%dT%H:%M:%S%.f")
188+
.unwrap();
189189

190190
let rate = PublishRateLimit {
191191
rate: Duration::from_millis(100),
@@ -204,7 +204,7 @@ mod tests {
204204
}
205205

206206
#[test]
207-
fn last_refill_always_advanced_by_multiple_of_rate() -> AppResult<()> {
207+
fn last_refill_always_advanced_by_multiple_of_rate() -> QueryResult<()> {
208208
let conn = pg_connection();
209209
let now = now();
210210

@@ -225,7 +225,7 @@ mod tests {
225225
}
226226

227227
#[test]
228-
fn zero_tokens_returned_when_user_has_no_tokens_left() -> AppResult<()> {
228+
fn zero_tokens_returned_when_user_has_no_tokens_left() -> QueryResult<()> {
229229
let conn = pg_connection();
230230
let now = now();
231231

@@ -248,7 +248,7 @@ mod tests {
248248
}
249249

250250
#[test]
251-
fn a_user_with_no_tokens_gets_a_token_after_exactly_rate() -> AppResult<()> {
251+
fn a_user_with_no_tokens_gets_a_token_after_exactly_rate() -> QueryResult<()> {
252252
let conn = pg_connection();
253253
let now = now();
254254

@@ -270,7 +270,7 @@ mod tests {
270270
}
271271

272272
#[test]
273-
fn tokens_never_refill_past_burst() -> AppResult<()> {
273+
fn tokens_never_refill_past_burst() -> QueryResult<()> {
274274
let conn = pg_connection();
275275
let now = now();
276276

@@ -292,7 +292,7 @@ mod tests {
292292
}
293293

294294
#[test]
295-
fn override_is_used_instead_of_global_burst_if_present() -> AppResult<()> {
295+
fn override_is_used_instead_of_global_burst_if_present() -> QueryResult<()> {
296296
let conn = pg_connection();
297297
let now = now();
298298

@@ -318,7 +318,7 @@ mod tests {
318318
Ok(())
319319
}
320320

321-
fn new_user(conn: &PgConnection, gh_login: &str) -> AppResult<i32> {
321+
fn new_user(conn: &PgConnection, gh_login: &str) -> QueryResult<i32> {
322322
use crate::models::NewUser;
323323

324324
let user = NewUser {
@@ -329,15 +329,18 @@ mod tests {
329329
Ok(user.id)
330330
}
331331

332-
fn new_user_bucket(conn: &PgConnection, tokens: i32, now: NaiveDateTime) -> AppResult<Bucket> {
332+
fn new_user_bucket(
333+
conn: &PgConnection,
334+
tokens: i32,
335+
now: NaiveDateTime,
336+
) -> QueryResult<Bucket> {
333337
diesel::insert_into(publish_limit_buckets::table)
334338
.values(Bucket {
335339
user_id: new_user(conn, "new_user")?,
336340
tokens,
337341
last_refill: now,
338342
})
339343
.get_result(conn)
340-
.map_err(Into::into)
341344
}
342345

343346
/// Strips ns precision from `Utc::now`. PostgreSQL only has microsecond

0 commit comments

Comments
 (0)