python sends socket heartbeat alarm in xml format
Requirements
- 1. Inline, the application sends heartbeat and xml to socker.
- 2. Use ps to obtain whether the program exists and determine whether to send a heartbeat.
- 3. Use xml.etree.ElementTree to format xml format
Implementation
- 1. Use socket link to create a socket link
- 2. Use xml.etree.ElementTree to format xml format
- 3. Monitor the process_name_list list to cycle as needed to determine whether to send heartbeats
- 4. Determine whether sending the socket is successful based on the information returned by sending the socket.
- 5. Record log information
Code
#!/usr/bin/python3
"""
need:
1. Inline, the application sends heartbeat and xml to socker.
2. Use ps to obtain whether the program exists and determine whether to send a heartbeat.
"""
import xml.etree.ElementTree as ET
import socket,os,time,logging
from logging.handlers import RotatingFileHandler
def build_xml_message(heartbeat_interval, name, SystemName, AlarmRes):
local_time1 = int(round(time.time() * 1000))
local_time2 = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
root = ET.Element("ALARMSTART")
name_element = ET.SubElement(root, "SystemName")
name_element.text = SystemName
age_element = ET.SubElement(root, "AlarmID")
age_element.text = str(local_time1)
city_element = ET.SubElement(root, "AlarmEquipment")
city_element.text = name
city_element = ET.SubElement(root, "AlarmRes")
city_element.text = AlarmRes
city_element = ET.SubElement(root, "EventTime")
city_element.text = local_time2
return ET.tostring(root, encoding="utf-8", method="xml").decode("utf-8")
def send_heartbeat(xml_message, server_address):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(10)
try:
# print(xml_message.encode("utf-8"))
sock.connect(server_address)
sock.sendall(xml_message.encode("utf-8"))
response = sock.recv(1024)
local_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
if "flag=true" in str(response):
log_info = str(local_time) + " info send info ok ,return info: " + str(response)
logger.debug(log_info)
# write_data_to_file_append(log_file_path, log_info)
else:
log_info = str(local_time) + " error send info fail ,return info: " + str(response)
logger.debug(log_info)
#write_data_to_file_app/end(log_file_path, log_info)
except Exception as e:
logger.error("Failed to send heartbeat to " + str(server_address) + " " + str(e))
finally:
sock.close()
def write_data_to_file_append(file_path, data):
with open(file_path, "a") as f:
f.write(data)
def is_process_running(ps_out,process_name):
# Traverse all processes and find the process with the given name
for line in ps_out.splitlines():
if process_name in line:
return True
return False
if __name__ == "__main__":
#Need to link to soket server
server_address = ("localhost", 10000)
# log_file_path = "send_socket.log"
heartbeat_interval = 5
#Program name to be monitored
process_name_list = ["nginx","mysqld","cdesk_nginx1"]
#xml name definition
SystemName = "ccc-test-001"
AlarmRes = "Customized interface server"
##### logger #####
log_file = "./log/heratbeat.log"
shell = "mkdir -p " + log_file.split("heratbeat.log")[0]
os.popen(shell)
# Create logger
logger = logging.getLogger("my_logger")
logger.setLevel(logging.DEBUG)
#Create RotatingFileHandler object
max_log_size = 100 * 1024 * 1024 # 100 MB
backup_count = 50
handler = RotatingFileHandler(log_file, maxBytes=max_log_size, backupCount=backup_count)
# Define log format
formatter = logging.Formatter("%(asctime)s %(thread)d %(filename)s:%(lineno)d %(levelname)s %(message)s")
handler.setFormatter(formatter)
# Add handler to logger
logger.addHandler(handler)
while True:
#Use os.popen to execute the ps command to obtain all processes
ps_output = os.popen("ps aux").read()
# ps_output = os.popen("ps -e").read()
for process_name in process_name_list:
if is_process_running(ps_output,process_name):
xml_message = build_xml_message(heartbeat_interval, process_name, SystemName, AlarmRes)
print(xml_message)
logger.debug(process_name + " is running.")
send_heartbeat(xml_message, server_address)
else:
logger.error(process_name + " is not running.")
time.sleep(heartbeat_interval)
# Close the handler when the program ends
logger.removeHandler(handler)
handler.close()
Resolve operating conditions
- Heartbeat monitoring running status
2023-09-19 10:32:49,369 140176451651392 heratbeat.py:103 DEBUG nginx is running.
2023-09-19 10:32:49,370 140176451651392 heratbeat.py:45 DEBUG 2023-09-19 10:32:49 error send info fail ,return info: b'<ALARMSTART><SystemName>ccc-test-001</ SystemName><AlarmID>1695090769369</AlarmID><AlarmEquipment>nginx</AlarmEquipment><AlarmRes>\xe5\xae\x9a\xe5\x88\xb6\xe6\x8e\xa5\xe5\x8f\xa3\xe6\x9c\ x8d\xe5\x8a\xa1\xe5\x99\xa8</AlarmRes><EventTime>2023-09-19 10:32:49</EventTime></ALARMSTART>'
2023-09-19 10:32:49,370 140176451651392 heratbeat.py:103 DEBUG mysqld is running.
2023-09-19 10:32:49,371 140176451651392 heratbeat.py:45 DEBUG 2023-09-19 10:32:49 error send info fail ,return info: b'<ALARMSTART><SystemName>ccc-test-001</ SystemName><AlarmID>1695090769370</AlarmID><AlarmEquipment>mysqld</AlarmEquipment><AlarmRes>\xe5\xae\x9a\xe5\x88\xb6\xe6\x8e\xa5\xe5\x8f\xa3\xe6\x9c\ x8d\xe5\x8a\xa1\xe5\x99\xa8</AlarmRes><EventTime>2023-09-19 10:32:49</EventTime></ALARMSTART>'
2023-09-19 10:32:49,371 140176451651392 heratbeat.py:106 ERROR cdesk_nginx1 is not running.
[devops@my-dev ABC]$ ./sock.py
Server started, waiting for client to connect...
Client ('127.0.0.1', 41400) is connected
Received data from ('127.0.0.1', 41400): <ALARMSTART><SystemName>ccc-test-001</SystemName><AlarmID>1695090684048</AlarmID><AlarmEquipment>nginx</AlarmEquipment><AlarmRes>Customized Interface Server</AlarmRes><EventTime>2023-09-19 10:31:24</EventTime></ALARMSTART>
Client ('127.0.0.1', 41402) is connected
Received data from ('127.0.0.1', 41402): <ALARMSTART><SystemName>ccc-test-001</SystemName><AlarmID>1695090684049</AlarmID><AlarmEquipment>mysqld</AlarmEquipment><AlarmRes>Customized Interface Server</AlarmRes><EventTime>2023-09-19 10:31:24</EventTime></ALARMSTART>