UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Drop mysql restarts

Browse Source
This commit is contained in:
Беклемишев Петр Алексеевич 2024-05-31 13:26:23 +07:00
parent e59e1f5049
commit ad32a2ef3c
1 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/apps/webui/internal/wrappers.py
View File

@ -1,8 +1,8 @@
from contextvars import ContextVar
from peewee import PostgresqlDatabase, InterfaceError as PeeWeeInterfaceError, MySQLDatabase, _ConnectionState
from peewee import PostgresqlDatabase, InterfaceError as PeeWeeInterfaceError, _ConnectionState
from playhouse.db_url import register_database
from playhouse.pool import PooledPostgresqlDatabase, PooledMySQLDatabase
from playhouse.pool import PooledPostgresqlDatabase
from playhouse.shortcuts import ReconnectMixin
from psycopg2 import OperationalError
from psycopg2.errors import InterfaceError
@ -26,7 +26,7 @@ class PeeweeConnectionState(_ConnectionState):
class CustomReconnectMixin(ReconnectMixin):
reconnect_errors = (
# default ReconnectMixin exceptions (MySQL specific)
# default ReconnectMixin exceptions
*ReconnectMixin.reconnect_errors,
# psycopg2
(OperationalError, 'termin'),
@ -44,16 +44,6 @@ class ReconnectingPooledPostgresqlDatabase(CustomReconnectMixin, PooledPostgresq
pass
class ReconnectingMySQLDatabase(CustomReconnectMixin, MySQLDatabase):
pass
class ReconnectingPooledMySQLDatabase(CustomReconnectMixin, PooledMySQLDatabase):
pass
def register_peewee_databases():
register_database(MySQLDatabase, 'mysql')
register_database(PooledMySQLDatabase, 'mysql+pool')
register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql')
register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool')