feat: delete custom_logger_handler

Author: harvey_xiang

src/memos/log.py

@@ -1,19 +1,13 @@
-import atexit
 import logging
-import os
-import threading
 
 from logging.config import dictConfig
 from pathlib import Path
 from sys import stdout
 
-import requests
-
 from dotenv import load_dotenv
 
 from memos import settings
 from memos.api.context.context import get_current_trace_id
-from memos.api.context.context_thread import ContextThreadPoolExecutor
 
 
 # Load environment variables
@@ -45,89 +39,6 @@ def filter(self, record):
         return True
 
 
-class CustomLoggerRequestHandler(logging.Handler):
-    _instance = None
-    _lock = threading.Lock()
-
-    def __new__(cls):
-        if cls._instance is None:
-            with cls._lock:
-                if cls._instance is None:
-                    cls._instance = super().__new__(cls)
-                    cls._instance._initialized = False
-                    cls._instance._executor = None
-                    cls._instance._session = None
-                    cls._instance._is_shutting_down = None
-        return cls._instance
-
-    def __init__(self):
-        """Initialize handler with minimal setup"""
-        if not self._initialized:
-            super().__init__()
-            workers = int(os.getenv("CUSTOM_LOGGER_WORKERS", "2"))
-            self._executor = ContextThreadPoolExecutor(
-                max_workers=workers, thread_name_prefix="log_sender"
-            )
-            self._is_shutting_down = threading.Event()
-            self._session = requests.Session()
-            self._initialized = True
-            atexit.register(self._cleanup)
-
-    def emit(self, record):
-        """Process log records of INFO or ERROR level (non-blocking)"""
-        if os.getenv("CUSTOM_LOGGER_URL") is None or self._is_shutting_down.is_set():
-            return
-
-        try:
-            trace_id = get_current_trace_id() or "no-trace-id"
-            self._executor.submit(self._send_log_sync, record.getMessage(), trace_id)
-        except Exception as e:
-            if not self._is_shutting_down.is_set():
-                print(f"Error sending log: {e}")
-
-    def _send_log_sync(self, message, trace_id):
-        """Send log message synchronously in a separate thread"""
-        try:
-            logger_url = os.getenv("CUSTOM_LOGGER_URL")
-            token = os.getenv("CUSTOM_LOGGER_TOKEN")
-
-            headers = {"Content-Type": "application/json"}
-            post_content = {"message": message, "trace_id": trace_id}
-
-            # Add auth token if exists
-            if token:
-                headers["Authorization"] = f"Bearer {token}"
-
-            # Add traceId to headers for consistency
-            headers["traceId"] = trace_id
-
-            # Add custom attributes from env
-            for key, value in os.environ.items():
-                if key.startswith("CUSTOM_LOGGER_ATTRIBUTE_"):
-                    attribute_key = key[len("CUSTOM_LOGGER_ATTRIBUTE_") :].lower()
-                    post_content[attribute_key] = value
-
-            self._session.post(logger_url, headers=headers, json=post_content, timeout=5)
-        except Exception:
-            # Silently ignore errors to avoid affecting main application
-            pass
-
-    def _cleanup(self):
-        """Clean up resources during program exit"""
-        if not self._initialized:
-            return
-
-        self._is_shutting_down.set()
-        try:
-            self._executor.shutdown(wait=False)
-            self._session.close()
-        except Exception as e:
-            print(f"Error during cleanup: {e}")
-
-    def close(self):
-        """Override close to prevent premature shutdown"""
-
-
 LOGGING_CONFIG = {
     "version": 1,
     "disable_existing_loggers": False,