def __install_plugin(self, region_name, user, tenant, plugin_template):
image = None
image_tag = None
if plugin_template["share_image"]:
image_and_tag = plugin_template["share_image"].rsplit(":", 1)
if len(image_and_tag) > 1:
image = image_and_tag[0]
image_tag = image_and_tag[1]
else:
image = image_and_tag[0]
image_tag = "latest"
status, msg, plugin_base_info = plugin_service.create_tenant_plugin(
tenant, user.user_id, region_name, plugin_template["desc"],
plugin_template["plugin_alias"], plugin_template["category"],
"image", image, plugin_template["code_repo"])
if status != 200:
return status, msg
plugin_base_info.origin = 'local_market'
plugin_base_info.origin_share_id = plugin_template.get("plugin_key")
plugin_base_info.save()
build_version = plugin_template.get('build_version')
min_memory = plugin_template.get('min_memory', 128)
plugin_build_version = plugin_version_service.create_build_version(
region_name,
plugin_base_info.plugin_id,
tenant.tenant_id,
user.user_id,
"",
"unbuild",
min_memory,
image_tag=image_tag,
code_version="",
build_version=build_version)
share_config_groups = plugin_template.get('config_groups', [])
plugin_config_service.create_config_groups(plugin_base_info.plugin_id,
build_version,
share_config_groups)
event_id = make_uuid()
plugin_build_version.event_id = event_id
plugin_build_version.plugin_version_status = "fixed"
plugin_service.create_region_plugin(region_name,
tenant,
plugin_base_info,
image_tag=image_tag)
ret = plugin_service.build_plugin(
region_name, plugin_base_info, plugin_build_version, user, tenant,
event_id, plugin_template.get("plugin_image", None))
plugin_build_version.build_status = ret.get('bean').get('status')
plugin_build_version.save()
return 200, "success"
def install_plugin(self, user, tenant, region_name, market_plugin):
plugin_template = json.loads(market_plugin.plugin_template)
share_plugin_info = plugin_template.get("share_plugin_info")
sid = transaction.savepoint()
try:
image = None
image_tag = None
if plugin_template["share_image"]:
image_and_tag = plugin_template["share_image"].rsplit(":", 1)
if len(image_and_tag) > 1:
image = image_and_tag[0]
image_tag = image_and_tag[1]
else:
image = image_and_tag[0]
image_tag = "latest"
status, msg, plugin_base_info = plugin_service.create_tenant_plugin(
tenant, user.user_id, region_name,
share_plugin_info.get("desc"), plugin_template["plugin_name"],
plugin_template["category"], "image", image,
plugin_template["code_repo"])
if status != 200:
return status, msg
plugin_base_info.origin = 'local_market' if market_plugin.source == 'local' else market_plugin.source
plugin_base_info.origin_share_id = share_plugin_info.get(
"plugin_key")
plugin_base_info.save()
build_version = plugin_template.get('build_version')
min_memory = build_version.get('min_memory')
plugin_build_version = plugin_version_service.create_build_version(
region_name,
plugin_base_info.plugin_id,
tenant.tenant_id,
user.user_id,
"",
"unbuild",
min_memory,
image_tag=image_tag,
code_version="")
config_groups, config_items = [], []
share_config_groups = share_plugin_info.get('config_groups', [])
for group in share_config_groups:
plugin_config_group = PluginConfigGroup(
plugin_id=plugin_base_info.plugin_id,
build_version=plugin_build_version.build_version,
config_name=group.get("config_name"),
service_meta_type=group.get("service_meta_type"),
injection=group.get("injection"))
config_groups.append(plugin_config_group)
share_config_items = group.get('config_items', [])
for item in share_config_items:
plugin_config_item = PluginConfigItems(
plugin_id=plugin_base_info.plugin_id,
build_version=plugin_build_version.build_version,
service_meta_type=item.get("service_meta_type"),
attr_name=item.get("attr_name"),
attr_alt_value=item.get("attr_alt_value"),
attr_type=item.get("attr_type", "string"),
attr_default_value=item.get("attr_default_value",
None),
is_change=item.get("is_change", False),
attr_info=item.get("attr_info", ""),
protocol=item.get("protocol", ""))
config_items.append(plugin_config_item)
PluginConfigGroup.objects.bulk_create(config_groups)
PluginConfigItems.objects.bulk_create(config_items)
event_id = make_uuid()
plugin_build_version.event_id = event_id
plugin_build_version.plugin_version_status = "fixed"
plugin_service.create_region_plugin(region_name,
tenant,
plugin_base_info,
image_tag=image_tag)
ret = plugin_service.build_plugin(
region_name, plugin_base_info, plugin_build_version, user,
tenant, event_id, share_plugin_info.get("plugin_image", None))
plugin_build_version.build_status = ret.get('bean').get('status')
plugin_build_version.save()
return 200, '安装成功'
except Exception as e:
logger.exception(e)
if sid:
transaction.savepoint_rollback(sid)
return 500, '插件安装失败'
def post(self, request, *args, **kwargs):
"""
插件创建
---
parameters:
- name: tenantName
description: 团队名
required: true
type: string
paramType: path
- name: plugin_alias
description: 插件名称
required: true
type: string
paramType: form
- name: build_source
description: 构建来源 dockerfile | image
required: true
type: string
paramType: form
- name: min_memory
description: 最小内存
required: true
type: integer
paramType: form
- name: category
description: 插件类别 net-plugin:down|net-plugin:up|net-plugin:in-and-out|
analyst-plugin:perf|init-plugin|general-plugin
required: false
type: string
paramType: form
- name: build_cmd
description: 构建命令
required: false
type: string
paramType: form
- name: code_repo
description: dockerfile构建代码仓库地址,选择dockerfile时必须
required: false
type: string
paramType: form
- name: code_version
description: 代码版本，默认master
required: false
type: string
paramType: form
- name: image
description: 镜像构建时镜像名称
required: false
type: string
paramType: form
- name: desc
description: 镜像说明
required: true
type: string
paramType: form
"""
# 必要参数
plugin_alias = request.data.get("plugin_alias", None)
build_source = request.data.get("build_source", None)
min_memory = request.data.get("min_memory", None)
category = request.data.get("category", None)
desc = request.data.get("desc", None)
# 非必要参数
build_cmd = request.data.get("build_cmd", None)
code_repo = request.data.get("code_repo", None)
code_version = request.data.get("code_version", None)
image = request.data.get("image", None)
# username and password is used for private docker hub or private git repo
username = request.data.get("username", None)
password = request.data.get("password", None)
tenant_plugin = None
plugin_build_version = None
try:
if not plugin_alias:
return Response(general_message(400, "plugin alias is null",
"插件名称未指明"),
status=400)
if not build_source:
return Response(general_message(400, "build source is null",
"构建来源未指明"),
status=400)
if not min_memory:
return Response(general_message(400,
"plugin min_memroy is null",
"插件内存大小未指明"),
status=400)
if not category:
return Response(general_message(400, "plugin category is null",
"插件类别未指明"),
status=400)
else:
if category not in allow_plugins:
return Response(general_message(
400, "plugin category is wrong",
"插件类别参数错误，详情请参数API说明"),
status=400)
if not desc:
return Response(general_message(400, "plugin desc is null",
"请填写插件描述"),
status=400)
image_tag = ""
if image and build_source == "image":
image_and_tag = image.rsplit(":", 1)
if len(image_and_tag) > 1:
image = image_and_tag[0]
image_tag = image_and_tag[1]
else:
image = image_and_tag[0]
image_tag = "latest"
# 创建基本信息
plugin_params = {
"tenant_id": self.tenant.tenant_id,
"region": self.response_region,
"create_user": self.user.user_id,
"desc": desc,
"plugin_alias": plugin_alias,
"category": category,
"build_source": build_source,
"image": image,
"code_repo": code_repo,
"username": username,
"password": password
}
code, msg, tenant_plugin = plugin_service.create_tenant_plugin(
plugin_params)
if code != 200:
return Response(general_message(code, "create plugin error",
msg),
status=code)
# 创建插件版本信息
plugin_build_version = plugin_version_service.create_build_version(
self.response_region, tenant_plugin.plugin_id,
self.tenant.tenant_id, self.user.user_id, "", "unbuild",
min_memory, build_cmd, image_tag, code_version)
# 数据中心创建插件
code, msg = plugin_service.create_region_plugin(
self.response_region, self.tenant, tenant_plugin, image_tag)
if code != 200:
plugin_service.delete_console_tenant_plugin(
tenant_plugin.plugin_id)
plugin_version_service.delete_build_version_by_id_and_version(
tenant_plugin.plugin_id,
plugin_build_version.build_version, True)
return Response(general_message(code, "create plugin error",
msg),
status=code)
bean = tenant_plugin.to_dict()
bean["build_version"] = plugin_build_version.build_version
bean["code_version"] = plugin_build_version.code_version
bean["build_status"] = plugin_build_version.build_status
bean["update_info"] = plugin_build_version.update_info
bean["image_tag"] = plugin_build_version.image_tag
result = general_message(200, "success", "创建成功", bean=bean)
except Exception as e:
logger.exception(e)
result = error_message(e.message)
if tenant_plugin:
plugin_service.delete_console_tenant_plugin(
tenant_plugin.plugin_id)
if plugin_build_version:
plugin_version_service.delete_build_version_by_id_and_version(
tenant_plugin.plugin_id,
plugin_build_version.build_version, True)
return Response(result, status=result["code"])
