mirror of
https://github.com/home-assistant/core
synced 2024-08-02 23:40:32 +02:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""A pool for sqlite connections."""
|
|
import threading
|
|
|
|
from sqlalchemy.pool import NullPool, StaticPool
|
|
|
|
|
|
class RecorderPool(StaticPool, NullPool):
|
|
"""A hybird of NullPool and StaticPool.
|
|
|
|
When called from the creating thread acts like StaticPool
|
|
When called from any other thread, acts like NullPool
|
|
"""
|
|
|
|
def __init__(self, *args, **kw): # pylint: disable=super-init-not-called
|
|
"""Create the pool."""
|
|
self._tid = threading.current_thread().ident
|
|
StaticPool.__init__(self, *args, **kw)
|
|
|
|
def _do_return_conn(self, conn):
|
|
if threading.current_thread().ident == self._tid:
|
|
return super()._do_return_conn(conn)
|
|
conn.close()
|
|
|
|
def dispose(self):
|
|
"""Dispose of the connection."""
|
|
if threading.current_thread().ident == self._tid:
|
|
return super().dispose()
|
|
|
|
def _do_get(self):
|
|
if threading.current_thread().ident == self._tid:
|
|
return super()._do_get()
|
|
return super( # pylint: disable=bad-super-call
|
|
NullPool, self
|
|
)._create_connection()
|