Code

Ticket #12234: charfield_index_opclass_fix_v1.diff

File charfield_index_opclass_fix_v1.diff, 2.2 KB (added by jbronn, 5 years ago)

Add extra indexes to include operator class.

Line 
1Index: django/db/backends/postgresql/creation.py
2===================================================================
3--- django/db/backends/postgresql/creation.py   (revision 11784)
4+++ django/db/backends/postgresql/creation.py   (working copy)
5@@ -34,3 +34,42 @@
6         if settings.TEST_DATABASE_CHARSET:
7             return "WITH ENCODING '%s'" % settings.TEST_DATABASE_CHARSET
8         return ''
9+
10+    def sql_indexes_for_field(self, model, f, style):
11+        from django.db.models.fields import CharField, TextField
12+
13+        if f.db_index and not f.unique:
14+            qn = self.connection.ops.quote_name
15+            tablespace = f.db_tablespace or model._meta.db_tablespace
16+            if tablespace:
17+                sql = self.connection.ops.tablespace_sql(tablespace)
18+                if sql:
19+                    tablespace_sql = ' ' + sql
20+                else:
21+                    tablespace_sql = ''
22+            else:
23+                tablespace_sql = ''
24+
25+            def get_index_sql(index_name, db_table, column, opclass=''):
26+                return (style.SQL_KEYWORD('CREATE INDEX') + ' ' +
27+                        style.SQL_TABLE(index_name) + ' ' +
28+                        style.SQL_KEYWORD('ON') + ' ' +
29+                        style.SQL_TABLE(db_table) + ' ' +
30+                        "(%s%s)" % (style.SQL_FIELD(column), opclass) +
31+                        "%s;" % tablespace_sql)
32+
33+            db_table = model._meta.db_table
34+            output = [get_index_sql(qn('%s_%s' % (db_table, f.column)),
35+                                    qn(db_table), qn(f.column))]
36+
37+            if isinstance(f, CharField):
38+                output.append(get_index_sql(qn('%s_%s_like' % (db_table, f.column)),
39+                                            qn(db_table), qn(f.column),
40+                                            ' varchar_pattern_ops'))
41+            elif isinstance(f, TextField):
42+                output.append(get_index_sql(qn('%s_%s_like' % (db_table, f.column)),
43+                                            qn(db_table), qn(f.column),
44+                                            ' text_pattern_ops'))               
45+        else:
46+            output = []
47+        return output