33 lines
969 B
Go
33 lines
969 B
Go
package server
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log/slog"
|
||
"net/http"
|
||
|
||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||
|
||
"aio-mcp/internal/config"
|
||
)
|
||
|
||
// Serve 按传输方式启动服务器，阻塞直到 stdin 关闭、监听失败或出错。
|
||
func Serve(s *mcp.Server, cfg *config.Config) error {
|
||
switch cfg.Transport {
|
||
case config.TransportStdio:
|
||
slog.Info("aio-mcp starting", "transport", "stdio")
|
||
// server.Run 会阻塞，直到 stdin 关闭或出错。
|
||
return s.Run(context.Background(), &mcp.StdioTransport{})
|
||
|
||
case config.TransportHTTP:
|
||
handler := mcp.NewStreamableHTTPHandler(func(*http.Request) *mcp.Server { return s }, nil)
|
||
slog.Info("aio-mcp starting", "transport", "http", "addr", cfg.Addr)
|
||
return http.ListenAndServe(cfg.Addr, handler)
|
||
|
||
default:
|
||
// config.Parse 已校验过，这里仅为防御性兜底。
|
||
return fmt.Errorf("未知 transport: %q（支持: %s, %s）",
|
||
cfg.Transport, config.TransportStdio, config.TransportHTTP)
|
||
}
|
||
}
|