Skip to content

Add internal spans to event repo #1678

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Feb 7, 2025
Merged

Add internal spans to event repo #1678

merged 1 commit into from
Feb 7, 2025

Conversation

ericallam
Copy link
Member

@ericallam ericallam commented Feb 7, 2025

Summary by CodeRabbit

  • New Features

    • Integrated unique identifiers into batch operations for enhanced operational tracking.
    • Added advanced tracing within event processing and export functions to improve monitoring and diagnostics.
  • Refactor

    • Streamlined logging outputs to display only essential run identifiers, simplifying log analysis.

Copy link

changeset-bot bot commented Feb 7, 2025

⚠️ No Changeset found

Latest commit: 5f2b70b

Merging this PR will not cause a version bump for any packages. If these changes should not result in a new version, you're good to go. If these changes should result in a version bump, you need to add a changeset.

This PR includes no changesets

When changesets are added to this PR, you'll see the packages that this PR includes changesets for and the associated semver types

Click here to learn what changesets are, and how to add one.

Click here if you're a maintainer who wants to add a changeset to this PR

Copy link
Contributor

coderabbitai bot commented Feb 7, 2025

Walkthrough

This pull request updates several server-side modules to improve batch tracking and observability. The callback signatures in the dynamic flush scheduler and event repository methods now include a new unique flushId generated by nanoid(). Tracing capabilities have been enhanced in the event repository and OTLP exporter by wrapping operations in spans and logging context-specific attributes. Additionally, logging in the finalize task run service has been streamlined to output only the run identifier. A new Tracer import has been added to support the tracing functionality.

Changes

File(s) Change Summary
apps/webapp/app/v3/dynamicFlushScheduler.server.ts, apps/webapp/app/v3/eventRepository.server.ts Updated flush-related methods and callback signatures to include a new flushId parameter generated via nanoid(), enhancing batch operation tracking.
apps/webapp/app/v3/eventRepository.server.ts, apps/webapp/app/v3/otlpExporter.server.ts Introduced and enhanced tracing functionality by wrapping export and flush operations in spans, adding attributes such as event counts and error details.
apps/webapp/app/v3/services/finalizeTaskRun.server.ts Adjusted log messages to output only run.id instead of the full run object, streamlining logging details.
apps/webapp/app/v3/tracer.server.ts Added an import for Tracer from @opentelemetry/api to support the new tracing mechanisms.

Sequence Diagram(s)

sequenceDiagram
  participant DS as DynamicFlushScheduler
  participant CB as Callback Function
  DS->>DS: Generate flushId via nanoid()
  DS->>CB: Invoke callback(flushId, batch)
Loading
sequenceDiagram
  participant ER as EventRepository
  participant Span as Tracing Span
  participant DB as Data Store
  ER->>ER: Generate flushId using nanoid()
  ER->>Span: Start span with startSpan()
  ER->>DB: Call flushBatch(flushId, batch)
  DB-->>ER: Return processed events
  ER->>Span: Set attributes (event count, errors)
  Span-->>ER: End span
Loading

Possibly related PRs

Poem

I'm a rabbit in the code, hopping down the lane,
With flushIds leaping high, no batch is quite the same.
Tracing spans guide my nibble through every line,
Logs now sing just IDs—oh, how sublime!
I munch on fresh updates with a twitch and a cheer 🐇
In this world of code, my ears stand high and clear!

Warning

There were issues while running some tools. Please review the errors and either fix the tool’s configuration or disable the tool if it’s a critical failure.

🔧 ESLint

If the error stems from missing dependencies, add them to the package.json file. For unrecoverable errors (e.g., due to private dependencies), disable the tool in the CodeRabbit configuration.

apps/webapp/app/v3/dynamicFlushScheduler.server.ts

