q792602257/agentic-coding-workflow
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(app): wire MCP module — repo/service/ratelimiter/server + transport mount + AuditWrap

Browse Source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Jerry Yan
2026-05-08 10:44:17 +08:00
parent 6ce21d178e
commit 5cccc97edd
1 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/app/app.go
+25 -3
View File
@@ -32,6 +32,7 @@ import (
"github.com/yan1h/agent-coding-workflow/internal/acp" "github.com/yan1h/agent-coding-workflow/internal/acp"
"github.com/yan1h/agent-coding-workflow/internal/audit" "github.com/yan1h/agent-coding-workflow/internal/audit"
"github.com/yan1h/agent-coding-workflow/internal/chat" "github.com/yan1h/agent-coding-workflow/internal/chat"
mcp "github.com/yan1h/agent-coding-workflow/internal/mcp"
"github.com/yan1h/agent-coding-workflow/internal/config" "github.com/yan1h/agent-coding-workflow/internal/config"
"github.com/yan1h/agent-coding-workflow/internal/infra/clock" "github.com/yan1h/agent-coding-workflow/internal/infra/clock"
"github.com/yan1h/agent-coding-workflow/internal/infra/crypto" "github.com/yan1h/agent-coding-workflow/internal/infra/crypto"
@@ -80,14 +81,15 @@ func New(ctx context.Context, cfg *config.Config, dist embed.FS) (*App, error) {
} }
auditRec := audit.NewPostgresRecorder(pool) auditRec := audit.NewPostgresRecorder(pool)
wrappedAudit := mcp.AuditWrap(auditRec) // 给业务 service 用，自动注入 via_mcp metadata
userRepo := user.NewPostgresRepository(pool) userRepo := user.NewPostgresRepository(pool)
userSvc := user.NewService(userRepo, auditRec) userSvc := user.NewService(userRepo, auditRec)
projectRepo := project.NewPostgresRepository(pool) projectRepo := project.NewPostgresRepository(pool)
projectSvc := project.NewProjectService(projectRepo, auditRec) projectSvc := project.NewProjectService(projectRepo, wrappedAudit)
requirementSvc := project.NewRequirementService(projectRepo, auditRec) requirementSvc := project.NewRequirementService(projectRepo, wrappedAudit)
issueSvc := project.NewIssueService(projectRepo, auditRec) issueSvc := project.NewIssueService(projectRepo, wrappedAudit)
notifyRepo := notify.NewPostgresRepository(pool) notifyRepo := notify.NewPostgresRepository(pool)
notifyDispatcher := notify.NewDispatcher(notifyRepo, log) notifyDispatcher := notify.NewDispatcher(notifyRepo, log)
@@ -251,6 +253,26 @@ func New(ctx context.Context, cfg *config.Config, dist embed.FS) (*App, error) {
chat.NewHandler(chatConvSvc, chatMsgSvc, chatTplSvc, chatEpSvc, chatUsageSvc, chat.NewHandler(chatConvSvc, chatMsgSvc, chatTplSvc, chatEpSvc, chatUsageSvc,
chatUploader, userSvc, userAdminAdapter{svc: userSvc}).Mount(r) chatUploader, userSvc, userAdminAdapter{svc: userSvc}).Mount(r)
// ===== MCP 模块装配 =====
mcpRepo := mcp.NewPostgresRepository(pool)
mcpTokenSvc := mcp.NewTokenService(mcpRepo, auditRec, nil) // mcp 自身 audit 不需要 via_mcp 标记
mcpRateLimiter := mcp.NewRateLimiter(mcp.RateLimitConfig{
PerTokenPerMinute: cfg.MCP.RateLimit.PerTokenPerMinute,
GlobalPerMinute: cfg.MCP.RateLimit.GlobalPerMinute,
}, nil)
mcpDeps := mcp.ServerDeps{
Caller: mcp.NewCallerResolver(userSvc),
Projects: projectSvc,
Reqs: requirementSvc,
Issues: issueSvc,
Workspaces: wsSvc,
Templates: chatTplSvc,
Messages: chatMsgSvc,
Conversations: chatConvSvc,
}
mcpServer := mcp.NewMCPServer(mcpDeps)
mcp.MountTransports(r, mcpServer, mcpTokenSvc, mcpRateLimiter)
// ===== ACP module wiring (Part 2: full SessionService + Supervisor) ===== // ===== ACP module wiring (Part 2: full SessionService + Supervisor) =====
acpRepo := acp.NewPostgresRepository(pool) acpRepo := acp.NewPostgresRepository(pool)
akSvc := acp.NewAgentKindService(acpRepo, enc, auditRec) akSvc := acp.NewAgentKindService(acpRepo, enc, auditRec)