You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

248 lines
12 KiB

from sqlalchemy.ext.asyncio import AsyncSession
from module_admin.entity.vo.metaprocessconfig_vo import MetaprocessconfigQueryModel
from module_admin.dao.metaprocessconfig_dao import MetaprocessconfigDao
class MetaprocessconfigService:
"""
参数配置管理模块服务层
"""
@classmethod
async def get_metaprocessconfig_list_services(
cls, query_db: AsyncSession, query_object: MetaprocessconfigQueryModel, is_page: bool = False
):
"""
获取参数配置列表信息service
:param query_db: orm对象
:param query_object: 查询参数对象
:return: 参数配置列表信息对象
"""
metaprocessconfig_list_result = await MetaprocessconfigDao.get_metaprocessconfig_list_all(query_db, query_object, is_page)
return metaprocessconfig_list_result
# @classmethod
# async def get_data_source_tree(cls,request: Request):
# url = 'http://47.121.207.11:12345/dolphinscheduler/datasources?pageNo=1&pageSize=100'
# token = await request.app.state.redis.get(f'{RedisInitKeyConfig.SYS_CONFIG.key}:sys.ds.token')
# headers = {'token': token}
# response = requests.get(url, headers=headers)
# if response.reason == 'OK':
# response_text = response.text
# data = json.loads(response_text)
# total_list = data["data"]["totalList"]
# # 解析 connectionParams 字符串为字典
# for item in total_list:
# item["connectionParams"] = json.loads(item["connectionParams"])
# # 使用 Pydantic 创建 DataSource 对象列表
# data_sources = [DataSource(**item) for item in total_list]
# return data_sources
# else:
# return {'error': f'Request failed with status code {response.status_code}'}
# @classmethod
# async def check_metaprocessconfig_name_unique_services(cls, query_db: AsyncSession, page_object: MetaprocessconfigModel):
# """
# 校验参数键名是否唯一service
# :param query_db: orm对象
# :param page_object: 参数配置对象
# :return: 校验结果
# """
# metaprocessconfig_id = -1 if page_object.metaprocessconfig_id is None else page_object.metaprocessconfig_id
# metaprocessconfig = await MetaprocessconfigDao.get_metaprocessconfig_detail_by_info(
# query_db, MetaprocessconfigModel(metaprocessconfigName=page_object.metaprocessconfig_name)
# )
# if metaprocessconfig and metaprocessconfig.metaprocessconfig_id != metaprocessconfig_id:
# return CommonConstant.NOT_UNIQUE
# return CommonConstant.UNIQUE
# @classmethod
# async def add_metaprocessconfig_services(cls, request: Request, query_db: AsyncSession, page_object: MetaprocessconfigModel):
# """
# 新增参数配置信息service
# :param request: Request对象
# :param query_db: orm对象
# :param page_object: 新增参数配置对象
# :return: 新增参数配置校验结果
# """
# if not await cls.check_metaprocessconfig_name_unique_services(query_db, page_object):
# raise ServiceException(message=f'新增元数据任务{page_object.metaprocessconfig_name}失败,任务名已存在')
# else:
# try:
# await MetaprocessconfigDao.add_metaprocessconfig_dao(query_db, page_object)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message='新增成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# @classmethod
# async def edit_metaprocessconfig_services(cls, request: Request, query_db: AsyncSession, page_object: MetaprocessconfigModel):
# """
# 编辑参数配置信息service
# :param request: Request对象
# :param query_db: orm对象
# :param page_object: 编辑参数配置对象
# :return: 编辑参数配置校验结果
# """
# edit_metaprocessconfig = page_object.model_dump(exclude_unset=True)
# metaprocessconfig_info = await cls.metaprocessconfig_detail_services(query_db, page_object.metaprocessconfig_id)
# if metaprocessconfig_info.metaprocessconfig_id:
# if not await cls.check_metaprocessconfig_name_unique_services(query_db, page_object):
# raise ServiceException(message=f'修改任务{page_object.metaprocessconfig_name}失败,任务名称已存在')
# else:
# try:
# await MetaprocessconfigDao.edit_metaprocessconfig_dao(query_db, edit_metaprocessconfig)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message='更新成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='更新失败')
# @classmethod
# async def up_or_down_metaprocessconfig_services(
# cls, request: Request, query_db: AsyncSession, current_user: CurrentUserModel, id: str, type: str
# ):
# metaprocessconfig_info = await cls.metaprocessconfig_detail_services(query_db, id)
# metaprocessconfig_info.update_by = current_user.user.user_name
# metaprocessconfig_info.update_time = datetime.now()
# type_str: str
# if type == 'down':
# # 下线
# type_str = '下线'
# metaprocessconfig_info.status = '1'
# else:
# # 上线
# type_str = '上线'
# metaprocessconfig_info.status = '2'
# edit_metaprocessconfig = metaprocessconfig_info.model_dump(exclude_unset=True)
# try:
# await MetaprocessconfigDao.edit_metaprocessconfig_dao(query_db, edit_metaprocessconfig)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message=metaprocessconfig_info.metaprocessconfig_name + '任务' + type_str + '成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='更新失败')
# @classmethod
# async def run_metaprocessconfig_services(
# cls, request: Request, query_db: AsyncSession, current_user: CurrentUserModel, id: str
# ):
# metaprocessconfig_info = await cls.metaprocessconfig_detail_services(query_db, id)
# metaprocessconfig_info.update_by = current_user.user.user_name
# metaprocessconfig_info.update_time = datetime.now()
# # 运行中
# metaprocessconfig_info.status = '3'
# edit_metaprocessconfig = metaprocessconfig_info.model_dump(exclude_unset=True)
# try:
# await MetaprocessconfigDao.edit_metaprocessconfig_dao(query_db, edit_metaprocessconfig)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message=metaprocessconfig_info.metaprocessconfig_name + '任务:' '运行成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='更新失败')
# @classmethod
# async def ds_metaprocessconfig_services(
# cls, request: Request, query_db: AsyncSession, current_user: CurrentUserModel, id: str
# ):
# metaprocessconfig_info = await cls.metaprocessconfig_detail_services(query_db, id)
# metaprocessconfig_info.update_by = current_user.user.user_name
# metaprocessconfig_info.update_time = datetime.now()
# # 运行中
# metaprocessconfig_info.status = '4'
# edit_metaprocessconfig = metaprocessconfig_info.model_dump(exclude_unset=True)
# try:
# await MetaprocessconfigDao.edit_metaprocessconfig_dao(query_db, edit_metaprocessconfig)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message=metaprocessconfig_info.metaprocessconfig_name + '任务:' '调度启动成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='更新失败')
# @classmethod
# async def get_metaprocessconfig_logs_services(
# cls, request: Request, current_user: CurrentUserModel, query_db: AsyncSession, id: str
# ):
# metaprocessconfig_info = await cls.metaprocessconfig_detail_services(query_db, id)
# metaprocessconfig_info.update_by = current_user.user.user_name
# metaprocessconfig_info.update_time = datetime.now()
# # 运行中
# metaprocessconfig_info = '3'
# edit_metaprocessconfig = metaprocessconfig_info.model_dump(exclude_unset=True)
# try:
# await MetaprocessconfigDao.edit_metaprocessconfig_dao(query_db, edit_metaprocessconfig)
# await query_db.commit()
# return CrudResponseModel(is_success=True, message=metaprocessconfig_info.metaprocessconfig_name + '任务:' '日志获取成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='更新失败')
# @classmethod
# async def delete_metaprocessconfig_services(cls, request: Request, query_db: AsyncSession, page_object: DeleteMetaprocessconfigModel):
# """
# 删除参数配置信息service
# :param request: Request对象
# :param query_db: orm对象
# :param page_object: 删除参数配置对象
# :return: 删除参数配置校验结果
# """
# # if page_object.metaprocessconfig_ids.split(','):
# # metaprocessconfig_id_list = page_object.metaprocessconfig_ids.split(',')
# # try:
# # for metaprocessconfig_id in metaprocessconfig_id_list:
# # metaprocessconfig_id_dict = dict(metaprocessconfig_id=metaprocessconfig_id)
# # MetaprocessconfigDao.delete_metaprocessconfig_dao(query_db, MetaprocessconfigModel(**metaprocessconfig_id_dict))
# # query_db.commit()
# # # await cls.init_cache_sys_metaprocessconfig_services(query_db, request.app.state.redis)s
# # result = dict(is_success=True, message='删除成功')
# # except Exception as e:
# # query_db.rollback()
# # result = dict(is_success=False, message=str(e))
# # else:
# # result = dict(is_success=False, message='传入字典数据id为空')
# # return CrudResponseModel(**result)
# if page_object.metaprocessconfig_ids:
# metaprocessconfig_id_list = page_object.metaprocessconfig_ids.split(',')
# try:
# for metaprocessconfig_id in metaprocessconfig_id_list:
# await MetaprocessconfigDao.delete_metaprocessconfig_dao(query_db, MetaprocessconfigModel(metaprocessconfigId=int(metaprocessconfig_id)))
# await query_db.commit()
# return CrudResponseModel(is_success=True, message='删除成功')
# except Exception as e:
# await query_db.rollback()
# raise e
# else:
# raise ServiceException(message='传入参数配置id为空')
# @classmethod
# async def metaprocessconfig_detail_services(cls, query_db: AsyncSession, metaprocessconfig_id: int):
# """
# 获取参数配置详细信息service
# :param query_db: orm对象
# :param metaprocessconfig_id: 参数配置id
# :return: 参数配置id对应的信息
# """
# metaprocessconfig = await MetaprocessconfigDao.get_metaprocessconfig_detail_by_id(query_db, metaprocessconfig_id=metaprocessconfig_id)
# if metaprocessconfig:
# result = MetaprocessconfigModel(**CamelCaseUtil.transform_result(metaprocessconfig))
# else:
# result = MetaprocessconfigModel(**dict())
# return result