From 99f557e19c461ab4d41fc3f4074ff2510b3ac2c9 Mon Sep 17 00:00:00 2001 From: iauhsnaiq Date: Fri, 10 Nov 2023 09:55:01 +0800 Subject: [PATCH] fix: modify database connection way modify parallelDatabase as Database Bug: 2042812 Change-Id: I33f282e69e7383a51c3e69b4c8e9273609fb988d --- skyline_apiserver/db/base.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/skyline_apiserver/db/base.py b/skyline_apiserver/db/base.py index 69f13a0..6ead183 100644 --- a/skyline_apiserver/db/base.py +++ b/skyline_apiserver/db/base.py @@ -16,7 +16,7 @@ from __future__ import annotations from contextvars import ContextVar -from databases import Database, DatabaseURL, core +from databases import Database, DatabaseURL from skyline_apiserver.config import CONF @@ -24,16 +24,11 @@ DATABASE = None DB: ContextVar = ContextVar("skyline_db") -class ParallelDatabase(Database): - def connection(self) -> core.Connection: - return core.Connection(self._backend) - - async def setup(): db_url = DatabaseURL(CONF.default.database_url) global DATABASE if db_url.scheme == "mysql": - DATABASE = ParallelDatabase( + DATABASE = Database( db_url, minsize=1, maxsize=100, @@ -42,7 +37,7 @@ async def setup(): client_flag=0, ) elif db_url.scheme == "sqlite": - DATABASE = ParallelDatabase(db_url) + DATABASE = Database(db_url) else: raise ValueError("Unsupported database backend") await DATABASE.connect()