Oops! Something went wrong! :(

ESLint: 8.45.0

ESLint couldn't find the config "custom" to extend from. Please check that the name of the config is correct.

The config "custom" was referenced from the config file in "/.eslintrc.js".

If you still have problems, please stop by https://eslint.org/chat/help to chat with the team.

apps/webapp/app/v3/eventRepository.server.ts

Oops! Something went wrong! :(

ESLint: 8.45.0

ESLint couldn't find the config "custom" to extend from. Please check that the name of the config is correct.

The config "custom" was referenced from the config file in "/.eslintrc.js".

If you still have problems, please stop by https://eslint.org/chat/help to chat with the team.

apps/webapp/app/v3/services/finalizeTaskRun.server.ts

Oops! Something went wrong! :(

ESLint: 8.45.0

ESLint couldn't find the config "custom" to extend from. Please check that the name of the config is correct.

The config "custom" was referenced from the config file in "/.eslintrc.js".

If you still have problems, please stop by https://eslint.org/chat/help to chat with the team.

  • 2 others
✨ Finishing Touches
  • 📝 Generate Docstrings (Beta)

Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Generate unit testing code for this file.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai generate unit testing code for this file.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and generate unit testing code.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai generate docstrings to generate docstrings for this PR. (Beta)
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 0

🧹 Nitpick comments (1)
apps/webapp/app/v3/eventRepository.server.ts (1)

243-243: Returning the promise without returning flushed events.

This immediate insertion also uses a flush ID. However, #flushBatch currently returns void, so consider whether the caller needs the inserted events.

📜 Review details

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 0b555fa and 5f2b70b.

📒 Files selected for processing (5)
  • apps/webapp/app/v3/dynamicFlushScheduler.server.ts (3 hunks)
  • apps/webapp/app/v3/eventRepository.server.ts (7 hunks)
  • apps/webapp/app/v3/otlpExporter.server.ts (1 hunks)
  • apps/webapp/app/v3/services/finalizeTaskRun.server.ts (1 hunks)
  • apps/webapp/app/v3/tracer.server.ts (1 hunks)
⏰ Context from checks skipped due to timeout of 90000ms (4)
  • GitHub Check: e2e / 🧪 CLI v3 tests (windows-latest - pnpm)
  • GitHub Check: e2e / 🧪 CLI v3 tests (windows-latest - npm)
  • GitHub Check: typecheck / typecheck
  • GitHub Check: Analyze (javascript-typescript)
🔇 Additional comments (17)
apps/webapp/app/v3/eventRepository.server.ts (7)

1-1: Added Tracer import for enhanced telemetry.

The addition of “Tracer” to the import is consistent with its subsequent usage to provide tracing functionality throughout the repository. No issues found.


35-36: New imports for spans and unique IDs.

Importing “startSpan” for tracing and “nanoid” for generating unique flush IDs is a solid approach. Both imports align well with the new flush and tracing logic below.


104-104: Optional tracer property in config.

Allowing an optional “tracer” in the repository config fosters flexibility and proper dependency injection. Implementation looks good.


208-209: Private properties for tracer and last flush time.

Introducing a private tracer reference and tracking the last flush time helps measure intervals between flushes. No immediate issues detected.


235-235: Immediate flush now generates a unique flush ID.

Using “nanoid” to create a flush ID for immediate event insertion improves observability. Looks good.


1028-1055: New #flushBatch method with tracing.

The tracing logic (startSpan, custom attributes) is well-implemented. The method sets “_lastFlushedAt,” logs differences in event counts, and publishes to Redis. Concurrency and partial-event handling appear sound.


1057-1142: Refined #doFlushBatch with split insertion fallback.

Splitting events on insert errors and logging them is a clever approach to handle problematic data while preserving partial successes. Consider verifying performance impacts for very large batches. Otherwise, nicely done.
[refactor_suggestion_good_to_have, verify]

apps/webapp/app/v3/dynamicFlushScheduler.server.ts (4)

1-1: Importing nanoid for unique flush IDs.

Bringing in “nanoid” sets the stage for generating unique flush identifiers in the scheduler’s callback.


6-6: Callback signature updated with flush ID.

Adding the “flushId” parameter in the config’s callback type fosters better tracking during flush operations. No concerns.


15-15: Revised callback property to receive flush ID.

The dynamic flush scheduler’s constructor now consistently uses the updated callback signature. Looks good.


62-62: Invoking callback with a new flush ID.

Calling the callback with “nanoid()” ensures each batch is tracked under a unique flush identifier, which is valuable for debugging. Implementation is correct.

apps/webapp/app/v3/tracer.server.ts (1)

11-11: Added Tracer import to support tracing.

Introducing “Tracer” aligns with the repository changes that make use of custom tracing features. No issues noted.

apps/webapp/app/v3/services/finalizeTaskRun.server.ts (1)

125-130: LGTM! Improved logging by reducing verbosity.

The changes to log only run.id instead of the entire run object are beneficial as they:

  • Reduce log verbosity
  • Minimize potential exposure of sensitive data
  • Improve log readability
apps/webapp/app/v3/otlpExporter.server.ts (4)

28-29: LGTM! Added OpenTelemetry imports.

The imports are correctly added to support the tracing functionality.


37-44: LGTM! Properly initialized the tracer.

The tracer is correctly initialized with a descriptive name "otlp-exporter" following OpenTelemetry best practices.


50-68: LGTM! Enhanced exportTraces with tracing capabilities.

The exportTraces method is now properly instrumented with:

  • Span creation using startSpan
  • Event count attribute for better observability
  • Existing verbose logging maintained within span context

75-93: LGTM! Enhanced exportLogs with tracing capabilities.

The exportLogs method is now properly instrumented with:

  • Span creation using startSpan
  • Event count attribute for better observability
  • Existing verbose logging maintained within span context

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant