NewsBlur/apps/monitor/views/newsblur_app_servers.py

import datetime

from django.conf import settings
from django.shortcuts import render
from django.views import View


class AppServers(View):
    def get(self, request):
        data = dict((("%s" % s["_id"].replace("-", ""), s["feeds"]) for s in self.stats))
        # total = self.total:
        # if total:
        #    data['total'] = total[0]['feeds']
        chart_name = "app_servers"
        chart_type = "counter"

        formatted_data = {}
        for k, v in data.items():
            formatted_data[k] = f'{chart_name}{{app_server="{k}"}} {v}'
        context = {
            "data": formatted_data,
            "chart_name": chart_name,
            "chart_type": chart_type,
        }
        return render(request, "monitor/prometheus_data.html", context, content_type="text/plain")

    @property
    def stats(self):
        stats = settings.MONGOANALYTICSDB.nbanalytics.page_loads.aggregate(
            [
                {
                    "$match": {
                        "date": {
                            "$gte": datetime.datetime.now() - datetime.timedelta(minutes=5),
                        },
                    },
                },
                {
                    "$group": {
                        "_id": "$server",
                        "feeds": {"$sum": 1},
                    },
                },
            ]
        )

        return list(stats)

    @property
    def total(self):
        stats = settings.MONGOANALYTICSDB.nbanalytics.page_loads.aggregate(
            [
                {
                    "$match": {
                        "date": {
                            "$gt": datetime.datetime.now() - datetime.timedelta(minutes=5),
                        },
                    },
                },
                {
                    "$group": {
                        "_id": 1,
                        "feeds": {"$sum": 1},
                    },
                },
            ]
        )

        return list(stats)