forked from TencentBlueKing/bk-nodeman
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: 卸载agent时停止插件并清理agent安装及运行过程中产生的目录(closed TencentBlueKing#2490)
- Loading branch information
1 parent
f14f01b
commit f12edb0
Showing
15 changed files
with
305 additions
and
12 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
86 changes: 86 additions & 0 deletions
86
apps/backend/components/collections/agent_new/stop_plugins.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,86 @@ | ||
# -*- coding: utf-8 -*- | ||
""" | ||
TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-节点管理(BlueKing-BK-NODEMAN) available. | ||
Copyright (C) 2017-2022 THL A29 Limited, a Tencent company. All rights reserved. | ||
Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License. | ||
You may obtain a copy of the License at https://opensource.org/licenses/MIT | ||
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on | ||
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the | ||
specific language governing permissions and limitations under the License. | ||
""" | ||
|
||
from collections import defaultdict | ||
from typing import List | ||
|
||
from django.conf import settings | ||
|
||
from apps.core.concurrent.retry import RetryHandler | ||
from apps.node_man import constants, models | ||
from apps.utils.batch_request import request_multi_thread | ||
from common.api import NodeApi | ||
from common.api.exception import DataAPIException | ||
|
||
from ..base import CommonData | ||
from ..subsubscription import SubSubscriptionBaseService | ||
from .base import AgentBaseService | ||
|
||
|
||
class StopPluginsService(SubSubscriptionBaseService, AgentBaseService): | ||
@staticmethod | ||
@RetryHandler(interval=1, retry_times=1, exception_types=[DataAPIException]) | ||
def call_create_subscription_api(params): | ||
return NodeApi.create_subscription(params) | ||
|
||
@classmethod | ||
def create_subscriptions(cls, common_data: CommonData) -> List[int]: | ||
|
||
host_ids_group_by_os = defaultdict(list) | ||
for host in common_data.host_id_obj_map.values(): | ||
host_ids_group_by_os[host.os_type.lower()].append(host.bk_host_id) | ||
|
||
installed_running_plugin_names = models.ProcessStatus.objects.filter( | ||
status=constants.ProcStateType.RUNNING, | ||
bk_host_id__in=common_data.bk_host_ids, | ||
proc_type=constants.ProcType.PLUGIN, | ||
).values_list("name", flat=True) | ||
|
||
plugin_name__os_type_set = set( | ||
models.Packages.objects.filter( | ||
project__in=installed_running_plugin_names, os__in=host_ids_group_by_os.keys() | ||
).values_list("project", "os") | ||
) | ||
params_list = [] | ||
for (plugin_name, os_type) in plugin_name__os_type_set: | ||
params_list.append( | ||
{ | ||
"params": { | ||
"run_immediately": True, | ||
"category": models.Subscription.CategoryType.ONCE, | ||
"bk_username": settings.SYSTEM_USE_API_ACCOUNT, | ||
"scope": { | ||
"node_type": models.Subscription.NodeType.INSTANCE, | ||
"object_type": models.Subscription.ObjectType.HOST, | ||
"nodes": [{"bk_host_id": bk_host_id} for bk_host_id in host_ids_group_by_os[os_type]], | ||
}, | ||
"steps": [ | ||
{ | ||
"id": plugin_name, | ||
"type": "PLUGIN", | ||
"config": { | ||
"job_type": constants.JobType.MAIN_STOP_PLUGIN, | ||
"plugin_name": plugin_name, | ||
"plugin_version": "latest", | ||
"config_templates": [ | ||
{"name": "{}.conf".format(plugin_name), "version": "latest", "is_main": True} | ||
], | ||
}, | ||
"params": {"context": {}}, | ||
} | ||
], | ||
} | ||
} | ||
) | ||
subscription_ids = request_multi_thread( | ||
cls.call_create_subscription_api, params_list, get_data=lambda x: [x["subscription_id"]] | ||
) | ||
return subscription_ids |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.