wip db migration

This commit is contained in:
Felix Ableitner 2024-09-24 14:44:07 +02:00
parent a4e2461819
commit 56a8f9fbaf
6 changed files with 89 additions and 26 deletions

View File

@ -276,8 +276,7 @@ mod tests {
private_key: None,
public_key: "pubkey".to_owned(),
last_refreshed_at: inserted_person.published,
inbox_url: inserted_person.inbox_url.clone(),
shared_inbox_url: None,
inbox_id: 0,
matrix_user_id: None,
ban_expires: None,
instance_id: inserted_instance.id,

View File

@ -321,6 +321,14 @@ diesel::table! {
}
}
diesel::table! {
inbox (id) {
id -> Int4,
#[max_length = 255]
url -> Varchar,
}
}
diesel::table! {
instance (id) {
id -> Int4,
@ -678,14 +686,11 @@ diesel::table! {
last_refreshed_at -> Timestamptz,
banner -> Nullable<Text>,
deleted -> Bool,
#[max_length = 255]
inbox_url -> Varchar,
#[max_length = 255]
shared_inbox_url -> Nullable<Varchar>,
matrix_user_id -> Nullable<Text>,
bot_account -> Bool,
ban_expires -> Nullable<Timestamptz>,
instance_id -> Int4,
inbox_id -> Int4,
}
}
@ -770,7 +775,7 @@ diesel::table! {
featured_local -> Bool,
url_content_type -> Nullable<Text>,
alt_text -> Nullable<Text>,
scheduled_publish_time -> Nullable<Timestamptz>
scheduled_publish_time -> Nullable<Timestamptz>,
}
}
@ -1043,6 +1048,7 @@ diesel::joinable!(mod_transfer_community -> community (community_id));
diesel::joinable!(oauth_account -> local_user (local_user_id));
diesel::joinable!(oauth_account -> oauth_provider (oauth_provider_id));
diesel::joinable!(password_reset_request -> local_user (local_user_id));
diesel::joinable!(person -> inbox (inbox_id));
diesel::joinable!(person -> instance (instance_id));
diesel::joinable!(person_aggregates -> person (person_id));
diesel::joinable!(person_ban -> person (person_id));
@ -1100,6 +1106,7 @@ diesel::allow_tables_to_appear_in_same_query!(
federation_blocklist,
federation_queue_state,
image_details,
inbox,
instance,
instance_block,
language,

View File

@ -3,7 +3,6 @@ use crate::schema::{person, person_follower};
use crate::{
newtypes::{DbUrl, InstanceId, PersonId},
sensitive::SensitiveString,
source::placeholder_apub_url,
};
use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
@ -45,11 +44,6 @@ pub struct Person {
pub banner: Option<DbUrl>,
/// Whether the person is deleted.
pub deleted: bool,
#[cfg_attr(feature = "full", ts(skip))]
#[serde(skip, default = "placeholder_apub_url")]
pub inbox_url: DbUrl,
#[serde(skip)]
pub shared_inbox_url: Option<DbUrl>,
/// A matrix id, usually given an @person:matrix.org
pub matrix_user_id: Option<String>,
/// Whether the person is a bot account.
@ -57,6 +51,9 @@ pub struct Person {
/// When their ban, if it exists, expires, if at all.
pub ban_expires: Option<DateTime<Utc>>,
pub instance_id: InstanceId,
#[cfg_attr(feature = "full", ts(skip))]
#[serde(skip, default)]
pub inbox_id: i32,
}
#[derive(Clone, derive_new::new)]
@ -91,9 +88,7 @@ pub struct PersonInsertForm {
#[new(default)]
pub deleted: Option<bool>,
#[new(default)]
pub inbox_url: Option<DbUrl>,
#[new(default)]
pub shared_inbox_url: Option<DbUrl>,
pub inbox_id: Option<i32>,
#[new(default)]
pub matrix_user_id: Option<String>,
#[new(default)]
@ -118,8 +113,7 @@ pub struct PersonUpdateForm {
pub last_refreshed_at: Option<DateTime<Utc>>,
pub banner: Option<Option<DbUrl>>,
pub deleted: Option<bool>,
pub inbox_url: Option<DbUrl>,
pub shared_inbox_url: Option<Option<DbUrl>>,
pub inbox_id: Option<i32>,
pub matrix_user_id: Option<Option<String>>,
pub bot_account: Option<bool>,
pub ban_expires: Option<Option<DateTime<Utc>>>,

View File

@ -4,13 +4,14 @@ use diesel::{
dsl::{count_star, not},
result::Error,
ExpressionMethods,
JoinOnDsl,
QueryDsl,
};
use diesel_async::RunQueryDsl;
use lemmy_db_schema::{
newtypes::{CommunityId, DbUrl, InstanceId, PersonId},
schema::{community, community_follower, person},
utils::{functions::coalesce, get_conn, DbPool},
schema::{community, community_follower, inbox, person},
utils::{get_conn, DbPool},
};
impl CommunityFollowerView {
@ -32,15 +33,13 @@ impl CommunityFollowerView {
community_follower::table
.inner_join(community::table)
.inner_join(person::table)
.inner_join(inbox::table.on(person::inbox_id.eq(inbox::id)))
.filter(person::instance_id.eq(instance_id))
.filter(community::local) // this should be a no-op since community_followers table only has
// local-person+remote-community or remote-person+local-community
.filter(not(person::local))
.filter(community_follower::published.gt(published_since.naive_utc()))
.select((
community::id,
coalesce(person::shared_inbox_url, person::inbox_url),
))
.select((community::id, inbox::url))
.distinct() // only need each community_id, inbox combination once
.load::<(CommunityId, DbUrl)>(conn)
.await
@ -51,10 +50,11 @@ impl CommunityFollowerView {
) -> Result<Vec<DbUrl>, Error> {
let conn = &mut get_conn(pool).await?;
let res = community_follower::table
.inner_join(person::table)
.inner_join(inbox::table.on(person::inbox_id.eq(inbox::id)))
.filter(community_follower::community_id.eq(community_id))
.filter(not(person::local))
.inner_join(person::table)
.select(coalesce(person::shared_inbox_url, person::inbox_url))
.select(inbox::url)
.distinct()
.load::<DbUrl>(conn)
.await?;

View File

@ -0,0 +1,11 @@
ALTER TABLE person
DROP CONSTRAINT person_inbox_id_fkey;
ALTER TABLE person
DROP COLUMN inbox_url;
DROP TABLE inbox;
ALTER TABLE person add COLUMN inbox_url character varying(255) not null default generate_unique_changeme();
ALTER TABLE person add COLUMN shared_inbox_url character varying(255) not null default generate_unique_changeme();

View File

@ -0,0 +1,52 @@
-- Remove text fields inbox_url, shared_inbox_url from person, community and site.
-- Instead use a foreign key for these urls so they can be deduplicated.
-- New inbox table
-- TODO: add trigger which removes unused inbox items
CREATE TABLE inbox (
id serial PRIMARY KEY,
url varchar(255) NOT NULL
);
-- Move existing inbox values to inbox table, and replace with foreign key
ALTER TABLE person
ADD COLUMN inbox_id int;
WITH inboxes AS (
SELECT
id AS person_id,
coalesce(shared_inbox_url, inbox_url) AS url
FROM
person
),
inserted AS (
INSERT INTO inbox (url)
SELECT
url
FROM
inboxes
ON CONFLICT
DO NOTHING
RETURNING id,
url)
UPDATE
person
SET
inbox_id = inserted.id
FROM
inboxes,
inserted
WHERE
person.id = inboxes.person_id
AND inserted.url = inboxes.url;
ALTER TABLE person
ADD CONSTRAINT person_inbox_id_fkey FOREIGN KEY (inbox_id) REFERENCES inbox(id) ON UPDATE CASCADE ON DELETE CASCADE;
ALTER TABLE person
ALTER COLUMN inbox_id SET NOT NULL;
-- Drop old columns and rename new one
ALTER TABLE person DROP COLUMN inbox_url;
ALTER TABLE person DROP COLUMN shared_inbox_url;
-- TODO: same thing for community and site