From 53cf26597dc249e9c5f90a29082b7648bdc802d8 Mon Sep 17 00:00:00 2001
From: Mengran Lan <lanmengran@qq.com>
Date: Fri, 19 Apr 2024 16:54:13 +0800
Subject: [PATCH] perf(db): add indexes for table prover_task, columns
 respectively: task_id, created_at (#1283)

---
 database/migrate/migrate_test.go               |  6 +++---
 ..._add_index_taskid_createdat_prover_task.sql | 18 ++++++++++++++++++
 2 files changed, 21 insertions(+), 3 deletions(-)
 create mode 100644 database/migrate/migrations/00018_add_index_taskid_createdat_prover_task.sql

diff --git a/database/migrate/migrate_test.go b/database/migrate/migrate_test.go
index d60671022..18c281ad8 100644
--- a/database/migrate/migrate_test.go
+++ b/database/migrate/migrate_test.go
@@ -59,20 +59,20 @@ func testResetDB(t *testing.T) {
 	cur, err := Current(pgDB)
 	assert.NoError(t, err)
 	// total number of tables.
-	assert.Equal(t, int64(17), cur)
+	assert.Equal(t, int64(18), cur)
 }
 
 func testMigrate(t *testing.T) {
 	assert.NoError(t, Migrate(pgDB))
 	cur, err := Current(pgDB)
 	assert.NoError(t, err)
-	assert.Equal(t, int64(17), cur)
+	assert.Equal(t, int64(18), cur)
 }
 
 func testRollback(t *testing.T) {
 	version, err := Current(pgDB)
 	assert.NoError(t, err)
-	assert.Equal(t, int64(17), version)
+	assert.Equal(t, int64(18), version)
 
 	assert.NoError(t, Rollback(pgDB, nil))
 
diff --git a/database/migrate/migrations/00018_add_index_taskid_createdat_prover_task.sql b/database/migrate/migrations/00018_add_index_taskid_createdat_prover_task.sql
new file mode 100644
index 000000000..f0751f2bb
--- /dev/null
+++ b/database/migrate/migrations/00018_add_index_taskid_createdat_prover_task.sql
@@ -0,0 +1,18 @@
+-- +goose Up
+-- +goose StatementBegin
+
+create index if not exists idx_prover_task_created_at on prover_task(created_at) where deleted_at IS NULL;
+
+create index if not exists idx_prover_task_task_id on prover_task(task_id) where deleted_at IS NULL;
+
+-- +goose StatementEnd
+
+-- +goose Down
+-- +goose StatementBegin
+
+drop index if exists idx_prover_task_created_at;
+
+drop index if exists idx_prover_task_task_id;
+
+
+-- +goose StatementEnd