Skip to content

Commit 6e310cc

Browse files
Ken LiaoKen Liao
authored andcommitted
remove customized logger and use the one provided by Python
1 parent 97ae477 commit 6e310cc

File tree

2 files changed

+12
-54
lines changed

2 files changed

+12
-54
lines changed

src/mcp_server_rabbitmq/logger.py

Lines changed: 0 additions & 43 deletions
This file was deleted.

src/mcp_server_rabbitmq/server.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
Tool,
66
)
77
import ssl
8+
import logging
89
from .models import (
910
Enqueue,
1011
Fanout,
@@ -16,7 +17,6 @@
1617
DeleteExchange,
1718
GetExchangeInfo
1819
)
19-
from .logger import Logger, LOG_LEVEL
2020
from .connection import RabbitMQConnection, validate_rabbitmq_name
2121
from .handlers import (
2222
handle_enqueue,
@@ -32,19 +32,20 @@
3232
from .admin import RabbitMQAdmin
3333

3434

35-
async def serve(rabbitmq_host: str, port: int, username: str, password: str, use_tls: bool, log_level: str = LOG_LEVEL.DEBUG.name, api_port: int = 15671) -> None:
35+
async def serve(rabbitmq_host: str, port: int, username: str, password: str, use_tls: bool, log_level: str = "INFO", api_port: int = 15671) -> None:
3636
# Setup server
3737
server = Server("mcp-rabbitmq")
3838
# Setup logger
39-
is_log_level_exception = False
40-
try:
41-
log_level = LOG_LEVEL[log_level]
42-
except Exception:
43-
is_log_level_exception = True
44-
log_level = LOG_LEVEL.WARNING
45-
logger = Logger("server.log", log_level)
46-
if is_log_level_exception:
47-
logger.warning("Wrong log_level received. Default to WARNING")
39+
logger = logging.getLogger("mcp-rabbitmq")
40+
logger.setLevel(log_level)
41+
# Configure logging with timestamp and file output
42+
logging.basicConfig(
43+
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
44+
handlers=[
45+
logging.FileHandler('server.log'),
46+
logging.StreamHandler()
47+
]
48+
)
4849

4950
@server.list_tools()
5051
async def list_tools() -> list[Tool]:

0 commit comments

Comments
 (0)