Add map indexes for O(1) lookups in Registry

Address review feedback to use maps for collections. Added lookup maps
(toolsByName, resourcesByURI, promptsByName) while keeping slices for
ordered iteration. This provides O(1) lookup for:

- FindToolByName
- filterToolsByName (used by ForMCPRequest)
- filterResourcesByURI
- filterPromptsByName

Maps are built once during Build() and shared in ForMCPRequest copies.

Author: SamMorrowDrums

.gitignore

@@ -19,4 +19,5 @@ bin/
 # binary
 github-mcp-server
 
-.historyconformance-report/
+.history
+conformance-report/

pkg/registry/builder.go

@@ -124,6 +124,9 @@ func (b *Builder) Build() *Registry {
 		featureChecker:    b.featureChecker,
 	}
 
+	// Note: lookup maps (toolsByName, resourcesByURI, promptsByName) are
+	// lazy-initialized on first use via getToolsByName(), etc.
+
 	// Process toolsets
 	r.enabledToolsets, r.unrecognizedToolsets = b.processToolsets()
 
@@ -147,16 +150,30 @@ func (b *Builder) Build() *Registry {
 // - enabledToolsets map (nil means all enabled)
 // - unrecognizedToolsets list for warnings
 func (b *Builder) processToolsets() (map[ToolsetID]bool, []string) {
-	// Build a set of valid toolset IDs for validation
+	// Single pass: collect valid IDs and default IDs together
 	validIDs := make(map[ToolsetID]bool)
-	for _, t := range b.tools {
-		validIDs[t.Toolset.ID] = true
+	defaultIDs := make(map[ToolsetID]bool)
+
+	for i := range b.tools {
+		id := b.tools[i].Toolset.ID
+		validIDs[id] = true
+		if b.tools[i].Toolset.Default {
+			defaultIDs[id] = true
+		}
 	}
-	for _, r := range b.resourceTemplates {
-		validIDs[r.Toolset.ID] = true
+	for i := range b.resourceTemplates {
+		id := b.resourceTemplates[i].Toolset.ID
+		validIDs[id] = true
+		if b.resourceTemplates[i].Toolset.Default {
+			defaultIDs[id] = true
+		}
 	}
-	for _, p := range b.prompts {
-		validIDs[p.Toolset.ID] = true
+	for i := range b.prompts {
+		id := b.prompts[i].Toolset.ID
+		validIDs[id] = true
+		if b.prompts[i].Toolset.Default {
+			defaultIDs[id] = true
+		}
 	}
 
 	toolsetIDs := b.toolsetIDs
@@ -184,7 +201,7 @@ func (b *Builder) processToolsets() (map[ToolsetID]bool, []string) {
 			continue
 		}
 		if trimmed == "default" {
-			for _, defaultID := range b.defaultToolsetIDs() {
+			for defaultID := range defaultIDs {
 				if !seen[defaultID] {
 					seen[defaultID] = true
 					expanded = append(expanded, defaultID)
@@ -213,29 +230,3 @@ func (b *Builder) processToolsets() (map[ToolsetID]bool, []string) {
 	}
 	return enabledToolsets, unrecognized
 }
-
-// defaultToolsetIDs returns toolset IDs marked as Default in their metadata.
-func (b *Builder) defaultToolsetIDs() []ToolsetID {
-	seen := make(map[ToolsetID]bool)
-	for i := range b.tools {
-		if b.tools[i].Toolset.Default {
-			seen[b.tools[i].Toolset.ID] = true
-		}
-	}
-	for i := range b.resourceTemplates {
-		if b.resourceTemplates[i].Toolset.Default {
-			seen[b.resourceTemplates[i].Toolset.ID] = true
-		}
-	}
-	for i := range b.prompts {
-		if b.prompts[i].Toolset.Default {
-			seen[b.prompts[i].Toolset.ID] = true
-		}
-	}
-
-	ids := make([]ToolsetID, 0, len(seen))
-	for id := range seen {
-		ids = append(ids, id)
-	}
-	return ids
-}

pkg/registry/filters.go

@@ -149,7 +149,7 @@ func (r *Registry) AvailablePrompts(ctx context.Context) []ServerPrompt {
 }
 
 // filterToolsByName returns tools matching the given name, checking deprecated aliases.
-// Returns from the current tools slice (respects existing filter chain).
+// Uses linear scan - optimized for single-lookup per-request scenarios (ForMCPRequest).
 func (r *Registry) filterToolsByName(name string) []ServerTool {
 	// First check for exact match
 	for i := range r.tools {
@@ -169,9 +169,9 @@ func (r *Registry) filterToolsByName(name string) []ServerTool {
 }
 
 // filterResourcesByURI returns resource templates matching the given URI pattern.
+// Uses linear scan - optimized for single-lookup per-request scenarios (ForMCPRequest).
 func (r *Registry) filterResourcesByURI(uri string) []ServerResourceTemplate {
 	for i := range r.resourceTemplates {
-		// Check if URI matches the template pattern (exact match on URITemplate string)
 		if r.resourceTemplates[i].Template.URITemplate == uri {
 			return []ServerResourceTemplate{r.resourceTemplates[i]}
 		}
@@ -180,6 +180,7 @@ func (r *Registry) filterResourcesByURI(uri string) []ServerResourceTemplate {
 }
 
 // filterPromptsByName returns prompts matching the given name.
+// Uses linear scan - optimized for single-lookup per-request scenarios (ForMCPRequest).
 func (r *Registry) filterPromptsByName(name string) []ServerPrompt {
 	for i := range r.prompts {
 		if r.prompts[i].Prompt.Name == name {

pkg/registry/registry.go

@@ -6,6 +6,7 @@ import (
 	"os"
 	"slices"
 	"sort"
+	"sync"
 
 	"github.com/modelcontextprotocol/go-sdk/mcp"
 )
@@ -25,11 +26,15 @@ import (
 //   - Lazy dependency injection during registration via RegisterAll()
 //   - Runtime toolset enabling for dynamic toolsets mode
 type Registry struct {
-	// tools holds all tools in this group
+	// tools holds all tools in this group (ordered for iteration)
 	tools []ServerTool
-	// resourceTemplates holds all resource templates in this group
+	// toolsByName provides O(1) lookup by tool name (lazy-initialized)
+	// Used by FindToolByName for repeated lookups in long-lived servers
+	toolsByName     map[string]*ServerTool
+	toolsByNameOnce sync.Once
+	// resourceTemplates holds all resource templates in this group (ordered for iteration)
 	resourceTemplates []ServerResourceTemplate
-	// prompts holds all prompts in this group
+	// prompts holds all prompts in this group (ordered for iteration)
 	prompts []ServerPrompt
 	// deprecatedAliases maps old tool names to new canonical names
 	deprecatedAliases map[string]string
@@ -57,6 +62,18 @@ func (r *Registry) UnrecognizedToolsets() []string {
 	return r.unrecognizedToolsets
 }
 
+// getToolsByName returns the toolsByName map, initializing it lazily on first call.
+// Used by FindToolByName for O(1) lookups in long-lived servers with repeated lookups.
+func (r *Registry) getToolsByName() map[string]*ServerTool {
+	r.toolsByNameOnce.Do(func() {
+		r.toolsByName = make(map[string]*ServerTool, len(r.tools))
+		for i := range r.tools {
+			r.toolsByName[r.tools[i].Tool.Name] = &r.tools[i]
+		}
+	})
+	return r.toolsByName
+}
+
 // MCP method constants for use with ForMCPRequest.
 const (
 	MCPMethodInitialize             = "initialize"
@@ -90,6 +107,8 @@ const (
 // All existing filters (read-only, toolsets, etc.) still apply to the returned items.
 func (r *Registry) ForMCPRequest(method string, itemName string) *Registry {
 	// Create a shallow copy with shared filter settings
+	// Note: lazy-init maps (toolsByName, etc.) are NOT copied - the new Registry
+	// will initialize its own maps on first use if needed
 	result := &Registry{
 		tools:                r.tools,
 		resourceTemplates:    r.resourceTemplates,
@@ -269,11 +288,8 @@ func (r *Registry) ResolveToolAliases(toolNames []string) (resolved []string, al
 // Returns the tool, its toolset ID, and an error if not found.
 // This searches ALL tools regardless of filters.
 func (r *Registry) FindToolByName(toolName string) (*ServerTool, ToolsetID, error) {
-	for i := range r.tools {
-		tool := &r.tools[i]
-		if tool.Tool.Name == toolName {
-			return tool, tool.Toolset.ID, nil
-		}
+	if tool, ok := r.getToolsByName()[toolName]; ok {
+		return tool, tool.Toolset.ID, nil
 	}
 	return nil, "", NewToolDoesNotExistError(toolName)
 }