mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-30 03:12:20 +03:00
80 lines
3.5 KiB
Python
80 lines
3.5 KiB
Python
|
#!/usr/bin/python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
# Author: Cemre ALPSOY <cemre.alpsoy@agem.com.tr>
|
|||
|
|
# Author: Emre Akkaya <emre.akkaya@agem.com.tr>
|
|||
|
|
|
|||
|
|
from base.plugin.abstract_plugin import AbstractPlugin
|
|||
|
|
from threading import Thread
|
|||
|
|
import time
|
|||
|
|
import psutil
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ResourceUsage(AbstractPlugin):
|
|||
|
|
def __init__(self, data, context):
|
|||
|
|
super(AbstractPlugin, self).__init__()
|
|||
|
|
self.data = data
|
|||
|
|
self.context = context
|
|||
|
|
self.logger = self.get_logger()
|
|||
|
|
self.message_code = self.get_message_code()
|
|||
|
|
self.is_running = False
|
|||
|
|
self._threadCep = None
|
|||
|
|
|
|||
|
|
def handle_task(self):
|
|||
|
|
try:
|
|||
|
|
action = self.data['action']
|
|||
|
|
self.logger.debug("Action: {0}".format(action))
|
|||
|
|
|
|||
|
|
if action == "start_timer":
|
|||
|
|
self.is_running = True
|
|||
|
|
with open('is_running.txt', 'w') as f:
|
|||
|
|
f.write("%s" % str('true'))
|
|||
|
|
self._threadCep = Thread(target=self.run_timer,
|
|||
|
|
args=(int(self.data['interval']), self.context.get('task_id')))
|
|||
|
|
self._threadCep.start()
|
|||
|
|
data = {"status": "started"}
|
|||
|
|
self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
|
|||
|
|
message='Kaynak kullanım bilgileri toplanmaya başlandı.', data=json.dumps(data), content_type=self.get_content_type().APPLICATION_JSON.value)
|
|||
|
|
elif action == "stop_timer":
|
|||
|
|
self.is_running = False
|
|||
|
|
with open('is_running.txt', 'w') as f:
|
|||
|
|
f.write("%s" % str('false'))
|
|||
|
|
data = {"action": "stop"}
|
|||
|
|
self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
|
|||
|
|
message='Kaynak kullanım bilgilerinin toplanması durduruldu.',
|
|||
|
|
data=json.dumps(data), content_type=self.get_content_type().APPLICATION_JSON.value)
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error(str(e))
|
|||
|
|
self.context.create_response(code=self.message_code.TASK_ERROR.value,
|
|||
|
|
message='Kaynak kullanım bilgileri işlenirken hata oluştu: ' + str(e))
|
|||
|
|
|
|||
|
|
def run_timer(self, interval, task_id):
|
|||
|
|
while self.is_running:
|
|||
|
|
self.gather_resource_usage(task_id)
|
|||
|
|
time.sleep(interval)
|
|||
|
|
with open('is_running.txt', 'r') as f:
|
|||
|
|
if f.read() == 'false':
|
|||
|
|
self.is_running = False
|
|||
|
|
|
|||
|
|
def gather_resource_usage(self, task_id):
|
|||
|
|
# Memory usage
|
|||
|
|
memory_usage = psutil.virtual_memory()
|
|||
|
|
self.logger.debug("Memory usage: {0}".format(memory_usage))
|
|||
|
|
# Disk usage
|
|||
|
|
disk_usage = psutil.disk_usage('/')
|
|||
|
|
self.logger.debug("Disk usage: {0}".format(disk_usage))
|
|||
|
|
# CPU usage
|
|||
|
|
cpu_percentage = psutil.cpu_percent(interval=1)
|
|||
|
|
self.logger.debug("CPU percentage: {0}".format(cpu_percentage))
|
|||
|
|
|
|||
|
|
data = {'memoryUsage': str(memory_usage), 'diskUsage': str(disk_usage), 'cpuPercentage': str(cpu_percentage)}
|
|||
|
|
command = 'python3 /opt/ahenk/ahenkd.py send -t {0} -m {1} -s'.format(task_id, json.dumps(str(data)))
|
|||
|
|
result_code, p_out, p_err = self.execute(command)
|
|||
|
|
if result_code != 0:
|
|||
|
|
self.logger.error("Error occurred while sending message: " + str(p_err))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def handle_task(task, context):
|
|||
|
|
plugin = ResourceUsage(task, context)
|
|||
|
|
plugin.handle_task()
|