address comment: fix telemetry init

Author: blva

src/server.ts

@@ -7,14 +7,17 @@ import logger, { initializeLogger } from "./logger.js";
 import { mongoLogId } from "mongodb-log-writer";
 import config from "./config.js";
 import { ObjectId } from "mongodb";
+import { Telemetry } from "./telemetry/telemetry.js";
 
 export class Server {
     public readonly session: Session;
     private readonly mcpServer: McpServer;
+    private readonly telemetry: Telemetry;
 
     constructor({ mcpServer, session }: { mcpServer: McpServer; session: Session }) {
         this.mcpServer = mcpServer;
         this.session = session;
+        this.telemetry = new Telemetry(session);
     }
 
     async connect(transport: Transport) {
@@ -45,7 +48,7 @@ export class Server {
 
     private registerTools() {
         for (const tool of [...AtlasTools, ...MongoDbTools]) {
-            new tool(this.session).register(this.mcpServer);
+            new tool(this.session, this.telemetry).register(this.mcpServer);
         }
     }
 

src/tools/atlas/atlasTool.ts

@@ -1,10 +1,14 @@
 import { ToolBase, ToolCategory } from "../tool.js";
 import { Session } from "../../session.js";
 import config from "../../config.js";
+import { Telemetry } from "../../telemetry/telemetry.js";
 
 export abstract class AtlasToolBase extends ToolBase {
-    constructor(protected readonly session: Session) {
-        super(session);
+    constructor(
+        protected readonly session: Session,
+        telemetry: Telemetry
+    ) {
+        super(session, telemetry);
     }
 
     protected category: ToolCategory = "atlas";

src/tools/mongodb/mongodbTool.ts

@@ -5,15 +5,16 @@ import { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { ErrorCodes, MongoDBError } from "../../errors.js";
 import config from "../../config.js";
+import { Telemetry } from "../../telemetry/telemetry.js";
 
 export const DbOperationArgs = {
     database: z.string().describe("Database name"),
     collection: z.string().describe("Collection name"),
 };
 
 export abstract class MongoDBToolBase extends ToolBase {
-    constructor(session: Session) {
-        super(session);
+    constructor(session: Session, telemetry: Telemetry) {
+        super(session, telemetry);
     }
 
     protected category: ToolCategory = "mongodb";

src/tools/tool.ts

@@ -1,12 +1,12 @@
 import { z, type ZodRawShape, type ZodNever } from "zod";
 import type { McpServer, ToolCallback } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { Session } from "../session.js";
 import logger from "../logger.js";
 import { mongoLogId } from "mongodb-log-writer";
 import config from "../config.js";
 import { Telemetry } from "../telemetry/telemetry.js";
-import { type ToolEvent, type TelemetryResult } from "../telemetry/types.js";
+import { type ToolEvent } from "../telemetry/types.js";
 
 export type ToolArgs<Args extends ZodRawShape> = z.objectOutputType<Args, ZodNever>;
 
@@ -24,13 +24,12 @@ export abstract class ToolBase {
 
     protected abstract argsShape: ZodRawShape;
 
-    private readonly telemetry: Telemetry;
-
     protected abstract execute(...args: Parameters<ToolCallback<typeof this.argsShape>>): Promise<CallToolResult>;
 
-    protected constructor(protected session: Session) {
-        this.telemetry = new Telemetry(session);
-    }
+    protected constructor(
+        protected session: Session,
+        protected readonly telemetry: Telemetry
+    ) {}
 
     /**
      * Creates and emits a tool telemetry event
@@ -40,11 +39,6 @@ export abstract class ToolBase {
      */
     private async emitToolEvent(startTime: number, result: CallToolResult): Promise<void> {
         const duration = Date.now() - startTime;
-        logger.info(
-            mongoLogId(1_000_007),
-            "tool",
-            `Tool ${this.name} executed in ${duration}ms with result: ${result}`
-        );
         const event: ToolEvent = {
             timestamp: new Date().toISOString(),
             source: "mdbmcp",