[AINODE] Fix logger level filter (apache#16195)

Author: liyuheng55555

iotdb-core/ainode/ainode/core/constant.py

@@ -73,12 +73,6 @@
 # AINode log
 LOG_FILE_TYPE = ["all", "info", "warning", "error"]
 AINODE_LOG_FILE_NAME_PREFIX = "log_ainode_"
-AINODE_LOG_FILE_NAMES = [
-    "log_ainode_all.log",
-    "log_ainode_info.log",
-    "log_ainode_warning.log",
-    "log_ainode_error.log",
-]
 AINODE_LOG_FILE_LEVELS = [logging.DEBUG, logging.INFO, logging.WARNING, logging.ERROR]
 DEFAULT_LOG_LEVEL = logging.INFO
 INFERENCE_LOG_FILE_NAME_PREFIX_TEMPLATE = (

iotdb-core/ainode/ainode/core/log.py

@@ -85,6 +85,9 @@ def __init__(self, log_file_name_prefix: str, log_dir=AINODE_LOG_DIR):
                 interval=1,
                 encoding="utf-8",
             )
+            if LOG_FILE_TYPE[i] != "all":
+                file_handler.setLevel(logging.NOTSET)
+                file_handler.addFilter(LogLevelFilter(file_level))
 
             # set renamer
             def universal_namer(
@@ -145,3 +148,13 @@ class Logger(BaseLogger):
 
     def __init__(self, log_file_name_prefix: str = AINODE_LOG_FILE_NAME_PREFIX):
         super().__init__(log_file_name_prefix=log_file_name_prefix)
+
+
+class LogLevelFilter(logging.Filter):
+
+    def __init__(self, level: int):
+        super().__init__()
+        self.level = level
+
+    def filter(self, record: logging.LogRecord) -> bool:
+        return record.levelno == self.level