fix: sync agent MCP payloads and query runtime state

This commit is contained in:
kyaring
2026-09-28 13:20:42 +08:00
parent 0c8ac02f82
commit 3ad7faba3b
3 changed files with 57 additions and 4 deletions
+2
View File
@@ -138,6 +138,8 @@ func run(hub, name, token string, insecure bool) error {
func handle(writeJSON func(cluster.Message) error, exec executor.Local, mcpRuntime *cluster.MCPRuntime, msg cluster.Message) error { func handle(writeJSON func(cluster.Message) error, exec executor.Local, mcpRuntime *cluster.MCPRuntime, msg cluster.Message) error {
switch msg.Action { switch msg.Action {
case "mcp_list":
return writeJSON(cluster.Message{ID: msg.ID, Status: "success", Payload: mcpRuntime.List()})
case "mcp_call": case "mcp_call":
var p struct { var p struct {
Server string `json:"server"` Server string `json:"server"`
+29
View File
@@ -9,6 +9,7 @@ import (
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
"net/url"
"os" "os"
"strings" "strings"
"time" "time"
@@ -147,3 +148,31 @@ func main() {
os.Stdout.WriteString("\n") os.Stdout.WriteString("\n")
} }
} }
func (h *HubRemoteBackend) ListAgentMCP(ctx context.Context, device string) (any, error) {
var res any
err := h.get(ctx, "/api/v1/mcp/list?device="+url.QueryEscape(device), &res)
return res, err
}
func (h *HubRemoteBackend) CallAgentMCP(ctx context.Context, clientIP, device, server, tool string, args any) (any, error) {
payload := map[string]any{"device": device, "server": server, "tool": tool, "arguments": args}
var res any
return res, h.post("/api/v1/mcp/call", payload, &res)
}
func (h *HubRemoteBackend) get(ctx context.Context, path string, out any) error {
req, err := http.NewRequestWithContext(ctx, http.MethodGet, strings.TrimRight(h.BaseURL, "/")+path, nil)
if err != nil {
return err
}
req.Header.Set("Authorization", "Bearer "+h.APIKey)
resp, err := h.HTTPClient.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode >= 400 {
b, _ := io.ReadAll(resp.Body)
return fmt.Errorf("HTTP %d: %s", resp.StatusCode, string(b))
}
return json.NewDecoder(resp.Body).Decode(out)
}
+26 -4
View File
@@ -332,13 +332,25 @@ func (m *Manager) syncMCP(n *Node) {
if n == nil || n.MCP == nil || m.Store == nil { if n == nil || n.MCP == nil || m.Store == nil {
return return
} }
services, err := m.Store.ListMCPServicesForNode(n.Name) stored, err := m.Store.ListMCPServicesForNode(n.Name)
if err != nil { if err != nil {
return return
} }
services := make([]MCPService, 0, len(stored))
for _, v := range stored {
services = append(services, decodeStoredMCPService(v))
}
_ = n.write(Message{Event: "mcp_sync", Payload: services, TS: time.Now().Unix()}) _ = n.write(Message{Event: "mcp_sync", Payload: services, TS: time.Now().Unix()})
} }
func decodeStoredMCPService(v storage.MCPService) MCPService {
out := MCPService{ID: v.ID, Name: v.Name, Description: v.Description, Transport: v.Transport, Command: v.Command, URL: v.URL}
_ = json.Unmarshal([]byte(v.ArgsJSON), &out.Args)
_ = json.Unmarshal([]byte(v.EnvJSON), &out.Env)
_ = json.Unmarshal([]byte(v.HeadersJSON), &out.Headers)
return out
}
func (m *Manager) SyncMCPServices() { func (m *Manager) SyncMCPServices() {
m.mu.RLock() m.mu.RLock()
nodes := make([]*Node, 0, len(m.nodes)) nodes := make([]*Node, 0, len(m.nodes))
@@ -357,13 +369,23 @@ func (m *Manager) ListMCP(name string) ([]MCPService, error) {
m.mu.RLock() m.mu.RLock()
n := m.nodes[name] n := m.nodes[name]
m.mu.RUnlock() m.mu.RUnlock()
if n == nil { if n == nil || !m.Online(name) {
return nil, fmt.Errorf("device %s not found", name) return nil, fmt.Errorf("device %s is offline", name)
}
msg, err := m.call(context.Background(), name, "mcp_list", nil)
if err != nil {
return nil, err
}
var out []MCPService
b, _ := json.Marshal(msg.Payload)
if err := json.Unmarshal(b, &out); err != nil {
return nil, err
} }
if n.MCP == nil { if n.MCP == nil {
n.MCP = NewMCPRuntime() n.MCP = NewMCPRuntime()
} }
return n.MCP.List(), nil n.MCP.Sync(out)
return out, nil
} }
func (m *Manager) CallMCP(ctx context.Context, name, server, tool string, args any) (MCPCallResult, error) { func (m *Manager) CallMCP(ctx context.Context, name, server, tool string, args any) (MCPCallResult, error) {