用python写一个syslog服务器接收网络设备的消息,并保存日志到文件,每个文件最大10M就自动分割存储
时间: 2023-03-04 20:56:04 浏览: 271
以下是一个使用Python编写的syslog服务器,用于接收网络设备的消息,并将日志保存到文件中。每个日志文件最大为10MB,并在达到此大小时自动分割存储到新的文件中。请注意,此代码使用了第三方模块`pygtail`,用于跟踪日志文件并读取其最新内容。
```python
import socket
import os
import time
import pygtail
LOG_DIR = "/var/log/syslog"
LOG_FILE = "syslog.log"
MAX_FILE_SIZE = 10485760 # 10MB
if not os.path.exists(LOG_DIR):
os.makedirs(LOG_DIR)
def create_log_file():
filename = os.path.join(LOG_DIR, LOG_FILE)
if not os.path.exists(filename):
with open(filename, 'w'):
pass
return filename
def rotate_log_file(filename):
timestamp = time.strftime('%Y%m%d%H%M%S', time.localtime())
new_filename = os.path.join(LOG_DIR, f"{LOG_FILE}.{timestamp}")
os.rename(filename, new_filename)
with open(filename, 'w'):
pass
return filename
def run():
filename = create_log_file()
current_size = os.path.getsize(filename)
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as sock:
sock.bind(('0.0.0.0', 514))
while True:
data, addr = sock.recvfrom(1024)
with open(filename, 'a') as f:
f.write(data.decode('utf-8'))
f.flush()
current_size += len(data)
if current_size >= MAX_FILE_SIZE:
filename = rotate_log_file(filename)
current_size = 0
for line in pygtail.Pygtail(filename):
print(line, end='')
if __name__ == '__main__':
run()
```
请注意,在生产环境中,您可能需要使用更健壮的代码来处理错误和异常情况,并确保文件系统和日志文件的安全性和完整性。此代码仅作为演示用途,应该谨慎使用。