xk6-mcp

用于测试 模型上下文协议 (MCP) 服务器的 k6 扩展。

安装

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

示例

import mcp from 'k6/x/mcp';

// 使用 stdio 传输初始化 MCP 客户端
const client = new mcp.StdioClient({
  path: 'npx',
  env: [],
  args: ['-y', '@modelcontextprotocol/server-everything', '/tmp'],
});

export default function () {
  console.log('Checking MCP server status...');
  console.log('MCP server running:', client.ping());

  // 列出可用的工具
  console.log('Tools available:');
  const tools = client.listTools().tools;
  tools.forEach(tool => console.log(`  - ${tool.name}`));

  // 列出可用的资源
  console.log('Resources available:');
  const resources = client.listResources().resources;
  resources.forEach(resource => console.log(`  - ${resource.uri}`));

  // 列出可用的提示
  console.log('Prompts available:');
  const prompts = client.listPrompts().prompts;
  prompts.forEach(prompt => console.log(`  - ${prompt.name}`));

  // 调用一个示例工具
  const toolResult = client.callTool({
    params: { name: 'echo', arguments: { message: 'Hello, world!' } }
  });
  console.log('Echo tool response:', toolResult.content[0].text);

  // 读取一个示例资源
  const resourceContent = client.readResource({
    params: { uri: 'test://static/resource/1' }
  });
  console.log('Resource content:', resourceContent.contents[0].text);

  // 获取一个示例提示
  const prompt = client.getPrompt({
    params: { name: 'simple_prompt' }
  });
  console.log('Prompt:', prompt.messages[0].content.text);
}

您还可以使用 SSE 传输连接到支持它的 MCP 服务器：

const client = new mcp.SSEClient({
  baseURL: 'http://localhost:3001/sse',
  headers: { Authorization: 'Bearer abc123' },
  timeout: '30s'
});

使用以下命令运行 SSE 服务器：

docker run -p 3001:3001 --rm -it tzolov/mcp-everything-server:v1