def get_add_services(self,
enterprise_id,
services,
group_key,
version,
market_name=None):
"""获取新增组件"""
app_template = None
if services:
service_keys = services.values_list('service_key', flat=True)
service_keys = set(service_keys) if service_keys else set()
if not market_name:
app = rainbond_app_repo.get_rainbond_app_by_key_version(
group_key=group_key, version=version)
if app:
app_template = app.app_template
else:
market = app_market_service.get_app_market_by_name(
enterprise_id, market_name, raise_exception=True)
app = app_market_service.get_market_app_model_version(
market, group_key, version, get_template=True)
if app:
app_template = app.template
if app_template:
return list(
self.get_new_services(
self.parse_app_template(app_template),
service_keys).values())
else:
return []
def get_add_services(self, services, group_key, version):
"""获取新增组件"""
service_keys = services.values_list('service_key', flat=True)
service_keys = set(service_keys) if service_keys else set()
app = rainbond_app_repo.get_rainbond_app_by_key_version(
group_key=group_key, version=version)
return self.get_new_services(self.parse_app_template(app.app_template),
service_keys).values()
def get_add_services(self, services, group_key, version):
"""获取新增组件"""
if services:
service_keys = services.values_list('service_key', flat=True)
service_keys = set(service_keys) if service_keys else set()
try:
app = rainbond_app_repo.get_rainbond_app_by_key_version(group_key=group_key, version=version)
if app:
app_template = app.app_template
else:
try:
app_template = market_api.get_app_template(services.first().tenant_id, group_key, version)
if app_template:
app_template = app_template["data"]["bean"]["template_content"]
except Exception:
return []
except AbortRequest:
try:
app_template = market_api.get_app_template(services.first().tenant_id, group_key, version)
if app_template:
app_template = app_template["data"]["bean"]["template_content"]
except Exception:
return []
return self.get_new_services(self.parse_app_template(app_template), service_keys).values()
def post(self, request, group_id, *args, **kwargs):
"""提交升级任务"""
rq_args = (
{
'key': 'upgrade_record_id',
'required': True,
'error': 'upgrade_record_id is a required parameter'
},
{
'key': 'group_key',
'required': True,
'error': 'group_key is a required parameter'
},
{
'key': 'version',
'required': True,
'error': 'version is a required parameter'
},
{
'key': 'services',
'required': True,
'error': 'services is a required parameter'
},
)
data = parse_date(request, rq_args)
group_key = data['group_key']
version = data['version']
app_record = get_object_or_404(
AppUpgradeRecord,
msg="Upgrade record not found",
tenant_id=self.tenant.tenant_id,
group_id=int(group_id),
group_key=group_key,
status=UpgradeStatus.NOT.value,
pk=data['upgrade_record_id'],
)
# 处理新增的服务
add_service_infos = {
service['service']['service_key']: service['upgrade_info']
for service in data['services']
if service['service']['type'] == UpgradeType.ADD.value
and service['upgrade_info']
}
install_info = {}
if add_service_infos:
old_app = rainbond_app_repo.get_rainbond_app_by_key_version(
group_key=group_key, version=version)
new_app = deepcopy(old_app)
# mock app信息
template = json.loads(new_app.app_template)
template['apps'] = add_service_infos.values()
new_app.app_template = json.dumps(template)
# 查询某一个云市应用下的所有服务
services = group_service.get_rainbond_services(
int(group_id), group_key)
try:
market_app_service.check_package_app_resource(
self.tenant, self.response_region, new_app)
install_info = market_app_service.install_service_when_upgrade_app(
self.tenant, self.response_region, self.user, group_id,
new_app, old_app, services, True)
except (ResourceNotEnoughException, AccountOverdueException) as re:
logger.exception(re)
return MessageResponse(msg="resource is not enough",
msg_show=re.message,
status_code=412,
error_code=10406)
upgrade_service.create_add_service_record(app_record,
install_info['events'],
add_service_infos)
# 处理需要升级的服务
upgrade_service_infos = {
service['service']['service_id']: service['upgrade_info']
for service in data['services']
if service['service']['type'] == UpgradeType.UPGRADE.value
and service['upgrade_info']
}
app_record.version = version
app_record.old_version = upgrade_service.get_old_version(
group_key, upgrade_service_infos.keys())
app_record.save()
services = service_repo.get_services_by_service_ids_and_group_key(
data['group_key'], upgrade_service_infos.keys())
market_services = [
upgrade_service.market_service_and_create_backup(
self.tenant, service, app_record.version)
for service in services
]
upgrade_service.upgrade_database(market_services)
upgrade_service.send_upgrade_request(market_services, self.tenant,
self.user, app_record,
upgrade_service_infos)
upgrade_repo.change_app_record_status(app_record,
UpgradeStatus.UPGRADING.value)
# 处理依赖关系
if add_service_infos:
market_app_service.save_service_deps_when_upgrade_app(
self.tenant,
install_info['service_key_dep_key_map'],
install_info['key_service_map'],
install_info['apps'],
install_info['app_map'],
)
return MessageResponse(
msg="success",
bean=upgrade_service.serialized_upgrade_record(app_record))
def post(self, request, group_id, *args, **kwargs):
"""提交升级任务"""
rq_args = (
{
'key': 'upgrade_record_id',
'required': True,
'error': 'upgrade_record_id is a required parameter'
},
{
'key': 'group_key',
'required': True,
'error': 'group_key is a required parameter'
},
{
'key': 'version',
'required': True,
'error': 'version is a required parameter'
},
{
'key': 'services',
'required': True,
'error': 'services is a required parameter'
},
)
data = parse_date(request, rq_args)
app_record = get_object_or_404(
AppUpgradeRecord,
msg="Upgrade record not found",
tenant_id=self.tenant.tenant_id,
group_id=int(group_id),
group_key=data['group_key'],
status=UpgradeStatus.NOT.value,
pk=data['upgrade_record_id'],
)
app_record.version = data['version']
app_record.save()
# 处理新增的服务
add_service_infos = [
service['upgrade_info'] for service in data['services']
if service['service']['type'] == UpgradeType.ADD.value
]
if add_service_infos:
app = rainbond_app_repo.get_rainbond_app_by_key_version(
group_key=data['group_key'], version=app_record.version)
# mock app信息
template = json.loads(app.app_template)
template['apps'] = add_service_infos
app.app_template = json.dumps(template)
market_app_service.check_package_app_resource(
self.tenant, self.response_region, app)
market_app_service.install_service(self.tenant,
self.response_region, self.user,
group_id, app, True)
# 处理需要升级的服务
upgrade_service_infos = {
service['service']['service_id']: service['upgrade_info']
for service in data['services']
if service['service']['type'] == UpgradeType.UPGRADE.value
}
services = service_repo.get_services_by_service_ids_and_group_key(
data['group_key'], upgrade_service_infos.keys())
market_services = [
upgrade_service.market_service_and_create_backup(
self.tenant, service, app_record.version)
for service in services
]
upgrade_service.upgrade_database(market_services)
upgrade_service.send_upgrade_request(market_services, self.tenant,
self.user, app_record,
upgrade_service_infos)
upgrade_repo.change_app_record_status(app_record,
UpgradeStatus.UPGRADING.value)
return MessageResponse(
msg="success",
bean=upgrade_service.serialized_upgrade_record(app_record))
