feat(grant): add unique index for active grants

raystack · Oct 20, 2023 · bcbbde8 · bcbbde8
1 parent 99e254e
commit bcbbde8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/internal/store/postgres/grant_repository_test.go b/internal/store/postgres/grant_repository_test.go
@@ -103,6 +103,17 @@ func (s *GrantRepositoryTestSuite) SetupSuite() {
 	s.Require().NoError(err)
 }
 
+func (s *GrantRepositoryTestSuite) AfterTest(suiteName, testName string) {
+	// clean grants table
+	db, err := s.store.DB().DB()
+	if err != nil {
+		s.T().Fatal(err)
+	}
+	if _, err := db.Exec("DELETE FROM grants"); err != nil {
+		s.T().Fatal(err)
+	}
+}
+
 func (s *GrantRepositoryTestSuite) TearDownSuite() {
 	// Clean tests
 	db, err := s.store.DB().DB()

diff --git a/internal/store/postgres/migrations/000019_create_grant_unique_index.down.sql b/internal/store/postgres/migrations/000019_create_grant_unique_index.down.sql
@@ -0,0 +1 @@
+DROP INDEX IF EXISTS "unique_active_grants_index";
diff --git a/internal/store/postgres/migrations/000019_create_grant_unique_index.up.sql b/internal/store/postgres/migrations/000019_create_grant_unique_index.up.sql
@@ -0,0 +1,3 @@
+CREATE UNIQUE INDEX IF NOT EXISTS "unique_active_grants_index" ON "grants" ("account_id", "account_type", "permissions")
+WHERE
+  "status" = 'active';