mirror of https://github.com/zulip/zulip.git
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
from django.db import connection
|
|
from zerver.lib.db import TimeTrackingConnection
|
|
|
|
import sqlalchemy
|
|
|
|
# This is a Pool that doesn't close connections. Therefore it can be used with
|
|
# existing Django database connections.
|
|
class NonClosingPool(sqlalchemy.pool.NullPool):
|
|
def status(self):
|
|
# type: () -> str
|
|
return "NonClosingPool"
|
|
|
|
def _do_return_conn(self, conn):
|
|
# type: (sqlalchemy.engine.base.Connection) -> None
|
|
pass
|
|
|
|
def recreate(self):
|
|
# type: () -> NonClosingPool
|
|
return self.__class__(creator=self._creator,
|
|
recycle=self._recycle,
|
|
use_threadlocal=self._use_threadlocal,
|
|
reset_on_return=self._reset_on_return,
|
|
echo=self.echo,
|
|
logging_name=self._orig_logging_name,
|
|
_dispatch=self.dispatch)
|
|
|
|
sqlalchemy_engine = None
|
|
def get_sqlalchemy_connection():
|
|
# type: () -> sqlalchemy.engine.base.Connection
|
|
global sqlalchemy_engine
|
|
if sqlalchemy_engine is None:
|
|
def get_dj_conn():
|
|
# type: () -> TimeTrackingConnection
|
|
connection.ensure_connection()
|
|
return connection.connection
|
|
sqlalchemy_engine = sqlalchemy.create_engine('postgresql://',
|
|
creator=get_dj_conn,
|
|
poolclass=NonClosingPool,
|
|
pool_reset_on_return=False)
|
|
sa_connection = sqlalchemy_engine.connect()
|
|
sa_connection.execution_options(autocommit=False)
|
|
return sa_connection
|