From ef3000e83bf2f308306d396d0ae29e3df7dc826c Mon Sep 17 00:00:00 2001 From: Rui Fu Date: Fri, 9 May 2025 00:05:54 +0800 Subject: [PATCH] fix features flag --- pkg/mcp/kafka_admin_connect_tools.go | 2 +- pkg/mcp/kafka_admin_sr_tools.go | 2 +- pkg/mcp/pulsar_admin_brokers_stats_tools.go | 2 +- pkg/mcp/pulsar_admin_brokers_tools.go | 2 +- pkg/mcp/pulsar_admin_cluster_tools.go | 2 +- pkg/mcp/pulsar_admin_functions_tools.go | 2 +- pkg/mcp/pulsar_admin_functions_worker_tools.go | 2 +- pkg/mcp/pulsar_admin_namespace_policy_tools.go | 2 +- pkg/mcp/pulsar_admin_namespace_tools.go | 2 +- .../pulsar_admin_nsisolationpolicy_tools.go | 2 +- pkg/mcp/pulsar_admin_packages_tools.go | 18 +++++++++--------- pkg/mcp/pulsar_admin_resourcequotas_tools.go | 2 +- pkg/mcp/pulsar_admin_schemas_tools.go | 2 +- pkg/mcp/pulsar_admin_sinks_tools.go | 2 +- pkg/mcp/pulsar_admin_sources_tools.go | 2 +- pkg/mcp/pulsar_admin_subscription_tools.go | 2 +- pkg/mcp/pulsar_admin_tenant_tools.go | 2 +- pkg/mcp/pulsar_admin_topic_policy_tools.go | 2 +- pkg/mcp/pulsar_admin_topic_tools.go | 2 +- 19 files changed, 27 insertions(+), 27 deletions(-) diff --git a/pkg/mcp/kafka_admin_connect_tools.go b/pkg/mcp/kafka_admin_connect_tools.go index ff76dac..7b4c9cf 100644 --- a/pkg/mcp/kafka_admin_connect_tools.go +++ b/pkg/mcp/kafka_admin_connect_tools.go @@ -30,7 +30,7 @@ import ( ) func KafkaAdminAddKafkaConnectTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeatureKafkaAdminKafkaConnect)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllKafka)) { + if !slices.Contains(features, string(FeatureKafkaAdminKafkaConnect)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllKafka)) && !slices.Contains(features, string(FeatureKafkaAdmin)) { return } diff --git a/pkg/mcp/kafka_admin_sr_tools.go b/pkg/mcp/kafka_admin_sr_tools.go index 06eac90..cdec21a 100644 --- a/pkg/mcp/kafka_admin_sr_tools.go +++ b/pkg/mcp/kafka_admin_sr_tools.go @@ -31,7 +31,7 @@ import ( ) func KafkaAdminAddSchemaRegistryTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeatureKafkaAdminSchemaRegistry)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllKafka)) { + if !slices.Contains(features, string(FeatureKafkaAdminSchemaRegistry)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllKafka)) && !slices.Contains(features, string(FeatureKafkaAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_brokers_stats_tools.go b/pkg/mcp/pulsar_admin_brokers_stats_tools.go index 8b692aa..6b04e51 100644 --- a/pkg/mcp/pulsar_admin_brokers_stats_tools.go +++ b/pkg/mcp/pulsar_admin_brokers_stats_tools.go @@ -31,7 +31,7 @@ import ( // PulsarAdminAddBrokerStatsTools adds broker-stats related tools to the MCP server func PulsarAdminAddBrokerStatsTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminBrokersStatus)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminBrokersStatus)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Unified broker stats tool diff --git a/pkg/mcp/pulsar_admin_brokers_tools.go b/pkg/mcp/pulsar_admin_brokers_tools.go index bebeecc..f343f5d 100644 --- a/pkg/mcp/pulsar_admin_brokers_tools.go +++ b/pkg/mcp/pulsar_admin_brokers_tools.go @@ -31,7 +31,7 @@ import ( // PulsarAdminAddBrokersTools adds broker-related tools to the MCP server func PulsarAdminAddBrokersTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminBrokers)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminBrokers)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Create a single unified broker tool to replace multiple individual tools diff --git a/pkg/mcp/pulsar_admin_cluster_tools.go b/pkg/mcp/pulsar_admin_cluster_tools.go index f670b05..07ac31e 100644 --- a/pkg/mcp/pulsar_admin_cluster_tools.go +++ b/pkg/mcp/pulsar_admin_cluster_tools.go @@ -32,7 +32,7 @@ import ( // PulsarAdminAddClusterTools adds cluster-related tools to the MCP server func PulsarAdminAddClusterTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminClusters)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminClusters)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Create a single unified cluster tool diff --git a/pkg/mcp/pulsar_admin_functions_tools.go b/pkg/mcp/pulsar_admin_functions_tools.go index fc97bdf..3af0a25 100644 --- a/pkg/mcp/pulsar_admin_functions_tools.go +++ b/pkg/mcp/pulsar_admin_functions_tools.go @@ -32,7 +32,7 @@ import ( // PulsarAdminAddFunctionsTools adds a unified function-related tool to the MCP server func PulsarAdminAddFunctionsTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminFunctions)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminFunctions)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Create a single unified tool for all function operations diff --git a/pkg/mcp/pulsar_admin_functions_worker_tools.go b/pkg/mcp/pulsar_admin_functions_worker_tools.go index 3ba319f..e5ed9db 100644 --- a/pkg/mcp/pulsar_admin_functions_worker_tools.go +++ b/pkg/mcp/pulsar_admin_functions_worker_tools.go @@ -31,7 +31,7 @@ import ( // PulsarAdminAddFunctionsWorkerTools adds functions worker-related tools to the MCP server func PulsarAdminAddFunctionsWorkerTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminFunctionsWorker)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminFunctionsWorker)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Create a single unified functions worker tool diff --git a/pkg/mcp/pulsar_admin_namespace_policy_tools.go b/pkg/mcp/pulsar_admin_namespace_policy_tools.go index 5b3beac..eb655e0 100644 --- a/pkg/mcp/pulsar_admin_namespace_policy_tools.go +++ b/pkg/mcp/pulsar_admin_namespace_policy_tools.go @@ -35,7 +35,7 @@ import ( // PulsarAdminAddNamespacePolicyTools adds namespace policy-related tools to the MCP server func PulsarAdminAddNamespacePolicyTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminNamespacePolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminNamespacePolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } namespaceGetPoliciesTool := mcp.NewTool("pulsar_admin_namespace_policy_get", diff --git a/pkg/mcp/pulsar_admin_namespace_tools.go b/pkg/mcp/pulsar_admin_namespace_tools.go index f4b6279..1539324 100644 --- a/pkg/mcp/pulsar_admin_namespace_tools.go +++ b/pkg/mcp/pulsar_admin_namespace_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddNamespaceTools adds namespace-related tools to the MCP server func PulsarAdminAddNamespaceTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminNamespaces)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminNamespaces)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_nsisolationpolicy_tools.go b/pkg/mcp/pulsar_admin_nsisolationpolicy_tools.go index 9a1bdf6..122d387 100644 --- a/pkg/mcp/pulsar_admin_nsisolationpolicy_tools.go +++ b/pkg/mcp/pulsar_admin_nsisolationpolicy_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddNsIsolationPolicyTools adds namespace isolation policy related tools to the MCP server func PulsarAdminAddNsIsolationPolicyTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminNsIsolationPolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminNsIsolationPolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_packages_tools.go b/pkg/mcp/pulsar_admin_packages_tools.go index cacd7bd..31b043a 100644 --- a/pkg/mcp/pulsar_admin_packages_tools.go +++ b/pkg/mcp/pulsar_admin_packages_tools.go @@ -1,12 +1,12 @@ -// 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 +// Licensed to Elasticsearch B.V. under one or more contributor +// license agreements. See the NOTICE file distributed with +// this work for additional information regarding copyright +// ownership. Elasticsearch B.V. 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 +// 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 @@ -32,7 +32,7 @@ import ( // PulsarAdminAddPackagesTools adds package-related tools to the MCP server func PulsarAdminAddPackagesTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminPackages)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminPackages)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_resourcequotas_tools.go b/pkg/mcp/pulsar_admin_resourcequotas_tools.go index e330f16..896cf2d 100644 --- a/pkg/mcp/pulsar_admin_resourcequotas_tools.go +++ b/pkg/mcp/pulsar_admin_resourcequotas_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddResourceQuotasTools adds resource quotas-related tools to the MCP server func PulsarAdminAddResourceQuotasTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminResourceQuotas)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminResourceQuotas)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } // Add unified resource quotas management tool diff --git a/pkg/mcp/pulsar_admin_schemas_tools.go b/pkg/mcp/pulsar_admin_schemas_tools.go index fca817e..e986e7c 100644 --- a/pkg/mcp/pulsar_admin_schemas_tools.go +++ b/pkg/mcp/pulsar_admin_schemas_tools.go @@ -35,7 +35,7 @@ import ( // PulsarAdminAddSchemasTools adds schema-related tools to the MCP server func PulsarAdminAddSchemasTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminSchemas)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminSchemas)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_sinks_tools.go b/pkg/mcp/pulsar_admin_sinks_tools.go index f2729b2..b6c138d 100644 --- a/pkg/mcp/pulsar_admin_sinks_tools.go +++ b/pkg/mcp/pulsar_admin_sinks_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddSinksTools adds a unified sink-related tool to the MCP server func PulsarAdminAddSinksTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminSinks)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminSinks)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_sources_tools.go b/pkg/mcp/pulsar_admin_sources_tools.go index 745e25c..157655e 100644 --- a/pkg/mcp/pulsar_admin_sources_tools.go +++ b/pkg/mcp/pulsar_admin_sources_tools.go @@ -32,7 +32,7 @@ import ( // PulsarAdminAddSourcesTools adds a unified source-related tool to the MCP server func PulsarAdminAddSourcesTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminSources)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminSources)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_subscription_tools.go b/pkg/mcp/pulsar_admin_subscription_tools.go index dfcd5ed..698d6e1 100644 --- a/pkg/mcp/pulsar_admin_subscription_tools.go +++ b/pkg/mcp/pulsar_admin_subscription_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddSubscriptionTools adds subscription-related tools to the MCP server func PulsarAdminAddSubscriptionTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminSubscriptions)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminSubscriptions)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_tenant_tools.go b/pkg/mcp/pulsar_admin_tenant_tools.go index de365da..302e48c 100644 --- a/pkg/mcp/pulsar_admin_tenant_tools.go +++ b/pkg/mcp/pulsar_admin_tenant_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddTenantTools adds tenant-related tools to the MCP server func PulsarAdminAddTenantTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminTenants)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminTenants)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_topic_policy_tools.go b/pkg/mcp/pulsar_admin_topic_policy_tools.go index 1325793..a3dc8fb 100644 --- a/pkg/mcp/pulsar_admin_topic_policy_tools.go +++ b/pkg/mcp/pulsar_admin_topic_policy_tools.go @@ -32,7 +32,7 @@ import ( // PulsarAdminAddTopicPolicyTools adds topic policy-related tools to the MCP server func PulsarAdminAddTopicPolicyTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminTopicPolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminTopicPolicy)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return } diff --git a/pkg/mcp/pulsar_admin_topic_tools.go b/pkg/mcp/pulsar_admin_topic_tools.go index bf2258e..805972c 100644 --- a/pkg/mcp/pulsar_admin_topic_tools.go +++ b/pkg/mcp/pulsar_admin_topic_tools.go @@ -33,7 +33,7 @@ import ( // PulsarAdminAddTopicTools adds topic-related tools to the MCP server func PulsarAdminAddTopicTools(s *server.MCPServer, readOnly bool, features []string) { - if !slices.Contains(features, string(FeaturePulsarAdminTopics)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) { + if !slices.Contains(features, string(FeaturePulsarAdminTopics)) && !slices.Contains(features, string(FeatureAll)) && !slices.Contains(features, string(FeatureAllPulsar)) && !slices.Contains(features, string(FeaturePulsarAdmin)) { return }