Starting to add post_view.
This commit is contained in:
parent
dabcfca67b
commit
eef93440d0
9 changed files with 980 additions and 5 deletions
|
@ -1,3 +1,4 @@
|
||||||
pub mod community_aggregates;
|
pub mod community_aggregates;
|
||||||
|
pub mod post_aggregates;
|
||||||
pub mod site_aggregates;
|
pub mod site_aggregates;
|
||||||
pub mod user_aggregates;
|
pub mod user_aggregates;
|
||||||
|
|
268
lemmy_db/src/aggregates/post_aggregates.rs
Normal file
268
lemmy_db/src/aggregates/post_aggregates.rs
Normal file
|
@ -0,0 +1,268 @@
|
||||||
|
use crate::schema::post_aggregates;
|
||||||
|
use diesel::{result::Error, *};
|
||||||
|
use serde::Serialize;
|
||||||
|
|
||||||
|
#[derive(Queryable, Associations, Identifiable, PartialEq, Debug, Serialize, Clone)]
|
||||||
|
#[table_name = "post_aggregates"]
|
||||||
|
pub struct PostAggregates {
|
||||||
|
pub id: i32,
|
||||||
|
pub post_id: i32,
|
||||||
|
pub comments: i64,
|
||||||
|
pub score: i64,
|
||||||
|
pub upvotes: i64,
|
||||||
|
pub downvotes: i64,
|
||||||
|
pub newest_comment_time: chrono::NaiveDateTime,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PostAggregates {
|
||||||
|
pub fn read(conn: &PgConnection, post_id: i32) -> Result<Self, Error> {
|
||||||
|
post_aggregates::table
|
||||||
|
.filter(post_aggregates::post_id.eq(post_id))
|
||||||
|
.first::<Self>(conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// #[cfg(test)]
|
||||||
|
// mod tests {
|
||||||
|
// use crate::{
|
||||||
|
// aggregates::community_aggregates::CommunityAggregates,
|
||||||
|
// comment::{Comment, CommentForm},
|
||||||
|
// community::{Community, CommunityFollower, CommunityFollowerForm, CommunityForm},
|
||||||
|
// post::{Post, PostForm},
|
||||||
|
// tests::establish_unpooled_connection,
|
||||||
|
// user::{UserForm, User_},
|
||||||
|
// Crud,
|
||||||
|
// Followable,
|
||||||
|
// ListingType,
|
||||||
|
// SortType,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// #[test]
|
||||||
|
// fn test_crud() {
|
||||||
|
// let conn = establish_unpooled_connection();
|
||||||
|
|
||||||
|
// let new_user = UserForm {
|
||||||
|
// name: "thommy_community_agg".into(),
|
||||||
|
// preferred_username: None,
|
||||||
|
// password_encrypted: "nope".into(),
|
||||||
|
// email: None,
|
||||||
|
// matrix_user_id: None,
|
||||||
|
// avatar: None,
|
||||||
|
// banner: None,
|
||||||
|
// admin: false,
|
||||||
|
// banned: Some(false),
|
||||||
|
// published: None,
|
||||||
|
// updated: None,
|
||||||
|
// show_nsfw: false,
|
||||||
|
// theme: "browser".into(),
|
||||||
|
// default_sort_type: SortType::Hot as i16,
|
||||||
|
// default_listing_type: ListingType::Subscribed as i16,
|
||||||
|
// lang: "browser".into(),
|
||||||
|
// show_avatars: true,
|
||||||
|
// send_notifications_to_email: false,
|
||||||
|
// actor_id: None,
|
||||||
|
// bio: None,
|
||||||
|
// local: true,
|
||||||
|
// private_key: None,
|
||||||
|
// public_key: None,
|
||||||
|
// last_refreshed_at: None,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let inserted_user = User_::create(&conn, &new_user).unwrap();
|
||||||
|
|
||||||
|
// let another_user = UserForm {
|
||||||
|
// name: "jerry_community_agg".into(),
|
||||||
|
// preferred_username: None,
|
||||||
|
// password_encrypted: "nope".into(),
|
||||||
|
// email: None,
|
||||||
|
// matrix_user_id: None,
|
||||||
|
// avatar: None,
|
||||||
|
// banner: None,
|
||||||
|
// admin: false,
|
||||||
|
// banned: Some(false),
|
||||||
|
// published: None,
|
||||||
|
// updated: None,
|
||||||
|
// show_nsfw: false,
|
||||||
|
// theme: "browser".into(),
|
||||||
|
// default_sort_type: SortType::Hot as i16,
|
||||||
|
// default_listing_type: ListingType::Subscribed as i16,
|
||||||
|
// lang: "browser".into(),
|
||||||
|
// show_avatars: true,
|
||||||
|
// send_notifications_to_email: false,
|
||||||
|
// actor_id: None,
|
||||||
|
// bio: None,
|
||||||
|
// local: true,
|
||||||
|
// private_key: None,
|
||||||
|
// public_key: None,
|
||||||
|
// last_refreshed_at: None,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let another_inserted_user = User_::create(&conn, &another_user).unwrap();
|
||||||
|
|
||||||
|
// let new_community = CommunityForm {
|
||||||
|
// name: "TIL_community_agg".into(),
|
||||||
|
// creator_id: inserted_user.id,
|
||||||
|
// title: "nada".to_owned(),
|
||||||
|
// description: None,
|
||||||
|
// category_id: 1,
|
||||||
|
// nsfw: false,
|
||||||
|
// removed: None,
|
||||||
|
// deleted: None,
|
||||||
|
// updated: None,
|
||||||
|
// actor_id: None,
|
||||||
|
// local: true,
|
||||||
|
// private_key: None,
|
||||||
|
// public_key: None,
|
||||||
|
// last_refreshed_at: None,
|
||||||
|
// published: None,
|
||||||
|
// icon: None,
|
||||||
|
// banner: None,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let inserted_community = Community::create(&conn, &new_community).unwrap();
|
||||||
|
|
||||||
|
// let another_community = CommunityForm {
|
||||||
|
// name: "TIL_community_agg_2".into(),
|
||||||
|
// creator_id: inserted_user.id,
|
||||||
|
// title: "nada".to_owned(),
|
||||||
|
// description: None,
|
||||||
|
// category_id: 1,
|
||||||
|
// nsfw: false,
|
||||||
|
// removed: None,
|
||||||
|
// deleted: None,
|
||||||
|
// updated: None,
|
||||||
|
// actor_id: None,
|
||||||
|
// local: true,
|
||||||
|
// private_key: None,
|
||||||
|
// public_key: None,
|
||||||
|
// last_refreshed_at: None,
|
||||||
|
// published: None,
|
||||||
|
// icon: None,
|
||||||
|
// banner: None,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let another_inserted_community = Community::create(&conn, &another_community).unwrap();
|
||||||
|
|
||||||
|
// let first_user_follow = CommunityFollowerForm {
|
||||||
|
// community_id: inserted_community.id,
|
||||||
|
// user_id: inserted_user.id,
|
||||||
|
// pending: false,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// CommunityFollower::follow(&conn, &first_user_follow).unwrap();
|
||||||
|
|
||||||
|
// let second_user_follow = CommunityFollowerForm {
|
||||||
|
// community_id: inserted_community.id,
|
||||||
|
// user_id: another_inserted_user.id,
|
||||||
|
// pending: false,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// CommunityFollower::follow(&conn, &second_user_follow).unwrap();
|
||||||
|
|
||||||
|
// let another_community_follow = CommunityFollowerForm {
|
||||||
|
// community_id: another_inserted_community.id,
|
||||||
|
// user_id: inserted_user.id,
|
||||||
|
// pending: false,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// CommunityFollower::follow(&conn, &another_community_follow).unwrap();
|
||||||
|
|
||||||
|
// let new_post = PostForm {
|
||||||
|
// name: "A test post".into(),
|
||||||
|
// url: None,
|
||||||
|
// body: None,
|
||||||
|
// creator_id: inserted_user.id,
|
||||||
|
// community_id: inserted_community.id,
|
||||||
|
// removed: None,
|
||||||
|
// deleted: None,
|
||||||
|
// locked: None,
|
||||||
|
// stickied: None,
|
||||||
|
// nsfw: false,
|
||||||
|
// updated: None,
|
||||||
|
// embed_title: None,
|
||||||
|
// embed_description: None,
|
||||||
|
// embed_html: None,
|
||||||
|
// thumbnail_url: None,
|
||||||
|
// ap_id: None,
|
||||||
|
// local: true,
|
||||||
|
// published: None,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let inserted_post = Post::create(&conn, &new_post).unwrap();
|
||||||
|
|
||||||
|
// let comment_form = CommentForm {
|
||||||
|
// content: "A test comment".into(),
|
||||||
|
// creator_id: inserted_user.id,
|
||||||
|
// post_id: inserted_post.id,
|
||||||
|
// removed: None,
|
||||||
|
// deleted: None,
|
||||||
|
// read: None,
|
||||||
|
// parent_id: None,
|
||||||
|
// published: None,
|
||||||
|
// updated: None,
|
||||||
|
// ap_id: None,
|
||||||
|
// local: true,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let inserted_comment = Comment::create(&conn, &comment_form).unwrap();
|
||||||
|
|
||||||
|
// let child_comment_form = CommentForm {
|
||||||
|
// content: "A test comment".into(),
|
||||||
|
// creator_id: inserted_user.id,
|
||||||
|
// post_id: inserted_post.id,
|
||||||
|
// removed: None,
|
||||||
|
// deleted: None,
|
||||||
|
// read: None,
|
||||||
|
// parent_id: Some(inserted_comment.id),
|
||||||
|
// published: None,
|
||||||
|
// updated: None,
|
||||||
|
// ap_id: None,
|
||||||
|
// local: true,
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let _inserted_child_comment = Comment::create(&conn, &child_comment_form).unwrap();
|
||||||
|
|
||||||
|
// let community_aggregates_before_delete =
|
||||||
|
// CommunityAggregates::read(&conn, inserted_community.id).unwrap();
|
||||||
|
|
||||||
|
// assert_eq!(2, community_aggregates_before_delete.subscribers);
|
||||||
|
// assert_eq!(1, community_aggregates_before_delete.posts);
|
||||||
|
// assert_eq!(2, community_aggregates_before_delete.comments);
|
||||||
|
|
||||||
|
// // Test the other community
|
||||||
|
// let another_community_aggs =
|
||||||
|
// CommunityAggregates::read(&conn, another_inserted_community.id).unwrap();
|
||||||
|
// assert_eq!(1, another_community_aggs.subscribers);
|
||||||
|
// assert_eq!(0, another_community_aggs.posts);
|
||||||
|
// assert_eq!(0, another_community_aggs.comments);
|
||||||
|
|
||||||
|
// // Unfollow test
|
||||||
|
// CommunityFollower::unfollow(&conn, &second_user_follow).unwrap();
|
||||||
|
// let after_unfollow = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
|
||||||
|
// assert_eq!(1, after_unfollow.subscribers);
|
||||||
|
|
||||||
|
// // Follow again just for the later tests
|
||||||
|
// CommunityFollower::follow(&conn, &second_user_follow).unwrap();
|
||||||
|
// let after_follow_again = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
|
||||||
|
// assert_eq!(2, after_follow_again.subscribers);
|
||||||
|
|
||||||
|
// // Remove a parent comment (the comment count should also be 0)
|
||||||
|
// Post::delete(&conn, inserted_post.id).unwrap();
|
||||||
|
// let after_parent_post_delete = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
|
||||||
|
// assert_eq!(0, after_parent_post_delete.comments);
|
||||||
|
// assert_eq!(0, after_parent_post_delete.posts);
|
||||||
|
|
||||||
|
// // Remove the 2nd user
|
||||||
|
// User_::delete(&conn, another_inserted_user.id).unwrap();
|
||||||
|
// let after_user_delete = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
|
||||||
|
// assert_eq!(1, after_user_delete.subscribers);
|
||||||
|
|
||||||
|
// // This should delete all the associated rows, and fire triggers
|
||||||
|
// let user_num_deleted = User_::delete(&conn, inserted_user.id).unwrap();
|
||||||
|
// assert_eq!(1, user_num_deleted);
|
||||||
|
|
||||||
|
// // Should be none found, since the creator was deleted
|
||||||
|
// let after_delete = CommunityAggregates::read(&conn, inserted_community.id);
|
||||||
|
// assert!(after_delete.is_err());
|
||||||
|
// }
|
||||||
|
// }
|
|
@ -8,9 +8,10 @@ use crate::{
|
||||||
Saveable,
|
Saveable,
|
||||||
};
|
};
|
||||||
use diesel::{dsl::*, result::Error, *};
|
use diesel::{dsl::*, result::Error, *};
|
||||||
|
use serde::Serialize;
|
||||||
use url::{ParseError, Url};
|
use url::{ParseError, Url};
|
||||||
|
|
||||||
#[derive(Queryable, Identifiable, PartialEq, Debug)]
|
#[derive(Clone, Queryable, Identifiable, PartialEq, Debug, Serialize)]
|
||||||
#[table_name = "post"]
|
#[table_name = "post"]
|
||||||
pub struct Post {
|
pub struct Post {
|
||||||
pub id: i32,
|
pub id: i32,
|
||||||
|
|
|
@ -329,6 +329,18 @@ table! {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
table! {
|
||||||
|
post_aggregates (id) {
|
||||||
|
id -> Int4,
|
||||||
|
post_id -> Int4,
|
||||||
|
comments -> Int8,
|
||||||
|
score -> Int8,
|
||||||
|
upvotes -> Int8,
|
||||||
|
downvotes -> Int8,
|
||||||
|
newest_comment_time -> Timestamp,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
table! {
|
table! {
|
||||||
post_aggregates_fast (id) {
|
post_aggregates_fast (id) {
|
||||||
id -> Int4,
|
id -> Int4,
|
||||||
|
@ -576,6 +588,7 @@ joinable!(mod_sticky_post -> user_ (mod_user_id));
|
||||||
joinable!(password_reset_request -> user_ (user_id));
|
joinable!(password_reset_request -> user_ (user_id));
|
||||||
joinable!(post -> community (community_id));
|
joinable!(post -> community (community_id));
|
||||||
joinable!(post -> user_ (creator_id));
|
joinable!(post -> user_ (creator_id));
|
||||||
|
joinable!(post_aggregates -> post (post_id));
|
||||||
joinable!(post_like -> post (post_id));
|
joinable!(post_like -> post (post_id));
|
||||||
joinable!(post_like -> user_ (user_id));
|
joinable!(post_like -> user_ (user_id));
|
||||||
joinable!(post_read -> post (post_id));
|
joinable!(post_read -> post (post_id));
|
||||||
|
@ -614,6 +627,7 @@ allow_tables_to_appear_in_same_query!(
|
||||||
mod_sticky_post,
|
mod_sticky_post,
|
||||||
password_reset_request,
|
password_reset_request,
|
||||||
post,
|
post,
|
||||||
|
post_aggregates,
|
||||||
post_aggregates_fast,
|
post_aggregates_fast,
|
||||||
post_like,
|
post_like,
|
||||||
post_read,
|
post_read,
|
||||||
|
|
|
@ -32,7 +32,7 @@ impl CommunityView {
|
||||||
// The left join below will return None in this case
|
// The left join below will return None in this case
|
||||||
let user_id_join = my_user_id.unwrap_or(-1);
|
let user_id_join = my_user_id.unwrap_or(-1);
|
||||||
|
|
||||||
let (community, creator, category, counts, subscribed) = community::table
|
let (community, creator, category, counts, follower) = community::table
|
||||||
.find(community_id)
|
.find(community_id)
|
||||||
.inner_join(user_::table)
|
.inner_join(user_::table)
|
||||||
.inner_join(category::table)
|
.inner_join(category::table)
|
||||||
|
@ -63,7 +63,7 @@ impl CommunityView {
|
||||||
community,
|
community,
|
||||||
creator,
|
creator,
|
||||||
category,
|
category,
|
||||||
subscribed: subscribed.is_some(),
|
subscribed: follower.is_some(),
|
||||||
counts,
|
counts,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,7 +2,6 @@ pub mod community_follower_view;
|
||||||
pub mod community_moderator_view;
|
pub mod community_moderator_view;
|
||||||
pub mod community_user_ban_view;
|
pub mod community_user_ban_view;
|
||||||
pub mod community_view;
|
pub mod community_view;
|
||||||
|
pub mod post_view;
|
||||||
pub mod site_view;
|
pub mod site_view;
|
||||||
pub mod user_view;
|
pub mod user_view;
|
||||||
|
|
||||||
// TODO Every single aggregate trigger is likely broken, you need to test every one of these out
|
|
||||||
|
|
576
lemmy_db/src/views/post_view.rs
Normal file
576
lemmy_db/src/views/post_view.rs
Normal file
|
@ -0,0 +1,576 @@
|
||||||
|
use crate::{
|
||||||
|
aggregates::post_aggregates::PostAggregates,
|
||||||
|
community::{Community, CommunityFollower, CommunitySafe, CommunityUserBan},
|
||||||
|
functions::hot_rank,
|
||||||
|
fuzzy_search,
|
||||||
|
limit_and_offset,
|
||||||
|
post::{Post, PostRead, PostSaved},
|
||||||
|
schema::{
|
||||||
|
community,
|
||||||
|
community_follower,
|
||||||
|
community_user_ban,
|
||||||
|
post,
|
||||||
|
post_aggregates,
|
||||||
|
post_like,
|
||||||
|
post_read,
|
||||||
|
post_saved,
|
||||||
|
user_,
|
||||||
|
},
|
||||||
|
user::{UserSafe, User_},
|
||||||
|
ListingType,
|
||||||
|
MaybeOptional,
|
||||||
|
SortType,
|
||||||
|
ToSafe,
|
||||||
|
};
|
||||||
|
use diesel::{result::Error, *};
|
||||||
|
use serde::Serialize;
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Clone)]
|
||||||
|
pub struct PostView {
|
||||||
|
pub post: Post,
|
||||||
|
pub creator: UserSafe,
|
||||||
|
pub community: CommunitySafe,
|
||||||
|
pub counts: PostAggregates,
|
||||||
|
pub subscribed: bool, // Left join to CommunityFollower
|
||||||
|
pub banned_from_community: bool, // Left Join to CommunityUserBan
|
||||||
|
pub saved: bool, // Left join to PostSaved
|
||||||
|
pub read: bool, // Left join to PostRead
|
||||||
|
pub my_vote: Option<i16>, // Left join to PostLike
|
||||||
|
}
|
||||||
|
|
||||||
|
type OutputTuple = (
|
||||||
|
Post,
|
||||||
|
UserSafe,
|
||||||
|
CommunitySafe,
|
||||||
|
PostAggregates,
|
||||||
|
Option<CommunityFollower>,
|
||||||
|
Option<CommunityUserBan>,
|
||||||
|
Option<PostSaved>,
|
||||||
|
Option<PostRead>,
|
||||||
|
Option<i16>,
|
||||||
|
);
|
||||||
|
|
||||||
|
impl PostView {
|
||||||
|
pub fn read(conn: &PgConnection, post_id: i32, my_user_id: Option<i32>) -> Result<Self, Error> {
|
||||||
|
// The left join below will return None in this case
|
||||||
|
let user_id_join = my_user_id.unwrap_or(-1);
|
||||||
|
|
||||||
|
let (post, creator, community, counts, follower, banned_from_community, saved, read, my_vote) =
|
||||||
|
post::table
|
||||||
|
.find(post_id)
|
||||||
|
.inner_join(user_::table)
|
||||||
|
.inner_join(community::table)
|
||||||
|
.inner_join(post_aggregates::table)
|
||||||
|
.left_join(
|
||||||
|
community_follower::table.on(
|
||||||
|
post::community_id
|
||||||
|
.eq(community_follower::community_id)
|
||||||
|
.and(community_follower::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
community_user_ban::table.on(
|
||||||
|
post::community_id
|
||||||
|
.eq(community_user_ban::community_id)
|
||||||
|
.and(community_user_ban::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_saved::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_saved::post_id)
|
||||||
|
.and(post_saved::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_read::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_read::post_id)
|
||||||
|
.and(post_read::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_like::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_like::post_id)
|
||||||
|
.and(post_like::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.select((
|
||||||
|
post::all_columns,
|
||||||
|
User_::safe_columns_tuple(),
|
||||||
|
Community::safe_columns_tuple(),
|
||||||
|
post_aggregates::all_columns,
|
||||||
|
community_follower::all_columns.nullable(),
|
||||||
|
community_user_ban::all_columns.nullable(),
|
||||||
|
post_saved::all_columns.nullable(),
|
||||||
|
post_read::all_columns.nullable(),
|
||||||
|
post_like::score.nullable(),
|
||||||
|
))
|
||||||
|
.first::<OutputTuple>(conn)?;
|
||||||
|
|
||||||
|
Ok(PostView {
|
||||||
|
post,
|
||||||
|
creator,
|
||||||
|
community,
|
||||||
|
counts,
|
||||||
|
subscribed: follower.is_some(),
|
||||||
|
banned_from_community: banned_from_community.is_some(),
|
||||||
|
saved: saved.is_some(),
|
||||||
|
read: read.is_some(),
|
||||||
|
my_vote,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod join_types {
|
||||||
|
use crate::schema::{
|
||||||
|
community,
|
||||||
|
community_follower,
|
||||||
|
community_user_ban,
|
||||||
|
post,
|
||||||
|
post_aggregates,
|
||||||
|
post_like,
|
||||||
|
post_read,
|
||||||
|
post_saved,
|
||||||
|
user_,
|
||||||
|
};
|
||||||
|
use diesel::{
|
||||||
|
pg::Pg,
|
||||||
|
query_builder::BoxedSelectStatement,
|
||||||
|
query_source::joins::{Inner, Join, JoinOn, LeftOuter},
|
||||||
|
sql_types::*,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// TODO awful, but necessary because of the boxed join
|
||||||
|
pub(super) type BoxedPostJoin<'a> = BoxedSelectStatement<
|
||||||
|
'a,
|
||||||
|
(
|
||||||
|
(
|
||||||
|
Integer,
|
||||||
|
Text,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Integer,
|
||||||
|
Integer,
|
||||||
|
Bool,
|
||||||
|
Bool,
|
||||||
|
Timestamp,
|
||||||
|
Nullable<Timestamp>,
|
||||||
|
Bool,
|
||||||
|
Bool,
|
||||||
|
Bool,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Text,
|
||||||
|
Bool,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
Integer,
|
||||||
|
Text,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Bool,
|
||||||
|
Bool,
|
||||||
|
Timestamp,
|
||||||
|
Nullable<Timestamp>,
|
||||||
|
Nullable<Text>,
|
||||||
|
Text,
|
||||||
|
Nullable<Text>,
|
||||||
|
Bool,
|
||||||
|
Nullable<Text>,
|
||||||
|
Bool,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
Integer,
|
||||||
|
Text,
|
||||||
|
Text,
|
||||||
|
Nullable<Text>,
|
||||||
|
Integer,
|
||||||
|
Integer,
|
||||||
|
Bool,
|
||||||
|
Timestamp,
|
||||||
|
Nullable<Timestamp>,
|
||||||
|
Bool,
|
||||||
|
Bool,
|
||||||
|
Text,
|
||||||
|
Bool,
|
||||||
|
Nullable<Text>,
|
||||||
|
Nullable<Text>,
|
||||||
|
),
|
||||||
|
(Integer, Integer, BigInt, BigInt, BigInt, BigInt, Timestamp),
|
||||||
|
Nullable<(Integer, Integer, Integer, Timestamp, Nullable<Bool>)>,
|
||||||
|
Nullable<(Integer, Integer, Integer, Timestamp)>,
|
||||||
|
Nullable<(Integer, Integer, Integer, Timestamp)>,
|
||||||
|
Nullable<(Integer, Integer, Integer, Timestamp)>,
|
||||||
|
Nullable<SmallInt>,
|
||||||
|
),
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<
|
||||||
|
JoinOn<
|
||||||
|
Join<post::table, user_::table, Inner>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
diesel::expression::nullable::Nullable<
|
||||||
|
post::columns::creator_id,
|
||||||
|
>,
|
||||||
|
diesel::expression::nullable::Nullable<user_::columns::id>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
community::table,
|
||||||
|
Inner,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
diesel::expression::nullable::Nullable<post::columns::community_id>,
|
||||||
|
diesel::expression::nullable::Nullable<community::columns::id>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
post_aggregates::table,
|
||||||
|
Inner,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
diesel::expression::nullable::Nullable<
|
||||||
|
post_aggregates::columns::post_id,
|
||||||
|
>,
|
||||||
|
diesel::expression::nullable::Nullable<post::columns::id>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
community_follower::table,
|
||||||
|
LeftOuter,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::And<
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
post::columns::community_id,
|
||||||
|
community_follower::columns::community_id,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
community_follower::columns::user_id,
|
||||||
|
diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
community_user_ban::table,
|
||||||
|
LeftOuter,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::And<
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
post::columns::community_id,
|
||||||
|
community_user_ban::columns::community_id,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
community_user_ban::columns::user_id,
|
||||||
|
diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
post_saved::table,
|
||||||
|
LeftOuter,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::And<
|
||||||
|
diesel::expression::operators::Eq<post::columns::id, post_saved::columns::post_id>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
post_saved::columns::user_id,
|
||||||
|
diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
post_read::table,
|
||||||
|
LeftOuter,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::And<
|
||||||
|
diesel::expression::operators::Eq<post::columns::id, post_read::columns::post_id>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
post_read::columns::user_id,
|
||||||
|
diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
post_like::table,
|
||||||
|
LeftOuter,
|
||||||
|
>,
|
||||||
|
diesel::expression::operators::And<
|
||||||
|
diesel::expression::operators::Eq<post::columns::id, post_like::columns::post_id>,
|
||||||
|
diesel::expression::operators::Eq<
|
||||||
|
post_like::columns::user_id,
|
||||||
|
diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
Pg,
|
||||||
|
>;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct PostQueryBuilder<'a> {
|
||||||
|
conn: &'a PgConnection,
|
||||||
|
query: join_types::BoxedPostJoin<'a>,
|
||||||
|
listing_type: &'a ListingType,
|
||||||
|
sort: &'a SortType,
|
||||||
|
for_creator_id: Option<i32>,
|
||||||
|
for_community_id: Option<i32>,
|
||||||
|
for_community_name: Option<String>,
|
||||||
|
search_term: Option<String>,
|
||||||
|
url_search: Option<String>,
|
||||||
|
show_nsfw: bool,
|
||||||
|
saved_only: bool,
|
||||||
|
unread_only: bool,
|
||||||
|
page: Option<i64>,
|
||||||
|
limit: Option<i64>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> PostQueryBuilder<'a> {
|
||||||
|
pub fn create(conn: &'a PgConnection, my_user_id: Option<i32>) -> Self {
|
||||||
|
// The left join below will return None in this case
|
||||||
|
let user_id_join = my_user_id.unwrap_or(-1);
|
||||||
|
|
||||||
|
let query = post::table
|
||||||
|
.inner_join(user_::table)
|
||||||
|
.inner_join(community::table)
|
||||||
|
.inner_join(post_aggregates::table)
|
||||||
|
.left_join(
|
||||||
|
community_follower::table.on(
|
||||||
|
post::community_id
|
||||||
|
.eq(community_follower::community_id)
|
||||||
|
.and(community_follower::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
community_user_ban::table.on(
|
||||||
|
post::community_id
|
||||||
|
.eq(community_user_ban::community_id)
|
||||||
|
.and(community_user_ban::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_saved::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_saved::post_id)
|
||||||
|
.and(post_saved::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_read::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_read::post_id)
|
||||||
|
.and(post_read::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.left_join(
|
||||||
|
post_like::table.on(
|
||||||
|
post::id
|
||||||
|
.eq(post_like::post_id)
|
||||||
|
.and(post_like::user_id.eq(user_id_join)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.select((
|
||||||
|
post::all_columns,
|
||||||
|
User_::safe_columns_tuple(),
|
||||||
|
Community::safe_columns_tuple(),
|
||||||
|
post_aggregates::all_columns,
|
||||||
|
community_follower::all_columns.nullable(),
|
||||||
|
community_user_ban::all_columns.nullable(),
|
||||||
|
post_saved::all_columns.nullable(),
|
||||||
|
post_read::all_columns.nullable(),
|
||||||
|
post_like::score.nullable(),
|
||||||
|
))
|
||||||
|
.into_boxed();
|
||||||
|
|
||||||
|
PostQueryBuilder {
|
||||||
|
conn,
|
||||||
|
query,
|
||||||
|
listing_type: &ListingType::All,
|
||||||
|
sort: &SortType::Hot,
|
||||||
|
for_creator_id: None,
|
||||||
|
for_community_id: None,
|
||||||
|
for_community_name: None,
|
||||||
|
search_term: None,
|
||||||
|
url_search: None,
|
||||||
|
show_nsfw: true,
|
||||||
|
saved_only: false,
|
||||||
|
unread_only: false,
|
||||||
|
page: None,
|
||||||
|
limit: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn listing_type(mut self, listing_type: &'a ListingType) -> Self {
|
||||||
|
self.listing_type = listing_type;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn sort(mut self, sort: &'a SortType) -> Self {
|
||||||
|
self.sort = sort;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn for_community_id<T: MaybeOptional<i32>>(mut self, for_community_id: T) -> Self {
|
||||||
|
self.for_community_id = for_community_id.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn for_community_name<T: MaybeOptional<String>>(mut self, for_community_name: T) -> Self {
|
||||||
|
self.for_community_name = for_community_name.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn for_creator_id<T: MaybeOptional<i32>>(mut self, for_creator_id: T) -> Self {
|
||||||
|
self.for_creator_id = for_creator_id.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn search_term<T: MaybeOptional<String>>(mut self, search_term: T) -> Self {
|
||||||
|
self.search_term = search_term.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn url_search<T: MaybeOptional<String>>(mut self, url_search: T) -> Self {
|
||||||
|
self.url_search = url_search.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn show_nsfw(mut self, show_nsfw: bool) -> Self {
|
||||||
|
self.show_nsfw = show_nsfw;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn saved_only(mut self, saved_only: bool) -> Self {
|
||||||
|
self.saved_only = saved_only;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn page<T: MaybeOptional<i64>>(mut self, page: T) -> Self {
|
||||||
|
self.page = page.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn limit<T: MaybeOptional<i64>>(mut self, limit: T) -> Self {
|
||||||
|
self.limit = limit.get_optional();
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn list(self) -> Result<Vec<PostView>, Error> {
|
||||||
|
use diesel::dsl::*;
|
||||||
|
|
||||||
|
let mut query = self.query;
|
||||||
|
|
||||||
|
query = match self.listing_type {
|
||||||
|
ListingType::Subscribed => query.filter(community_follower::user_id.is_not_null()), // TODO could be this: and(community_follower::user_id.eq(user_id_join)),
|
||||||
|
ListingType::Local => query.filter(community::local.eq(true)),
|
||||||
|
_ => query,
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(for_community_id) = self.for_community_id {
|
||||||
|
query = query
|
||||||
|
.filter(post::community_id.eq(for_community_id))
|
||||||
|
.then_order_by(post::stickied.desc());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(for_community_name) = self.for_community_name {
|
||||||
|
query = query
|
||||||
|
.filter(community::name.eq(for_community_name))
|
||||||
|
.filter(community::local.eq(true))
|
||||||
|
.then_order_by(post::stickied.desc());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(url_search) = self.url_search {
|
||||||
|
query = query.filter(post::url.eq(url_search));
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(search_term) = self.search_term {
|
||||||
|
let searcher = fuzzy_search(&search_term);
|
||||||
|
query = query.filter(
|
||||||
|
post::name
|
||||||
|
.ilike(searcher.to_owned())
|
||||||
|
.or(post::body.ilike(searcher)),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
query = match self.sort {
|
||||||
|
SortType::Active => query
|
||||||
|
.then_order_by(
|
||||||
|
hot_rank(post_aggregates::score, post_aggregates::newest_comment_time).desc(),
|
||||||
|
)
|
||||||
|
.then_order_by(post::published.desc()),
|
||||||
|
SortType::Hot => query
|
||||||
|
.then_order_by(hot_rank(post_aggregates::score, post::published).desc())
|
||||||
|
.then_order_by(post::published.desc()),
|
||||||
|
SortType::New => query.then_order_by(post::published.desc()),
|
||||||
|
SortType::TopAll => query.then_order_by(post_aggregates::score.desc()),
|
||||||
|
SortType::TopYear => query
|
||||||
|
.filter(post::published.gt(now - 1.years()))
|
||||||
|
.then_order_by(post_aggregates::score.desc()),
|
||||||
|
SortType::TopMonth => query
|
||||||
|
.filter(post::published.gt(now - 1.months()))
|
||||||
|
.then_order_by(post_aggregates::score.desc()),
|
||||||
|
SortType::TopWeek => query
|
||||||
|
.filter(post::published.gt(now - 1.weeks()))
|
||||||
|
.then_order_by(post_aggregates::score.desc()),
|
||||||
|
SortType::TopDay => query
|
||||||
|
.filter(post::published.gt(now - 1.days()))
|
||||||
|
.then_order_by(post_aggregates::score.desc()),
|
||||||
|
};
|
||||||
|
|
||||||
|
// If its for a specific user, show the removed / deleted
|
||||||
|
if let Some(for_creator_id) = self.for_creator_id {
|
||||||
|
query = query.filter(post::creator_id.eq(for_creator_id));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !self.show_nsfw {
|
||||||
|
query = query
|
||||||
|
.filter(post::nsfw.eq(false))
|
||||||
|
.filter(community::nsfw.eq(false));
|
||||||
|
};
|
||||||
|
|
||||||
|
// TODO These two might be wrong
|
||||||
|
if self.saved_only {
|
||||||
|
query = query.filter(post_saved::id.is_not_null());
|
||||||
|
};
|
||||||
|
|
||||||
|
if self.unread_only {
|
||||||
|
query = query.filter(post_read::id.is_not_null());
|
||||||
|
};
|
||||||
|
|
||||||
|
let (limit, offset) = limit_and_offset(self.page, self.limit);
|
||||||
|
|
||||||
|
let res = query
|
||||||
|
.limit(limit)
|
||||||
|
.offset(offset)
|
||||||
|
.filter(post::removed.eq(false))
|
||||||
|
.filter(post::deleted.eq(false))
|
||||||
|
.filter(community::removed.eq(false))
|
||||||
|
.filter(community::deleted.eq(false))
|
||||||
|
.load::<OutputTuple>(self.conn)?;
|
||||||
|
|
||||||
|
Ok(to_vec(res))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO turn this into a trait with an associated type
|
||||||
|
fn to_vec(posts: Vec<OutputTuple>) -> Vec<PostView> {
|
||||||
|
posts
|
||||||
|
.iter()
|
||||||
|
.map(|a| PostView {
|
||||||
|
post: a.0.to_owned(),
|
||||||
|
creator: a.1.to_owned(),
|
||||||
|
community: a.2.to_owned(),
|
||||||
|
counts: a.3.to_owned(),
|
||||||
|
subscribed: a.4.is_some(),
|
||||||
|
banned_from_community: a.5.is_some(),
|
||||||
|
saved: a.6.is_some(),
|
||||||
|
read: a.7.is_some(),
|
||||||
|
my_vote: a.8,
|
||||||
|
})
|
||||||
|
.collect::<Vec<PostView>>()
|
||||||
|
}
|
|
@ -0,0 +1,9 @@
|
||||||
|
-- post aggregates
|
||||||
|
drop table post_aggregates;
|
||||||
|
drop trigger post_aggregates_post on post;
|
||||||
|
drop trigger post_aggregates_comment_count on comment;
|
||||||
|
drop trigger post_aggregates_score on post_like;
|
||||||
|
drop function
|
||||||
|
post_aggregates_post,
|
||||||
|
post_aggregates_comment_count,
|
||||||
|
post_aggregates_score;
|
107
migrations/2020-12-10-152350_create_post_aggregates/up.sql
Normal file
107
migrations/2020-12-10-152350_create_post_aggregates/up.sql
Normal file
|
@ -0,0 +1,107 @@
|
||||||
|
-- Add post aggregates
|
||||||
|
create table post_aggregates (
|
||||||
|
id serial primary key,
|
||||||
|
post_id int references post on update cascade on delete cascade not null,
|
||||||
|
comments bigint not null default 0,
|
||||||
|
score bigint not null default 0,
|
||||||
|
upvotes bigint not null default 0,
|
||||||
|
downvotes bigint not null default 0,
|
||||||
|
newest_comment_time timestamp not null default now(),
|
||||||
|
unique (post_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
insert into post_aggregates (post_id, comments, score, upvotes, downvotes, newest_comment_time)
|
||||||
|
select
|
||||||
|
p.id,
|
||||||
|
coalesce(ct.comments, 0::bigint) as comments,
|
||||||
|
coalesce(pl.score, 0::bigint) as score,
|
||||||
|
coalesce(pl.upvotes, 0::bigint) as upvotes,
|
||||||
|
coalesce(pl.downvotes, 0::bigint) as downvotes,
|
||||||
|
greatest(ct.recent_comment_time, p.published) as newest_activity_time
|
||||||
|
from post p
|
||||||
|
left join (
|
||||||
|
select comment.post_id,
|
||||||
|
count(*) as comments,
|
||||||
|
max(comment.published) as recent_comment_time
|
||||||
|
from comment
|
||||||
|
group by comment.post_id
|
||||||
|
) ct on ct.post_id = p.id
|
||||||
|
left join (
|
||||||
|
select post_like.post_id,
|
||||||
|
sum(post_like.score) as score,
|
||||||
|
sum(post_like.score) filter (where post_like.score = 1) as upvotes,
|
||||||
|
-sum(post_like.score) filter (where post_like.score = '-1'::integer) as downvotes
|
||||||
|
from post_like
|
||||||
|
group by post_like.post_id
|
||||||
|
) pl on pl.post_id = p.id;
|
||||||
|
|
||||||
|
-- Add community aggregate triggers
|
||||||
|
|
||||||
|
-- initial post add
|
||||||
|
create function post_aggregates_post()
|
||||||
|
returns trigger language plpgsql
|
||||||
|
as $$
|
||||||
|
begin
|
||||||
|
IF (TG_OP = 'INSERT') THEN
|
||||||
|
insert into post_aggregates (post_id) values (NEW.id);
|
||||||
|
ELSIF (TG_OP = 'DELETE') THEN
|
||||||
|
delete from post_aggregates where post_id = OLD.id;
|
||||||
|
END IF;
|
||||||
|
return null;
|
||||||
|
end $$;
|
||||||
|
|
||||||
|
create trigger post_aggregates_post
|
||||||
|
after insert or delete on post
|
||||||
|
for each row
|
||||||
|
execute procedure post_aggregates_post();
|
||||||
|
|
||||||
|
-- comment count
|
||||||
|
create function post_aggregates_comment_count()
|
||||||
|
returns trigger language plpgsql
|
||||||
|
as $$
|
||||||
|
begin
|
||||||
|
IF (TG_OP = 'INSERT') THEN
|
||||||
|
update post_aggregates pa
|
||||||
|
set comments = comments + 1,
|
||||||
|
newest_comment_time = NEW.published
|
||||||
|
where pa.post_id = NEW.post_id;
|
||||||
|
ELSIF (TG_OP = 'DELETE') THEN
|
||||||
|
update post_aggregates pa
|
||||||
|
set comments = comments - 1
|
||||||
|
where pa.post_id = OLD.post_id;
|
||||||
|
END IF;
|
||||||
|
return null;
|
||||||
|
end $$;
|
||||||
|
|
||||||
|
create trigger post_aggregates_comment_count
|
||||||
|
after insert or delete on comment
|
||||||
|
for each row
|
||||||
|
execute procedure post_aggregates_comment_count();
|
||||||
|
|
||||||
|
-- post score
|
||||||
|
create function post_aggregates_score()
|
||||||
|
returns trigger language plpgsql
|
||||||
|
as $$
|
||||||
|
begin
|
||||||
|
IF (TG_OP = 'INSERT') THEN
|
||||||
|
update post_aggregates pa
|
||||||
|
set score = score + NEW.score,
|
||||||
|
upvotes = case when NEW.score = 1 then upvotes + 1 else upvotes end,
|
||||||
|
downvotes = case when NEW.score = -1 then downvotes + 1 else downvotes end
|
||||||
|
where pa.post_id = NEW.post_id;
|
||||||
|
|
||||||
|
ELSIF (TG_OP = 'DELETE') THEN
|
||||||
|
update post_aggregates pa
|
||||||
|
set score = score - OLD.score,
|
||||||
|
upvotes = case when OLD.score = 1 then upvotes - 1 else upvotes end,
|
||||||
|
downvotes = case when OLD.score = -1 then downvotes - 1 else downvotes end
|
||||||
|
where pa.post_id = OLD.post_id;
|
||||||
|
|
||||||
|
END IF;
|
||||||
|
return null;
|
||||||
|
end $$;
|
||||||
|
|
||||||
|
create trigger post_aggregates_score
|
||||||
|
after insert or delete on post_like
|
||||||
|
for each row
|
||||||
|
execute procedure post_aggregates_score();
|
Loading…
Reference in a new issue