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('检查 MCP 服务器状态...');
  console.log('MCP 服务器运行中:', client.ping());

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

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

  // 列出可用提示
  console.log('可用提示:');
  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 工具响应:', toolResult.content[0].text);

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

  // 获取一个示例提示
  const prompt = client.getPrompt({
    params: { name: 'simple_prompt' }
  });
  console.log('提示:', 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