a88a5cf15a
Puts into place the DeploymentConfiguration yaml that provides the options that should be configured by the site design to the deployment (and update) workflows. This change additionally refactors reused parts to common modules as related to info passing (xcom) Change-Id: Ib6470899b204dbc18d2a9a2e4f95540b3b0032b0
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# Copyright 2018 AT&T Intellectual Property. All other rights reserved.
|
|
#
|
|
# 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.
|
|
|
|
import logging
|
|
import time
|
|
|
|
from airflow.plugins_manager import AirflowPlugin
|
|
from airflow.exceptions import AirflowException
|
|
|
|
from promenade_base_operator import PromenadeBaseOperator
|
|
|
|
|
|
class PromenadeCheckEtcdOperator(PromenadeBaseOperator):
|
|
|
|
"""Promenade Check ETCD Operator
|
|
|
|
This operator will trigger promenade to retrieve the current
|
|
state of etcd.
|
|
|
|
"""
|
|
|
|
def do_execute(self):
|
|
# Placeholder function. Updates will be made when the Promenade
|
|
# API is ready for consumption.
|
|
|
|
# TODO(bryan-strassner) use:
|
|
# self.dc['kubernetes_provisioner.etcd_ready_timeout']
|
|
# self.dc['kubernetes_provisioner.remove_etcd_timeout']
|
|
logging.info("Performing health check on etcd...")
|
|
time.sleep(5)
|
|
|
|
check_etcd = True
|
|
|
|
if check_etcd:
|
|
logging.info("The etcd cluster is healthy and ready")
|
|
else:
|
|
raise AirflowException('Please check the state of etcd!')
|
|
|
|
|
|
class PromenadeCheckEtcdOperatorPlugin(AirflowPlugin):
|
|
|
|
"""Creates PromenadeCheckEtcdOperator in Airflow."""
|
|
|
|
name = 'promenade_check_etcd_operator'
|
|
operators = [PromenadeCheckEtcdOperator]
|