9f2bbdbd26
There are a couple changes in this Patch Set: 1) There has been recent changes in the genesis process and the Tiller pod now resides in the ucp namespace instead of kube-system. As such it no longer takes on the IP of the Genesis Node. The YAMLs continue to be hosted on the nginx server which resides on the genesis node. Hence we will need to get the IP of mass-rack pod instead in order to resolve to the correct IP of the nginx server. Note that this logic will change in near future when Armada is integrated with DeckHand. The nginx server concept will go away at that time. 2) Bug fix in 'cluster_join_check_backoff_time' variable type in DryDock Operator. That variable needs to take on integer value. 3) Updates to the 'get_k8s_pod_port_ip' Operator. Note that it is possible for pod to have an IP with no port. For instance the maas-rack pod takes on genesis node IP and has no port associated with it. We will assign the value 'None' to the port value in such cases. Change-Id: I07dd922c98c63e03a8963f7767b457c932ee772b
111 lines
4.5 KiB
Python
111 lines
4.5 KiB
Python
# Copyright 2017 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
|
|
from functools import wraps
|
|
|
|
from airflow.exceptions import AirflowException
|
|
from kubernetes import client, config
|
|
|
|
|
|
def get_pod_port_ip(*pods):
|
|
def get_k8s_pod_port_ip(func):
|
|
@wraps(func)
|
|
def k8s_pod_port_ip_get(self, context, *args):
|
|
"""This function retrieves Kubernetes Pod Port and IP
|
|
information. It can be used in different areas. For
|
|
instance, we can use this to retrieve the tiller pod
|
|
IP and port information for usage in the Armada Operator
|
|
|
|
:param context: Information on the current workflow
|
|
|
|
Example::
|
|
|
|
from get_k8s_pod_port_ip import get_pod_port_ip
|
|
|
|
@get_pod_port_ip('tiller', 'drydock')
|
|
def get_pod_info(self, context, *args):
|
|
# Get IP and port information of Pods from context
|
|
k8s_pods_ip_port = context['pods_ip_port']
|
|
|
|
tiller_ip = k8s_pods_ip_port['tiller'].get('ip')
|
|
drydock_ip = k8s_pods_ip_port['drydock'].get('ip')
|
|
|
|
tiller_port = k8s_pods_ip_port['tiller'].get('port')
|
|
drydock_port = k8s_pods_ip_port['drydock'].get('port')
|
|
"""
|
|
# Initialize variable
|
|
k8s_pods = {}
|
|
|
|
# The function allows us to query information on multiple
|
|
# pods
|
|
for pod_name in pods:
|
|
# Initialize variables
|
|
pod_attr = {}
|
|
pod_attr[pod_name] = {}
|
|
|
|
# Make use of kubernetes client to retrieve pod IP
|
|
# and port information
|
|
# Note that we should use 'in_cluster_config'
|
|
config.load_incluster_config()
|
|
v1 = client.CoreV1Api()
|
|
ret = v1.list_pod_for_all_namespaces(watch=False)
|
|
|
|
# Loop through items to extract port and IP information
|
|
# of the pod
|
|
for i in ret.items:
|
|
if pod_name in i.metadata.name:
|
|
# Get pod IP
|
|
logging.info("Retrieving %s IP", pod_name)
|
|
pod_attr[pod_name]['ip'] = i.status.pod_ip
|
|
logging.info("%s IP is %s", pod_name,
|
|
pod_attr[pod_name]['ip'])
|
|
|
|
# Get pod port
|
|
logging.info("Retrieving %s Port", pod_name)
|
|
|
|
# It is possible for a pod to have an IP with no
|
|
# port. For instance maas-rack takes on genesis
|
|
# node IP and has no port associated with it. We
|
|
# will assign the value 'None' to the port value
|
|
# in such cases.
|
|
try:
|
|
specs_dict = i.spec.containers[0].__dict__
|
|
ports_dict = specs_dict['_ports'][0].__dict__
|
|
pod_attr[pod_name]['port'] = (
|
|
ports_dict['_container_port'])
|
|
logging.info("%s Port is %s", pod_name,
|
|
pod_attr[pod_name]['port'])
|
|
except:
|
|
pod_attr[pod_name]['port'] = 'None'
|
|
logging.info("%s Port is None", pod_name)
|
|
|
|
# Update k8s_pods with new entry
|
|
k8s_pods.update(pod_attr)
|
|
|
|
break
|
|
|
|
# Raise Execptions if the pod does not exits in the
|
|
# Kubernetes cluster
|
|
if not pod_attr[pod_name]:
|
|
raise AirflowException("Unable to locate", pod_name)
|
|
|
|
# Assign pods IP and ports information to context
|
|
context['pods_ip_port'] = k8s_pods
|
|
|
|
return func(self, context, *args)
|
|
|
|
return k8s_pod_port_ip_get
|
|
return get_k8s_pod_port_ip
|