monitor_converter.py

import com.vmware.nsx_policy.model_client as model_client

def get_alb_response_codes(response_codes):
    if not response_codes:
        return None
    HttpResponseCode = model_client.ALBHealthMonitorHttp
    codes = list()
    for code in response_codes:
        if code<200:
            if HttpResponseCode.HTTP_RESPONSE_CODE_1XX not in codes:
                codes.append(HttpResponseCode.HTTP_RESPONSE_CODE_1XX)
        elif code>199 and code<300:
            if HttpResponseCode.HTTP_RESPONSE_CODE_2XX not in codes:
                codes.append(HttpResponseCode.HTTP_RESPONSE_CODE_2XX)
        elif code>299 and code<400:
            if HttpResponseCode.HTTP_RESPONSE_CODE_3XX not in codes:
                codes.append(HttpResponseCode.HTTP_RESPONSE_CODE_3XX)
        elif code>399 and code<500:
            if HttpResponseCode.HTTP_RESPONSE_CODE_4XX not in codes:
                codes.append(HttpResponseCode.HTTP_RESPONSE_CODE_4XX)
        elif code>499 and code<600:
            if HttpResponseCode.HTTP_RESPONSE_CODE_5XX not in codes:
                codes.append(HttpResponseCode.HTTP_RESPONSE_CODE_5XX)
    return codes


def update_alb_type(lb_hm, alb_hm):

    if lb_hm['resource_type'] == 'LBHttpMonitorProfile':
        alb_hm['type'] = 'HEALTH_MONITOR_HTTP'
        alb_hm['http_monitor'] = dict(
            http_request=lb_hm['request_url'],
            http_request_body=lb_hm.get('request_body'),
            http_response=lb_hm.get('response_body'),
            http_response_code=get_alb_response_codes(lb_hm['response_status_codes']),
        )
    elif lb_hm['resource_type'] == 'LBHttpsMonitorProfile':
        alb_hm['type'] = 'HEALTH_MONITOR_HTTPS'
        alb_hm['https_monitor'] = dict(
            http_request=lb_hm['request_url'],
            http_request_body=lb_hm.get('request_body'),
            http_response=lb_hm.get('response_body'),
            http_response_code=get_alb_response_codes(lb_hm['response_status_codes']),
        )
    elif lb_hm['resource_type'] == 'LBIcmpMonitorProfile':
        alb_hm['type'] = 'HEALTH_MONITOR_PING'
    elif lb_hm['resource_type'] == 'LBTcpMonitorProfile':
        alb_hm['type'] = 'HEALTH_MONITOR_TCP'
    elif lb_hm['resource_type'] == 'LbUdpMonitorProfile':
        alb_hm['type'] = 'HEALTH_MONITOR_UDP'


def convert(alb_config, nsx_lb_config):
    alb_config['HealthMonitor'] = list()

    for lb_hm in nsx_lb_config['LbMonitorProfiles']:
        if lb_hm['resource_type'] == 'LBPassiveMonitorProfile':
            continue
        alb_hm = dict(
            name=lb_hm['display_name'],
            failed_checks=lb_hm['fall_count'],
            receive_timeout=lb_hm['timeout'],
            send_interval=lb_hm['interval'],
            monitor_port=lb_hm.get('monitor_port', None),
        )
        update_alb_type(lb_hm, alb_hm)

        alb_config['HealthMonitor'].append(alb_hm)