xk6-mcp

xk6-mcp

用于测试 Model Context Protocol (MCP) 服务的 k6 扩展。

注意

此扩展处于实验阶段，且未获得 Grafana Labs 的官方支持。

安装

1. 首先，确保已安装 xk6：

go install go.k6.io/xk6/cmd/xk6@latest

2. 使用 xk6-mcp 扩展构建 k6 二进制文件：

xk6 build --with github.com/dgzlopes/xk6-mcp

3. 在测试脚本顶部导入 mcp 模块：

import mcp from 'k6/x/mcp';

4. 构建的二进制文件将位于当前目录中。你可以将其移至 PATH，或直接使用：

./k6 run script.js

示例

⚠️ 此示例依赖于 mcp-example-server
你可以从发布页面下载最新版本。

import mcp from 'k6/x/mcp';

export default function () {
  // Initialize MCP Client with stdio transport
  const client = new mcp.StdioClient({
    path: './mcp-example-server',
  });

  // Check connection to MCP server
  console.log('MCP server running:', client.ping());

  // List all available tools
  console.log('Tools available:');
  const tools = client.listAllTools().tools;
  tools.forEach(tool => console.log(`  - ${tool.name}`));

  // List all available resources
  console.log('Resources available:');
  const resources = client.listAllResources().resources;
  resources.forEach(resource => console.log(`  - ${resource.uri}`));

  // List all available prompts
  console.log('Prompts available:');
  const prompts = client.listAllPrompts().prompts;
  prompts.forEach(prompt => console.log(`  - ${prompt.name}`));

  // Call a sample tool
  const toolResult = client.callTool({ name: 'greet', arguments: { name: 'Grafana k6' } });
  console.log(`Greet tool response: "${toolResult.content[0].text}"`);

  // Read a sample resource
  const resourceContent = client.readResource({ uri: 'embedded:info' });
  console.log(`Resource content: ${resourceContent.contents[0].text}`);

  // Get a sample prompt
  const prompt = client.getPrompt({ name: 'greet' });
  console.log(`Prompt: ${prompt.messages[0].content.text}`);
}

常见问题

非 stdio 传输方式呢？

SSE 和 Streamable HTTP 也都受支持！

你可以这样使用：

// SSE
const client = new mcp.SSEClient({
    base_url: 'http://localhost:3002',
});

// Streamable HTTP
const client = new mcp.StreamableHTTPClient({
    base_url: 'http://localhost:3001',
});

分页呢？

此扩展提供两种方式来列出资源、工具和提示词：

// With All: Handles pagination automatically
const allTools = client.listAllTools();
const allResources = client.listAllResources();
const allPrompts = client.listAllPrompts();

// Without All: Requires manual pagination
const first = client.listTools()
const second = client.listTools({ cursor: first.next_cursor });

指标呢？

此扩展会自动跟踪每个 MCP 操作的 RED 风格指标：

• mcp_request_duration（trend）：每个 MCP 请求的耗时（毫秒）。
• mcp_request_count（counter）：发出的 MCP 请求数量。
• mcp_request_errors（counter）：失败的 MCP 请求数量。

每个指标都带有以下标签：

• method：调用的 MCP 方法（例如，GetPrompt、ListTools）。