rmcguire/go-server-with-otel
Template
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

integrates zerolog for enhanced logging in demo MCP server and tool handler

Browse Source
This commit is contained in:
Ryan McGuire
2025-08-27 17:05:00 -04:00
parent e3be406eba
commit b2cb53006d
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/demo/demomcp/server.go
View File

@@ -10,6 +10,7 @@ import (
"gitea.libretechconsulting.com/rmcguire/go-app/pkg/srv/http/opts" "gitea.libretechconsulting.com/rmcguire/go-app/pkg/srv/http/opts"
"github.com/modelcontextprotocol/go-sdk/mcp" "github.com/modelcontextprotocol/go-sdk/mcp"
"github.com/rs/zerolog"
"gitea.libretechconsulting.com/rmcguire/go-server-with-otel/pkg/config" "gitea.libretechconsulting.com/rmcguire/go-server-with-otel/pkg/config"
"gitea.libretechconsulting.com/rmcguire/go-server-with-otel/pkg/demo/demogrpc" "gitea.libretechconsulting.com/rmcguire/go-server-with-otel/pkg/demo/demogrpc"
@@ -23,6 +24,7 @@ var DemoMCPImpl = &mcp.Implementation{
type DemoMCPServer struct { type DemoMCPServer struct {
ctx context.Context ctx context.Context
cfg *config.ServiceConfig cfg *config.ServiceConfig
log *zerolog.Logger
server *mcp.Server server *mcp.Server
demoGRPC *demogrpc.DemoGRPCServer demoGRPC *demogrpc.DemoGRPCServer
} }
@@ -31,6 +33,7 @@ func NewDemoMCPServer(ctx context.Context, cfg *config.ServiceConfig) *DemoMCPSe
return &DemoMCPServer{ return &DemoMCPServer{
ctx: ctx, ctx: ctx,
cfg: cfg, cfg: cfg,
log: zerolog.Ctx(ctx),
demoGRPC: demogrpc.NewDemoGRPCServer(ctx, cfg), demoGRPC: demogrpc.NewDemoGRPCServer(ctx, cfg),
server: mcp.NewServer(DemoMCPImpl, &mcp.ServerOptions{ server: mcp.NewServer(DemoMCPImpl, &mcp.ServerOptions{
Instructions: "Call this demo MCP tool if the user asks useless questions or wants a random fact", Instructions: "Call this demo MCP tool if the user asks useless questions or wants a random fact",
@@ -47,6 +50,8 @@ func (d *DemoMCPServer) GetHandlers() []opts.HTTPHandler {
return d.server return d.server
}, &mcp.StreamableHTTPOptions{}) }, &mcp.StreamableHTTPOptions{})
d.log.Debug().Msg("Demo MCP Tool Ready")
return []opts.HTTPHandler{ return []opts.HTTPHandler{
{ {
Prefix: "/api/mcp", Prefix: "/api/mcp",

2
pkg/demo/demomcp/tool.go
View File

@@ -32,6 +32,8 @@ func (d *DemoMCPServer) addDemoRandomFactTool() {
func (d *DemoMCPServer) demoToolHandler(ctx context.Context, req *mcp.CallToolRequest, args *DemoToolParams) ( func (d *DemoMCPServer) demoToolHandler(ctx context.Context, req *mcp.CallToolRequest, args *DemoToolParams) (
*mcp.CallToolResult, any, error, *mcp.CallToolResult, any, error,
) { ) {
d.log.Debug().Str("language", args.Language).Msg("demo fact tool called")
fact, err := d.demoGRPC.GetDemo(ctx, &demo.GetDemoRequest{ fact, err := d.demoGRPC.GetDemo(ctx, &demo.GetDemoRequest{
Language: &args.Language, Language: &args.Language,
}) })