feat: timeout

Author: XiaJunjie2020

backend/apps/datasource/crud/datasource.py

@@ -41,7 +41,7 @@ def get_ds(session: SessionDep, id: int):
 
 
 def check_status(session: SessionDep, ds: CoreDatasource, is_raise: bool = False):
-    conn = get_engine(ds)
+    conn = get_engine(ds, 5)
     try:
         with conn.connect() as connection:
             SQLBotLogUtil.info("success")

backend/apps/db/db.py

@@ -52,10 +52,10 @@ def get_uri_from_config(type: str, conf: DatasourceConf) -> str:
     return db_url
 
 
-def get_engine(ds: CoreDatasource) -> Engine:
+def get_engine(ds: CoreDatasource, timeout: int = 30) -> Engine:
     conf = DatasourceConf(**json.loads(aes_decrypt(ds.configuration))) if ds.type != "excel" else get_engine_config()
     if conf.timeout is None:
-        conf.timeout = 30
+        conf.timeout = timeout
     if ds.type == "pg" and (conf.dbSchema is not None and conf.dbSchema != ""):
         engine = create_engine(get_uri(ds),
                                connect_args={"options": f"-c search_path={urllib.parse.quote(conf.dbSchema)}",