Alanxtl commented on code in PR #757: URL: https://github.com/apache/dubbo-go-pixiu/pull/757#discussion_r2336401240
########## pkg/model/mcpserver.go: ########## @@ -0,0 +1,225 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package model + +import ( + "regexp" + "time" +) + +// McpServerConfig MCP Server Filter configuration +type McpServerConfig struct { + ServerInfo ServerInfo `yaml:"server_info" json:"server_info"` + Endpoint string `yaml:"endpoint" json:"endpoint" default:"/mcp"` + Tools []ToolConfig `yaml:"tools,omitempty" json:"tools,omitempty"` + Resources []ResourceConfig `yaml:"resources,omitempty" json:"resources,omitempty"` + ResourceTemplates []ResourceTemplateConfig `yaml:"resource_templates,omitempty" json:"resource_templates,omitempty"` + Prompts []PromptConfig `yaml:"prompts,omitempty" json:"prompts,omitempty"` +} + +// ServerInfo server information +type ServerInfo struct { + Name string `yaml:"name" json:"name" default:"Pixiu MCP Server"` + Version string `yaml:"version" json:"version" default:"1.0.0"` + Description string `yaml:"description,omitempty" json:"description,omitempty" default:"MCP Server powered by Apache Dubbo-go-pixiu"` + Instructions string `yaml:"instructions,omitempty" json:"instructions,omitempty" default:"Use the provided tools to interact with backend services."` +} + +// ToolConfig tool configuration +type ToolConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description" json:"description"` + Cluster string `yaml:"cluster" json:"cluster"` + BackendURL string `yaml:"backend_url,omitempty" json:"backend_url,omitempty"` + Request RequestConfig `yaml:"request" json:"request"` + Args []ArgConfig `yaml:"args,omitempty" json:"args,omitempty"` +} + +// RequestConfig request configuration +type RequestConfig struct { + Method string `yaml:"method" json:"method" default:"GET"` + Path string `yaml:"path" json:"path"` + Headers map[string]string `yaml:"headers,omitempty" json:"headers,omitempty"` + Timeout string `yaml:"timeout,omitempty" json:"timeout,omitempty" default:"30s"` +} + +// ArgConfig parameter configuration (simplified) +type ArgConfig struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + In string `yaml:"in" json:"in"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` +} + +// ResourceConfig resource configuration +type ResourceConfig struct { + Name string `yaml:"name" json:"name"` + URI string `yaml:"uri" json:"uri"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Source ResourceSource `yaml:"source" json:"source"` +} + +// ResourceSource resource source configuration (simplified) +type ResourceSource struct { + Type string `yaml:"type" json:"type"` + Path string `yaml:"path,omitempty" json:"path,omitempty"` // for file type + URL string `yaml:"url,omitempty" json:"url,omitempty"` // for url type + Content string `yaml:"content,omitempty" json:"content,omitempty"` // for inline type + Template string `yaml:"template,omitempty" json:"template,omitempty"` // for template type +} + +// ResourceTemplateConfig resource template configuration +type ResourceTemplateConfig struct { + Name string `yaml:"name" json:"name"` + URITemplate string `yaml:"uri_template" json:"uri_template"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Parameters []ResourceTemplateParameter `yaml:"parameters,omitempty" json:"parameters,omitempty"` + Annotations *ResourceTemplateAnnotations `yaml:"annotations,omitempty" json:"annotations,omitempty"` +} + +// ResourceTemplateParameter resource template parameter +type ResourceTemplateParameter struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty" default:"false"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` +} + +// ResourceTemplateAnnotations resource template annotations +type ResourceTemplateAnnotations struct { + Audience []string `yaml:"audience,omitempty" json:"audience,omitempty"` + Priority *float64 `yaml:"priority,omitempty" json:"priority,omitempty"` + LastModified string `yaml:"last_modified,omitempty" json:"last_modified,omitempty"` +} + +// ComputedParameter computed parameter (for internal processing, simplified) +type ComputedParameter struct { + Name string + Type string + In string + Description string + Required bool + Enum []string + Default any +} + +// PromptConfig prompt configuration +type PromptConfig struct { + Name string `yaml:"name" json:"name"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Arguments []PromptArgumentConfig `yaml:"arguments,omitempty" json:"arguments,omitempty"` + Messages []PromptMessageConfig `yaml:"messages" json:"messages"` +} + +// PromptArgumentConfig prompt argument configuration +type PromptArgumentConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` +} + +// PromptMessageConfig prompt message configuration +type PromptMessageConfig struct { + Role string `yaml:"role" json:"role"` // "user" or "assistant" + Content string `yaml:"content" json:"content"` +} + +// RegistryConfig 注册表配置 +type RegistryConfig struct { + ToolConfigs map[string]ToolConfig `yaml:"toolConfigs"` + ResourceConfigs map[string]ResourceConfig `yaml:"resourceConfigs"` + PromptConfigs map[string]PromptConfig `yaml:"promptConfigs"` + LastUpdated time.Time `yaml:"lastUpdated"` +} + +// GetAllParameters gets all parameters of the tool +func (tc *ToolConfig) GetAllParameters() ([]ComputedParameter, error) { + var allParams []ComputedParameter + + // 1. Automatically extract path parameters + pathParams := GetPathParameterNames(tc.Request.Path) + for _, paramName := range pathParams { + // Find corresponding arg configuration + var argConfig *ArgConfig + for _, arg := range tc.Args { + if arg.Name == paramName && arg.In == "path" { + argConfig = &arg + break + } + } + + // Create computed parameter + computed := ComputedParameter{ + Name: paramName, + Type: "string", // Default type + In: "path", + Required: true, // Path parameters are always required + } + + // Apply arg configuration + if argConfig != nil { + computed.Type = argConfig.Type + computed.Description = argConfig.Description + // Simplified: removed Pattern and Format fields Review Comment: Delete redundant comment ########## pkg/filter/mcp/mcpserver/dynamic.go: ########## @@ -0,0 +1,65 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package mcpserver + +import ( + "github.com/apache/dubbo-go-pixiu/pkg/logger" + "github.com/apache/dubbo-go-pixiu/pkg/model" +) + +var ( + globalRegistry *ToolRegistry + globalDynamic *DynamicConsumer +) + +// GetOrInitRegistry returns a singleton ToolRegistry +func GetOrInitRegistry() *ToolRegistry { + if globalRegistry == nil { + globalRegistry = NewToolRegistry() + } + return globalRegistry Review Comment: use sync.Once ########## pkg/adapter/mcpserver/common/url.go: ########## @@ -0,0 +1,62 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package common + +import ( + "net" + "net/url" + "strconv" + "strings" +) + +// ParseHostPortFromURL extracts host and port from a raw URL or host:port string. +// It supports forms like: +// - http://host:port/path?query +// - https://host:port +// - host:port +// Returns ("", 0) if parsing fails or no valid port is present. +func ParseHostPortFromURL(raw string) (string, int) { + if raw == "" { + return "", 0 + } + + addr := strings.TrimSpace(raw) + + if strings.Contains(addr, "://") { Review Comment: ```suggestion if strings.Contains(addr, constant.ProtocolSlash) { ``` ########## pkg/adapter/mcpserver/registry/nacos/client.go: ########## @@ -0,0 +1,866 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package nacos + +import ( + "encoding/json" + "fmt" + "regexp" + "strings" + "sync" +) + +import ( + "github.com/nacos-group/nacos-sdk-go/v2/clients" + "github.com/nacos-group/nacos-sdk-go/v2/clients/config_client" + "github.com/nacos-group/nacos-sdk-go/v2/clients/naming_client" + "github.com/nacos-group/nacos-sdk-go/v2/common/constant" + nacosmodel "github.com/nacos-group/nacos-sdk-go/v2/model" + "github.com/nacos-group/nacos-sdk-go/v2/vo" +) + +import ( + "github.com/apache/dubbo-go-pixiu/pkg/logger" + "github.com/apache/dubbo-go-pixiu/pkg/model" +) + +// Nacos configuration group constants +const ( + McpServerVersionGroup = "mcp-server-versions" + McpServerSpecGroup = "mcp-server" + McpToolSpecGroup = "mcp-tools" + DefaultNacosListConfigMode = "blur" + DefaultNacosListConfigPageSize = 50 + ListMcpServerConfigIdPattern = "*mcp-versions.json" +) + +// Configuration key name constants +const ( + SystemConfigKeyPrefix = "system::" + CredentialConfigKeyPrefix = "cred::" + ConfigKeySeparator = "::" +) + +// Supported MCP protocol types +var supportedMcpProtocols = map[string]bool{ + "mcp-sse": false, + "mcp-streamable": true, +} + +var templateRegex = regexp.MustCompile(`\$\{nacos\.([a-zA-Z0-9-_:\\.]+/[a-zA-Z0-9-_:\\.]+)}`) + +// ==================== Data Structure Definitions ==================== + +// ServerSpecInfo server specification information +type ServerSpecInfo struct { + RemoteServerConfig *RemoteServerConfig `json:"remoteServerConfig"` +} + +// RemoteServerConfig remote server configuration +type RemoteServerConfig struct { + ServiceRef *ServiceRef `json:"serviceRef"` +} + +// ServiceRef service reference information +type ServiceRef struct { + ServiceName string `json:"serviceName"` + GroupName string `json:"groupName"` + NamespaceId string `json:"namespaceId"` +} + +// NacosRegistryClient Nacos registry client +// Responsible for communicating with Nacos configuration center and service discovery +type NacosRegistryClient struct { + mu sync.RWMutex // Read-write lock, protects concurrent access to servers map + namespaceId string + configClient config_client.IConfigClient + namingClient naming_client.INamingClient + servers map[string]*ServerContext +} + +// VersionedMcpServerInfo MCP server information with version +type VersionedMcpServerInfo struct { + serverInfo *BasicMcpServerInfo + version string +} + +// ServerContext server context, manages all states of a single MCP server +type ServerContext struct { + mu sync.Mutex + id string + versionedMcpServerInfo *VersionedMcpServerInfo + serverChangeListener McpServerListener + configsMap map[string]*ConfigListenerWrap + serviceInfo *nacosmodel.Service + namingCallback func(services []nacosmodel.Instance, err error) +} + +// McpServerConfig MCP server configuration +type McpServerConfig struct { + ServerSpecConfig string + ToolsSpecConfig string + ServiceInfo *nacosmodel.Service + Credentials map[string]interface{} + ToolConfigs []model.ToolConfig // Converted tool configurations +} + +// ConfigListenerWrap configuration listener wrapper +type ConfigListenerWrap struct { + dataId string + group string + data string + listener func(namespace, group, dataId, data string) +} + +// BasicMcpServerInfo basic MCP server information +type BasicMcpServerInfo struct { + Name string `json:"name"` + Id string `json:"id"` + FrontProtocol string `json:"frontProtocol"` + Protocol string `json:"protocol"` +} + +// VersionsMcpServerInfo MCP server information with version details +type VersionsMcpServerInfo struct { + BasicMcpServerInfo + LatestPublishedVersion string `json:"latestPublishedVersion"` + Versions []*VersionDetail `json:"versionDetails"` +} + +// VersionDetail version detail +type VersionDetail struct { + Version string `json:"version"` + IsLatest bool `json:"is_latest"` +} + +// McpServerListener MCP server change listener +type McpServerListener func(info *McpServerConfig) + +// ==================== Constructors ==================== + +// NewMcpRegistryClient creates a new Nacos MCP registry client +func NewMcpRegistryClient(clientConfig *constant.ClientConfig, serverConfig []constant.ServerConfig, namespaceId string) (*NacosRegistryClient, error) { + logger.Infof("[dubbo-go-pixiu] nacos registry creating MCP registry client for namespace: %s", namespaceId) + + clientParam := vo.NacosClientParam{ + ClientConfig: clientConfig, + ServerConfigs: serverConfig, + } + + configClient, err := clients.NewConfigClient(clientParam) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to create config client: %v", err) + return nil, fmt.Errorf("failed to create config client: %w", err) + } + + namingClient, err := clients.NewNamingClient(clientParam) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to create naming client: %v", err) + return nil, fmt.Errorf("failed to create naming client: %w", err) + } + + client := &NacosRegistryClient{ + namespaceId: namespaceId, + configClient: configClient, + namingClient: namingClient, + servers: make(map[string]*ServerContext), + } + + return client, nil +} + +// ==================== Public API ==================== + +// ListMcpServer lists all MCP servers from Nacos MCP registry +func (n *NacosRegistryClient) ListMcpServer() ([]BasicMcpServerInfo, error) { + + configs, err := n.listMcpServerConfigs() + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to list MCP server configs: %v", err) + return nil, err + } + + result := make([]BasicMcpServerInfo, 0, len(configs)) + skippedCount := 0 + + for _, config := range configs { + + mcpServerBasicConfig, err := n.configClient.GetConfig(vo.ConfigParam{ + Group: McpServerVersionGroup, + DataId: config.DataId, + }) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to get MCP server version config (dataId: %s): %v", config.DataId, err) + continue + } + + if mcpServerBasicConfig == "" { + skippedCount++ + continue + } + + mcpServer := BasicMcpServerInfo{} + if err := json.Unmarshal([]byte(mcpServerBasicConfig), &mcpServer); err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to parse MCP server version config (dataId: %s): %v", config.DataId, err) + skippedCount++ + continue + } + + if !isMcpServerShouldBeDiscoveryForGateway(mcpServer) { + skippedCount++ + continue + } + + result = append(result, mcpServer) + } + + logger.Debugf("[dubbo-go-pixiu] nacos registry successfully listed %d MCP servers, skipped %d invalid/unsupported servers", + len(result), skippedCount) + return result, nil +} + +// ListenToMcpServer listens to MCP server configuration and backend services +func (n *NacosRegistryClient) ListenToMcpServer(id string, listener McpServerListener) error { + logger.Infof("[dubbo-go-pixiu] nacos registry starting to listen to MCP server: %s", id) + + versionConfigId := fmt.Sprintf("%s-mcp-versions.json", id) + + // Get initial version configuration (network call, may block) + serverVersionConfig, err := n.configClient.GetConfig(vo.ConfigParam{ + Group: McpServerVersionGroup, + DataId: versionConfigId, + }) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to get initial version config for MCP server %s: %v", id, err) + } + + // Version callback implementation: parse, update ctx.version and handle version changes asynchronously + versionConfigCallBack := func(namespace, group, dataId, content string) { + var info VersionsMcpServerInfo + if err := json.Unmarshal([]byte(content), &info); err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to parse version config for MCP server %s: %v", id, err) + return + } + + // Get ctx pointer (short-term global lock) + n.mu.RLock() + ctx := n.servers[id] + n.mu.RUnlock() + if ctx == nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry server context not found for MCP server %s", id) + return + } + + // Update ctx.versionedMcpServerInfo (local lock) + ctx.mu.Lock() + if ctx.versionedMcpServerInfo == nil { + ctx.versionedMcpServerInfo = &VersionedMcpServerInfo{} + } + changed := ctx.versionedMcpServerInfo.version != info.LatestPublishedVersion + ctx.versionedMcpServerInfo.serverInfo = &info.BasicMcpServerInfo + if changed { + ctx.versionedMcpServerInfo.version = info.LatestPublishedVersion + } + ctx.mu.Unlock() + + if changed { + // Asynchronously execute time-consuming/network operations to avoid blocking SDK callbacks + go func() { + n.onServerVersionChanged(ctx) + n.triggerMcpServerChange(id) + }() + } + } + + // Create server context (short-term global lock) + ctx := &ServerContext{ + id: id, + serverChangeListener: listener, + configsMap: make(map[string]*ConfigListenerWrap), + } + // Use unified system key to store version configuration + ctx.configsMap[buildSystemConfigKey(id, McpServerVersionGroup)] = &ConfigListenerWrap{ + dataId: versionConfigId, + group: McpServerVersionGroup, + listener: versionConfigCallBack, + } + + n.mu.Lock() + n.servers[id] = ctx + n.mu.Unlock() + + // Manually trigger initial callback (callback will self-lock and asynchronously handle onServerVersionChanged) + versionConfigCallBack(n.namespaceId, McpServerVersionGroup, versionConfigId, serverVersionConfig) + + // Start listening to configuration changes (network call) + err = n.configClient.ListenConfig(vo.ConfigParam{ + Group: McpServerVersionGroup, + DataId: versionConfigId, + OnChange: versionConfigCallBack, + }) + + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to start listening to MCP server %s: %v", id, err) + // If ListenConfig fails, clean up the ctx that was just added + n.mu.Lock() + delete(n.servers, id) + n.mu.Unlock() + return err + } + + return nil +} + +// CancelListenToServer cancels listening to MCP server +func (n *NacosRegistryClient) CancelListenToServer(id string) error { + logger.Infof("[dubbo-go-pixiu] nacos registry canceling listen to MCP server: %s", id) + + // First briefly take out and delete ctx to avoid holding lock for long time + n.mu.Lock() + server, exist := n.servers[id] + if !exist || server == nil { + n.mu.Unlock() + logger.Warnf("[dubbo-go-pixiu] nacos registry server context not found for MCP server %s", id) + return nil + } + delete(n.servers, id) + n.mu.Unlock() + + // Cancel all configuration listeners (not within global lock) + // collect wraps under ctx.mu to avoid concurrent map mutation + server.mu.Lock() + wraps := make([]*ConfigListenerWrap, 0, len(server.configsMap)) + for _, wrap := range server.configsMap { + if wrap != nil { + wraps = append(wraps, wrap) + } + } + server.configsMap = make(map[string]*ConfigListenerWrap) // clear map + oldService := server.serviceInfo + oldCallback := server.namingCallback + server.serviceInfo = nil + server.namingCallback = nil + server.mu.Unlock() + + for _, wrap := range wraps { + n.cancelListenToConfig(wrap) + } + + // Cancel service subscription (network operation) + if oldService != nil && oldCallback != nil { + n.namingClient.Unsubscribe(&vo.SubscribeParam{ + GroupName: oldService.GroupName, + ServiceName: oldService.Name, + SubscribeCallback: oldCallback, + }) + } + + return nil +} + +// CloseClient closes the client and cleans up all resources +func (n *NacosRegistryClient) CloseClient() { + logger.Infof("[dubbo-go-pixiu] nacos registry closing MCP registry client") + + // First collect all server IDs and release lock + n.mu.RLock() + ids := make([]string, 0, len(n.servers)) + for id := range n.servers { + ids = append(ids, id) + } + n.mu.RUnlock() + + // Now safely cancel all server listeners + for _, id := range ids { + _ = n.CancelListenToServer(id) + } + + // Close Nacos clients + if n.namingClient != nil { + n.namingClient.CloseClient() + } + if n.configClient != nil { + n.configClient.CloseClient() + } +} + +// ==================== Internal Helper Methods ==================== + +// listMcpServerConfigs paginates to get all MCP server configurations +func (n *NacosRegistryClient) listMcpServerConfigs() ([]nacosmodel.ConfigItem, error) { + + currentPageNum := 1 + result := make([]nacosmodel.ConfigItem, 0) + + for currentPageNum <= 100 { + + configPage, err := n.configClient.SearchConfig(vo.SearchConfigParam{ + Search: DefaultNacosListConfigMode, + DataId: ListMcpServerConfigIdPattern, + Group: McpServerVersionGroup, + PageNo: currentPageNum, + PageSize: DefaultNacosListConfigPageSize, + }) + + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to list MCP server configs for page %d: %v", currentPageNum, err) + return result, err + } + + if configPage == nil { + break + } + + result = append(result, configPage.PageItems...) + + if configPage.PageNumber >= configPage.PagesAvailable { + break + } + + currentPageNum++ + } + + return result, nil +} + +// onServerVersionChanged handles server version changes +func (n *NacosRegistryClient) onServerVersionChanged(ctx *ServerContext) { + // Read necessary information + ctx.mu.Lock() + if ctx.versionedMcpServerInfo == nil || ctx.versionedMcpServerInfo.serverInfo == nil { + ctx.mu.Unlock() + logger.Errorf("[dubbo-go-pixiu] nacos registry missing version/serverInfo for ctx") + return + } + id := ctx.versionedMcpServerInfo.serverInfo.Id + version := ctx.versionedMcpServerInfo.version + ctx.mu.Unlock() + + logger.Infof("[dubbo-go-pixiu] nacos registry processing version change for MCP server %s to version %s", id, version) + + configsMap := map[string]string{ + McpServerSpecGroup: fmt.Sprintf("%s-%s-mcp-server.json", id, version), + McpToolSpecGroup: fmt.Sprintf("%s-%s-mcp-tools.json", id, version), + } + + // For each group: cancel old wrap (if any), create new Listener (network) and update ctx.configsMap under ctx.mu + for group, dataId := range configsMap { + configsKey := buildSystemConfigKey(id, group) + + // If there is an old wrap, remove reference and cancel after unlocking + var oldWrap *ConfigListenerWrap + ctx.mu.Lock() + if w, ok := ctx.configsMap[configsKey]; ok && w != nil { + oldWrap = w + delete(ctx.configsMap, configsKey) + } + ctx.mu.Unlock() + + if oldWrap != nil { + n.cancelListenToConfig(oldWrap) + } + + // Create new version listener (network, may block) + newWrap, err := n.ListenToConfig(ctx, dataId, group) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to listen to config %s for MCP server %s: %v", dataId, id, err) + continue + } + + // Save new wrap under ctx.mu + ctx.mu.Lock() + ctx.configsMap[configsKey] = newWrap + ctx.mu.Unlock() + } +} + +// triggerMcpServerChange triggers MCP server change notification +func (n *NacosRegistryClient) triggerMcpServerChange(id string) { + // Get ctx pointer + n.mu.RLock() + ctx := n.servers[id] + n.mu.RUnlock() + if ctx == nil { + logger.Warnf("[dubbo-go-pixiu] nacos registry server context not found for MCP server %s when triggering change", id) + return + } + + // Generate configuration using ctx.mu protection + cfg := mapConfigMapToServerConfig(ctx) // Internally acquires ctx.mu + if cfg != nil { + // Call user callback after releasing lock + ctx.mu.Lock() + listener := ctx.serverChangeListener + ctx.mu.Unlock() + + if listener != nil { + listener(cfg) + } + } else { + logger.Warnf("[dubbo-go-pixiu] nacos registry failed to generate config for MCP server %s", id) + } +} + +// ListenToConfig listens to configuration changes +func (n *NacosRegistryClient) ListenToConfig(ctx *ServerContext, dataId string, group string) (*ConfigListenerWrap, error) { + + wrap := &ConfigListenerWrap{ + dataId: dataId, + group: group, + } + + // Create configListener (only use ctx.mu for local state updates, put time-consuming operations outside lock) + configListener := func(namespace, group, dataId, data string) { + // Quickly atomically update wrap.data (within ctx.mu) + ctx.mu.Lock() + changed := wrap.data != data + if changed { + wrap.data = data + } + ctx.mu.Unlock() + + if !changed { + return + } + + // Execute time-consuming or network-related logic outside lock + if group == McpToolSpecGroup { + n.resetNacosTemplateConfigs(ctx, wrap) + } else if group == McpServerSpecGroup { + n.refreshServiceListenerIfNeeded(ctx, data) + } + + n.triggerMcpServerChange(ctx.id) + } + + // Get initial configuration (network call) + config, err := n.configClient.GetConfig(vo.ConfigParam{ + DataId: dataId, + Group: group, + }) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to get initial config for MCP server %s (dataId: %s, group: %s): %v", + ctx.id, dataId, group, err) + return nil, err + } + + wrap.listener = configListener + wrap.data = config + + // Process initial configuration (placed outside lock, as these operations may perform network calls) + if group == McpToolSpecGroup { + n.resetNacosTemplateConfigs(ctx, wrap) + } else if group == McpServerSpecGroup { + n.refreshServiceListenerIfNeeded(ctx, config) + } + + // Start listening to configuration changes (network call) + err = n.configClient.ListenConfig(vo.ConfigParam{ + DataId: dataId, + Group: group, + OnChange: configListener, + }) + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to start listening to config for MCP server %s (dataId: %s, group: %s): %v", + ctx.id, dataId, group, err) + return nil, err + } + + return wrap, nil +} + +// cancelListenToConfig cancels configuration listening +func (n *NacosRegistryClient) cancelListenToConfig(wrap *ConfigListenerWrap) error { + if wrap == nil { + return nil + } + + err := n.configClient.CancelListenConfig(vo.ConfigParam{ + DataId: wrap.dataId, + Group: wrap.group, + OnChange: wrap.listener, + }) + + if err != nil { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to cancel config listener (dataId: %s, group: %s): %v", + wrap.dataId, wrap.group, err) + } + + return err +} + +// ==================== Configuration Conversion ==================== + +// mapConfigMapToServerConfig converts configuration map to MCP server configuration +// Internally acquires ctx.mu to protect reading ctx fields +func mapConfigMapToServerConfig(ctx *ServerContext) *McpServerConfig { + + ctx.mu.Lock() + defer ctx.mu.Unlock() + + result := &McpServerConfig{ + Credentials: make(map[string]interface{}), + } + + // Process system configurations + for key, data := range ctx.configsMap { + if data == nil { + continue + } + + if isSystemConfigKey(key) { + _, _, group, _ := parseConfigKey(key) + if group == McpServerSpecGroup { + result.ServerSpecConfig = data.data + } else if group == McpToolSpecGroup { + result.ToolsSpecConfig = data.data + } + } else if isCredentialConfigKey(key) { + _, _, group, dataId := parseConfigKey(key) + credentialId := group + "_" + dataId + + var credData interface{} + if err := json.Unmarshal([]byte(data.data), &credData); err != nil { + result.Credentials[credentialId] = data.data + } else { + result.Credentials[credentialId] = credData + } + } + } + + // Directly reference serviceInfo (caller must note that this object may be modified) + result.ServiceInfo = ctx.serviceInfo + + // Convert Nacos format to ToolConfig + if result.ToolsSpecConfig != "" { + toolsSpec := &ToolsSpec{} + if err := json.Unmarshal([]byte(result.ToolsSpecConfig), toolsSpec); err == nil { + toolConfigs, err := ConvertNacosToolsToToolConfig(toolsSpec) + if err == nil { + result.ToolConfigs = toolConfigs + } else { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to convert tools spec for MCP server %s: %v", ctx.id, err) + } + } else { + logger.Errorf("[dubbo-go-pixiu] nacos registry failed to parse tools spec JSON for MCP server %s: %v", ctx.id, err) + } + } + + return result +} + +// parseTemplatePlaceholders parses template placeholders and returns replaced content and required (dataId,group) list +func parseTemplatePlaceholders(content string) (string, [][2]string) { + all := templateRegex.FindAllStringSubmatch(content, -1) + if len(all) == 0 { + return content, nil + } + placeholders := make([][2]string, 0, len(all)) + newContent := content + seen := make(map[string]bool) + for _, m := range all { + if len(m) < 2 { + continue + } + p := m[1] // dataId/group + parts := strings.Split(p, "/") + if len(parts) != 2 { + continue + } + dataId := strings.TrimSpace(parts[0]) + group := strings.TrimSpace(parts[1]) + key := group + "::" + dataId + if seen[key] { + // replace duplicates as well + newContent = strings.ReplaceAll(newContent, "${nacos."+p+"}", ".config.credentials."+group+"_"+dataId) + continue + } + seen[key] = true + placeholders = append(placeholders, [2]string{dataId, group}) + newContent = strings.ReplaceAll(newContent, "${nacos."+p+"}", ".config.credentials."+group+"_"+dataId) + } + return newContent, placeholders +} + +// replaceTemplateAndExactConfigsItems has been split, old logic is called by resetNacosTemplateConfigs +// resetNacosTemplateConfigs resets Nacos template configurations +func (n *NacosRegistryClient) resetNacosTemplateConfigs(ctx *ServerContext, config *ConfigListenerWrap) { + // Parse out placeholders and replaced text + newContent, placeholders := parseTemplatePlaceholders(config.data) + + // For each placeholder, create ListenToConfig (network call), collect newWraps + newWraps := make(map[string]*ConfigListenerWrap) + for _, ph := range placeholders { + dataId := ph[0] + group := ph[1] Review Comment: does access to ph have out of range problem? ########## pkg/adapter/mcpserver/registry/nacos/converter.go: ########## @@ -0,0 +1,214 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package nacos + +import ( + "encoding/json" + "fmt" + "net/url" + "regexp" + "strings" +) + +import ( + "github.com/apache/dubbo-go-pixiu/pkg/logger" + "github.com/apache/dubbo-go-pixiu/pkg/model" +) + +// ConvertNacosToolsToToolConfig converts Nacos Tools to the Filter's ToolConfig +func ConvertNacosToolsToToolConfig(toolsSpec *ToolsSpec) ([]model.ToolConfig, error) { + var toolConfigs []model.ToolConfig + + for _, nacosTool := range toolsSpec.Tools { + meta := toolsSpec.ToolsMeta[nacosTool.Name] + if !meta.Enabled { + continue + } + + // Extract json-go-template + templateData, ok := meta.Templates["json-go-template"] + if !ok { + logger.Warnf("Tool %s has no json-go-template, skipping", nacosTool.Name) + continue + } + + toolConfig, err := convertSingleTool(nacosTool, templateData) + if err != nil { + return nil, fmt.Errorf("failed to convert tool %s: %w", nacosTool.Name, err) + } + + toolConfigs = append(toolConfigs, toolConfig) + } + + return toolConfigs, nil +} + +func convertSingleTool(nacosTool NacosTool, templateData interface{}) (model.ToolConfig, error) { + // Parse template data + templateBytes, err := json.Marshal(templateData) + if err != nil { + return model.ToolConfig{}, err + } + + var template JsonGoTemplate + if err := json.Unmarshal(templateBytes, &template); err != nil { + return model.ToolConfig{}, err + } + + toolConfig := model.ToolConfig{ + Name: nacosTool.Name, + Description: nacosTool.Description, + Cluster: nacosTool.Name, // Directly use the tool name as the cluster name + BackendURL: template.RequestTemplate.URL, + Request: model.RequestConfig{ + Method: template.RequestTemplate.Method, + Path: extractPathFromURL(template.RequestTemplate.URL), + Headers: convertHeaders(template.RequestTemplate.Headers), + }, + Args: func() []model.ArgConfig { + args, err := convertInputSchemaToArgs(nacosTool.InputSchema, template.RequestTemplate) + if err != nil { + logger.Warnf("Failed to convert args for tool %s: %v", nacosTool.Name, err) + return []model.ArgConfig{} + } + return args + }(), + } + + return toolConfig, nil +} + +func extractPathFromURL(raw string) string { + if raw == "" { + return "/" + } + s := strings.TrimSpace(raw) + + // Prefer url.Parse to extract the path + if i := strings.Index(s, "://"); i >= 0 { Review Comment: ```suggestion if i := strings.Index(s, constant.ProtocolSlash); i >= 0 { ``` ########## pkg/adapter/mcpserver/registry/nacos/converter.go: ########## @@ -0,0 +1,214 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package nacos + +import ( + "encoding/json" + "fmt" + "net/url" + "regexp" + "strings" +) + +import ( + "github.com/apache/dubbo-go-pixiu/pkg/logger" + "github.com/apache/dubbo-go-pixiu/pkg/model" +) + +// ConvertNacosToolsToToolConfig converts Nacos Tools to the Filter's ToolConfig +func ConvertNacosToolsToToolConfig(toolsSpec *ToolsSpec) ([]model.ToolConfig, error) { + var toolConfigs []model.ToolConfig + + for _, nacosTool := range toolsSpec.Tools { + meta := toolsSpec.ToolsMeta[nacosTool.Name] + if !meta.Enabled { + continue + } + + // Extract json-go-template + templateData, ok := meta.Templates["json-go-template"] + if !ok { + logger.Warnf("Tool %s has no json-go-template, skipping", nacosTool.Name) + continue + } + + toolConfig, err := convertSingleTool(nacosTool, templateData) + if err != nil { + return nil, fmt.Errorf("failed to convert tool %s: %w", nacosTool.Name, err) + } + + toolConfigs = append(toolConfigs, toolConfig) + } + + return toolConfigs, nil +} + +func convertSingleTool(nacosTool NacosTool, templateData interface{}) (model.ToolConfig, error) { + // Parse template data + templateBytes, err := json.Marshal(templateData) + if err != nil { + return model.ToolConfig{}, err + } + + var template JsonGoTemplate + if err := json.Unmarshal(templateBytes, &template); err != nil { + return model.ToolConfig{}, err + } + + toolConfig := model.ToolConfig{ + Name: nacosTool.Name, + Description: nacosTool.Description, + Cluster: nacosTool.Name, // Directly use the tool name as the cluster name + BackendURL: template.RequestTemplate.URL, + Request: model.RequestConfig{ + Method: template.RequestTemplate.Method, + Path: extractPathFromURL(template.RequestTemplate.URL), + Headers: convertHeaders(template.RequestTemplate.Headers), + }, + Args: func() []model.ArgConfig { + args, err := convertInputSchemaToArgs(nacosTool.InputSchema, template.RequestTemplate) + if err != nil { + logger.Warnf("Failed to convert args for tool %s: %v", nacosTool.Name, err) + return []model.ArgConfig{} + } + return args + }(), + } + + return toolConfig, nil +} + +func extractPathFromURL(raw string) string { + if raw == "" { + return "/" + } + s := strings.TrimSpace(raw) + + // Prefer url.Parse to extract the path + if i := strings.Index(s, "://"); i >= 0 { + if u, err := url.Parse(s); err == nil { + path := u.Path + if path == "" { + path = "/" Review Comment: ```suggestion path = constant.PathSlash ``` ########## pkg/model/mcpserver.go: ########## @@ -0,0 +1,225 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package model + +import ( + "regexp" + "time" +) + +// McpServerConfig MCP Server Filter configuration +type McpServerConfig struct { + ServerInfo ServerInfo `yaml:"server_info" json:"server_info"` + Endpoint string `yaml:"endpoint" json:"endpoint" default:"/mcp"` + Tools []ToolConfig `yaml:"tools,omitempty" json:"tools,omitempty"` + Resources []ResourceConfig `yaml:"resources,omitempty" json:"resources,omitempty"` + ResourceTemplates []ResourceTemplateConfig `yaml:"resource_templates,omitempty" json:"resource_templates,omitempty"` + Prompts []PromptConfig `yaml:"prompts,omitempty" json:"prompts,omitempty"` +} + +// ServerInfo server information +type ServerInfo struct { + Name string `yaml:"name" json:"name" default:"Pixiu MCP Server"` + Version string `yaml:"version" json:"version" default:"1.0.0"` + Description string `yaml:"description,omitempty" json:"description,omitempty" default:"MCP Server powered by Apache Dubbo-go-pixiu"` + Instructions string `yaml:"instructions,omitempty" json:"instructions,omitempty" default:"Use the provided tools to interact with backend services."` +} + +// ToolConfig tool configuration +type ToolConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description" json:"description"` + Cluster string `yaml:"cluster" json:"cluster"` + BackendURL string `yaml:"backend_url,omitempty" json:"backend_url,omitempty"` + Request RequestConfig `yaml:"request" json:"request"` + Args []ArgConfig `yaml:"args,omitempty" json:"args,omitempty"` +} + +// RequestConfig request configuration +type RequestConfig struct { + Method string `yaml:"method" json:"method" default:"GET"` + Path string `yaml:"path" json:"path"` + Headers map[string]string `yaml:"headers,omitempty" json:"headers,omitempty"` + Timeout string `yaml:"timeout,omitempty" json:"timeout,omitempty" default:"30s"` +} + +// ArgConfig parameter configuration (simplified) +type ArgConfig struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + In string `yaml:"in" json:"in"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` +} + +// ResourceConfig resource configuration +type ResourceConfig struct { + Name string `yaml:"name" json:"name"` + URI string `yaml:"uri" json:"uri"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Source ResourceSource `yaml:"source" json:"source"` +} + +// ResourceSource resource source configuration (simplified) +type ResourceSource struct { + Type string `yaml:"type" json:"type"` + Path string `yaml:"path,omitempty" json:"path,omitempty"` // for file type + URL string `yaml:"url,omitempty" json:"url,omitempty"` // for url type + Content string `yaml:"content,omitempty" json:"content,omitempty"` // for inline type + Template string `yaml:"template,omitempty" json:"template,omitempty"` // for template type +} + +// ResourceTemplateConfig resource template configuration +type ResourceTemplateConfig struct { + Name string `yaml:"name" json:"name"` + URITemplate string `yaml:"uri_template" json:"uri_template"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Parameters []ResourceTemplateParameter `yaml:"parameters,omitempty" json:"parameters,omitempty"` + Annotations *ResourceTemplateAnnotations `yaml:"annotations,omitempty" json:"annotations,omitempty"` +} + +// ResourceTemplateParameter resource template parameter +type ResourceTemplateParameter struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty" default:"false"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` +} + +// ResourceTemplateAnnotations resource template annotations +type ResourceTemplateAnnotations struct { + Audience []string `yaml:"audience,omitempty" json:"audience,omitempty"` + Priority *float64 `yaml:"priority,omitempty" json:"priority,omitempty"` + LastModified string `yaml:"last_modified,omitempty" json:"last_modified,omitempty"` +} + +// ComputedParameter computed parameter (for internal processing, simplified) +type ComputedParameter struct { + Name string + Type string + In string + Description string + Required bool + Enum []string + Default any +} + +// PromptConfig prompt configuration +type PromptConfig struct { + Name string `yaml:"name" json:"name"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Arguments []PromptArgumentConfig `yaml:"arguments,omitempty" json:"arguments,omitempty"` + Messages []PromptMessageConfig `yaml:"messages" json:"messages"` +} + +// PromptArgumentConfig prompt argument configuration +type PromptArgumentConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` +} + +// PromptMessageConfig prompt message configuration +type PromptMessageConfig struct { + Role string `yaml:"role" json:"role"` // "user" or "assistant" + Content string `yaml:"content" json:"content"` +} + +// RegistryConfig 注册表配置 Review Comment: use English ########## pkg/model/mcpserver.go: ########## @@ -0,0 +1,225 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package model + +import ( + "regexp" + "time" +) + +// McpServerConfig MCP Server Filter configuration +type McpServerConfig struct { + ServerInfo ServerInfo `yaml:"server_info" json:"server_info"` + Endpoint string `yaml:"endpoint" json:"endpoint" default:"/mcp"` + Tools []ToolConfig `yaml:"tools,omitempty" json:"tools,omitempty"` + Resources []ResourceConfig `yaml:"resources,omitempty" json:"resources,omitempty"` + ResourceTemplates []ResourceTemplateConfig `yaml:"resource_templates,omitempty" json:"resource_templates,omitempty"` + Prompts []PromptConfig `yaml:"prompts,omitempty" json:"prompts,omitempty"` +} + +// ServerInfo server information +type ServerInfo struct { + Name string `yaml:"name" json:"name" default:"Pixiu MCP Server"` + Version string `yaml:"version" json:"version" default:"1.0.0"` + Description string `yaml:"description,omitempty" json:"description,omitempty" default:"MCP Server powered by Apache Dubbo-go-pixiu"` + Instructions string `yaml:"instructions,omitempty" json:"instructions,omitempty" default:"Use the provided tools to interact with backend services."` +} + +// ToolConfig tool configuration +type ToolConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description" json:"description"` + Cluster string `yaml:"cluster" json:"cluster"` + BackendURL string `yaml:"backend_url,omitempty" json:"backend_url,omitempty"` + Request RequestConfig `yaml:"request" json:"request"` + Args []ArgConfig `yaml:"args,omitempty" json:"args,omitempty"` +} + +// RequestConfig request configuration +type RequestConfig struct { + Method string `yaml:"method" json:"method" default:"GET"` + Path string `yaml:"path" json:"path"` + Headers map[string]string `yaml:"headers,omitempty" json:"headers,omitempty"` + Timeout string `yaml:"timeout,omitempty" json:"timeout,omitempty" default:"30s"` +} + +// ArgConfig parameter configuration (simplified) +type ArgConfig struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + In string `yaml:"in" json:"in"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` +} + +// ResourceConfig resource configuration +type ResourceConfig struct { + Name string `yaml:"name" json:"name"` + URI string `yaml:"uri" json:"uri"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Source ResourceSource `yaml:"source" json:"source"` +} + +// ResourceSource resource source configuration (simplified) +type ResourceSource struct { + Type string `yaml:"type" json:"type"` + Path string `yaml:"path,omitempty" json:"path,omitempty"` // for file type + URL string `yaml:"url,omitempty" json:"url,omitempty"` // for url type + Content string `yaml:"content,omitempty" json:"content,omitempty"` // for inline type + Template string `yaml:"template,omitempty" json:"template,omitempty"` // for template type +} + +// ResourceTemplateConfig resource template configuration +type ResourceTemplateConfig struct { + Name string `yaml:"name" json:"name"` + URITemplate string `yaml:"uri_template" json:"uri_template"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + MIMEType string `yaml:"mime_type,omitempty" json:"mime_type,omitempty"` + Parameters []ResourceTemplateParameter `yaml:"parameters,omitempty" json:"parameters,omitempty"` + Annotations *ResourceTemplateAnnotations `yaml:"annotations,omitempty" json:"annotations,omitempty"` +} + +// ResourceTemplateParameter resource template parameter +type ResourceTemplateParameter struct { + Name string `yaml:"name" json:"name"` + Type string `yaml:"type" json:"type" default:"string"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty" default:"false"` + Enum []string `yaml:"enum,omitempty" json:"enum,omitempty"` + Default any `yaml:"default,omitempty" json:"default,omitempty"` +} + +// ResourceTemplateAnnotations resource template annotations +type ResourceTemplateAnnotations struct { + Audience []string `yaml:"audience,omitempty" json:"audience,omitempty"` + Priority *float64 `yaml:"priority,omitempty" json:"priority,omitempty"` + LastModified string `yaml:"last_modified,omitempty" json:"last_modified,omitempty"` +} + +// ComputedParameter computed parameter (for internal processing, simplified) +type ComputedParameter struct { + Name string + Type string + In string + Description string + Required bool + Enum []string + Default any +} + +// PromptConfig prompt configuration +type PromptConfig struct { + Name string `yaml:"name" json:"name"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Arguments []PromptArgumentConfig `yaml:"arguments,omitempty" json:"arguments,omitempty"` + Messages []PromptMessageConfig `yaml:"messages" json:"messages"` +} + +// PromptArgumentConfig prompt argument configuration +type PromptArgumentConfig struct { + Name string `yaml:"name" json:"name"` + Description string `yaml:"description,omitempty" json:"description,omitempty"` + Required bool `yaml:"required,omitempty" json:"required,omitempty"` +} + +// PromptMessageConfig prompt message configuration +type PromptMessageConfig struct { + Role string `yaml:"role" json:"role"` // "user" or "assistant" + Content string `yaml:"content" json:"content"` +} + +// RegistryConfig 注册表配置 +type RegistryConfig struct { + ToolConfigs map[string]ToolConfig `yaml:"toolConfigs"` + ResourceConfigs map[string]ResourceConfig `yaml:"resourceConfigs"` + PromptConfigs map[string]PromptConfig `yaml:"promptConfigs"` + LastUpdated time.Time `yaml:"lastUpdated"` +} + +// GetAllParameters gets all parameters of the tool +func (tc *ToolConfig) GetAllParameters() ([]ComputedParameter, error) { + var allParams []ComputedParameter + + // 1. Automatically extract path parameters + pathParams := GetPathParameterNames(tc.Request.Path) + for _, paramName := range pathParams { + // Find corresponding arg configuration + var argConfig *ArgConfig + for _, arg := range tc.Args { + if arg.Name == paramName && arg.In == "path" { + argConfig = &arg + break + } + } + + // Create computed parameter + computed := ComputedParameter{ + Name: paramName, + Type: "string", // Default type + In: "path", + Required: true, // Path parameters are always required + } + + // Apply arg configuration + if argConfig != nil { + computed.Type = argConfig.Type + computed.Description = argConfig.Description + // Simplified: removed Pattern and Format fields + } + + allParams = append(allParams, computed) + } + + // 2. Add non-path parameters + for _, arg := range tc.Args { + if arg.In != "path" { + computed := ComputedParameter{ + Name: arg.Name, + Type: arg.Type, + In: arg.In, + Description: arg.Description, + Required: arg.Required, + Enum: arg.Enum, + Default: arg.Default, + // Simplified: removed complex validation fields Review Comment: Delete redundant comment -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: notifications-unsubscr...@dubbo.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: notifications-unsubscr...@dubbo.apache.org For additional commands, e-mail: notifications-h...@dubbo.apache.org
