from django_celery_beat.models import PeriodicTask, IntervalSchedule, CrontabSchedule from rest_framework.request import Request from .filters import ServerFilter, MonitorFilter from .models import Server, Monitor, SysFiles from .serializers import ServerSerializer, MonitorSerializer, UpdateServerSerializer from ..op_drf.response import SuccessResponse, ErrorResponse from ..op_drf.viewsets import CustomModelViewSet from ..permission.permissions import CommonPermission from ..system.models import ConfigSettings class ServerModelViewSet(CustomModelViewSet): """ 服务器信息 模型的CRUD视图 """ queryset = Server.objects.all() serializer_class = ServerSerializer update_serializer_class = UpdateServerSerializer # extra_filter_backends = [DataLevelPermissionsFilter] filter_class = ServerFilter update_extra_permission_classes = (CommonPermission,) destroy_extra_permission_classes = (CommonPermission,) create_extra_permission_classes = (CommonPermission,) ordering = '-create_datetime' # 默认排序 class MonitorModelViewSet(CustomModelViewSet): """ 服务器监控信息 模型的CRUD视图 """ queryset = Monitor.objects.all() serializer_class = MonitorSerializer # extra_filter_backends = [DataLevelPermissionsFilter] filter_class = MonitorFilter update_extra_permission_classes = (CommonPermission,) destroy_extra_permission_classes = (CommonPermission,) create_extra_permission_classes = (CommonPermission,) ordering = '-create_datetime' # 默认排序 def get_rate_info(self, request: Request, *args, **kwargs): """ 获取使用率 监控信息 :param request: :param args: :param kwargs: :return: """ pk = kwargs.get("pk") queryset = self.filter_queryset(self.get_queryset()) queryset = queryset.filter(server__id=pk).order_by("-create_datetime") # 间隔取值 queryset_count = queryset.count() Interval_num = 1 if queryset_count < 200 else int(queryset_count / 100) queryset = queryset.values('cpu_sys', 'mem_num', 'mem_sys')[::Interval_num][:100] data = { "cpu": [], "memory": [], } for ele in queryset: data["cpu"].append(float(ele.get('cpu_sys', 0))) data["memory"].append(float(ele.get('mem_num', 0)) and round(float(ele.get('mem_sys', 0)) / float(ele.get('mem_num', 0)), 4)) return SuccessResponse(data=data) def get_monitor_info(self, request: Request, *args, **kwargs): """ 最新的服务器状态信息 :param request: :param args: :param kwargs: :return: """ pk = kwargs.get("pk") instance = Monitor.objects.filter(server__id=pk).order_by("-create_datetime").first() if not instance: return ErrorResponse(msg="未找到服务器信息id") serializer = self.get_serializer(instance) data = serializer.data return SuccessResponse(data={ "cpu": { "total": int(data.get('cpu_num'), 0), "used": "", # cpu核心 可不传，如指cpu当前主频，该值可以传 "rate": float(data.get('cpu_sys', 0)) / 100, "unit": "核心", # 默认单位 核心 }, "memory": { "total": int(int(data.get('mem_num', 0)) / 1024), "used": int(int(data.get('mem_sys', 0)) / 1024), "rate": int(data.get('mem_num', 0)) and round(int(data.get('mem_sys', 0)) / int(data.get('mem_num', 0)), 4), "unit": "MB", # 默认单位 MB }, "disk": [{ "dir_name": ele.get('dir_name'), "total": int(int(ele.get('total', 0)) / 1024 / 1024 / 1024), "used": int(int(ele.get('disk_sys', 0)) / 1024 / 1024 / 1024), "rate": int(ele.get('total', 0)) and round(int(ele.get('disk_sys', 0)) / int(ele.get('total', 0)), 4), "unit": "GB", # 默认单位 GB } for ele in SysFiles.objects.filter(monitor=instance).values('dir_name', 'total', 'disk_sys')] }) def enabled_monitor_info(self, request: Request, *args, **kwargs): """ 更新和获取监控信息 :param request: :param args: :param kwargs: :return: """ enabled = request.query_params.get('enabled', None) save_days = request.query_params.get('save_days', None) interval = request.query_params.get('interval', None) # 定时获取系统监控信息 periodictask_obj = PeriodicTask.objects.filter(task='apps.vadmin.monitor.tasks.get_monitor_info').first() if not periodictask_obj: intervalschedule_obj, _ = IntervalSchedule.objects.get_or_create(every=5, period="seconds") periodictask_obj = PeriodicTask() periodictask_obj.task = "apps.vadmin.monitor.tasks.get_monitor_info" periodictask_obj.name = "定时获取系统监控信息" periodictask_obj.interval = intervalschedule_obj periodictask_obj.enabled = False periodictask_obj.save() # 定时清理多余 系统监控信息 clean_task_obj = PeriodicTask.objects.filter( task='apps.vadmin.monitor.tasks.clean_surplus_monitor_info').first() if not clean_task_obj: crontab_obj, _ = CrontabSchedule.objects.get_or_create(day_of_month="*", day_of_week="*", hour="1", minute="0", month_of_year="*") clean_task_obj = PeriodicTask() clean_task_obj.task = "apps.vadmin.monitor.tasks.clean_surplus_monitor_info" clean_task_obj.name = "定时清理多余-系统监控信息" clean_task_obj.crontab = crontab_obj clean_task_obj.enabled = True clean_task_obj.save() # 配置添加 config_obj = ConfigSettings.objects.filter(configKey='sys.monitor.info.save_days').first() if not config_obj: config_obj = ConfigSettings() config_obj.configKey = "sys.monitor.info.save_days" config_obj.configName = "定时清理多余系统监控信息" config_obj.configValue = "30" config_obj.configType = "Y" config_obj.status = "1" config_obj.remark = "定时清理多余-系统监控信息，默认30天" config_obj.save() if enabled: # 更新监控状态 periodictask_obj.enabled = True if enabled == "1" else False periodictask_obj.save() # 更新 定时清理多余 系统监控信息 状态 clean_task_obj.enabled = True if enabled == "1" else False clean_task_obj.save() # 更新保留天数 if save_days and config_obj: config_obj.configValue = save_days config_obj.save() # 更新监控获取频率 if interval: periodictask_obj.interval.every = interval periodictask_obj.interval.save() return SuccessResponse(data={ "enabled": periodictask_obj.enabled, "interval": periodictask_obj.interval.every, "save_days": config_obj.configValue if config_obj else "30", }) def clean_all(self, request: Request, *args, **kwargs): """ 清空监控信息 :param request: :param args: :param kwargs: :return: """ self.get_queryset().delete() return SuccessResponse(msg="清空成功")