eb3870a4b3
This patchset is there to change the code structure. The objective is to flatten the project file tree by merging 'api/' and 'framework/' into a single package. This also contains some tidy ups in package naming (like using only singular nouns). This should only affect file/folder names and their subsequent import paths wherever they were used. Change-Id: Ie903ba20ca5cf03b0b42efa60131d1b919b0c2c9
83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
# -*- encoding: utf-8 -*-
|
|
# Copyright (c) 2015 b<>com
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# 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 oslo_log import log
|
|
|
|
from watcher.common.messaging.events.event import Event
|
|
from watcher.decision_engine.messaging.command.base import \
|
|
BaseDecisionEngineCommand
|
|
from watcher.decision_engine.messaging.events import Events
|
|
from watcher.decision_engine.planner.default import DefaultPlanner
|
|
from watcher.decision_engine.strategy.context.default import StrategyContext
|
|
from watcher.objects.audit import Audit
|
|
from watcher.objects.audit import AuditStatus
|
|
from watcher.objects.audit_template import AuditTemplate
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
|
|
class TriggerAuditCommand(BaseDecisionEngineCommand):
|
|
def __init__(self, messaging, model_collector):
|
|
self.messaging = messaging
|
|
self.model_collector = model_collector
|
|
self.strategy_context = StrategyContext()
|
|
|
|
def notify(self, audit_uuid, event_type, status):
|
|
event = Event()
|
|
event.set_type(event_type)
|
|
event.set_data({})
|
|
payload = {'audit_uuid': audit_uuid,
|
|
'audit_status': status}
|
|
self.messaging.topic_status.publish_event(event.get_type().name,
|
|
payload)
|
|
|
|
def update_audit(self, request_context, audit_uuid, state):
|
|
LOG.debug("update audit {0} ".format(state))
|
|
audit = Audit.get_by_uuid(request_context, audit_uuid)
|
|
audit.state = state
|
|
audit.save()
|
|
self.notify(audit_uuid, Events.TRIGGER_AUDIT, state)
|
|
return audit
|
|
|
|
def execute(self, audit_uuid, request_context):
|
|
try:
|
|
LOG.debug("Execute TriggerAuditCommand ")
|
|
|
|
# 1 - change status to ONGOING
|
|
audit = self.update_audit(request_context, audit_uuid,
|
|
AuditStatus.ONGOING)
|
|
|
|
# 3 - Retrieve cluster-data-model
|
|
cluster = self.model_collector.get_latest_cluster_data_model()
|
|
|
|
# 4 - Select appropriate strategy
|
|
audit_template = AuditTemplate.get_by_id(request_context,
|
|
audit.audit_template_id)
|
|
|
|
self.strategy_context.set_goal(audit_template.goal)
|
|
|
|
# 5 - compute change requests
|
|
solution = self.strategy_context.execute_strategy(cluster)
|
|
|
|
# 6 - create an action plan
|
|
planner = DefaultPlanner()
|
|
planner.schedule(request_context, audit.id, solution)
|
|
|
|
# 7 - change status to SUCCESS and notify
|
|
self.update_audit(request_context, audit_uuid, AuditStatus.SUCCESS)
|
|
except Exception as e:
|
|
self.update_audit(request_context, audit_uuid, AuditStatus.FAILED)
|
|
LOG.error("Execute audit command {0} ".format(unicode(e)))
|