WIP - add index

api/edge_api/identities/migrations/0001_add_search_by_traits_capability_for_edge_identities.py

@@ -1,5 +1,6 @@
-# Generated by Django 3.2.25 on 2024-08-13 20:04
+# Generated by Django 3.2.25 on 2024-08-14 11:11
 
+import django.contrib.postgres.indexes
 from django.db import migrations, models
 import django.db.models.deletion
 
@@ -25,4 +26,8 @@ class Migration(migrations.Migration):
                 ('environment', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='environments.environment')),
             ],
         ),
+        migrations.AddIndex(
+            model_name='edgeidentitymeta',
+            index=django.contrib.postgres.indexes.GinIndex(fields=['searchable_traits'], name='edge_api_id_searcha_816c05_gin'),
+        ),
     ]

api/edge_api/identities/models.py

@@ -2,6 +2,7 @@
 import typing
 from contextlib import suppress
 
+from django.contrib.postgres.indexes import GinIndex
 from django.db import models
 from django.db.models import Prefetch, Q
 from flag_engine.features.models import FeatureStateModel
@@ -314,11 +315,15 @@ class EdgeIdentityMeta(models.Model):
     _SEPARATOR = "<separator>"
     _EQUALITY_CHARS = "<equals>"
 
+    class Meta:
+        indexes = [GinIndex(fields=["searchable_traits"])]
+
     def update_searchable_traits(self, traits: typing.List[TraitModel]) -> None:
-        searchable_trait_keys = self.environment.trait_meta_items.filter(
-            is_searchable=True
-        ).values_list("trait_key", flat=True)
+        # searchable_trait_keys = self.environment.trait_meta_items.filter(
+        #     is_searchable=True
+        # ).values_list("trait_key", flat=True)
         self.searchable_traits = self._SEPARATOR.join(
             f"{trait.trait_key}{self._EQUALITY_CHARS}{trait.trait_value}"
-            for trait in filter(lambda t: t.trait_key in searchable_trait_keys, traits)
+            # for trait in filter(lambda t: t.trait_key in searchable_trait_keys, traits)
+            for trait in traits
         )