tinix-ai
146 lines
4.7 KiB
Python
146 lines
4.7 KiB
Python
"""
|
|
Mô-đun theo dõi và nhật ký - Hệ thống nhật ký cấp sản xuất
|
|
|
|
Bản quyền © 2026 Công ty TNHH Công nghệ An ninh mạng Huyễn Thành Tân Cương (Công nghệ Huyễn Thành)
|
|
Tác giả: Huyễn Thành
|
|
"""
|
|
import logging
|
|
import logging.handlers
|
|
import os
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
LOG_DIR = "logs"
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|
|
|
# Cấu hình tệp nhật ký
|
|
LOG_FILE = os.path.join(LOG_DIR, f"novel_tool_{datetime.now().strftime('%Y%m%d')}.log")
|
|
ERROR_LOG_FILE = os.path.join(LOG_DIR, f"errors_{datetime.now().strftime('%Y%m%d')}.log")
|
|
|
|
|
|
def setup_logger(
|
|
name: str,
|
|
log_level: int = logging.INFO,
|
|
log_to_file: bool = True,
|
|
force_reconfigure: bool = False
|
|
) -> logging.Logger:
|
|
"""
|
|
Khởi tạo logger cấp sản xuất
|
|
|
|
Args:
|
|
name: Tên logger
|
|
log_level: Cấp độ nhật ký
|
|
log_to_file: Có xuất ra tệp hay không
|
|
force_reconfigure: Có bắt buộc cấu hình lại không (xóa các handler cũ)
|
|
|
|
Returns:
|
|
Biến logger đã được cấu hình
|
|
"""
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(log_level)
|
|
|
|
# Nếu bắt buộc cấu hình lại, xóa bỏ handler hiện có
|
|
if force_reconfigure:
|
|
for handler in logger.handlers[:]:
|
|
logger.removeHandler(handler)
|
|
# Tránh thêm handler lặp lại
|
|
elif logger.handlers:
|
|
return logger
|
|
|
|
# Định dạng đầu ra bảng điều khiển
|
|
console_formatter = logging.Formatter(
|
|
'[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
)
|
|
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(logging.INFO)
|
|
console_handler.setFormatter(console_formatter)
|
|
logger.addHandler(console_handler)
|
|
|
|
# Đầu ra tệp (nếu được bật)
|
|
if log_to_file:
|
|
file_formatter = logging.Formatter(
|
|
'[%(asctime)s] [%(name)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
)
|
|
|
|
# Nhật ký chung
|
|
file_handler = logging.handlers.RotatingFileHandler(
|
|
LOG_FILE,
|
|
maxBytes=10*1024*1024, # 10MB
|
|
backupCount=5,
|
|
encoding='utf-8'
|
|
)
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler.setFormatter(file_formatter)
|
|
logger.addHandler(file_handler)
|
|
|
|
# Nhật ký lỗi - Luôn thêm vào, đảm bảo chắc chắn sẽ thu thập được nhật ký cấp độ ERROR
|
|
error_handler = logging.handlers.RotatingFileHandler(
|
|
ERROR_LOG_FILE,
|
|
maxBytes=10*1024*1024,
|
|
backupCount=5,
|
|
encoding='utf-8'
|
|
)
|
|
error_handler.setLevel(logging.ERROR)
|
|
error_handler.setFormatter(file_formatter)
|
|
logger.addHandler(error_handler)
|
|
|
|
return logger
|
|
|
|
|
|
class PerformanceMonitor:
|
|
"""Công cụ theo dõi hiệu suất"""
|
|
|
|
def __init__(self):
|
|
self.logger = setup_logger("PerformanceMonitor")
|
|
self.metrics = {}
|
|
|
|
def record_metric(self, name: str, value: float, unit: str = "ms") -> None:
|
|
"""Ghi lại số liệu hiệu suất"""
|
|
if name not in self.metrics:
|
|
self.metrics[name] = []
|
|
|
|
self.metrics[name].append(value)
|
|
|
|
# Chỉ lưu 1000 bản ghi gần nhất
|
|
if len(self.metrics[name]) > 1000:
|
|
self.metrics[name] = self.metrics[name][-1000:]
|
|
|
|
def get_average(self, name: str) -> Optional[float]:
|
|
"""Lấy giá trị trung bình"""
|
|
if name not in self.metrics or not self.metrics[name]:
|
|
return None
|
|
return sum(self.metrics[name]) / len(self.metrics[name])
|
|
|
|
def report(self) -> str:
|
|
"""Tạo báo cáo hiệu suất"""
|
|
if not self.metrics:
|
|
return "No performance data yet"
|
|
|
|
report = "=== Performance Report ===\n"
|
|
for name, values in self.metrics.items():
|
|
if values:
|
|
avg = sum(values) / len(values)
|
|
max_val = max(values)
|
|
min_val = min(values)
|
|
report += f"{name}: avg={avg:.2f}ms, max={max_val:.2f}ms, min={min_val:.2f}ms, count={len(values)}\n"
|
|
|
|
return report
|
|
|
|
|
|
# Phiên bản toàn cục
|
|
_logger = setup_logger("NovelTool")
|
|
_performance_monitor = PerformanceMonitor()
|
|
|
|
|
|
def get_logger(name: str = "NovelTool") -> logging.Logger:
|
|
"""Lấy biến instance của logger"""
|
|
return logging.getLogger(name)
|
|
|
|
|
|
def get_performance_monitor() -> PerformanceMonitor:
|
|
"""Lấy trình theo dõi hiệu suất"""
|
|
return _performance_monitor
|