Переглянути джерело

# 手动数据库关闭连接 ai_word_engin.dispose()

liyuyi@c-top.com.cn 3 роки тому
батько
коміт
56e1360d6a
1 змінених файлів з 15 додано та 0 видалено
  1. 15 0
      routers/script_config.py

+ 15 - 0
routers/script_config.py

@@ -174,6 +174,9 @@ def export_script_file(item: List[QueryWordItem]):
             script_df = pd.read_sql(sql, ai_word_engine)
             out_df = video_query_word_df.merge(script_df, on='signature', how='inner')
 
+        # 手动数据库关闭连接
+        ai_word_engine.dispose()
+
         # 3 返回流数据
         if not out_df.empty:
             bio = BytesIO()
@@ -279,6 +282,10 @@ def get_query_word_task_info_lst(item: QueryWordTaskInfoLst):
                   f"and ('{item.source_code}' = '[]' or source_code in {tuple(source_code_lst)}) " \
                   f"and ('{item.search_word}' = '' or query_word = '{item.search_word}')"
             df = pd.read_sql(sql, ai_word_engine)
+
+        # 手动数据库关闭连接
+        ai_word_engine.dispose()
+
         if not df.empty:
             df['source_name'] = df['source_code'].apply(lambda x: source_name_map[x]['name'])
             df = df[['source_name', 'query_word', 'stat_date', 'script_num', 'task_status']]
@@ -338,6 +345,10 @@ def add_script_config(item: AddScriptConfig):
                                con=ai_word_engine,
                                if_exists='append',
                                index=False)
+
+        # 手动数据库关闭连接
+        ai_word_engine.dispose()
+
         logger.info(f"request body: {item}, code:0, message: add_script_config success")
         return {"code": 0,
                 "message": "add success",
@@ -389,6 +400,10 @@ def delete_script_config(item: DeleteScriptConfig):
                    dtype=column_type)
         else:
             response.message = f'没有获取到 {item.config_id} 对应的数据'
+
+        # 手动数据库关闭连接
+        ai_word_engine.dispose()
+
         logger.info(f"request body: {item}, response: {response}")
     except:
         response.code = -1