easy-mcp-nest 0.3.0 → 0.4.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +176 -0
- package/dist/EasyMCP.js +0 -3
- package/dist/EasyMCP.js.map +1 -1
- package/dist/adapters/express/express-adapter.js +10 -0
- package/dist/adapters/express/express-adapter.js.map +1 -1
- package/dist/core/batch/batch-executor.service.d.ts +21 -0
- package/dist/core/batch/batch-executor.service.js +101 -0
- package/dist/core/batch/batch-executor.service.js.map +1 -0
- package/dist/core/documentation/openapi-generator.service.d.ts +13 -0
- package/dist/core/documentation/openapi-generator.service.js +116 -0
- package/dist/core/documentation/openapi-generator.service.js.map +1 -0
- package/dist/core/errors/error-handler.interface.d.ts +11 -0
- package/dist/core/errors/error-handler.interface.js +3 -0
- package/dist/core/errors/error-handler.interface.js.map +1 -0
- package/dist/core/health/health-check.service.d.ts +33 -0
- package/dist/core/health/health-check.service.js +71 -0
- package/dist/core/health/health-check.service.js.map +1 -0
- package/dist/core/mcp-server/mcp-server.service.d.ts +14 -1
- package/dist/core/mcp-server/mcp-server.service.js +196 -3
- package/dist/core/mcp-server/mcp-server.service.js.map +1 -1
- package/dist/core/middleware/middleware-executor.service.d.ts +8 -0
- package/dist/core/middleware/middleware-executor.service.js +46 -0
- package/dist/core/middleware/middleware-executor.service.js.map +1 -0
- package/dist/core/middleware/middleware.interface.d.ts +6 -0
- package/dist/core/middleware/middleware.interface.js +3 -0
- package/dist/core/middleware/middleware.interface.js.map +1 -0
- package/dist/core/observability/metrics.service.d.ts +29 -0
- package/dist/core/observability/metrics.service.js +124 -0
- package/dist/core/observability/metrics.service.js.map +1 -0
- package/dist/core/observability/tracing.service.d.ts +12 -0
- package/dist/core/observability/tracing.service.js +88 -0
- package/dist/core/observability/tracing.service.js.map +1 -0
- package/dist/core/progress/progress-notifier.service.d.ts +16 -0
- package/dist/core/progress/progress-notifier.service.js +96 -0
- package/dist/core/progress/progress-notifier.service.js.map +1 -0
- package/dist/core/rate-limiting/rate-limit.interface.d.ts +11 -0
- package/dist/core/rate-limiting/rate-limit.interface.js +3 -0
- package/dist/core/rate-limiting/rate-limit.interface.js.map +1 -0
- package/dist/core/rate-limiting/rate-limiter.service.d.ts +13 -0
- package/dist/core/rate-limiting/rate-limiter.service.js +147 -0
- package/dist/core/rate-limiting/rate-limiter.service.js.map +1 -0
- package/dist/core/resilience/circuit-breaker.interface.d.ts +15 -0
- package/dist/core/resilience/circuit-breaker.interface.js +3 -0
- package/dist/core/resilience/circuit-breaker.interface.js.map +1 -0
- package/dist/core/resilience/circuit-breaker.service.d.ts +11 -0
- package/dist/core/resilience/circuit-breaker.service.js +117 -0
- package/dist/core/resilience/circuit-breaker.service.js.map +1 -0
- package/dist/core/resilience/retry.service.d.ts +6 -0
- package/dist/core/resilience/retry.service.js +66 -0
- package/dist/core/resilience/retry.service.js.map +1 -0
- package/dist/core/utils/decorator-scanner.d.ts +4 -0
- package/dist/core/utils/decorator-scanner.js +112 -0
- package/dist/core/utils/decorator-scanner.js.map +1 -0
- package/dist/core/utils/sanitize.util.d.ts +1 -0
- package/dist/core/utils/sanitize.util.js +19 -0
- package/dist/core/utils/sanitize.util.js.map +1 -1
- package/dist/decorators/index.d.ts +3 -0
- package/dist/decorators/index.js +12 -1
- package/dist/decorators/index.js.map +1 -1
- package/dist/decorators/mcp-error-handler.decorator.d.ts +4 -0
- package/dist/decorators/mcp-error-handler.decorator.js +15 -0
- package/dist/decorators/mcp-error-handler.decorator.js.map +1 -0
- package/dist/decorators/mcp-middleware.decorator.d.ts +4 -0
- package/dist/decorators/mcp-middleware.decorator.js +17 -0
- package/dist/decorators/mcp-middleware.decorator.js.map +1 -0
- package/dist/decorators/mcp-tool.decorator.d.ts +28 -0
- package/dist/decorators/mcp-tool.decorator.js +36 -0
- package/dist/decorators/mcp-tool.decorator.js.map +1 -0
- package/dist/index.d.ts +2 -2
- package/dist/index.js +4 -1
- package/dist/index.js.map +1 -1
- package/dist/interface/interface.interface.d.ts +2 -0
- package/dist/interface/mcp-protocol.interface.d.ts +14 -0
- package/dist/interface/mcp-protocol.interface.js.map +1 -1
- package/dist/tooling/tool-registry/tool-registry.service.d.ts +7 -1
- package/dist/tooling/tool-registry/tool-registry.service.js +40 -3
- package/dist/tooling/tool-registry/tool-registry.service.js.map +1 -1
- package/dist/tooling/tool.interface.d.ts +7 -1
- package/dist/tsconfig.build.tsbuildinfo +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1,124 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.MetricsService = void 0;
|
|
10
|
+
const common_1 = require("@nestjs/common");
|
|
11
|
+
const logger_util_1 = require("../utils/logger.util");
|
|
12
|
+
let MetricsService = class MetricsService {
|
|
13
|
+
toolMetrics = new Map();
|
|
14
|
+
requestCount = 0;
|
|
15
|
+
activeRequests = 0;
|
|
16
|
+
startTime = Date.now();
|
|
17
|
+
onModuleInit() {
|
|
18
|
+
logger_util_1.logger.info("MetricsService", "Metrics collection initialized", {
|
|
19
|
+
component: "Observability",
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
recordToolStart(toolName) {
|
|
23
|
+
const startTime = Date.now();
|
|
24
|
+
this.activeRequests++;
|
|
25
|
+
this.requestCount++;
|
|
26
|
+
return () => {
|
|
27
|
+
const executionTime = Date.now() - startTime;
|
|
28
|
+
this.activeRequests--;
|
|
29
|
+
const metrics = this.toolMetrics.get(toolName) || {
|
|
30
|
+
toolName,
|
|
31
|
+
executionCount: 0,
|
|
32
|
+
totalExecutionTime: 0,
|
|
33
|
+
errorCount: 0,
|
|
34
|
+
};
|
|
35
|
+
metrics.executionCount++;
|
|
36
|
+
metrics.totalExecutionTime += executionTime;
|
|
37
|
+
metrics.lastExecutionTime = Date.now();
|
|
38
|
+
this.toolMetrics.set(toolName, metrics);
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
recordToolError(toolName) {
|
|
42
|
+
const metrics = this.toolMetrics.get(toolName) || {
|
|
43
|
+
toolName,
|
|
44
|
+
executionCount: 0,
|
|
45
|
+
totalExecutionTime: 0,
|
|
46
|
+
errorCount: 0,
|
|
47
|
+
};
|
|
48
|
+
metrics.errorCount++;
|
|
49
|
+
this.toolMetrics.set(toolName, metrics);
|
|
50
|
+
}
|
|
51
|
+
getToolMetrics(toolName) {
|
|
52
|
+
return this.toolMetrics.get(toolName);
|
|
53
|
+
}
|
|
54
|
+
getAllToolMetrics() {
|
|
55
|
+
return Array.from(this.toolMetrics.values());
|
|
56
|
+
}
|
|
57
|
+
getAverageExecutionTime(toolName) {
|
|
58
|
+
const metrics = this.toolMetrics.get(toolName);
|
|
59
|
+
if (!metrics || metrics.executionCount === 0) {
|
|
60
|
+
return 0;
|
|
61
|
+
}
|
|
62
|
+
return metrics.totalExecutionTime / metrics.executionCount;
|
|
63
|
+
}
|
|
64
|
+
getErrorRate(toolName) {
|
|
65
|
+
const metrics = this.toolMetrics.get(toolName);
|
|
66
|
+
if (!metrics || metrics.executionCount === 0) {
|
|
67
|
+
return 0;
|
|
68
|
+
}
|
|
69
|
+
return metrics.errorCount / metrics.executionCount;
|
|
70
|
+
}
|
|
71
|
+
getServerMetrics() {
|
|
72
|
+
return {
|
|
73
|
+
totalRequests: this.requestCount,
|
|
74
|
+
activeRequests: this.activeRequests,
|
|
75
|
+
uptime: Date.now() - this.startTime,
|
|
76
|
+
toolCount: this.toolMetrics.size,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
getPrometheusMetrics() {
|
|
80
|
+
const lines = [];
|
|
81
|
+
const serverMetrics = this.getServerMetrics();
|
|
82
|
+
lines.push(`# HELP mcp_server_total_requests Total number of requests`);
|
|
83
|
+
lines.push(`# TYPE mcp_server_total_requests counter`);
|
|
84
|
+
lines.push(`mcp_server_total_requests ${serverMetrics.totalRequests}`);
|
|
85
|
+
lines.push(`# HELP mcp_server_active_requests Current number of active requests`);
|
|
86
|
+
lines.push(`# TYPE mcp_server_active_requests gauge`);
|
|
87
|
+
lines.push(`mcp_server_active_requests ${serverMetrics.activeRequests}`);
|
|
88
|
+
lines.push(`# HELP mcp_server_uptime_seconds Server uptime in seconds`);
|
|
89
|
+
lines.push(`# TYPE mcp_server_uptime_seconds gauge`);
|
|
90
|
+
lines.push(`mcp_server_uptime_seconds ${Math.floor(serverMetrics.uptime / 1000)}`);
|
|
91
|
+
for (const metrics of this.toolMetrics.values()) {
|
|
92
|
+
const toolName = metrics.toolName.replace(/[^a-zA-Z0-9_]/g, "_");
|
|
93
|
+
lines.push(`# HELP mcp_tool_executions_total Total number of tool executions`);
|
|
94
|
+
lines.push(`# TYPE mcp_tool_executions_total counter`);
|
|
95
|
+
lines.push(`mcp_tool_executions_total{tool="${toolName}"} ${metrics.executionCount}`);
|
|
96
|
+
lines.push(`# HELP mcp_tool_execution_time_seconds Total execution time in seconds`);
|
|
97
|
+
lines.push(`# TYPE mcp_tool_execution_time_seconds counter`);
|
|
98
|
+
lines.push(`mcp_tool_execution_time_seconds{tool="${toolName}"} ${metrics.totalExecutionTime / 1000}`);
|
|
99
|
+
lines.push(`# HELP mcp_tool_errors_total Total number of tool errors`);
|
|
100
|
+
lines.push(`# TYPE mcp_tool_errors_total counter`);
|
|
101
|
+
lines.push(`mcp_tool_errors_total{tool="${toolName}"} ${metrics.errorCount}`);
|
|
102
|
+
const avgTime = this.getAverageExecutionTime(metrics.toolName);
|
|
103
|
+
lines.push(`# HELP mcp_tool_avg_execution_time_seconds Average execution time in seconds`);
|
|
104
|
+
lines.push(`# TYPE mcp_tool_avg_execution_time_seconds gauge`);
|
|
105
|
+
lines.push(`mcp_tool_avg_execution_time_seconds{tool="${toolName}"} ${avgTime / 1000}`);
|
|
106
|
+
const errorRate = this.getErrorRate(metrics.toolName);
|
|
107
|
+
lines.push(`# HELP mcp_tool_error_rate Error rate (0.0 to 1.0)`);
|
|
108
|
+
lines.push(`# TYPE mcp_tool_error_rate gauge`);
|
|
109
|
+
lines.push(`mcp_tool_error_rate{tool="${toolName}"} ${errorRate}`);
|
|
110
|
+
}
|
|
111
|
+
return lines.join("\n");
|
|
112
|
+
}
|
|
113
|
+
reset() {
|
|
114
|
+
this.toolMetrics.clear();
|
|
115
|
+
this.requestCount = 0;
|
|
116
|
+
this.activeRequests = 0;
|
|
117
|
+
this.startTime = Date.now();
|
|
118
|
+
}
|
|
119
|
+
};
|
|
120
|
+
exports.MetricsService = MetricsService;
|
|
121
|
+
exports.MetricsService = MetricsService = __decorate([
|
|
122
|
+
(0, common_1.Injectable)()
|
|
123
|
+
], MetricsService);
|
|
124
|
+
//# sourceMappingURL=metrics.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.service.js","sourceRoot":"","sources":["../../../src/core/observability/metrics.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA0D;AAC1D,sDAA8C;AAsBvC,IAAM,cAAc,GAApB,MAAM,cAAc;IACjB,WAAW,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC7C,YAAY,GAAG,CAAC,CAAC;IACjB,cAAc,GAAG,CAAC,CAAC;IACnB,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE/B,YAAY;QAEV,oBAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,gCAAgC,EAAE;YAC9D,SAAS,EAAE,eAAe;SAC3B,CAAC,CAAC;IACL,CAAC;IAKD,eAAe,CAAC,QAAgB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,OAAO,GAAG,EAAE;YAEV,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,cAAc,EAAE,CAAC;YAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI;gBAChD,QAAQ;gBACR,cAAc,EAAE,CAAC;gBACjB,kBAAkB,EAAE,CAAC;gBACrB,UAAU,EAAE,CAAC;aACd,CAAC;YAEF,OAAO,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,CAAC,kBAAkB,IAAI,aAAa,CAAC;YAC5C,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC,CAAC;IACJ,CAAC;IAKD,eAAe,CAAC,QAAgB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI;YAChD,QAAQ;YACR,cAAc,EAAE,CAAC;YACjB,kBAAkB,EAAE,CAAC;YACrB,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAKD,cAAc,CAAC,QAAgB;QAC7B,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAKD,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;IAKD,uBAAuB,CAAC,QAAgB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,OAAO,CAAC,kBAAkB,GAAG,OAAO,CAAC,cAAc,CAAC;IAC7D,CAAC;IAKD,YAAY,CAAC,QAAgB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC,cAAc,CAAC;IACrD,CAAC;IAKD,gBAAgB;QAMd,OAAO;YACL,aAAa,EAAE,IAAI,CAAC,YAAY;YAChC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS;YACnC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;SACjC,CAAC;IACJ,CAAC;IAKD,oBAAoB;QAClB,MAAM,KAAK,GAAa,EAAE,CAAC;QAG3B,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9C,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;QACxE,KAAK,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QACvD,KAAK,CAAC,IAAI,CAAC,6BAA6B,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC;QAEvE,KAAK,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;QAClF,KAAK,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;QACtD,KAAK,CAAC,IAAI,CAAC,8BAA8B,aAAa,CAAC,cAAc,EAAE,CAAC,CAAC;QAEzE,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;QACxE,KAAK,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QACrD,KAAK,CAAC,IAAI,CAAC,6BAA6B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC;QAGnF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;YAEjE,KAAK,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;YAC/E,KAAK,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YACvD,KAAK,CAAC,IAAI,CAAC,mCAAmC,QAAQ,MAAM,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;YAEtF,KAAK,CAAC,IAAI,CAAC,wEAAwE,CAAC,CAAC;YACrF,KAAK,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC7D,KAAK,CAAC,IAAI,CACR,yCAAyC,QAAQ,MAAM,OAAO,CAAC,kBAAkB,GAAG,IAAI,EAAE,CAC3F,CAAC;YAEF,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;YACvE,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;YACnD,KAAK,CAAC,IAAI,CAAC,+BAA+B,QAAQ,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;YAE9E,MAAM,OAAO,GAAG,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,8EAA8E,CAAC,CAAC;YAC3F,KAAK,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,6CAA6C,QAAQ,MAAM,OAAO,GAAG,IAAI,EAAE,CAAC,CAAC;YAExF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtD,KAAK,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;YACjE,KAAK,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC/C,KAAK,CAAC,IAAI,CAAC,6BAA6B,QAAQ,MAAM,SAAS,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAKD,KAAK;QACH,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;CACF,CAAA;AA1KY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;GACA,cAAc,CA0K1B"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export interface TraceContext {
|
|
2
|
+
traceId: string;
|
|
3
|
+
spanId: string;
|
|
4
|
+
parentSpanId?: string;
|
|
5
|
+
startTime: number;
|
|
6
|
+
}
|
|
7
|
+
export declare class TracingService {
|
|
8
|
+
createTraceContext(parentTraceId?: string, parentSpanId?: string): TraceContext;
|
|
9
|
+
private validateTraceId;
|
|
10
|
+
extractTraceContext(headers?: Record<string, any>): TraceContext | undefined;
|
|
11
|
+
createChildSpan(parent: TraceContext): TraceContext;
|
|
12
|
+
}
|
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.TracingService = void 0;
|
|
10
|
+
const common_1 = require("@nestjs/common");
|
|
11
|
+
const crypto_1 = require("crypto");
|
|
12
|
+
function generateUUID() {
|
|
13
|
+
if (typeof crypto_1.randomUUID === "function") {
|
|
14
|
+
return (0, crypto_1.randomUUID)();
|
|
15
|
+
}
|
|
16
|
+
if (typeof crypto !== "undefined" && crypto.getRandomValues) {
|
|
17
|
+
const bytes = new Uint8Array(16);
|
|
18
|
+
crypto.getRandomValues(bytes);
|
|
19
|
+
bytes[6] = (bytes[6] & 0x0f) | 0x40;
|
|
20
|
+
bytes[8] = (bytes[8] & 0x3f) | 0x80;
|
|
21
|
+
const hex = Array.from(bytes)
|
|
22
|
+
.map((b) => b.toString(16).padStart(2, "0"))
|
|
23
|
+
.join("");
|
|
24
|
+
return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20, 32)}`;
|
|
25
|
+
}
|
|
26
|
+
return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
|
|
27
|
+
const r = (Math.random() * 16) | 0;
|
|
28
|
+
const v = c === "x" ? r : (r & 0x3) | 0x8;
|
|
29
|
+
return v.toString(16);
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
let TracingService = class TracingService {
|
|
33
|
+
createTraceContext(parentTraceId, parentSpanId) {
|
|
34
|
+
return {
|
|
35
|
+
traceId: parentTraceId || generateUUID(),
|
|
36
|
+
spanId: generateUUID(),
|
|
37
|
+
parentSpanId,
|
|
38
|
+
startTime: Date.now(),
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
validateTraceId(traceId) {
|
|
42
|
+
if (!traceId || typeof traceId !== "string") {
|
|
43
|
+
return null;
|
|
44
|
+
}
|
|
45
|
+
let sanitized = traceId.replace(/[\x00-\x1F\x7F-\x9F\n\r]/g, "");
|
|
46
|
+
const MAX_TRACE_ID_LENGTH = 128;
|
|
47
|
+
if (sanitized.length > MAX_TRACE_ID_LENGTH) {
|
|
48
|
+
return null;
|
|
49
|
+
}
|
|
50
|
+
if (!/^[a-zA-Z0-9_-]{1,128}$/.test(sanitized)) {
|
|
51
|
+
return null;
|
|
52
|
+
}
|
|
53
|
+
return sanitized;
|
|
54
|
+
}
|
|
55
|
+
extractTraceContext(headers) {
|
|
56
|
+
if (!headers) {
|
|
57
|
+
return undefined;
|
|
58
|
+
}
|
|
59
|
+
const traceId = headers["x-trace-id"] || headers["trace-id"];
|
|
60
|
+
const spanId = headers["x-span-id"] || headers["span-id"];
|
|
61
|
+
const parentSpanId = headers["x-parent-span-id"] || headers["parent-span-id"];
|
|
62
|
+
const validatedTraceId = traceId ? this.validateTraceId(traceId) : null;
|
|
63
|
+
if (!validatedTraceId) {
|
|
64
|
+
return undefined;
|
|
65
|
+
}
|
|
66
|
+
const validatedSpanId = spanId ? this.validateTraceId(spanId) : null;
|
|
67
|
+
const validatedParentSpanId = parentSpanId ? this.validateTraceId(parentSpanId) : null;
|
|
68
|
+
return {
|
|
69
|
+
traceId: validatedTraceId,
|
|
70
|
+
spanId: validatedSpanId || generateUUID(),
|
|
71
|
+
parentSpanId: validatedParentSpanId || undefined,
|
|
72
|
+
startTime: Date.now(),
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
createChildSpan(parent) {
|
|
76
|
+
return {
|
|
77
|
+
traceId: parent.traceId,
|
|
78
|
+
spanId: generateUUID(),
|
|
79
|
+
parentSpanId: parent.spanId,
|
|
80
|
+
startTime: Date.now(),
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
};
|
|
84
|
+
exports.TracingService = TracingService;
|
|
85
|
+
exports.TracingService = TracingService = __decorate([
|
|
86
|
+
(0, common_1.Injectable)()
|
|
87
|
+
], TracingService);
|
|
88
|
+
//# sourceMappingURL=tracing.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracing.service.js","sourceRoot":"","sources":["../../../src/core/observability/tracing.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,mCAAoC;AAMpC,SAAS,YAAY;IAEnB,IAAI,OAAO,mBAAU,KAAK,UAAU,EAAE,CAAC;QACrC,OAAO,IAAA,mBAAU,GAAE,CAAC;IACtB,CAAC;IAGD,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAC;QACjC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAE9B,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;QACpC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;QAGpC,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;aAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;aAC3C,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;IACjH,CAAC;IAID,OAAO,sCAAsC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;QACnE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QAC1C,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC;AAgBM,IAAM,cAAc,GAApB,MAAM,cAAc;IAIzB,kBAAkB,CAAC,aAAsB,EAAE,YAAqB;QAC9D,OAAO;YACL,OAAO,EAAE,aAAa,IAAI,YAAY,EAAE;YACxC,MAAM,EAAE,YAAY,EAAE;YACtB,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;IAOO,eAAe,CAAC,OAAY;QAClC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC;QAGjE,MAAM,mBAAmB,GAAG,GAAG,CAAC;QAChC,IAAI,SAAS,CAAC,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC;QACd,CAAC;QAID,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAMD,mBAAmB,CAAC,OAA6B;QAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,YAAY,GAAG,OAAO,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAG9E,MAAM,gBAAgB,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACxE,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QAGD,MAAM,eAAe,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,MAAM,qBAAqB,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAEvF,OAAO;YACL,OAAO,EAAE,gBAAgB;YACzB,MAAM,EAAE,eAAe,IAAI,YAAY,EAAE;YACzC,YAAY,EAAE,qBAAqB,IAAI,SAAS;YAChD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;IAKD,eAAe,CAAC,MAAoB;QAClC,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,YAAY,EAAE;YACtB,YAAY,EAAE,MAAM,CAAC,MAAM;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;CACF,CAAA;AAnFY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;GACA,cAAc,CAmF1B"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { IInterfaceLayer } from "../../interface/interface.interface";
|
|
2
|
+
export type ProgressCallback = (progress: {
|
|
3
|
+
progress: number;
|
|
4
|
+
total?: number;
|
|
5
|
+
message?: string;
|
|
6
|
+
}) => void;
|
|
7
|
+
export declare class ProgressNotifierService {
|
|
8
|
+
private interfaceLayer?;
|
|
9
|
+
private progressTokens;
|
|
10
|
+
private nextTokenId;
|
|
11
|
+
constructor(interfaceLayer?: IInterfaceLayer | undefined);
|
|
12
|
+
createProgressCallback(requestId: string | number | null): ProgressCallback;
|
|
13
|
+
private getOrCreateProgressToken;
|
|
14
|
+
private sendProgress;
|
|
15
|
+
cleanup(requestId: string | number | null): void;
|
|
16
|
+
}
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
15
|
+
exports.ProgressNotifierService = void 0;
|
|
16
|
+
const common_1 = require("@nestjs/common");
|
|
17
|
+
const constants_1 = require("../../config/constants");
|
|
18
|
+
const logger_util_1 = require("../utils/logger.util");
|
|
19
|
+
let ProgressNotifierService = class ProgressNotifierService {
|
|
20
|
+
interfaceLayer;
|
|
21
|
+
progressTokens = new Map();
|
|
22
|
+
nextTokenId = 0;
|
|
23
|
+
constructor(interfaceLayer) {
|
|
24
|
+
this.interfaceLayer = interfaceLayer;
|
|
25
|
+
}
|
|
26
|
+
createProgressCallback(requestId) {
|
|
27
|
+
if (requestId === null || requestId === undefined) {
|
|
28
|
+
return () => { };
|
|
29
|
+
}
|
|
30
|
+
const progressToken = this.getOrCreateProgressToken(requestId);
|
|
31
|
+
return (progress) => {
|
|
32
|
+
this.sendProgress(requestId, progressToken, progress);
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
getOrCreateProgressToken(requestId) {
|
|
36
|
+
if (!this.progressTokens.has(requestId)) {
|
|
37
|
+
const token = `progress_${this.nextTokenId++}_${Date.now()}`;
|
|
38
|
+
this.progressTokens.set(requestId, token);
|
|
39
|
+
}
|
|
40
|
+
return this.progressTokens.get(requestId);
|
|
41
|
+
}
|
|
42
|
+
sendProgress(requestId, progressToken, progress) {
|
|
43
|
+
if (!this.interfaceLayer) {
|
|
44
|
+
logger_util_1.logger.debug("ProgressNotifierService", "Progress update", {
|
|
45
|
+
component: "ProgressNotifier",
|
|
46
|
+
requestId,
|
|
47
|
+
progress: progress.progress,
|
|
48
|
+
message: progress.message,
|
|
49
|
+
});
|
|
50
|
+
return;
|
|
51
|
+
}
|
|
52
|
+
const clampedProgress = Math.max(0, Math.min(1, progress.progress));
|
|
53
|
+
const notification = {
|
|
54
|
+
jsonrpc: "2.0",
|
|
55
|
+
method: "notifications/progress",
|
|
56
|
+
params: {
|
|
57
|
+
progressToken,
|
|
58
|
+
progress: clampedProgress,
|
|
59
|
+
...(progress.total !== undefined && { total: progress.total }),
|
|
60
|
+
...(progress.message && { message: progress.message }),
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
try {
|
|
64
|
+
if (this.interfaceLayer.sendNotification) {
|
|
65
|
+
this.interfaceLayer.sendNotification(notification);
|
|
66
|
+
}
|
|
67
|
+
else {
|
|
68
|
+
logger_util_1.logger.debug("ProgressNotifierService", "Progress notification", {
|
|
69
|
+
component: "ProgressNotifier",
|
|
70
|
+
requestId,
|
|
71
|
+
notification,
|
|
72
|
+
});
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
catch (error) {
|
|
76
|
+
logger_util_1.logger.error("ProgressNotifierService", "Failed to send progress notification", {
|
|
77
|
+
component: "ProgressNotifier",
|
|
78
|
+
requestId,
|
|
79
|
+
error: error instanceof Error ? error.message : String(error),
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
cleanup(requestId) {
|
|
84
|
+
if (requestId !== null && requestId !== undefined) {
|
|
85
|
+
this.progressTokens.delete(requestId);
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
};
|
|
89
|
+
exports.ProgressNotifierService = ProgressNotifierService;
|
|
90
|
+
exports.ProgressNotifierService = ProgressNotifierService = __decorate([
|
|
91
|
+
(0, common_1.Injectable)(),
|
|
92
|
+
__param(0, (0, common_1.Optional)()),
|
|
93
|
+
__param(0, (0, common_1.Inject)(constants_1.INTERFACE_LAYER_TOKEN)),
|
|
94
|
+
__metadata("design:paramtypes", [Object])
|
|
95
|
+
], ProgressNotifierService);
|
|
96
|
+
//# sourceMappingURL=progress-notifier.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"progress-notifier.service.js","sourceRoot":"","sources":["../../../src/core/progress/progress-notifier.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAA8D;AAG9D,sDAA+D;AAC/D,sDAA8C;AAevC,IAAM,uBAAuB,GAA7B,MAAM,uBAAuB;IAKmB;IAJ7C,cAAc,GAAG,IAAI,GAAG,EAA2B,CAAC;IACpD,WAAW,GAAG,CAAC,CAAC;IAExB,YACqD,cAAgC;QAAhC,mBAAc,GAAd,cAAc,CAAkB;IAClF,CAAC;IASJ,sBAAsB,CAAC,SAAiC;QACtD,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAElD,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;QAE/D,OAAO,CAAC,QAAgE,EAAE,EAAE;YAC1E,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAC;IACJ,CAAC;IAKO,wBAAwB,CAAC,SAA0B;QACzD,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,YAAY,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC7D,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;IAC7C,CAAC;IAKO,YAAY,CAClB,SAA0B,EAC1B,aAAqB,EACrB,QAAgE;QAEhE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAEzB,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,iBAAiB,EAAE;gBACzD,SAAS,EAAE,kBAAkB;gBAC7B,SAAS;gBACT,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAGD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEpE,MAAM,YAAY,GAAyB;YACzC,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,wBAAwB;YAChC,MAAM,EAAE;gBACN,aAAa;gBACb,QAAQ,EAAE,eAAe;gBACzB,GAAG,CAAC,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC9D,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;aACvD;SACF,CAAC;QAKF,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,CAAC;gBACzC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBAEN,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,uBAAuB,EAAE;oBAC/D,SAAS,EAAE,kBAAkB;oBAC7B,SAAS;oBACT,YAAY;iBACb,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oBAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,sCAAsC,EAAE;gBAC9E,SAAS,EAAE,kBAAkB;gBAC7B,SAAS;gBACT,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAKD,OAAO,CAAC,SAAiC;QACvC,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAClD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF,CAAA;AAvGY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;IAMR,WAAA,IAAA,iBAAQ,GAAE,CAAA;IAAE,WAAA,IAAA,eAAM,EAAC,iCAAqB,CAAC,CAAA;;GALjC,uBAAuB,CAuGnC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { RateLimitConfig } from "../../decorators/mcp-tool.decorator";
|
|
2
|
+
export type { RateLimitConfig };
|
|
3
|
+
export interface RateLimitEntry {
|
|
4
|
+
count: number;
|
|
5
|
+
resetTime: number;
|
|
6
|
+
}
|
|
7
|
+
export interface RateLimitResult {
|
|
8
|
+
allowed: boolean;
|
|
9
|
+
remaining: number;
|
|
10
|
+
resetTime: number;
|
|
11
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limit.interface.js","sourceRoot":"","sources":["../../../src/core/rate-limiting/rate-limit.interface.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { OnModuleInit, OnModuleDestroy } from "@nestjs/common";
|
|
2
|
+
import { RateLimitConfig, RateLimitResult } from "./rate-limit.interface";
|
|
3
|
+
export declare class RateLimiterService implements OnModuleInit, OnModuleDestroy {
|
|
4
|
+
private rateLimiters;
|
|
5
|
+
private cleanupInterval?;
|
|
6
|
+
private static readonly MAX_ENTRIES_PER_TOOL;
|
|
7
|
+
private static readonly CLEANUP_INTERVAL_MS;
|
|
8
|
+
onModuleInit(): void;
|
|
9
|
+
onModuleDestroy(): void;
|
|
10
|
+
checkRateLimit(toolName: string, identifier: string, config: RateLimitConfig): RateLimitResult;
|
|
11
|
+
cleanup(): void;
|
|
12
|
+
reset(toolName?: string): void;
|
|
13
|
+
}
|
|
@@ -0,0 +1,147 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var RateLimiterService_1;
|
|
9
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
10
|
+
exports.RateLimiterService = void 0;
|
|
11
|
+
const common_1 = require("@nestjs/common");
|
|
12
|
+
const logger_util_1 = require("../utils/logger.util");
|
|
13
|
+
const sanitize_util_1 = require("../utils/sanitize.util");
|
|
14
|
+
function parseTimeWindow(window) {
|
|
15
|
+
const match = window.match(/^(\d+)([smhd])$/);
|
|
16
|
+
if (!match) {
|
|
17
|
+
logger_util_1.logger.warn("RateLimiterService", `Invalid time window format: ${window}. Using default 1 minute`, {
|
|
18
|
+
component: "RateLimiter",
|
|
19
|
+
invalidWindow: window,
|
|
20
|
+
});
|
|
21
|
+
return 60 * 1000;
|
|
22
|
+
}
|
|
23
|
+
const value = parseInt(match[1], 10);
|
|
24
|
+
const unit = match[2];
|
|
25
|
+
if (isNaN(value) || value <= 0) {
|
|
26
|
+
logger_util_1.logger.warn("RateLimiterService", `Invalid time window value: ${value}. Using default 1 minute`, {
|
|
27
|
+
component: "RateLimiter",
|
|
28
|
+
invalidValue: value,
|
|
29
|
+
});
|
|
30
|
+
return 60 * 1000;
|
|
31
|
+
}
|
|
32
|
+
const multipliers = {
|
|
33
|
+
s: 1000,
|
|
34
|
+
m: 60 * 1000,
|
|
35
|
+
h: 60 * 60 * 1000,
|
|
36
|
+
d: 24 * 60 * 60 * 1000,
|
|
37
|
+
};
|
|
38
|
+
const multiplier = multipliers[unit];
|
|
39
|
+
if (!multiplier) {
|
|
40
|
+
logger_util_1.logger.warn("RateLimiterService", `Invalid time window unit: ${unit}. Using default 1 minute`, {
|
|
41
|
+
component: "RateLimiter",
|
|
42
|
+
invalidUnit: unit,
|
|
43
|
+
});
|
|
44
|
+
return 60 * 1000;
|
|
45
|
+
}
|
|
46
|
+
return value * multiplier;
|
|
47
|
+
}
|
|
48
|
+
let RateLimiterService = class RateLimiterService {
|
|
49
|
+
static { RateLimiterService_1 = this; }
|
|
50
|
+
rateLimiters = new Map();
|
|
51
|
+
cleanupInterval;
|
|
52
|
+
static MAX_ENTRIES_PER_TOOL = 10000;
|
|
53
|
+
static CLEANUP_INTERVAL_MS = 5 * 60 * 1000;
|
|
54
|
+
onModuleInit() {
|
|
55
|
+
this.cleanupInterval = setInterval(() => {
|
|
56
|
+
this.cleanup();
|
|
57
|
+
}, RateLimiterService_1.CLEANUP_INTERVAL_MS);
|
|
58
|
+
}
|
|
59
|
+
onModuleDestroy() {
|
|
60
|
+
if (this.cleanupInterval) {
|
|
61
|
+
clearInterval(this.cleanupInterval);
|
|
62
|
+
this.cleanupInterval = undefined;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
checkRateLimit(toolName, identifier, config) {
|
|
66
|
+
try {
|
|
67
|
+
const windowMs = parseTimeWindow(config.window);
|
|
68
|
+
const now = Date.now();
|
|
69
|
+
if (!this.rateLimiters.has(toolName)) {
|
|
70
|
+
this.rateLimiters.set(toolName, new Map());
|
|
71
|
+
}
|
|
72
|
+
const toolLimiter = this.rateLimiters.get(toolName);
|
|
73
|
+
if (toolLimiter.size >= RateLimiterService_1.MAX_ENTRIES_PER_TOOL) {
|
|
74
|
+
const entries = Array.from(toolLimiter.entries());
|
|
75
|
+
const expired = entries.filter(([, entry]) => entry.resetTime < now);
|
|
76
|
+
if (expired.length > 0) {
|
|
77
|
+
expired.forEach(([id]) => toolLimiter.delete(id));
|
|
78
|
+
}
|
|
79
|
+
else {
|
|
80
|
+
const sorted = entries.sort(([, a], [, b]) => a.resetTime - b.resetTime);
|
|
81
|
+
const toRemove = Math.floor(sorted.length * 0.1);
|
|
82
|
+
sorted.slice(0, toRemove).forEach(([id]) => toolLimiter.delete(id));
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
let entry = toolLimiter.get(identifier);
|
|
86
|
+
if (!entry || entry.resetTime < now) {
|
|
87
|
+
entry = {
|
|
88
|
+
count: 0,
|
|
89
|
+
resetTime: now + windowMs,
|
|
90
|
+
};
|
|
91
|
+
toolLimiter.set(identifier, entry);
|
|
92
|
+
}
|
|
93
|
+
if (entry.count >= config.max) {
|
|
94
|
+
return {
|
|
95
|
+
allowed: false,
|
|
96
|
+
remaining: 0,
|
|
97
|
+
resetTime: entry.resetTime,
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
entry.count++;
|
|
101
|
+
return {
|
|
102
|
+
allowed: true,
|
|
103
|
+
remaining: config.max - entry.count,
|
|
104
|
+
resetTime: entry.resetTime,
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
catch (error) {
|
|
108
|
+
logger_util_1.logger.error("RateLimiterService", "Rate limit check failed", {
|
|
109
|
+
component: "RateLimiter",
|
|
110
|
+
toolName: (0, sanitize_util_1.sanitizeActorId)(toolName),
|
|
111
|
+
identifier: (0, sanitize_util_1.sanitizeActorId)(identifier),
|
|
112
|
+
error: error instanceof Error ? error.message : String(error),
|
|
113
|
+
});
|
|
114
|
+
return {
|
|
115
|
+
allowed: false,
|
|
116
|
+
remaining: 0,
|
|
117
|
+
resetTime: Date.now() + 60000,
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
cleanup() {
|
|
122
|
+
const now = Date.now();
|
|
123
|
+
for (const [toolName, toolLimiter] of this.rateLimiters.entries()) {
|
|
124
|
+
for (const [identifier, entry] of toolLimiter.entries()) {
|
|
125
|
+
if (entry.resetTime < now) {
|
|
126
|
+
toolLimiter.delete(identifier);
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
if (toolLimiter.size === 0) {
|
|
130
|
+
this.rateLimiters.delete(toolName);
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
reset(toolName) {
|
|
135
|
+
if (toolName) {
|
|
136
|
+
this.rateLimiters.delete(toolName);
|
|
137
|
+
}
|
|
138
|
+
else {
|
|
139
|
+
this.rateLimiters.clear();
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
};
|
|
143
|
+
exports.RateLimiterService = RateLimiterService;
|
|
144
|
+
exports.RateLimiterService = RateLimiterService = RateLimiterService_1 = __decorate([
|
|
145
|
+
(0, common_1.Injectable)()
|
|
146
|
+
], RateLimiterService);
|
|
147
|
+
//# sourceMappingURL=rate-limiter.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.service.js","sourceRoot":"","sources":["../../../src/core/rate-limiting/rate-limiter.service.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAA2E;AAE3E,sDAA8C;AAC9C,0DAAyD;AAMzD,SAAS,eAAe,CAAC,MAAc;IACrC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,+BAA+B,MAAM,0BAA0B,EAAE;YACjG,SAAS,EAAE,aAAa;YACxB,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAEtB,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QAC/B,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,8BAA8B,KAAK,0BAA0B,EAAE;YAC/F,SAAS,EAAE,aAAa;YACxB,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAA2B;QAC1C,CAAC,EAAE,IAAI;QACP,CAAC,EAAE,EAAE,GAAG,IAAI;QACZ,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI;QACjB,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI;KACvB,CAAC;IAEF,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,oBAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,6BAA6B,IAAI,0BAA0B,EAAE;YAC7F,SAAS,EAAE,aAAa;YACxB,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,OAAO,KAAK,GAAG,UAAU,CAAC;AAC5B,CAAC;AAMM,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;;IACrB,YAAY,GAAG,IAAI,GAAG,EAAuC,CAAC;IAC9D,eAAe,CAAkB;IAEjC,MAAM,CAAU,oBAAoB,GAAG,KAAK,CAAC;IAE7C,MAAM,CAAU,mBAAmB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IAM5D,YAAY;QAEV,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,EAAE,oBAAkB,CAAC,mBAAmB,CAAC,CAAC;IAC7C,CAAC;IAMD,eAAe;QACb,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACnC,CAAC;IACH,CAAC;IAKD,cAAc,CACZ,QAAgB,EAChB,UAAkB,EAClB,MAAuB;QAEvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAGvB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAC7C,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;YAGrD,IAAI,WAAW,CAAC,IAAI,IAAI,oBAAkB,CAAC,oBAAoB,EAAE,CAAC;gBAEhE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;gBAClD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;gBACrE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAEvB,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBAEN,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;oBACzE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;oBACjD,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAGD,IAAI,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBAEpC,KAAK,GAAG;oBACN,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,GAAG,GAAG,QAAQ;iBAC1B,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrC,CAAC;YAGD,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;gBAC9B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,SAAS,EAAE,CAAC;oBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B,CAAC;YACJ,CAAC;YAGD,KAAK,CAAC,KAAK,EAAE,CAAC;YAEd,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,MAAM,CAAC,GAAG,GAAG,KAAK,CAAC,KAAK;gBACnC,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAIf,oBAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,yBAAyB,EAAE;gBAC5D,SAAS,EAAE,aAAa;gBACxB,QAAQ,EAAE,IAAA,+BAAe,EAAC,QAAQ,CAAC;gBACnC,UAAU,EAAE,IAAA,+BAAe,EAAC,UAAU,CAAC;gBACvC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YAGH,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;aAC9B,CAAC;QACJ,CAAC;IACH,CAAC;IAKD,OAAO;QACL,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;oBAC1B,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;YACD,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,QAAiB;QACrB,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;;AA1IU,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,mBAAU,GAAE;GACA,kBAAkB,CA2I9B"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
export type CircuitState = "closed" | "open" | "half-open";
|
|
2
|
+
export interface CircuitBreakerConfig {
|
|
3
|
+
errorThreshold: number;
|
|
4
|
+
timeWindow: number;
|
|
5
|
+
minRequests: number;
|
|
6
|
+
halfOpenTimeout: number;
|
|
7
|
+
}
|
|
8
|
+
export interface CircuitBreakerState {
|
|
9
|
+
state: CircuitState;
|
|
10
|
+
failures: number;
|
|
11
|
+
successes: number;
|
|
12
|
+
lastFailureTime?: number;
|
|
13
|
+
lastSuccessTime?: number;
|
|
14
|
+
nextAttemptTime?: number;
|
|
15
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"circuit-breaker.interface.js","sourceRoot":"","sources":["../../../src/core/resilience/circuit-breaker.interface.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { CircuitBreakerConfig, CircuitState } from "./circuit-breaker.interface";
|
|
2
|
+
export declare class CircuitBreakerService {
|
|
3
|
+
private circuits;
|
|
4
|
+
private configs;
|
|
5
|
+
private getCircuit;
|
|
6
|
+
isOpen(toolName: string, config?: CircuitBreakerConfig): boolean;
|
|
7
|
+
recordSuccess(toolName: string): void;
|
|
8
|
+
recordFailure(toolName: string): void;
|
|
9
|
+
getState(toolName: string): CircuitState;
|
|
10
|
+
reset(toolName?: string): void;
|
|
11
|
+
}
|