add index for key

bricks-cloud · Jul 30, 2024 · 9b1fcae · 9b1fcae
1 parent 4db5063
commit 9b1fcae
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/cmd/bricksllm/main.go b/cmd/bricksllm/main.go
@@ -96,6 +96,11 @@ func main() {
 		log.Sugar().Fatalf("error create create at index for keys: %v", err)
 	}
 
+	err = store.CreateKeyIndexForKeys()
+	if err != nil {
+		log.Sugar().Fatalf("error create create at index for keys: %v", err)
+	}
+
 	err = store.CreateEventsTable()
 	if err != nil {
 		log.Sugar().Fatalf("error creating events table: %v", err)

diff --git a/internal/storage/postgresql/key.go b/internal/storage/postgresql/key.go
@@ -83,6 +83,21 @@ func (s *Store) CreateCreateAtIndexForKeys() error {
 	return nil
 }
 
+func (s *Store) CreateKeyIndexForKeys() error {
+	createIndexQuery := `
+	CREATE INDEX IF NOT EXISTS key_idx ON keys(key);
+	`
+
+	ctxTimeout, cancel := context.WithTimeout(context.Background(), s.wt)
+	defer cancel()
+	_, err := s.db.ExecContext(ctxTimeout, createIndexQuery)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func (s *Store) GetKeys(tags, keyIds []string, provider string) ([]*key.ResponseKey, error) {
 	ctxTimeout, cancel := context.WithTimeout(context.Background(), s.rt)
 	defer cancel()