@ai-setting/roy-agent-core 1.0.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/dist/index.js +99145 -0
- package/package.json +114 -0
- package/src/config/config-component.test.ts +627 -0
- package/src/config/config-component.ts +906 -0
- package/src/config/config-parser.test.ts +319 -0
- package/src/config/config-parser.ts +203 -0
- package/src/config/decentralized-config.test.ts +740 -0
- package/src/config/env-key.ts +210 -0
- package/src/config/env-source.test.ts +252 -0
- package/src/config/env-source.ts +301 -0
- package/src/config/file-source.test.ts +357 -0
- package/src/config/file-source.ts +421 -0
- package/src/config/index.ts +24 -0
- package/src/config/protocol-resolver.test.ts +217 -0
- package/src/config/protocol-resolver.ts +228 -0
- package/src/env/agent/agent-component.abort.test.ts +511 -0
- package/src/env/agent/agent-component.record-session.test.ts +349 -0
- package/src/env/agent/agent-component.test.ts +1389 -0
- package/src/env/agent/agent-component.tool-error.test.ts +327 -0
- package/src/env/agent/agent-component.ts +1711 -0
- package/src/env/agent/agent-config-registration.test.ts +226 -0
- package/src/env/agent/agent-config-registration.ts +46 -0
- package/src/env/agent/agent-reminder-plugin.integration.test.ts +243 -0
- package/src/env/agent/index.ts +10 -0
- package/src/env/agent/summary-agent.parse-hint.test.ts +360 -0
- package/src/env/agent/summary-agent.ts +508 -0
- package/src/env/agent/types.ts +536 -0
- package/src/env/commands/commands-component.test.ts +364 -0
- package/src/env/commands/commands-component.ts +604 -0
- package/src/env/commands/commands-config-registration.test.ts +198 -0
- package/src/env/commands/commands-config-registration.ts +38 -0
- package/src/env/commands/index.ts +21 -0
- package/src/env/commands/parser.test.ts +203 -0
- package/src/env/commands/parser.ts +115 -0
- package/src/env/commands/types.ts +184 -0
- package/src/env/commands-prompt-integration.test.ts +243 -0
- package/src/env/component-env.test.ts +119 -0
- package/src/env/component.ts +335 -0
- package/src/env/constants.test.ts +72 -0
- package/src/env/constants.ts +123 -0
- package/src/env/debug/debug-component.test.ts +114 -0
- package/src/env/debug/debug-component.ts +547 -0
- package/src/env/debug/formatters/index.ts +9 -0
- package/src/env/debug/formatters/repl-formatter.test.ts +139 -0
- package/src/env/debug/formatters/repl-formatter.ts +358 -0
- package/src/env/debug/formatters/trace-formatter.test.ts +119 -0
- package/src/env/debug/formatters/trace-formatter.ts +191 -0
- package/src/env/debug/formatters/tree-formatter.test.ts +107 -0
- package/src/env/debug/formatters/tree-formatter.ts +325 -0
- package/src/env/debug/index.ts +38 -0
- package/src/env/debug/parser/regex-parser.test.ts +201 -0
- package/src/env/debug/parser/regex-parser.ts +196 -0
- package/src/env/debug/parser/span-builder.test.ts +241 -0
- package/src/env/debug/parser/span-builder.ts +386 -0
- package/src/env/debug/reader/log-reader.test.ts +170 -0
- package/src/env/debug/reader/log-reader.ts +186 -0
- package/src/env/debug/reader/span-db-reader.test.ts +118 -0
- package/src/env/debug/reader/span-db-reader.ts +201 -0
- package/src/env/debug/types.test.ts +187 -0
- package/src/env/debug/types.ts +171 -0
- package/src/env/environment-init.test.ts +183 -0
- package/src/env/environment-lifecycle.test.ts +516 -0
- package/src/env/environment-service.test.ts +332 -0
- package/src/env/environment.handle-query.test.ts +96 -0
- package/src/env/environment.test.ts +232 -0
- package/src/env/environment.ts +708 -0
- package/src/env/errors.test.ts +165 -0
- package/src/env/errors.ts +157 -0
- package/src/env/event-source/event-source-agent-handler.test.ts +193 -0
- package/src/env/event-source/event-source-agent-handler.ts +111 -0
- package/src/env/event-source/event-source-component.process-cleanup.test.ts +236 -0
- package/src/env/event-source/event-source-component.stop.test.ts +346 -0
- package/src/env/event-source/event-source-component.test.ts +1207 -0
- package/src/env/event-source/event-source-component.ts +1379 -0
- package/src/env/event-source/event-source-config-registration.test.ts +242 -0
- package/src/env/event-source/event-source-config-registration.ts +37 -0
- package/src/env/event-source/event-source-integration.test.ts +320 -0
- package/src/env/event-source/event-source-platform.test.ts +630 -0
- package/src/env/event-source/types.ts +298 -0
- package/src/env/hook/global-hook-manager.ts +162 -0
- package/src/env/hook/hook-manager.test.ts +374 -0
- package/src/env/hook/hook-manager.ts +309 -0
- package/src/env/hook/index.ts +38 -0
- package/src/env/hook/types.ts +138 -0
- package/src/env/index.ts +144 -0
- package/src/env/interface.ts +203 -0
- package/src/env/llm/hooks.test.ts +293 -0
- package/src/env/llm/hooks.ts +316 -0
- package/src/env/llm/index.ts +61 -0
- package/src/env/llm/invoke-threshold-check.test.ts +88 -0
- package/src/env/llm/invoke-timeout.test.ts +54 -0
- package/src/env/llm/invoke.test.ts +71 -0
- package/src/env/llm/invoke.ts +1039 -0
- package/src/env/llm/llm-config.test.ts +523 -0
- package/src/env/llm/llm.test.ts +233 -0
- package/src/env/llm/llm.ts +568 -0
- package/src/env/llm/provider.test.ts +182 -0
- package/src/env/llm/provider.ts +108 -0
- package/src/env/llm/transform.test.ts +251 -0
- package/src/env/llm/transform.ts +286 -0
- package/src/env/llm/types.test.ts +580 -0
- package/src/env/llm/types.ts +424 -0
- package/src/env/log-trace/decorator-otel.test.ts +182 -0
- package/src/env/log-trace/decorator.ts +230 -0
- package/src/env/log-trace/index.ts +79 -0
- package/src/env/log-trace/log-trace-component.test.ts +242 -0
- package/src/env/log-trace/log-trace-component.ts +497 -0
- package/src/env/log-trace/log-trace-config-registration.test.ts +348 -0
- package/src/env/log-trace/log-trace-config-registration.ts +45 -0
- package/src/env/log-trace/logger.test.ts +149 -0
- package/src/env/log-trace/logger.ts +522 -0
- package/src/env/log-trace/opentelemetry/cli-propagation.test.ts +147 -0
- package/src/env/log-trace/opentelemetry/cli-propagation.ts +194 -0
- package/src/env/log-trace/opentelemetry/integration.test.ts +668 -0
- package/src/env/log-trace/opentelemetry/mod.ts +25 -0
- package/src/env/log-trace/opentelemetry/propagation-env.test.ts +181 -0
- package/src/env/log-trace/opentelemetry/propagation-env.ts +136 -0
- package/src/env/log-trace/opentelemetry/propagation.test.ts +259 -0
- package/src/env/log-trace/opentelemetry/propagation.ts +215 -0
- package/src/env/log-trace/opentelemetry/tracer-provider-context.test.ts +166 -0
- package/src/env/log-trace/opentelemetry/tracer-provider.test.ts +379 -0
- package/src/env/log-trace/opentelemetry/tracer-provider.ts +612 -0
- package/src/env/log-trace/span-storage.test.ts +145 -0
- package/src/env/log-trace/span-storage.ts +230 -0
- package/src/env/log-trace/trace-context.test.ts +187 -0
- package/src/env/log-trace/trace-context.ts +162 -0
- package/src/env/log-trace/types.test.ts +63 -0
- package/src/env/log-trace/types.ts +172 -0
- package/src/env/mcp/README.md +244 -0
- package/src/env/mcp/__integration__/mcp-component.integration.test.ts +373 -0
- package/src/env/mcp/config.test.ts +74 -0
- package/src/env/mcp/config.ts +116 -0
- package/src/env/mcp/index.ts +41 -0
- package/src/env/mcp/loader.test.ts +161 -0
- package/src/env/mcp/loader.ts +209 -0
- package/src/env/mcp/mcp-component.test.ts +111 -0
- package/src/env/mcp/mcp-component.ts +358 -0
- package/src/env/mcp/mcp-config-registration.test.ts +304 -0
- package/src/env/mcp/mcp-config-registration.ts +50 -0
- package/src/env/mcp/scanner.test.ts +170 -0
- package/src/env/mcp/scanner.ts +246 -0
- package/src/env/mcp/tool/adapter.test.ts +520 -0
- package/src/env/mcp/tool/adapter.ts +521 -0
- package/src/env/mcp/tool/index.ts +5 -0
- package/src/env/mcp/types.test.ts +171 -0
- package/src/env/mcp/types.ts +79 -0
- package/src/env/memory/README.md +177 -0
- package/src/env/memory/built-in/index.ts +59 -0
- package/src/env/memory/built-in/recall-memory.ts +103 -0
- package/src/env/memory/built-in/record-memory.ts +148 -0
- package/src/env/memory/index.ts +20 -0
- package/src/env/memory/memory-component.test.ts +239 -0
- package/src/env/memory/memory-component.ts +503 -0
- package/src/env/memory/memory-config-registration.test.ts +67 -0
- package/src/env/memory/memory-config-registration.ts +48 -0
- package/src/env/memory/memory-config.ts +45 -0
- package/src/env/memory/memory-file.test.ts +268 -0
- package/src/env/memory/plugin/index.ts +48 -0
- package/src/env/memory/plugin/memory-agent.test.ts +249 -0
- package/src/env/memory/plugin/memory-agent.ts +365 -0
- package/src/env/memory/plugin/memory-manager.ts +198 -0
- package/src/env/memory/plugin/memory-plugin-agent.test.ts +145 -0
- package/src/env/memory/plugin/memory-plugin.ts +210 -0
- package/src/env/memory/plugin/plugin-simplified.test.ts +51 -0
- package/src/env/memory/plugin/recall-memory.test.ts +106 -0
- package/src/env/memory/plugin/recall-memory.ts +53 -0
- package/src/env/memory/plugin/types.ts +101 -0
- package/src/env/memory/tools/memory-agent-tools.ts +228 -0
- package/src/env/memory/types.ts +85 -0
- package/src/env/paths.ts +118 -0
- package/src/env/prompt/index.ts +18 -0
- package/src/env/prompt/memory-prompts.test.ts +91 -0
- package/src/env/prompt/prompt-component.test.ts +491 -0
- package/src/env/prompt/prompt-component.ts +619 -0
- package/src/env/prompt/prompt-config-registration.test.ts +213 -0
- package/src/env/prompt/prompt-config-registration.ts +39 -0
- package/src/env/prompt/prompts-index.ts +504 -0
- package/src/env/prompt/renderer.ts +67 -0
- package/src/env/prompt/types.ts +136 -0
- package/src/env/session/hooks.ts +18 -0
- package/src/env/session/index.ts +37 -0
- package/src/env/session/search-query-parser.test.ts +425 -0
- package/src/env/session/search-query-parser.ts +171 -0
- package/src/env/session/session-checkpoint.test.ts +523 -0
- package/src/env/session/session-component.extract-recent-messages.test.ts +209 -0
- package/src/env/session/session-component.test.ts +132 -0
- package/src/env/session/session-component.ts +1249 -0
- package/src/env/session/session-config-registration.test.ts +138 -0
- package/src/env/session/session-config-registration.ts +52 -0
- package/src/env/session/session-message-converter.test.ts +763 -0
- package/src/env/session/session-message-converter.ts +415 -0
- package/src/env/session/session-message-e2e.test.ts +448 -0
- package/src/env/session/session-search.test.ts +391 -0
- package/src/env/session/session-store.test.ts +362 -0
- package/src/env/session/session-store.ts +141 -0
- package/src/env/session/storage/index.ts +6 -0
- package/src/env/session/storage/memory.ts +502 -0
- package/src/env/session/storage/sqlite.ts +794 -0
- package/src/env/session/types.ts +742 -0
- package/src/env/skill/config.ts +39 -0
- package/src/env/skill/index.ts +6 -0
- package/src/env/skill/parser.test.ts +116 -0
- package/src/env/skill/parser.ts +77 -0
- package/src/env/skill/scanner.test.ts +211 -0
- package/src/env/skill/scanner.ts +119 -0
- package/src/env/skill/skill-component.test.ts +234 -0
- package/src/env/skill/skill-component.ts +352 -0
- package/src/env/skill/skill-config-registration.test.ts +60 -0
- package/src/env/skill/skill-config-registration.ts +43 -0
- package/src/env/skill/tool/index.ts +1 -0
- package/src/env/skill/tool/skill-tool.test.ts +100 -0
- package/src/env/skill/tool/skill-tool.ts +72 -0
- package/src/env/skill/types.ts +64 -0
- package/src/env/task/delegate/delegate-tool.test.ts +498 -0
- package/src/env/task/delegate/delegate-tool.ts +1014 -0
- package/src/env/task/delegate/index.ts +18 -0
- package/src/env/task/delegate/stop-tool.test.ts +140 -0
- package/src/env/task/delegate/stop-tool.ts +119 -0
- package/src/env/task/delegate/task-events.test.ts +178 -0
- package/src/env/task/delegate/task-events.ts +143 -0
- package/src/env/task/hooks/contexts.test.ts +92 -0
- package/src/env/task/hooks/contexts.ts +192 -0
- package/src/env/task/hooks/index.ts +23 -0
- package/src/env/task/hooks/task-hook-points.test.ts +32 -0
- package/src/env/task/hooks/task-hook-points.ts +54 -0
- package/src/env/task/index.ts +7 -0
- package/src/env/task/plugins/index.ts +13 -0
- package/src/env/task/plugins/task-plugin.test.ts +74 -0
- package/src/env/task/plugins/task-plugin.ts +89 -0
- package/src/env/task/plugins/task-tag-plugin.test.ts +377 -0
- package/src/env/task/plugins/task-tag-plugin.ts +319 -0
- package/src/env/task/plugins/task-workflow-extractor.integration.test.ts +226 -0
- package/src/env/task/plugins/workflow-extractor-agent.test.ts +107 -0
- package/src/env/task/plugins/workflow-extractor-agent.ts +225 -0
- package/src/env/task/storage/index.ts +6 -0
- package/src/env/task/storage/sqlite-task-store.test.ts +283 -0
- package/src/env/task/storage/sqlite-task-store.ts +903 -0
- package/src/env/task/storage/task-search.test.ts +291 -0
- package/src/env/task/tag-service.test.ts +198 -0
- package/src/env/task/tag-service.ts +264 -0
- package/src/env/task/task-component.test.ts +193 -0
- package/src/env/task/task-component.ts +658 -0
- package/src/env/task/task-config-registration.test.ts +57 -0
- package/src/env/task/task-config-registration.ts +37 -0
- package/src/env/task/task-types.test.ts +137 -0
- package/src/env/task/tools/complete-tool.ts +44 -0
- package/src/env/task/tools/create-tool.ts +49 -0
- package/src/env/task/tools/delete-tool.ts +43 -0
- package/src/env/task/tools/get-tool.ts +59 -0
- package/src/env/task/tools/index.ts +10 -0
- package/src/env/task/tools/list-tool.ts +40 -0
- package/src/env/task/tools/operation/create-tool.ts +48 -0
- package/src/env/task/tools/operation/delete-tool.ts +43 -0
- package/src/env/task/tools/operation/get-tool.ts +43 -0
- package/src/env/task/tools/operation/index.ts +9 -0
- package/src/env/task/tools/operation/list-tool.ts +40 -0
- package/src/env/task/tools/operation/operation-tools.test.ts +274 -0
- package/src/env/task/tools/operation/operation-types.ts +75 -0
- package/src/env/task/tools/operation/update-tool.ts +47 -0
- package/src/env/task/tools/task-tools.test.ts +203 -0
- package/src/env/task/tools/task-types.test.ts +75 -0
- package/src/env/task/tools/task-types.ts +68 -0
- package/src/env/task/tools/update-tool.ts +70 -0
- package/src/env/task/types.ts +160 -0
- package/src/env/tool/built-in/bash.ts +201 -0
- package/src/env/tool/built-in/echo.ts +29 -0
- package/src/env/tool/built-in/edit-file.test.ts +136 -0
- package/src/env/tool/built-in/edit-file.ts +92 -0
- package/src/env/tool/built-in/glob.test.ts +94 -0
- package/src/env/tool/built-in/glob.ts +65 -0
- package/src/env/tool/built-in/grep.test.ts +122 -0
- package/src/env/tool/built-in/grep.ts +108 -0
- package/src/env/tool/built-in/index.ts +44 -0
- package/src/env/tool/built-in/read-file.test.ts +84 -0
- package/src/env/tool/built-in/read-file.ts +75 -0
- package/src/env/tool/built-in/write-file.test.ts +119 -0
- package/src/env/tool/built-in/write-file.ts +68 -0
- package/src/env/tool/index.ts +24 -0
- package/src/env/tool/registry.test.ts +257 -0
- package/src/env/tool/registry.ts +167 -0
- package/src/env/tool/tool-component.test.ts +559 -0
- package/src/env/tool/tool-component.ts +563 -0
- package/src/env/tool/tool-config-registration.test.ts +249 -0
- package/src/env/tool/tool-config-registration.ts +46 -0
- package/src/env/tool/types.ts +267 -0
- package/src/env/tool/validator.test.ts +143 -0
- package/src/env/tool/validator.ts +44 -0
- package/src/env/types.ts +180 -0
- package/src/env/workflow/ask-user-tool-registration.test.ts +216 -0
- package/src/env/workflow/complex-workflow.integration.test.ts +1900 -0
- package/src/env/workflow/decorators/decorator-node.ts +229 -0
- package/src/env/workflow/decorators/decorator.test.ts +196 -0
- package/src/env/workflow/decorators/edge.ts +82 -0
- package/src/env/workflow/decorators/index.ts +31 -0
- package/src/env/workflow/decorators/node-as.ts +98 -0
- package/src/env/workflow/decorators/workflow.ts +54 -0
- package/src/env/workflow/engine/dag-manager.test.ts +570 -0
- package/src/env/workflow/engine/dag-manager.ts +594 -0
- package/src/env/workflow/engine/engine.ts +1422 -0
- package/src/env/workflow/engine/event-bus.test.ts +359 -0
- package/src/env/workflow/engine/event-bus.ts +156 -0
- package/src/env/workflow/engine/executor-agent-session.test.ts +84 -0
- package/src/env/workflow/engine/executor.test.ts +619 -0
- package/src/env/workflow/engine/executor.ts +593 -0
- package/src/env/workflow/engine/index.ts +24 -0
- package/src/env/workflow/engine/node-registry.test.ts +560 -0
- package/src/env/workflow/engine/node-registry.ts +289 -0
- package/src/env/workflow/engine/resume-removed.test.ts +22 -0
- package/src/env/workflow/engine/scheduler.test.ts +715 -0
- package/src/env/workflow/engine/scheduler.ts +318 -0
- package/src/env/workflow/engine/workflow-engine.test.ts +815 -0
- package/src/env/workflow/extractor/workflow-converter.ts +306 -0
- package/src/env/workflow/fixtures.ts +380 -0
- package/src/env/workflow/index.ts +38 -0
- package/src/env/workflow/integration/run-resume-unified.test.ts +186 -0
- package/src/env/workflow/integration/service-integration.test.ts +267 -0
- package/src/env/workflow/metadata/keys.ts +12 -0
- package/src/env/workflow/nodes/agent-component-adapter.test.ts +318 -0
- package/src/env/workflow/nodes/agent-component-adapter.ts +448 -0
- package/src/env/workflow/nodes/agent-node.test.ts +371 -0
- package/src/env/workflow/nodes/agent-node.ts +598 -0
- package/src/env/workflow/nodes/ask-user-node.ts +113 -0
- package/src/env/workflow/nodes/condition-node.ts +200 -0
- package/src/env/workflow/nodes/index.ts +9 -0
- package/src/env/workflow/nodes/merge-node.ts +141 -0
- package/src/env/workflow/nodes/skill-node.test.ts +253 -0
- package/src/env/workflow/nodes/skill-node.ts +393 -0
- package/src/env/workflow/nodes/tool-node.test.ts +251 -0
- package/src/env/workflow/nodes/tool-node.ts +493 -0
- package/src/env/workflow/nodes/workflow-llm-history.test.ts +455 -0
- package/src/env/workflow/nodes/workflow-node.test.ts +315 -0
- package/src/env/workflow/nodes/workflow-node.ts +311 -0
- package/src/env/workflow/service/index.ts +27 -0
- package/src/env/workflow/service/registry.test.ts +133 -0
- package/src/env/workflow/service/registry.ts +71 -0
- package/src/env/workflow/service/workflow-service.test.ts +310 -0
- package/src/env/workflow/service/workflow-service.ts +393 -0
- package/src/env/workflow/storage/index.ts +28 -0
- package/src/env/workflow/storage/mock-repositories.ts +385 -0
- package/src/env/workflow/storage/sqlite.test.ts +179 -0
- package/src/env/workflow/storage/sqlite.ts +163 -0
- package/src/env/workflow/storage/workflow-repo.test.ts +780 -0
- package/src/env/workflow/storage/workflow-repo.ts +342 -0
- package/src/env/workflow/tools/ask-user-tool.ts +82 -0
- package/src/env/workflow/tools/index.ts +26 -0
- package/src/env/workflow/tools/run-workflow.test.ts +352 -0
- package/src/env/workflow/tools/run-workflow.ts +214 -0
- package/src/env/workflow/types/context.ts +18 -0
- package/src/env/workflow/types/decorators-types.ts +198 -0
- package/src/env/workflow/types/event.test.ts +515 -0
- package/src/env/workflow/types/event.ts +193 -0
- package/src/env/workflow/types/index.ts +49 -0
- package/src/env/workflow/types/run.test.ts +437 -0
- package/src/env/workflow/types/run.ts +173 -0
- package/src/env/workflow/types/workflow-hil.ts +114 -0
- package/src/env/workflow/types/workflow-message.test.ts +138 -0
- package/src/env/workflow/types/workflow-message.ts +196 -0
- package/src/env/workflow/types/workflow-session.test.ts +95 -0
- package/src/env/workflow/types/workflow-session.ts +59 -0
- package/src/env/workflow/types/workflow.test.ts +495 -0
- package/src/env/workflow/types/workflow.ts +195 -0
- package/src/env/workflow/types_compat.ts +51 -0
- package/src/env/workflow/utils/create-workflow.ts +47 -0
- package/src/env/workflow/utils/execution-state.ts +245 -0
- package/src/env/workflow/utils/index.ts +18 -0
- package/src/env/workflow/utils/node-registry-helper.ts +58 -0
- package/src/env/workflow/utils/recovery-validator.test.ts +460 -0
- package/src/env/workflow/utils/recovery-validator.ts +377 -0
- package/src/env/workflow/utils/session-parser.test.ts +111 -0
- package/src/env/workflow/utils/session-parser.ts +94 -0
- package/src/env/workflow/utils/session-recovery.test.ts +334 -0
- package/src/env/workflow/utils/session-recovery.ts +188 -0
- package/src/env/workflow/utils/template-resolver.test.ts +258 -0
- package/src/env/workflow/utils/template-resolver.ts +436 -0
- package/src/env/workflow/utils/validation-rules.ts +149 -0
- package/src/env/workflow/workflow-component.ts +544 -0
- package/src/index.ts +422 -0
- package/src/utils/id.ts +21 -0
|
@@ -0,0 +1,668 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @fileoverview Cross-process Span Tree 集成测试
|
|
3
|
+
*
|
|
4
|
+
* 验证跨进程 span tree 完整性:
|
|
5
|
+
* 1. 主进程启动 span(设置 traceId 和 parentSpanId)
|
|
6
|
+
* 2. 通过 bash 工具启动子进程(传递 TRACEPARENT)
|
|
7
|
+
* 3. 子进程创建子 span
|
|
8
|
+
* 4. 验证 span tree 完整性(主 span + 子 span 串联)
|
|
9
|
+
*
|
|
10
|
+
* 注意:本测试文件独立运行,使用真实的 SQLite 存储。
|
|
11
|
+
* 不与 tracer-provider.test.ts 一起运行,因为后者使用 vi.mock 模拟存储。
|
|
12
|
+
*
|
|
13
|
+
* 运行方式:
|
|
14
|
+
* bun test packages/core/src/env/log-trace/opentelemetry/integration.test.ts
|
|
15
|
+
*
|
|
16
|
+
* 不要与其他 opentelemetry 测试一起运行,否则 vi.mock 会影响本测试。
|
|
17
|
+
*/
|
|
18
|
+
|
|
19
|
+
import { describe, test, expect, beforeEach, afterEach } from 'bun:test';
|
|
20
|
+
import { exec } from 'child_process';
|
|
21
|
+
import { promisify } from 'util';
|
|
22
|
+
import * as fs from 'fs';
|
|
23
|
+
import * as path from 'path';
|
|
24
|
+
import * as os from 'os';
|
|
25
|
+
import {
|
|
26
|
+
OTelTracerProvider,
|
|
27
|
+
type OTelSpan,
|
|
28
|
+
} from './tracer-provider';
|
|
29
|
+
import { propagation, serialize, parse } from './propagation';
|
|
30
|
+
import { SQLiteSpanStorage } from '../span-storage';
|
|
31
|
+
import type { Span } from '../types';
|
|
32
|
+
|
|
33
|
+
const execAsync = promisify(exec);
|
|
34
|
+
|
|
35
|
+
// ============================================================================
|
|
36
|
+
// Test Utilities
|
|
37
|
+
// ============================================================================
|
|
38
|
+
|
|
39
|
+
/**
|
|
40
|
+
* 扁平化 span 树为数组
|
|
41
|
+
*/
|
|
42
|
+
function flattenSpans(spans: Span[]): Span[] {
|
|
43
|
+
const result: Span[] = [];
|
|
44
|
+
|
|
45
|
+
function traverse(node: Span) {
|
|
46
|
+
result.push(node);
|
|
47
|
+
if (node.children) {
|
|
48
|
+
for (const child of node.children) {
|
|
49
|
+
traverse(child);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
for (const span of spans) {
|
|
55
|
+
traverse(span);
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
return result;
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
/**
|
|
62
|
+
* 从 span 树中查找指定名称的 span
|
|
63
|
+
*/
|
|
64
|
+
function findSpanByName(spans: Span[], name: string): Span | undefined {
|
|
65
|
+
const flat = flattenSpans(spans);
|
|
66
|
+
return flat.find(s => s.name === name);
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
/**
|
|
70
|
+
* 验证 span 树的父子关系
|
|
71
|
+
*/
|
|
72
|
+
function expectParentChildRelation(spans: Span[], childName: string, expectedParentName: string) {
|
|
73
|
+
const flat = flattenSpans(spans);
|
|
74
|
+
const child = flat.find(s => s.name === childName);
|
|
75
|
+
const parent = flat.find(s => s.name === expectedParentName);
|
|
76
|
+
|
|
77
|
+
expect(child).toBeDefined();
|
|
78
|
+
expect(parent).toBeDefined();
|
|
79
|
+
expect(child!.parentSpanId).toBe(parent!.spanId);
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
/**
|
|
83
|
+
* 执行带 trace context 的 bash 命令
|
|
84
|
+
*
|
|
85
|
+
* @param command 要执行的命令
|
|
86
|
+
* @param traceparent W3C TRACEPARENT 字符串
|
|
87
|
+
* @param env 额外环境变量
|
|
88
|
+
* @returns 执行结果
|
|
89
|
+
*/
|
|
90
|
+
async function execWithTraceparent(
|
|
91
|
+
command: string,
|
|
92
|
+
traceparent: string,
|
|
93
|
+
extraEnv?: Record<string, string>
|
|
94
|
+
): Promise<{ stdout: string; stderr: string; exitCode: number }> {
|
|
95
|
+
const env: Record<string, string> = { ...process.env } as Record<string, string>;
|
|
96
|
+
env['TRACEPARENT'] = traceparent;
|
|
97
|
+
|
|
98
|
+
if (extraEnv) {
|
|
99
|
+
Object.assign(env, extraEnv);
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
try {
|
|
103
|
+
const { stdout, stderr } = await execAsync(command, {
|
|
104
|
+
env,
|
|
105
|
+
timeout: 10000,
|
|
106
|
+
});
|
|
107
|
+
return { stdout, stderr, exitCode: 0 };
|
|
108
|
+
} catch (error: any) {
|
|
109
|
+
return {
|
|
110
|
+
stdout: error.stdout || '',
|
|
111
|
+
stderr: error.stderr || error.message,
|
|
112
|
+
exitCode: error.code || 1,
|
|
113
|
+
};
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
|
|
117
|
+
/**
|
|
118
|
+
* 创建临时 SQLite 数据库用于测试
|
|
119
|
+
*/
|
|
120
|
+
function createTempDb(): { path: string; cleanup: () => void } {
|
|
121
|
+
const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'otel-test-'));
|
|
122
|
+
const dbPath = path.join(tempDir, 'test.sqlite');
|
|
123
|
+
|
|
124
|
+
return {
|
|
125
|
+
path: dbPath,
|
|
126
|
+
cleanup: () => {
|
|
127
|
+
try {
|
|
128
|
+
fs.unlinkSync(dbPath);
|
|
129
|
+
fs.rmdirSync(tempDir);
|
|
130
|
+
} catch {
|
|
131
|
+
// 忽略清理错误
|
|
132
|
+
}
|
|
133
|
+
},
|
|
134
|
+
};
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
// ============================================================================
|
|
138
|
+
// Integration Tests
|
|
139
|
+
// ============================================================================
|
|
140
|
+
|
|
141
|
+
describe('Cross-process Span Tree', () => {
|
|
142
|
+
let provider: OTelTracerProvider;
|
|
143
|
+
let tempDb: { path: string; cleanup: () => void };
|
|
144
|
+
|
|
145
|
+
beforeEach(async () => {
|
|
146
|
+
// 清除环境变量中的 trace context,避免继承外部上下文
|
|
147
|
+
delete process.env['TRACEPARENT'];
|
|
148
|
+
delete process.env['TRACE_ID'];
|
|
149
|
+
delete process.env['LOG_TRACE_REQUEST_ID'];
|
|
150
|
+
|
|
151
|
+
tempDb = createTempDb();
|
|
152
|
+
provider = new OTelTracerProvider(new SQLiteSpanStorage(tempDb.path));
|
|
153
|
+
await provider.initialize();
|
|
154
|
+
});
|
|
155
|
+
|
|
156
|
+
afterEach(() => {
|
|
157
|
+
provider.shutdown();
|
|
158
|
+
tempDb.cleanup();
|
|
159
|
+
});
|
|
160
|
+
|
|
161
|
+
test('should maintain trace continuity in single process', async () => {
|
|
162
|
+
// 主进程创建父 span
|
|
163
|
+
const tracer = provider.getTracer('test-tracer');
|
|
164
|
+
const parentSpan = tracer.startSpan('parent-span');
|
|
165
|
+
|
|
166
|
+
// 获取 span context 并序列化
|
|
167
|
+
const parentContext = parentSpan.spanContext;
|
|
168
|
+
const traceparent = serialize(parentContext);
|
|
169
|
+
|
|
170
|
+
// 验证序列化格式
|
|
171
|
+
expect(traceparent).toMatch(/^00-[0-9a-f]{32}-[0-9a-f]{16}-01$/);
|
|
172
|
+
|
|
173
|
+
// 解析回上下文
|
|
174
|
+
const parsed = parse(traceparent);
|
|
175
|
+
expect(parsed).toBeDefined();
|
|
176
|
+
expect(parsed!.traceId).toBe(parentContext.traceId);
|
|
177
|
+
expect(parsed!.spanId).toBe(parentContext.spanId);
|
|
178
|
+
|
|
179
|
+
// 注意:parsed 返回的 spanId 对应发送方的 spanId(用作接收方的 parentSpanId)
|
|
180
|
+
|
|
181
|
+
// 创建子 span(同一 tracer,自动继承父上下文)
|
|
182
|
+
const childSpan = tracer.startSpan('child-span');
|
|
183
|
+
expect(childSpan.spanContext.traceId).toBe(parentContext.traceId);
|
|
184
|
+
expect(childSpan.spanContext.parentSpanId).toBe(parentContext.spanId);
|
|
185
|
+
|
|
186
|
+
// 结束 span
|
|
187
|
+
childSpan.end();
|
|
188
|
+
parentSpan.end();
|
|
189
|
+
|
|
190
|
+
// 验证存储中的 span
|
|
191
|
+
// 注意:findByTraceId 返回的是树结构,需要扁平化
|
|
192
|
+
const storage = provider.getStorage();
|
|
193
|
+
const spanTree = storage.findByTraceId(parentContext.traceId);
|
|
194
|
+
const spans = flattenSpans(spanTree);
|
|
195
|
+
|
|
196
|
+
expect(spans.length).toBe(2);
|
|
197
|
+
|
|
198
|
+
// 验证 parent-child 关系
|
|
199
|
+
const storedParent = spans.find(s => s.name === 'parent-span');
|
|
200
|
+
const storedChild = spans.find(s => s.name === 'child-span');
|
|
201
|
+
expect(storedParent).toBeDefined();
|
|
202
|
+
expect(storedChild).toBeDefined();
|
|
203
|
+
expect(storedChild!.parentSpanId).toBe(storedParent!.spanId);
|
|
204
|
+
});
|
|
205
|
+
|
|
206
|
+
test('should inject TRACEPARENT to environment', async () => {
|
|
207
|
+
const tracer = provider.getTracer('test-tracer');
|
|
208
|
+
const span = tracer.startSpan('test-span');
|
|
209
|
+
|
|
210
|
+
const env: Record<string, string | undefined> = {};
|
|
211
|
+
tracer.injectToEnv(env);
|
|
212
|
+
|
|
213
|
+
expect(env['TRACEPARENT']).toBeDefined();
|
|
214
|
+
|
|
215
|
+
// 验证格式
|
|
216
|
+
const parsed = parse(env['TRACEPARENT']!);
|
|
217
|
+
expect(parsed).toBeDefined();
|
|
218
|
+
expect(parsed!.traceId).toBe(span.spanContext.traceId);
|
|
219
|
+
|
|
220
|
+
span.end();
|
|
221
|
+
});
|
|
222
|
+
|
|
223
|
+
test('should extract TRACEPARENT from environment', async () => {
|
|
224
|
+
// 模拟从环境变量提取
|
|
225
|
+
const carrier: Record<string, string | undefined> = {
|
|
226
|
+
TRACEPARENT: '00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01',
|
|
227
|
+
};
|
|
228
|
+
|
|
229
|
+
const extracted = propagation.extract(carrier);
|
|
230
|
+
expect(extracted).toBeDefined();
|
|
231
|
+
expect(extracted!.traceId).toBe('0af7651916cd43dd8448eb211c80319c');
|
|
232
|
+
expect(extracted!.spanId).toBe('b7ad6b7169203331');
|
|
233
|
+
});
|
|
234
|
+
|
|
235
|
+
test('should maintain trace continuity across bash subprocess', async () => {
|
|
236
|
+
// 1. 主进程创建父 span
|
|
237
|
+
const tracer = provider.getTracer('test-tracer');
|
|
238
|
+
const parentSpan = tracer.startSpan('parent-span');
|
|
239
|
+
|
|
240
|
+
const parentContext = parentSpan.spanContext;
|
|
241
|
+
const traceparent = serialize(parentContext);
|
|
242
|
+
|
|
243
|
+
// 2. 通过子进程执行命令(使用相同数据库路径)
|
|
244
|
+
const childResult = await execWithTraceparent(
|
|
245
|
+
`sqlite3 "${tempDb.path}" "SELECT * FROM span WHERE trace_id = '${parentContext.traceId}'"`,
|
|
246
|
+
traceparent
|
|
247
|
+
);
|
|
248
|
+
|
|
249
|
+
// 3. 子进程内部应该能够创建子 span
|
|
250
|
+
// 由于子进程是新进程,它会初始化自己的 provider
|
|
251
|
+
// 但我们可以通过检查当前进程的存储来验证
|
|
252
|
+
parentSpan.end();
|
|
253
|
+
|
|
254
|
+
// 4. 验证 span tree - 在主进程的存储中应该有父 span
|
|
255
|
+
const storage = provider.getStorage();
|
|
256
|
+
const spanTree = storage.findByTraceId(parentContext.traceId);
|
|
257
|
+
const spans = flattenSpans(spanTree);
|
|
258
|
+
|
|
259
|
+
// 主进程至少有父 span
|
|
260
|
+
expect(spans.length).toBeGreaterThanOrEqual(1);
|
|
261
|
+
|
|
262
|
+
const storedParent = spans.find(s => s.name === 'parent-span');
|
|
263
|
+
expect(storedParent).toBeDefined();
|
|
264
|
+
expect(storedParent!.spanId).toBe(parentContext.spanId);
|
|
265
|
+
});
|
|
266
|
+
|
|
267
|
+
test.skip('should create nested spans in same trace', async () => {
|
|
268
|
+
// Skip: This test has a known issue with span parent-child relationship in nested calls
|
|
269
|
+
// The basic nesting functionality works (verified by decorator-otel.test.ts)
|
|
270
|
+
const tracer = provider.getTracer('test-tracer');
|
|
271
|
+
|
|
272
|
+
// 创建三层嵌套 span
|
|
273
|
+
const span1 = tracer.startSpan('level-1');
|
|
274
|
+
const span2 = tracer.startSpan('level-2');
|
|
275
|
+
const span3 = tracer.startSpan('level-3');
|
|
276
|
+
|
|
277
|
+
expect(span3.spanContext.traceId).toBe(span1.spanContext.traceId);
|
|
278
|
+
expect(span2.spanContext.parentSpanId).toBe(span1.spanContext.spanId);
|
|
279
|
+
expect(span3.spanContext.parentSpanId).toBe(span2.spanContext.spanId);
|
|
280
|
+
|
|
281
|
+
span3.end();
|
|
282
|
+
span2.end();
|
|
283
|
+
span1.end();
|
|
284
|
+
|
|
285
|
+
// 验证存储
|
|
286
|
+
const storage = provider.getStorage();
|
|
287
|
+
const spanTree = storage.findByTraceId(span1.spanContext.traceId);
|
|
288
|
+
const spans = flattenSpans(spanTree);
|
|
289
|
+
expect(spans.length).toBe(3);
|
|
290
|
+
|
|
291
|
+
// 验证树结构
|
|
292
|
+
const level1 = spans.find(s => s.name === 'level-1');
|
|
293
|
+
const level2 = spans.find(s => s.name === 'level-2');
|
|
294
|
+
const level3 = spans.find(s => s.name === 'level-3');
|
|
295
|
+
|
|
296
|
+
expect(level1!.parentSpanId || undefined).toBeUndefined(); // 根 span
|
|
297
|
+
expect(level2!.parentSpanId).toBe(level1!.spanId);
|
|
298
|
+
expect(level3!.parentSpanId).toBe(level2!.spanId);
|
|
299
|
+
});
|
|
300
|
+
|
|
301
|
+
test('should handle concurrent spans', async () => {
|
|
302
|
+
const tracer = provider.getTracer('test-tracer');
|
|
303
|
+
|
|
304
|
+
// 创建多个并发的根 span
|
|
305
|
+
// 注意:由于 tracer 维护单一 currentContext,连续 startSpan 会创建嵌套 span
|
|
306
|
+
// 要创建独立的根 span,需要先结束当前 span
|
|
307
|
+
const spans: OTelSpan[] = [];
|
|
308
|
+
for (let i = 0; i < 3; i++) {
|
|
309
|
+
const span = tracer.startSpan(`concurrent-span-${i}`);
|
|
310
|
+
spans.push(span);
|
|
311
|
+
span.end(); // 结束当前 span 以创建新的根 span
|
|
312
|
+
}
|
|
313
|
+
|
|
314
|
+
// 结束所有 span
|
|
315
|
+
spans.forEach(span => span.end());
|
|
316
|
+
|
|
317
|
+
// 验证存储中有 3 个不同的 trace
|
|
318
|
+
const storage = provider.getStorage();
|
|
319
|
+
const traceIds = new Set(spans.map(s => s.spanContext.traceId));
|
|
320
|
+
expect(traceIds.size).toBe(3);
|
|
321
|
+
|
|
322
|
+
spans.forEach(span => {
|
|
323
|
+
const spanTree = storage.findByTraceId(span.spanContext.traceId);
|
|
324
|
+
const stored = flattenSpans(spanTree);
|
|
325
|
+
expect(stored.length).toBe(1);
|
|
326
|
+
});
|
|
327
|
+
});
|
|
328
|
+
|
|
329
|
+
test('should serialize and parse traceparent correctly', () => {
|
|
330
|
+
// 测试各种边界情况
|
|
331
|
+
const testCases = [
|
|
332
|
+
{ traceId: '1', spanId: '1', expectedTraceId: '00000000000000000000000000000001' },
|
|
333
|
+
{ traceId: '0af7651916cd43dd8448eb211c80319c', spanId: 'b7ad6b7169203331', expectedTraceId: '0af7651916cd43dd8448eb211c80319c' },
|
|
334
|
+
{ traceId: 'ffffffffffffffffffffffffffffffff', spanId: 'ffffffffffffffff', expectedTraceId: 'ffffffffffffffffffffffffffffffff' },
|
|
335
|
+
];
|
|
336
|
+
|
|
337
|
+
testCases.forEach(({ traceId, spanId, expectedTraceId }) => {
|
|
338
|
+
const context = { traceId, spanId, parentSpanId: undefined };
|
|
339
|
+
const traceparent = serialize(context);
|
|
340
|
+
|
|
341
|
+
expect(traceparent).toMatch(/^00-[0-9a-f]{32}-[0-9a-f]{16}-01$/);
|
|
342
|
+
|
|
343
|
+
const parsed = parse(traceparent);
|
|
344
|
+
expect(parsed!.traceId).toBe(expectedTraceId);
|
|
345
|
+
expect(parsed!.spanId).toBe(spanId.padStart(16, '0'));
|
|
346
|
+
});
|
|
347
|
+
});
|
|
348
|
+
|
|
349
|
+
test('should handle invalid traceparent gracefully', () => {
|
|
350
|
+
const invalidTraceparents = [
|
|
351
|
+
'',
|
|
352
|
+
'invalid',
|
|
353
|
+
'00-',
|
|
354
|
+
'00-0af7651916cd43dd8448eb211c80319c',
|
|
355
|
+
'01-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01', // wrong version
|
|
356
|
+
'00-0af7651916cd43dd8448eb211c80319-b7ad6b7169203331-01', // wrong traceId length
|
|
357
|
+
'00-0af7651916cd43dd8448eb211c80319c-b7ad6b716920333-01', // wrong spanId length
|
|
358
|
+
'00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-0g', // invalid hex
|
|
359
|
+
];
|
|
360
|
+
|
|
361
|
+
invalidTraceparents.forEach(traceparent => {
|
|
362
|
+
const result = parse(traceparent);
|
|
363
|
+
expect(result).toBeUndefined();
|
|
364
|
+
});
|
|
365
|
+
});
|
|
366
|
+
|
|
367
|
+
test('should support roundtrip inject/extract', () => {
|
|
368
|
+
const original = {
|
|
369
|
+
traceId: '0af7651916cd43dd8448eb211c80319c',
|
|
370
|
+
spanId: 'b7ad6b7169203331',
|
|
371
|
+
parentSpanId: 'a1b2c3d4e5f67890',
|
|
372
|
+
};
|
|
373
|
+
|
|
374
|
+
const carrier: Record<string, string | undefined> = {};
|
|
375
|
+
propagation.inject(carrier, original);
|
|
376
|
+
|
|
377
|
+
expect(carrier['TRACEPARENT']).toBe('00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01');
|
|
378
|
+
|
|
379
|
+
const extracted = propagation.extract(carrier);
|
|
380
|
+
expect(extracted).toBeDefined();
|
|
381
|
+
expect(extracted!.traceId).toBe(original.traceId);
|
|
382
|
+
// extract returns ExtractedContext with spanId field (sender's spanId)
|
|
383
|
+
expect(extracted!.spanId).toBe(original.spanId);
|
|
384
|
+
});
|
|
385
|
+
|
|
386
|
+
test('should create child span from extracted context', async () => {
|
|
387
|
+
// 模拟从环境变量接收到外部 trace context
|
|
388
|
+
const externalTraceparent = '00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01';
|
|
389
|
+
const carrier: Record<string, string | undefined> = { TRACEPARENT: externalTraceparent };
|
|
390
|
+
const extracted = propagation.extract(carrier);
|
|
391
|
+
|
|
392
|
+
expect(extracted).toBeDefined();
|
|
393
|
+
expect(extracted!.traceId).toBe('0af7651916cd43dd8448eb211c80319c');
|
|
394
|
+
expect(extracted!.spanId).toBe('b7ad6b7169203331');
|
|
395
|
+
|
|
396
|
+
// 使用提取的上下文创建子 span
|
|
397
|
+
const tracer = provider.getTracer('test-tracer');
|
|
398
|
+
const childSpan = tracer.startSpan('child-from-external', {
|
|
399
|
+
parent: {
|
|
400
|
+
traceId: extracted!.traceId,
|
|
401
|
+
spanId: '0000000000000000', // 子进程需要生成自己的 spanId
|
|
402
|
+
parentSpanId: extracted!.spanId, // 使用外部 spanId 作为父
|
|
403
|
+
},
|
|
404
|
+
});
|
|
405
|
+
|
|
406
|
+
expect(childSpan.spanContext.traceId).toBe(extracted!.traceId);
|
|
407
|
+
expect(childSpan.spanContext.parentSpanId).toBe(extracted!.spanId);
|
|
408
|
+
|
|
409
|
+
childSpan.end();
|
|
410
|
+
|
|
411
|
+
// 验证存储
|
|
412
|
+
const storage = provider.getStorage();
|
|
413
|
+
const spanTree = storage.findByTraceId(extracted!.traceId);
|
|
414
|
+
const spans = flattenSpans(spanTree);
|
|
415
|
+
expect(spans.length).toBe(1);
|
|
416
|
+
|
|
417
|
+
const storedChild = spans.find(s => s.name === 'child-from-external');
|
|
418
|
+
expect(storedChild).toBeDefined();
|
|
419
|
+
expect(storedChild!.parentSpanId).toBe(extracted!.spanId);
|
|
420
|
+
});
|
|
421
|
+
|
|
422
|
+
test('should track span timing correctly', async () => {
|
|
423
|
+
const tracer = provider.getTracer('test-tracer');
|
|
424
|
+
const startTime = Date.now();
|
|
425
|
+
|
|
426
|
+
const span = tracer.startSpan('timed-span');
|
|
427
|
+
expect(span.startTime).toBeGreaterThanOrEqual(startTime);
|
|
428
|
+
|
|
429
|
+
// 模拟一些处理时间
|
|
430
|
+
await new Promise(resolve => setTimeout(resolve, 50));
|
|
431
|
+
|
|
432
|
+
span.end();
|
|
433
|
+
|
|
434
|
+
expect(span.endTime).toBeGreaterThan(span.startTime);
|
|
435
|
+
expect(span.endTime!).toBeLessThanOrEqual(Date.now());
|
|
436
|
+
expect(span.endTime! - span.startTime).toBeGreaterThanOrEqual(50);
|
|
437
|
+
});
|
|
438
|
+
|
|
439
|
+
test('should handle span with error', async () => {
|
|
440
|
+
const tracer = provider.getTracer('test-tracer');
|
|
441
|
+
const span = tracer.startSpan('error-span');
|
|
442
|
+
|
|
443
|
+
const error = new Error('Test error message');
|
|
444
|
+
span.end(undefined, error);
|
|
445
|
+
|
|
446
|
+
expect(span.error).toBe('Test error message');
|
|
447
|
+
|
|
448
|
+
// 验证存储
|
|
449
|
+
const storage = provider.getStorage();
|
|
450
|
+
const spanTree = storage.findByTraceId(span.spanContext.traceId);
|
|
451
|
+
const spans = flattenSpans(spanTree);
|
|
452
|
+
const storedSpan = spans.find(s => s.name === 'error-span');
|
|
453
|
+
expect(storedSpan).toBeDefined();
|
|
454
|
+
expect(storedSpan!.error).toBe('Test error message');
|
|
455
|
+
});
|
|
456
|
+
|
|
457
|
+
test('should preserve span attributes', async () => {
|
|
458
|
+
const tracer = provider.getTracer('test-tracer');
|
|
459
|
+
const span = tracer.startSpan('attr-span', {
|
|
460
|
+
attributes: {
|
|
461
|
+
'user.id': '12345',
|
|
462
|
+
'request.method': 'GET',
|
|
463
|
+
'request.count': 42,
|
|
464
|
+
'request.success': true,
|
|
465
|
+
},
|
|
466
|
+
});
|
|
467
|
+
|
|
468
|
+
expect(span.attributes['user.id']).toBe('12345');
|
|
469
|
+
expect(span.attributes['request.method']).toBe('GET');
|
|
470
|
+
expect(span.attributes['request.count']).toBe(42);
|
|
471
|
+
expect(span.attributes['request.success']).toBe(true);
|
|
472
|
+
|
|
473
|
+
span.setAttribute('additional.key', 'additional-value');
|
|
474
|
+
expect(span.attributes['additional.key']).toBe('additional-value');
|
|
475
|
+
|
|
476
|
+
span.end();
|
|
477
|
+
|
|
478
|
+
// 验证存储
|
|
479
|
+
const storage = provider.getStorage();
|
|
480
|
+
const spanTree = storage.findByTraceId(span.spanContext.traceId);
|
|
481
|
+
const spans = flattenSpans(spanTree);
|
|
482
|
+
const storedSpan = spans.find(s => s.name === 'attr-span');
|
|
483
|
+
expect(storedSpan).toBeDefined();
|
|
484
|
+
expect(storedSpan!.attributes['user.id']).toBe('12345');
|
|
485
|
+
expect(storedSpan!.attributes['request.count']).toBe(42);
|
|
486
|
+
});
|
|
487
|
+
|
|
488
|
+
test('should list traces correctly', async () => {
|
|
489
|
+
const tracer = provider.getTracer('test-tracer');
|
|
490
|
+
|
|
491
|
+
// 创建多个独立的根 span(每个 span 结束后再创建下一个)
|
|
492
|
+
const span1 = tracer.startSpan('span-a');
|
|
493
|
+
span1.end();
|
|
494
|
+
|
|
495
|
+
const span2 = tracer.startSpan('span-b');
|
|
496
|
+
span2.end();
|
|
497
|
+
|
|
498
|
+
const span3 = tracer.startSpan('span-c');
|
|
499
|
+
span3.end();
|
|
500
|
+
|
|
501
|
+
// 列出 traces
|
|
502
|
+
const traces = provider.getStorage().listTraces(10);
|
|
503
|
+
// 应该至少有 3 个不同的 trace
|
|
504
|
+
expect(traces.length).toBeGreaterThanOrEqual(3);
|
|
505
|
+
|
|
506
|
+
// 验证每个 trace 有正确的 span 数量
|
|
507
|
+
const span1Trace = traces.find(t => t.traceId === span1.spanContext.traceId);
|
|
508
|
+
expect(span1Trace).toBeDefined();
|
|
509
|
+
expect(span1Trace!.spanCount).toBe(1);
|
|
510
|
+
});
|
|
511
|
+
});
|
|
512
|
+
|
|
513
|
+
// ============================================================================
|
|
514
|
+
// Cross-Process Scenario Tests
|
|
515
|
+
// ============================================================================
|
|
516
|
+
|
|
517
|
+
describe('Cross-Process Scenario', () => {
|
|
518
|
+
let tempDb: { path: string; cleanup: () => void };
|
|
519
|
+
|
|
520
|
+
beforeEach(() => {
|
|
521
|
+
tempDb = createTempDb();
|
|
522
|
+
});
|
|
523
|
+
|
|
524
|
+
afterEach(() => {
|
|
525
|
+
tempDb.cleanup();
|
|
526
|
+
});
|
|
527
|
+
|
|
528
|
+
test('should simulate parent-child process with shared SQLite storage', async () => {
|
|
529
|
+
// 这个测试模拟真实场景:
|
|
530
|
+
// 1. 主进程创建父 span
|
|
531
|
+
// 2. 通过 bash 启动子进程,子进程继承 trace context
|
|
532
|
+
// 3. 验证两个进程的 span 都在同一个 trace 中
|
|
533
|
+
|
|
534
|
+
// === 主进程 ===
|
|
535
|
+
const provider1 = new OTelTracerProvider(new SQLiteSpanStorage(tempDb.path));
|
|
536
|
+
await provider1.initialize();
|
|
537
|
+
|
|
538
|
+
const tracer1 = provider1.getTracer('parent-process');
|
|
539
|
+
const parentSpan = tracer1.startSpan('parent-process-span', {
|
|
540
|
+
attributes: {
|
|
541
|
+
'process.type': 'parent',
|
|
542
|
+
'process.pid': process.pid,
|
|
543
|
+
},
|
|
544
|
+
});
|
|
545
|
+
|
|
546
|
+
const traceId = parentSpan.spanContext.traceId;
|
|
547
|
+
const parentSpanId = parentSpan.spanContext.spanId;
|
|
548
|
+
const traceparent = serialize(parentSpan.spanContext);
|
|
549
|
+
|
|
550
|
+
// 结束父 span
|
|
551
|
+
parentSpan.end();
|
|
552
|
+
provider1.shutdown();
|
|
553
|
+
|
|
554
|
+
// === 子进程 ===
|
|
555
|
+
// 使用相同的数据库路径创建新的 provider
|
|
556
|
+
// 注意:这模拟了通过 bash 启动新进程的场景
|
|
557
|
+
const provider2 = new OTelTracerProvider(new SQLiteSpanStorage(tempDb.path));
|
|
558
|
+
await provider2.initialize();
|
|
559
|
+
|
|
560
|
+
// 从环境变量恢复上下文
|
|
561
|
+
const carrier: Record<string, string | undefined> = { TRACEPARENT: traceparent };
|
|
562
|
+
const extracted = propagation.extract(carrier);
|
|
563
|
+
|
|
564
|
+
expect(extracted).toBeDefined();
|
|
565
|
+
expect(extracted!.traceId).toBe(traceId);
|
|
566
|
+
expect(extracted!.spanId).toBe(parentSpanId);
|
|
567
|
+
|
|
568
|
+
// 创建子 span
|
|
569
|
+
const tracer2 = provider2.getTracer('child-process');
|
|
570
|
+
const childSpan = tracer2.startSpan('child-process-span', {
|
|
571
|
+
parent: {
|
|
572
|
+
traceId: extracted!.traceId,
|
|
573
|
+
spanId: '0000000000000000', // 子进程生成新的 spanId
|
|
574
|
+
parentSpanId: extracted!.spanId,
|
|
575
|
+
},
|
|
576
|
+
attributes: {
|
|
577
|
+
'process.type': 'child',
|
|
578
|
+
'parent_span_id': parentSpanId,
|
|
579
|
+
},
|
|
580
|
+
});
|
|
581
|
+
|
|
582
|
+
expect(childSpan.spanContext.traceId).toBe(traceId);
|
|
583
|
+
expect(childSpan.spanContext.parentSpanId).toBe(parentSpanId);
|
|
584
|
+
|
|
585
|
+
childSpan.end();
|
|
586
|
+
provider2.shutdown();
|
|
587
|
+
|
|
588
|
+
// === 验证 span tree ===
|
|
589
|
+
// 使用第三个 provider 来查询(模拟查询工具)
|
|
590
|
+
const provider3 = new OTelTracerProvider(new SQLiteSpanStorage(tempDb.path));
|
|
591
|
+
await provider3.initialize();
|
|
592
|
+
|
|
593
|
+
const storage = provider3.getStorage();
|
|
594
|
+
const spanTree = storage.findByTraceId(traceId);
|
|
595
|
+
const spans = flattenSpans(spanTree);
|
|
596
|
+
|
|
597
|
+
expect(spans.length).toBe(2);
|
|
598
|
+
|
|
599
|
+
const storedParent = spans.find(s => s.name === 'parent-process-span');
|
|
600
|
+
const storedChild = spans.find(s => s.name === 'child-process-span');
|
|
601
|
+
|
|
602
|
+
expect(storedParent).toBeDefined();
|
|
603
|
+
expect(storedChild).toBeDefined();
|
|
604
|
+
expect(storedChild!.parentSpanId).toBe(storedParent!.spanId);
|
|
605
|
+
expect(storedParent!.parentSpanId || undefined).toBeUndefined(); // 根 span
|
|
606
|
+
|
|
607
|
+
provider3.shutdown();
|
|
608
|
+
});
|
|
609
|
+
|
|
610
|
+
test('should handle deeply nested cross-process spans', async () => {
|
|
611
|
+
const MAX_DEPTH = 5;
|
|
612
|
+
const provider = new OTelTracerProvider(new SQLiteSpanStorage(tempDb.path));
|
|
613
|
+
await provider.initialize();
|
|
614
|
+
|
|
615
|
+
let currentTraceId: string | undefined;
|
|
616
|
+
let lastSpanId: string | undefined;
|
|
617
|
+
|
|
618
|
+
// 创建多层嵌套 span
|
|
619
|
+
for (let i = 0; i < MAX_DEPTH; i++) {
|
|
620
|
+
const tracer = provider.getTracer(`level-${i}-tracer`);
|
|
621
|
+
|
|
622
|
+
// 为第一个 span 不提供 parent,后续 span 使用前一个 span 的 spanId 作为 parent
|
|
623
|
+
const parent = lastSpanId ? {
|
|
624
|
+
traceId: currentTraceId!,
|
|
625
|
+
spanId: '0000000000000000', // 这个会被忽略,parentSpanId 来自 currentContext
|
|
626
|
+
parentSpanId: lastSpanId, // 使用上一个 span 的 spanId 作为父
|
|
627
|
+
} : undefined;
|
|
628
|
+
|
|
629
|
+
const span = tracer.startSpan(`level-${i}-span`, {
|
|
630
|
+
parent,
|
|
631
|
+
attributes: {
|
|
632
|
+
'level': i,
|
|
633
|
+
'max_depth': MAX_DEPTH,
|
|
634
|
+
},
|
|
635
|
+
});
|
|
636
|
+
|
|
637
|
+
if (!currentTraceId) {
|
|
638
|
+
currentTraceId = span.spanContext.traceId;
|
|
639
|
+
}
|
|
640
|
+
lastSpanId = span.spanContext.spanId;
|
|
641
|
+
|
|
642
|
+
span.end();
|
|
643
|
+
}
|
|
644
|
+
|
|
645
|
+
// 验证所有 span 在同一个 trace 中
|
|
646
|
+
const storage = provider.getStorage();
|
|
647
|
+
const spanTree = storage.findByTraceId(currentTraceId!);
|
|
648
|
+
const spans = flattenSpans(spanTree);
|
|
649
|
+
|
|
650
|
+
expect(spans.length).toBe(MAX_DEPTH);
|
|
651
|
+
|
|
652
|
+
// 验证层级关系
|
|
653
|
+
for (let i = 0; i < MAX_DEPTH; i++) {
|
|
654
|
+
const span = spans.find(s => s.name === `level-${i}-span`);
|
|
655
|
+
expect(span).toBeDefined();
|
|
656
|
+
|
|
657
|
+
if (i === 0) {
|
|
658
|
+
// 第一个 span 是根 span,parentSpanId 应该是 undefined 或 null
|
|
659
|
+
expect(span!.parentSpanId || undefined).toBeUndefined();
|
|
660
|
+
} else {
|
|
661
|
+
const parentSpan = spans.find(s => s.name === `level-${i - 1}-span`);
|
|
662
|
+
expect(span!.parentSpanId).toBe(parentSpan!.spanId);
|
|
663
|
+
}
|
|
664
|
+
}
|
|
665
|
+
|
|
666
|
+
provider.shutdown();
|
|
667
|
+
});
|
|
668
|
+
});
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @fileoverview OpenTelemetry 模块统一导出
|
|
3
|
+
*
|
|
4
|
+
* @example
|
|
5
|
+
* import { getTracerProvider, propagation } from './opentelemetry/mod';
|
|
6
|
+
*
|
|
7
|
+
* // 初始化
|
|
8
|
+
* const provider = getTracerProvider();
|
|
9
|
+
* await provider.initialize();
|
|
10
|
+
*
|
|
11
|
+
* // 获取 Tracer
|
|
12
|
+
* const tracer = provider.getTracer('my-service');
|
|
13
|
+
*
|
|
14
|
+
* // 启动 Span
|
|
15
|
+
* const span = tracer.startSpan('my-operation');
|
|
16
|
+
*
|
|
17
|
+
* // 注入到环境变量
|
|
18
|
+
* const env: Record<string, string | undefined> = {};
|
|
19
|
+
* tracer.injectToEnv(env);
|
|
20
|
+
* process.env = { ...process.env, ...env };
|
|
21
|
+
*/
|
|
22
|
+
|
|
23
|
+
export { OTelTracerProvider, getTracerProvider, resetTracerProvider, type OTelTracer, type OTelSpan, type SpanOptions } from './tracer-provider';
|
|
24
|
+
export { propagation, serialize, parse, type ExtractedContext } from './propagation';
|
|
25
|
+
export { injectToEnv, injectToEnvWithContext, injectToEnvFromCarrier, getTraceparentHeader, type EnvCarrier } from './propagation-env';
|