PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Création d'un index Gin avec Trigram (gin_trgm_ops) dans le modèle Django

J'ai trouvé un article du 12/2020 qui utilise la dernière version de Django ORM en tant que tel :

class Author(models.Model):
    first_name = models.CharField(max_length=100)
    last_name = models.CharField(max_length=100)

    class Meta:
        indexes = [
            GinIndex(
                name='review_author_ln_gin_idx', 
                fields=['last_name'], 
                opclasses=['gin_trgm_ops'],
            )
        ]

Si, comme l'affiche originale, vous cherchiez à créer un index qui fonctionne avec icontains, vous devrez indexer le UPPER() de la colonne, ce qui nécessite une manipulation spéciale de OpClass :

from django.db.models.functions import Upper
from django.contrib.postgres.indexes import GinIndex, OpClass

class Author(models.Model):
        indexes = [
            GinIndex(
                OpClass(Upper('last_name'), name='gin_trgm_ops'),
                name='review_author_ln_gin_idx',
            )
        ]

Inspiré d'un ancien article à ce sujet, j'ai atterri sur un actuel qui donne la solution suivante pour un GistIndex :

Mise à jour :Depuis Django-1.11, les choses semblent être plus simples, comme cette réponse et docs Django suggestion :

from django.contrib.postgres.indexes import GinIndex

class MyModel(models.Model):
    the_field = models.CharField(max_length=512, db_index=True)

    class Meta:
        indexes = [GinIndex(fields=['the_field'])]

De Django-2.2 , un attribut opclasses sera disponible dans la classe class Index(fields=(), name=None, db_tablespace=None, opclasses=()) à cet effet.

from django.contrib.postgres.indexes import GistIndex

class GistIndexTrgrmOps(GistIndex):
    def create_sql(self, model, schema_editor):
        # - this Statement is instantiated by the _create_index_sql()
        #   method of django.db.backends.base.schema.BaseDatabaseSchemaEditor.
        #   using sql_create_index template from
        #   django.db.backends.postgresql.schema.DatabaseSchemaEditor
        # - the template has original value:
        #   "CREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s)%(extra)s"
        statement = super().create_sql(model, schema_editor)
        # - however, we want to use a GIST index to accelerate trigram
        #   matching, so we want to add the gist_trgm_ops index operator
        #   class
        # - so we replace the template with:
        #   "CREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s gist_trgrm_ops)%(extra)s"
        statement.template =\
            "CREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s gist_trgm_ops)%(extra)s"

        return statement

Que vous pouvez ensuite utiliser dans votre classe de modèle comme ceci :

class YourModel(models.Model):
    some_field = models.TextField(...)

    class Meta:
        indexes = [
            GistIndexTrgrmOps(fields=['some_field'])
        ]