测试

app/upgrade/upgrade_service.py

@@ -78,10 +78,13 @@ docker-compose -f /data/upgrade_tools_data/backup/{upgrade_no}/docker-compose.ym
                 if err:
                     logger.info(f"有报错信息：{err}")
                 logger.info(f'回滚{app_list}成功：{out}')
+
+        with db_driver as session:
             # 对应的版本信息修改为当前版本
             session.exec_sql(f'update app_upgrade_log set default_version=1 where id={log_id}')
-            return {'err': err, "out": out}
-            # 注意异常情况下的处理
+
+        return {'err': err, "out": out}
+        # 注意异常情况下的处理
 
     def find_compose_by_upgrade_no(self, upgrade_no):
         '''

app/utils/db_tools.py

@@ -71,7 +71,7 @@ class DBDriver:
             return res
 
         res_dict = dict(map(lambda x, y: [x, y], fields, res))
-        # cur.close()
+        cur.close()
         return res_dict
 
     def query_sql(self, sql, *args, **kwargs):
@@ -82,7 +82,7 @@ class DBDriver:
         for r in res:
             res_dict = OrderedDict(map(lambda x, y: [x, y], fields, r))
             res_list.append(res_dict)
-        # cur.close()
+        cur.close()
         return res_list
 
     def __enter__(self):