UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-20 12:00:22 +00:00
Code Issues Actions 11 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 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.db_url import register_database
from playhouse.pool import PooledPostgresqlDatabase, PooledMySQLDatabase from playhouse.pool import PooledPostgresqlDatabase
from playhouse.shortcuts import ReconnectMixin from playhouse.shortcuts import ReconnectMixin
from psycopg2 import OperationalError from psycopg2 import OperationalError
from psycopg2.errors import InterfaceError from psycopg2.errors import InterfaceError
@ -26,7 +26,7 @@ class PeeweeConnectionState(_ConnectionState):
class CustomReconnectMixin(ReconnectMixin): class CustomReconnectMixin(ReconnectMixin):
reconnect_errors = ( reconnect_errors = (
# default ReconnectMixin exceptions (MySQL specific) # default ReconnectMixin exceptions
*ReconnectMixin.reconnect_errors, *ReconnectMixin.reconnect_errors,
# psycopg2 # psycopg2
(OperationalError, 'termin'), (OperationalError, 'termin'),
@ -44,16 +44,6 @@ class ReconnectingPooledPostgresqlDatabase(CustomReconnectMixin, PooledPostgresq
pass pass
class ReconnectingMySQLDatabase(CustomReconnectMixin, MySQLDatabase):
pass
class ReconnectingPooledMySQLDatabase(CustomReconnectMixin, PooledMySQLDatabase):
pass
def register_peewee_databases(): def register_peewee_databases():
register_database(MySQLDatabase, 'mysql')
register_database(PooledMySQLDatabase, 'mysql+pool')
register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql') register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql')
register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool') register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool')