mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2026-01-29 00:51:53 +08:00
refactoring
This commit is contained in:
@@ -1,13 +1,5 @@
|
||||
import sys
|
||||
import re
|
||||
import asyncio
|
||||
import logging
|
||||
import time
|
||||
|
||||
from typing import Dict
|
||||
from typing import AsyncGenerator
|
||||
|
||||
import systemd.journal
|
||||
|
||||
|
||||
# =====
|
||||
@@ -21,42 +13,3 @@ def get_logger(depth: int=1) -> logging.Logger:
|
||||
break
|
||||
name = frames[depth].f_globals["__name__"]
|
||||
return logging.getLogger(name)
|
||||
|
||||
|
||||
class Log:
|
||||
def __init__(self, loop: asyncio.AbstractEventLoop) -> None:
|
||||
self.__loop = loop
|
||||
|
||||
async def poll_log(self, seek: int, follow: bool) -> AsyncGenerator[Dict, None]:
|
||||
reader = systemd.journal.Reader()
|
||||
reader.this_boot()
|
||||
reader.this_machine()
|
||||
reader.log_level(systemd.journal.LOG_DEBUG)
|
||||
|
||||
services = set(
|
||||
service
|
||||
for service in systemd.journal.Reader().query_unique("_SYSTEMD_UNIT")
|
||||
if re.match(r"kvmd(-\w+)?\.service", service)
|
||||
).union(["kvmd.service"])
|
||||
|
||||
for service in services:
|
||||
reader.add_match(_SYSTEMD_UNIT=service)
|
||||
if seek > 0:
|
||||
reader.seek_realtime(float(time.time() - seek))
|
||||
|
||||
for entry in reader:
|
||||
yield self.__entry_to_record(entry)
|
||||
|
||||
while follow:
|
||||
entry = reader.get_next()
|
||||
if entry:
|
||||
yield self.__entry_to_record(entry)
|
||||
else:
|
||||
await asyncio.sleep(1)
|
||||
|
||||
def __entry_to_record(self, entry: Dict) -> Dict[str, Dict]:
|
||||
return {
|
||||
"dt": entry["__REALTIME_TIMESTAMP"],
|
||||
"service": entry["_SYSTEMD_UNIT"],
|
||||
"msg": entry["MESSAGE"].rstrip(),
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user