From daec9ab10e192f82155ecf46d4ab257ca6655c24 Mon Sep 17 00:00:00 2001
From: Blackle Morisanchetto <isabelle@blackle-mori.com>
Date: Wed, 31 Aug 2022 05:27:39 -0400
Subject: [feature] Sort follow requests, followers, and following by
 updated_at (#774)

* Sort follow requests, followers, and following by updated_at

* Add migration to regenerate indexes for follows and follow requests
---
 ...0220830014732_regenerate_indexes_for_follows.go | 114 +++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 internal/db/bundb/migrations/20220830014732_regenerate_indexes_for_follows.go

(limited to 'internal/db/bundb/migrations')

diff --git a/internal/db/bundb/migrations/20220830014732_regenerate_indexes_for_follows.go b/internal/db/bundb/migrations/20220830014732_regenerate_indexes_for_follows.go
new file mode 100644
index 000000000..259d757f0
--- /dev/null
+++ b/internal/db/bundb/migrations/20220830014732_regenerate_indexes_for_follows.go
@@ -0,0 +1,114 @@
+/*
+   GoToSocial
+   Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU Affero General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU Affero General Public License for more details.
+
+   You should have received a copy of the GNU Affero General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package migrations
+
+import (
+	"context"
+
+	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+	"github.com/uptrace/bun"
+)
+
+func init() {
+	up := func(ctx context.Context, db *bun.DB) error {
+		return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
+			// drop indexes for these tables
+			if _, err := tx.
+				NewDropIndex().
+				Index("follow_requests_account_id_idx").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewDropIndex().
+				Index("follow_requests_target_account_id_idx").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewDropIndex().
+				Index("follows_account_id_idx").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewDropIndex().
+				Index("follows_target_account_id_idx").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// now re-create them with an ordering
+			if _, err := tx.
+				NewCreateIndex().
+				Model(&gtsmodel.FollowRequest{}).
+				Index("follow_requests_account_id_idx").
+				Column("account_id").
+				ColumnExpr("updated_at DESC").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewCreateIndex().
+				Model(&gtsmodel.FollowRequest{}).
+				Index("follow_requests_target_account_id_idx").
+				Column("target_account_id").
+				ColumnExpr("updated_at DESC").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewCreateIndex().
+				Model(&gtsmodel.Follow{}).
+				Index("follows_account_id_idx").
+				Column("account_id").
+				ColumnExpr("updated_at DESC").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			if _, err := tx.
+				NewCreateIndex().
+				Model(&gtsmodel.Follow{}).
+				Index("follows_target_account_id_idx").
+				Column("target_account_id").
+				ColumnExpr("updated_at DESC").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			return nil
+		})
+	}
+
+	down := func(ctx context.Context, db *bun.DB) error {
+		return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
+			return nil
+		})
+	}
+
+	if err := Migrations.Register(up, down); err != nil {
+		panic(err)
+	}
+}
-- 
cgit v1.3