当前位置: 首页 > news >正文

go构建streamablehttp mcp服务

package mainimport ("context""fmt""log""os""os/signal""syscall""time""github.com/ThinkInAIXYZ/go-mcp/protocol""github.com/ThinkInAIXYZ/go-mcp/server""github.com/ThinkInAIXYZ/go-mcp/transport"
)type currentTimeReq struct {
}var srv *server.Serverfunc main() {var err errorstateMode := "stateful"addr := "127.0.0.1:8080"log.Printf("start current time mcp server with streamable_http transport, listen %s", addr)opts := transport.WithStreamableHTTPServerTransportOptionStateMode(transport.StateMode(stateMode))t := transport.NewStreamableHTTPServerTransport(addr, opts)srv, err = server.NewServer(t,server.WithServerInfo(protocol.Implementation{Name:    "current-time-v2-server",Version: "1.0.0",}),)if err != nil {log.Fatalf("Failed to create server: %v", err)}tool1, err := protocol.NewTool("current_time", "Get current time with timezone, Asia/Shanghai is default", currentTimeReq{})if err != nil {log.Fatalf("Failed to create tool: %v", err)return}srv.RegisterTool(tool1, currentTime)errCh := make(chan error)go func() {errCh <- srv.Run()}()if err = signalWaiter(errCh); err != nil {log.Fatalf("signal waiter: %v", err)return}ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)defer cancel()if err := srv.Shutdown(ctx); err != nil {log.Fatalf("Shutdown error: %v", err)}
}func currentTime(ctx context.Context, request *protocol.CallToolRequest) (*protocol.CallToolResult, error) {text := fmt.Sprintf(`%s`, time.Now().Format(time.DateTime))return &protocol.CallToolResult{Content: []protocol.Content{&protocol.TextContent{Type: "text",Text: text,},},}, nil
}func signalWaiter(errCh chan error) error {signalToNotify := []os.Signal{syscall.SIGINT, syscall.SIGHUP, syscall.SIGTERM}if signal.Ignored(syscall.SIGHUP) {signalToNotify = []os.Signal{syscall.SIGINT, syscall.SIGTERM}}signals := make(chan os.Signal, 1)signal.Notify(signals, signalToNotify...)select {case sig := <-signals:switch sig {case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM:log.Printf("Received signal: %s\n", sig)// graceful shutdownreturn nil}case err := <-errCh:return err}return nil
}
http://www.jsqmd.com/news/24051/

相关文章:

  • 算法分析--分治--2.归并排序
  • Vue2随笔-251027
  • Java: Jdk17 异步或同步或并行发邮件
  • Day5div与span标签以及字体实体
  • CCUT应用OJ题解——贪吃的松鼠
  • Java流程控制——Scanner进阶使用
  • go构建sse服务端
  • 结对编程心得
  • 关于结对编程的一些感悟
  • 10.27博客
  • 关联子查询
  • AI概念
  • 工控modBus TCP, 服务端或客户端, 均可以与PHP 通讯
  • 1027
  • 20232421 2025-2026-1 《网络与系统攻防技术》实验三实验报告
  • [Mirror] LinuxMirrors: Linux 一键换源项目
  • 10月27日日记
  • 20232326 2025-2026-1 《网络与系统攻防技术》实验三实验报告
  • 20232429 马成栋 2025-2026-1 《网络与系统攻防技术》实验三实验报告
  • 做题记录 #4
  • 扩散模型
  • 20232303 2025-2026-1 《网络与系统攻防技术》实验三实验报告
  • 衡量模型生成图片质量的指标
  • 代码大全2 第一章 与第二章
  • 第二十一天
  • 面试谈薪资指南:掌握主动权的关键策略
  • CF2018B
  • 10/27
  • 第7天(中等题 滑动窗口)
  • C++ 获取 const char* 字符串长度