Example #1
0
def graph(request, name):
    """Handle the graph request.

    :param request: The original resmon GRAPH request.
    :type request: HttpRequest
    :param name: The name of the resmon RRD whose graph is desired.
    :type name: str
    :return: The response to the graph request.
    :rtype: HttpResponse
    """
    try:
        period = request.GET.get('period')
        width = request.GET.get('width')
        height = request.GET.get('height')
        dataset = request.GET.get('dataset')  # Space delimited data set
        end_ts = int(time.time())
        start_ts = end_ts - parse_timespec(period)

        if not dataset:
            raise HttpResponseBadRequest('No dataset given')

        resp = RestCmd('resmon/graph', '')

        img = tempfile.mkstemp(suffix=".png", prefix="resmongraph")
        try:
            close(img[0])

            execute_graph_request(name, dataset.split(), img[1], start_ts, end_ts, 'AVERAGE', width, height)

            # Now read in the image file and prepare for the response.
            with open(img[1], 'r') as f:
                img_size = os.path.getsize(img[1])
                if img_size <= 2095104:  # Read maximum 2Mb
                    resp.data = f.read()

                    if 'xml' in request.META.get('HTTP_ACCEPT'):
                        return HttpResponse(ElementTree.tostring(resp.export_xml()), content_type='application/xml')

                    return HttpResponse(resp.export_json(), content_type='application/json')

                else:
                    logger.warning('Image file size {0} exceeded the maximum limit of 2Mb, please retry with smaller dimensions.'
                                   .format(img_size))
                    return HttpResponseServerError('Image file too large')

        finally:
            os.remove(img[1])  # Clean up the temp file.

    except Exception as e:
        tb = sys.exc_info()[2]
        logger.error('Encountered unexpected {0} at {1}'.format(e, traceback.format_tb(tb)))
        return HttpResponseServerError('Unexpected internal server error')
Example #2
0
def fetch(request, name, res_type, consolidation):
    """Fetch the data from a specific RRD, in some time frame, with a particular resolution.

    :param request: The original request.
    :type request: HttpRequest
    :param name: The name of the target resmon database.
    :type name: str
    :param res_type: The type of data: cpu | memory
    :type res_type: str
    :param consolidation: The optional consolidation function used. If specified must be one of: average | last.
     The default is last.
    :type consolidation; str
    :return: The requested data set.
    """
    try:
        resolution = request.GET.get('resolution')
        start = request.GET.get('start')
        end = request.GET.get('end')
        if not resolution:
            resolution = '1s'  # Default to the fastest resolution

        if not end:
            end = time.time()  # Default to now

        if not start:
            start = end - (24 * 3600)  # Default to one day worth of data

        ts = time.time()
        data = None
        if consolidation == 'average':
            data = rrdutil.fetch_avg_data(name + res_type + '.rrd', start_ts=int(start), end_ts=int(end), resolution=parse_timespec(resolution))
        else:
            data = rrdutil.fetch_last_data(name + res_type + '.rrd', start_ts=int(start), end_ts=int(end), resolution=parse_timespec(resolution))

        resp = RestCmd('/resmon/fetch', _process_fetch_data(data))

        if 'xml' in request.META.get('HTTP_ACCEPT'):
            return HttpResponse(ElementTree.tostring(resp.export_xml()), content_type='application/xml')

        return HttpResponse(resp.export_json(), content_type='application/json')

    except Exception as e:
        tb = sys.exc_info()[2]
        logger.error('Encountered unexpected {0} at {1}'.format(e, traceback.format_tb(tb)))
        return HttpResponseServerError('Encountered unexpected error, please check service log for more details')