summaryrefslogtreecommitdiffstats
path: root/setup.py
blob: d556de77284d060bbf859624f9e475c84ac910f0 (plain) 
1
2
3
4
5
6
7
8
9

#!/usr/bin/env python

from setuptools import setup

setup(
    name="daisy",
    version="master",
    url="https://www.opnfv.org",
)
ght .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */ .highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */ .highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */ .highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */ .highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */ .highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */ .highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */ .highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Double */ .highlight .se { color: #0044dd; background-color: #fff0f0 } /* Literal.String.Escape */ .highlight .sh { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Heredoc */ .highlight .si { color: #3333bb; background-color: #fff0f0 } /* Literal.String.Interpol */ .highlight .sx { color: #22bb22; background-color: #f0fff0 } /* Literal.String.Other */ .highlight .sr { color: #008800; background-color: #fff0ff } /* Literal.String.Regex */ .highlight .s1 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Single */ .highlight .ss { color: #aa6600; background-color: #fff0f0 } /* Literal.String.Symbol */ .highlight .bp { color: #003388 } /* Name.Builtin.Pseudo */ .highlight .fm { color: #0066bb; font-weight: bold } /* Name.Function.Magic */ .highlight .vc { color: #336699 } /* Name.Variable.Class */ .highlight .vg { color: #dd7700 } /* Name.Variable.Global */ .highlight .vi { color: #3333bb } /* Name.Variable.Instance */ .highlight .vm { color: #336699 } /* Name.Variable.Magic */ .highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */ } 
##############################################################################
# Copyright (c) 2017 ZTE Corporation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from datetime import datetime
import json
import requests
import socket
from threading import Thread
import time

from doctor_tests.identity_auth import get_session
from doctor_tests.monitor.base import BaseMonitor


class SampleMonitor(BaseMonitor):
    event_type = "compute.host.down"

    def __init__(self, conf, inspector_url, log):
        super(SampleMonitor, self).__init__(conf, inspector_url, log)
        self.session = get_session()
        self.pinger = None

    def start(self, host):
        self.log.info('sample monitor start......')
        self.pinger = Pinger(host.name, host.ip, self, self.log)
        self.pinger.start()

    def stop(self):
        self.log.info('sample monitor stop......')
        if self.pinger is not None:
            self.pinger.stop()
            self.pinger.join()

    def report_error(self, hostname):
        self.log.info('sample monitor report error......')
        data = [
            {
                'id': 'monitor_sample_id1',
                'time': datetime.now().isoformat(),
                'type': self.event_type,
                'details': {
                    'hostname': hostname,
                    'status': 'down',
                    'monitor': 'monitor_sample',
                    'monitor_event_id': 'monitor_sample_event1'
                },
            },
        ]

        auth_token = self.session.get_token() if \
                     self.conf.inspector.type != 'sample' else None
        headers = {
            'Content-Type': 'application/json',
            'Accept': 'application/json',
            'X-Auth-Token': auth_token,
        }

        url = '%s%s' % (self.inspector_url, 'events') \
            if self.inspector_url.endswith('/') else \
            '%s%s' % (self.inspector_url, '/events')
        requests.put(url, data=json.dumps(data), headers=headers)


class Pinger(Thread):
    interval = 0.1  # second
    timeout = 0.1   # second
    ICMP_ECHO_MESSAGE = bytes([0x08, 0x00, 0xf7, 0xff, 0x00, 0x00, 0x00, 0x00])

    def __init__(self, host_name, host_ip, monitor, log):
        Thread.__init__(self)
        self.monitor = monitor
        self.hostname = host_name
        self.ip_addr = host_ip or socket.gethostbyname(self.hostname)
        self.log = log
        self._stopped = False

    def run(self):
        self.log.info("Starting Pinger host_name(%s), host_ip(%s)"
                      % (self.hostname, self.ip_addr))

        sock = socket.socket(socket.AF_INET, socket.SOCK_RAW,
                             socket.IPPROTO_ICMP)
        sock.settimeout(self.timeout)
        while True:
            if self._stopped:
                return
            try:
                sock.sendto(self.ICMP_ECHO_MESSAGE, (self.ip_addr, 0))
                sock.recv(4096)
            except socket.timeout:
                self.log.info("doctor monitor detected at %s" % time.time())
                self.monitor.report_error(self.hostname)
                self.log.info("ping timeout, quit monitoring...")
                self._stopped = True
                return
            time.sleep(self.interval)

    def stop(self):
        self.log.info("Stopping Pinger host_name(%s), host_ip(%s)"
                      % (self.hostname, self.ip_addr))
        self._stopped = True