From 5554d3002a3ff768791fd7683e5d02c7f0c4319a Mon Sep 17 00:00:00 2001 From: Rudy Fraser Date: Sun, 8 Dec 2024 04:02:24 -0500 Subject: [PATCH] Add indexes and properly store quoteUris for mediaquotes --- rsky-feedgen/Cargo.toml | 2 +- rsky-feedgen/migrations/2024-12-01-074204_021/down.sql | 3 +++ rsky-feedgen/migrations/2024-12-01-074204_021/up.sql | 4 ++++ rsky-feedgen/src/apis/mod.rs | 3 ++- 4 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 rsky-feedgen/migrations/2024-12-01-074204_021/down.sql create mode 100644 rsky-feedgen/migrations/2024-12-01-074204_021/up.sql diff --git a/rsky-feedgen/Cargo.toml b/rsky-feedgen/Cargo.toml index 8e5c4f41..f12e0d98 100644 --- a/rsky-feedgen/Cargo.toml +++ b/rsky-feedgen/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "rsky-feedgen" -version = "1.0.1" +version = "1.0.2" authors = ["Rudy Fraser "] description = "A framework for building AT Protocol feed generators, in Rust." license = "Apache-2.0" diff --git a/rsky-feedgen/migrations/2024-12-01-074204_021/down.sql b/rsky-feedgen/migrations/2024-12-01-074204_021/down.sql new file mode 100644 index 00000000..3485f26f --- /dev/null +++ b/rsky-feedgen/migrations/2024-12-01-074204_021/down.sql @@ -0,0 +1,3 @@ +-- This file should undo anything in `up.sql` +DROP INDEX IF EXISTS idx_like_subjecturi_indexedat; +DROP INDEX IF EXISTS idx_post_createdAt_cid; diff --git a/rsky-feedgen/migrations/2024-12-01-074204_021/up.sql b/rsky-feedgen/migrations/2024-12-01-074204_021/up.sql new file mode 100644 index 00000000..1457315d --- /dev/null +++ b/rsky-feedgen/migrations/2024-12-01-074204_021/up.sql @@ -0,0 +1,4 @@ +-- Your SQL goes here +CREATE INDEX idx_like_subjecturi_indexedat ON public.like ("subjectUri", "indexedAt"); +CREATE INDEX idx_post_createdAt_cid ON "post" ("createdAt" DESC, "cid" DESC) + WHERE "replyParent" IS NULL AND "replyRoot" IS NULL; \ No newline at end of file diff --git a/rsky-feedgen/src/apis/mod.rs b/rsky-feedgen/src/apis/mod.rs index 3f25acb6..38a702f8 100644 --- a/rsky-feedgen/src/apis/mod.rs +++ b/rsky-feedgen/src/apis/mod.rs @@ -174,7 +174,6 @@ pub async fn get_blacksky_trending( connection: ReadReplicaConn, config: &State, ) -> Result { - use std::env; let trending_percentile_min = config.trending_percentile_min.clone(); let params_cursor = match params_cursor { @@ -610,6 +609,8 @@ pub async fn queue_creation( }; } Embeds::RecordWithMedia(e) => { + new_post.quote_cid = Some(e.record.record.cid); + new_post.quote_uri = Some(e.record.record.uri); match e.media { MediaUnion::Images(m) => { for image in m.images {