CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

chg: Use connection pool in indexing

pull/251/head
Raphaël Vinot 2021-08-18 17:31:17 +02:00
parent 046387a9f4
commit 7495e5b4f1
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lookyloo/indexing.py
View File

@ -7,7 +7,8 @@ from typing import List, Tuple, Set, Dict, Optional, Iterable
from collections import defaultdict from collections import defaultdict
import re import re
from redis import Redis from redis import Redis, ConnectionPool
from redis.connection import UnixDomainSocketConnection
from har2tree import CrawledTree from har2tree import CrawledTree
from .helpers import get_socket_path, get_public_suffix_list from .helpers import get_socket_path, get_public_suffix_list
@ -16,11 +17,16 @@ from .helpers import get_socket_path, get_public_suffix_list
class Indexing(): class Indexing():
def __init__(self) -> None: def __init__(self) -> None:
self.redis: Redis = Redis(unix_socket_path=get_socket_path('indexing'), decode_responses=True) self.redis_pool: ConnectionPool = ConnectionPool(connection_class=UnixDomainSocketConnection,
path=get_socket_path('indexing'), decode_responses=True)
def clear_indexes(self): def clear_indexes(self):
self.redis.flushdb() self.redis.flushdb()
@property
def redis(self):
return Redis(connection_pool=self.redis_pool)
# ###### Cookies ###### # ###### Cookies ######
@property @property