Skip to content

Commit 32f0ab8

Browse files
committed
models/team: Remove sync fns
1 parent e3f3c5b commit 32f0ab8

File tree

4 files changed

+11
-37
lines changed

4 files changed

+11
-37
lines changed

src/models/team.rs

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use bon::Builder;
2-
use diesel_async::AsyncPgConnection;
2+
use diesel::prelude::*;
3+
use diesel_async::{AsyncPgConnection, RunQueryDsl};
34
use http::StatusCode;
45

56
use crate::app::App;
@@ -10,9 +11,6 @@ use oauth2::AccessToken;
1011

1112
use crate::models::{Crate, CrateOwner, Owner, OwnerKind, User};
1213
use crate::schema::{crate_owners, teams};
13-
use crate::sql::lower;
14-
use crate::util::diesel::prelude::*;
15-
use crate::util::diesel::Conn;
1614

1715
/// For now, just a Github Team. Can be upgraded to other teams
1816
/// later if desirable.
@@ -46,9 +44,8 @@ pub struct NewTeam<'a> {
4644
}
4745

4846
impl<'a> NewTeam<'a> {
49-
pub async fn async_create_or_update(&self, conn: &mut AsyncPgConnection) -> QueryResult<Team> {
47+
pub async fn create_or_update(&self, conn: &mut AsyncPgConnection) -> QueryResult<Team> {
5048
use diesel::insert_into;
51-
use diesel_async::RunQueryDsl;
5249

5350
insert_into(teams::table)
5451
.values(self)
@@ -58,30 +55,9 @@ impl<'a> NewTeam<'a> {
5855
.get_result(conn)
5956
.await
6057
}
61-
62-
pub fn create_or_update(&self, conn: &mut impl Conn) -> QueryResult<Team> {
63-
use diesel::insert_into;
64-
use diesel::RunQueryDsl;
65-
66-
insert_into(teams::table)
67-
.values(self)
68-
.on_conflict(teams::github_id)
69-
.do_update()
70-
.set(self)
71-
.get_result(conn)
72-
}
7358
}
7459

7560
impl Team {
76-
pub fn find_by_login(conn: &mut impl Conn, login: &str) -> QueryResult<Self> {
77-
use diesel::RunQueryDsl;
78-
79-
teams::table
80-
.filter(lower(teams::login).eq(&login.to_lowercase()))
81-
.first(conn)
82-
.map_err(Into::into)
83-
}
84-
8561
/// Tries to create the Team in the DB (assumes a `:` has already been found).
8662
///
8763
/// # Panics
@@ -178,7 +154,7 @@ impl Team {
178154
.maybe_name(team.name.as_deref())
179155
.maybe_avatar(org.avatar_url.as_deref())
180156
.build()
181-
.async_create_or_update(conn)
157+
.create_or_update(conn)
182158
.await
183159
.map_err(Into::into)
184160
}
@@ -199,8 +175,6 @@ impl Team {
199175
}
200176

201177
pub async fn owning(krate: &Crate, conn: &mut AsyncPgConnection) -> QueryResult<Vec<Owner>> {
202-
use diesel_async::RunQueryDsl;
203-
204178
let base_query = CrateOwner::belonging_to(krate).filter(crate_owners::deleted.eq(false));
205179
let teams = base_query
206180
.inner_join(teams::table)

src/tests/owners.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ async fn check_ownership_two_crates() {
286286
let user = user.as_model();
287287

288288
let team = new_team("team_foo")
289-
.async_create_or_update(&mut conn)
289+
.create_or_update(&mut conn)
290290
.await
291291
.unwrap();
292292
let krate_owned_by_team = CrateBuilder::new("foo", user.id)
@@ -326,7 +326,7 @@ async fn check_ownership_one_crate() {
326326
let user = user.as_model();
327327

328328
let team = new_team("github:test_org:team_sloth")
329-
.async_create_or_update(&mut conn)
329+
.create_or_update(&mut conn)
330330
.await
331331
.unwrap();
332332
let krate = CrateBuilder::new("best_crate", user.id)
@@ -360,7 +360,7 @@ async fn add_existing_team() {
360360
let user = user.as_model();
361361

362362
let t = new_team("github:test_org:bananas")
363-
.async_create_or_update(&mut conn)
363+
.create_or_update(&mut conn)
364364
.await
365365
.unwrap();
366366
let krate = CrateBuilder::new("best_crate", user.id)

src/tests/team.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ async fn add_renamed_team() {
115115
.github_id(2001)
116116
.build();
117117

118-
new_team.async_create_or_update(&mut conn).await.unwrap();
118+
new_team.create_or_update(&mut conn).await.unwrap();
119119

120120
assert_eq!(
121121
teams::table
@@ -465,7 +465,7 @@ async fn crates_by_team_id() {
465465
let user = user.as_model();
466466

467467
let t = new_team("github:test-org:team")
468-
.async_create_or_update(&mut conn)
468+
.create_or_update(&mut conn)
469469
.await
470470
.unwrap();
471471
let krate = CrateBuilder::new("foo", user.id)
@@ -492,7 +492,7 @@ async fn crates_by_team_id_not_including_deleted_owners() {
492492
.github_id(2001)
493493
.build();
494494

495-
let t = new_team.async_create_or_update(&mut conn).await.unwrap();
495+
let t = new_team.create_or_update(&mut conn).await.unwrap();
496496

497497
let krate = CrateBuilder::new("foo", user.id)
498498
.expect_build(&mut conn)

src/typosquat/test_util.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ pub mod faker {
3737
.name(team)
3838
.build();
3939

40-
Ok(team.async_create_or_update(conn).await?)
40+
Ok(team.create_or_update(conn).await?)
4141
}
4242

4343
pub fn user(conn: &mut PgConnection, login: &str) -> QueryResult<User> {

0 commit comments

Comments
 (0)