nyanogang/nyanoblog
2
1
Fork 0
Code Issues Pull requests Projects 1 Releases Packages Wiki Activity

data: return newly crated entries in SQL

Browse source
This commit is contained in:
anna 2022-12-25 05:19:08 +01:00
parent fb8bf0f316
commit 93182705d9
Signed by: fef
GPG key ID: EC22E476DC2D3D84
2 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/data/follow/pg.rs
View file

@ -28,12 +28,13 @@ impl PgFollowDataSource {
T: Into<NewFollow> + Send, T: Into<NewFollow> + Send,
{ {
let new = new.into(); let new = new.into();
let follow: Follow = let follow: Follow = sqlx::query_as(
sqlx::query_as("INSERT INTO follows (follower_id, followee_id) VALUES ($1, $2)") "INSERT INTO follows (follower_id, followee_id) VALUES ($1, $2) RETURNING *",
.bind(new.follower_id) )
.bind(new.followee_id) .bind(new.follower_id)
.fetch_one(&self.pool) .bind(new.followee_id)
.await?; .fetch_one(&self.pool)
.await?;
Ok(follow) Ok(follow)
} }

11
src/data/like/pg.rs
View file

@ -30,11 +30,12 @@ impl PgLikeDataSource {
E: Into<Error>, E: Into<Error>,
{ {
let new = new.try_into().map_err(|e| e.into())?.inner(); let new = new.try_into().map_err(|e| e.into())?.inner();
let like: Like = sqlx::query_as("INSERT INTO likes (note_id, account_id) VALUES ($1, $2)") let like: Like =
.bind(new.note_id) sqlx::query_as("INSERT INTO likes (note_id, account_id) VALUES ($1, $2) RETURNING *")
.bind(new.account_id) .bind(new.note_id)
.fetch_one(&self.pool) .bind(new.account_id)
.await?; .fetch_one(&self.pool)
.await?;
Ok(like) Ok(like)
} }