decentral1se/pandora
1
0
Fork 0
forked from 0x2620/pandora
Code Issues Pull requests Projects Releases Wiki Activity

make sure existing index is using gin

Browse source
This commit is contained in:
j 2016-06-26 16:55:58 +02:00
parent ab0dfddf31
commit de9b062d63
1 changed files with 24 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
pandora/item/management/commands/sqlfindindex.py
View file

@ -21,10 +21,11 @@ class Command(BaseCommand):
def add_arguments(self, parser): def add_arguments(self, parser):
parser.add_argument('--debug', action='store_true', dest='debug', parser.add_argument('--debug', action='store_true', dest='debug',
default=False, help='print sql commans') default=False, help='print sql commans')
def handle(self, **options): def handle(self, **options):
cursor = connection.cursor() cursor = connection.cursor()
def create_index(index, table, key): def create_index(index, table, key):
sql = 'CREATE INDEX "%s" ON "%s" USING gin ("%s" gin_trgm_ops)' % (index, table, key) sql = 'CREATE INDEX "%s" ON "%s" USING gin ("%s" gin_trgm_ops)' % (index, table, key)
if options['debug']: if options['debug']:
@ -33,13 +34,28 @@ class Command(BaseCommand):
if settings.DB_GIN_TRGM: if settings.DB_GIN_TRGM:
import entity.models import entity.models
for table_name, name in ( for table, column in (
(models.ItemFind._meta.db_table, 'value'), # Item Find (models.ItemFind._meta.db_table, 'value'), # Item Find
(models.Clip._meta.db_table, 'findvalue'), # Clip Find (models.Clip._meta.db_table, 'findvalue'), # Clip Find
(entity.models.Find._meta.db_table, 'value'),# Entity Find (entity.models.Find._meta.db_table, 'value'), # Entity Find
): ):
cursor = connection.cursor() cursor = connection.cursor()
indexes = connection.introspection.get_indexes(cursor, table_name) indexes = connection.introspection.get_indexes(cursor, table)
if name not in indexes: drop = []
create_index("%s_%s_idx"%(table_name, name), table_name, name) if column in indexes:
sql = "SELECT indexname, indexdef FROM pg_catalog.pg_indexes " + \
"WHERE indexname LIKE '%{table}%' AND indexname LIKE '%{column}%'".format(table=table, column=column)
cursor.execute(sql)
for r in cursor:
if 'USING gin' not in r[1]:
drop.append(r[0])
if drop:
for idx in drop:
sql = 'DROP INDEX ' + idx
if options['debug']:
print(sql)
cursor.execute(sql)
indexes = connection.introspection.get_indexes(cursor, table)
if column not in indexes:
create_index("%s_%s_idx" % (table, column), table, column)
transaction.commit() transaction.commit()