feat(profiling): Route raw profiles through objectstore service#6025
Conversation
Extract StoreRawProfile into the objectstore service and replace inline raw profile blob with objectstore key reference in the Kafka message. Co-Authored-By: Claude <noreply@anthropic.com>
Align the objectstore usecase identifier with the naming convention so it can be easily distinguished from a future profiles_v2 usecase. Co-Authored-By: Claude <noreply@anthropic.com>
Add try_send_to_objectstore to StoreHandle, which returns the message when the objectstore service is not configured. Profile chunks fall back to sending the StoreProfileChunk directly to Store, preventing data loss. Also log session creation errors matching the pattern in other objectstore handlers. Co-Authored-By: Claude <noreply@anthropic.com>
The rename in 3a83bc5 only updated MessageKind::as_str() but missed the Usecase::new() call, causing raw profiles to be stored under the generic "profiles" namespace instead of "profiles_raw". Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
| let trace_attachments = Usecase::new("trace_attachments") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); | ||
| let profiles = Usecase::new("profiles") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); |
There was a problem hiding this comment.
Note that DEFAULT_ATTACHMENT_RETENTION means event attachments here. Even if that happens to be the correct retention you actually want, you likely want to use a different constant here to not mix things up.
Here you likely want to use the profiles retention instead.
There was a problem hiding this comment.
Good point, I was wondering about this as well. Some quick research showed me that the monolith builds a project config, which then gets propagate to relay. And that config is coming from getsentry. At this point I feel like pulling on a thread a bit 😅 but I guess we need this, maybe @jjbayer could chime in here as well, before I start adding support for this.
We would need to
- [getsentry] Propagate the config based on the plan, looks like this already happens automatically as it iterates over all data categories
- [sentry] Extend the RETENTIONS_CONFIG_MAPPING to also include
DataCategory.PROFILE_DURATIONandDataCategory.PROFILE_DURATION_UI, so those get propagated to relay - [relay] Extend RetentionsConfig and use that config here
There was a problem hiding this comment.
That sounds sensible, yes.
There was a problem hiding this comment.
The retention is already read from passed in by forward_store via StoreRawProfile::retention, and used in handle_raw_profile.
The fact that profiles don't have a custom retention config is out of scope of this PR IMO.
TL;DR: A hard-coded constant is fine for the fallback, but I would define it in a custom constant and set it to 90 days to match the default event retention.
Co-Authored-By: Claude <noreply@anthropic.com>
… feat/markushi/perfetto-profiling-support-pipeline-storage-mechanism
| let trace_attachments = Usecase::new("trace_attachments") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); | ||
| let profiles = Usecase::new("profiles") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); |
There was a problem hiding this comment.
That sounds sensible, yes.
| Self::Envelope => "envelope", | ||
| Self::EventAttachment => "attachment", | ||
| Self::TraceAttachment => "attachment_v2", | ||
| Self::RawProfile => "profiles_raw", |
There was a problem hiding this comment.
Other variants use singular:
| Self::RawProfile => "profiles_raw", | |
| Self::RawProfile => "raw_profile", |
| if let Some(unsent) = s.try_send_to_objectstore(msg) { | ||
| s.send_to_store(unsent.map(|profile, _| profile.store_message)); | ||
| } |
There was a problem hiding this comment.
We only need this branch if we want to support perfetto profiles in self-hosted, which has no objectstore yet by default. In that case, we also need the kafka consumer to handler the raw_profile_object_store_key correctly.
So the simpler approach would be to never send raw profiles via kafka, and accept that perfetto is not enabled in self-hosted.
There was a problem hiding this comment.
As relay takes care of parsing the perfetto profile into a sample v2 format I think it's still worth sending the profile chunk kafka message, no matter if object store is available or not. If it's not available, the kafka message would still be sent, just without a raw_profile_object_store_key set.
So the simpler approach would be to never send raw profiles via kafka
Maybe I'm misunderstanding something here, but that's the intention of this PR. Instead of packing the raw profile into the kafka message, we reference using raw_profile_object_store_key instead, which is only set when objectstore is available and successfully stored the raw profile.
| raw_profile: chunk.raw_profile, | ||
| })); | ||
| if chunk.raw_profile.is_some() { | ||
| let msg = chunk.map(|chunk, _| { |
There was a problem hiding this comment.
It should be possible to move the if into the map and then use if let Some(raw_profile) = chunk.raw_profile.
| /// Objectstore key where the raw profile blob is stored. | ||
| pub raw_profile_object_store_key: Option<String>, | ||
| /// Content type of the raw profile (e.g. Perfetto trace). | ||
| pub raw_profile_content_type: Option<ContentType>, |
There was a problem hiding this comment.
Since a StoreProfileChunk can only have either a byte payload or an objectstore key, I would encode this into the type system as something like
pub payload: StoreProfileChunkPayloadwhere
enum StoreProfileChunkPayload {
Bytes(Bytes),
ObjectstoreKey(String)
}| raw_profile: message.raw_profile.as_ref().map(|r| r.payload.clone()), | ||
| raw_profile_content_type: message.raw_profile.map(|r| r.content_type), | ||
| raw_profile_object_store_key: message.raw_profile_object_store_key, | ||
| raw_profile_content_type: message.raw_profile_content_type, |
There was a problem hiding this comment.
Why do we need to provide the content type in the kafka message in the first place? Isn't it always ContentType::PerfettoTrace?
There was a problem hiding this comment.
Yes, as of now it can only be perfetto. I added it nevertheless to make it easier to add other profiles formats in future. Also it will be helpful downstream to know what format raw_profile_object_store_key refers to, otherwise we'd need to make at least a HEAD request towards objectstore every time we want to process / surface the data (e.g. a "Download Perfetto Trace" button on the frontend).
An alternative would be to simply rename raw_profile_object_store_key to perfetto_profile_object_store_key.
No strong opinions on my end, happy for any directions here as I'm still new to the codebase.
| /// Content type of the raw profile. | ||
| pub content_type: ContentType, | ||
| /// The profile chunk message to forward to Kafka after upload. | ||
| pub store_message: StoreProfileChunk, |
There was a problem hiding this comment.
This looks unnecessarily nested. The objectstore service should be able to construct the StoreProfileChunk message from scratch when it sends the data to the store service.
| let trace_attachments = Usecase::new("trace_attachments") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); | ||
| let profiles = Usecase::new("profiles") | ||
| .with_expiration_policy(ExpirationPolicy::TimeToLive(DEFAULT_ATTACHMENT_RETENTION)); |
There was a problem hiding this comment.
The retention is already read from passed in by forward_store via StoreRawProfile::retention, and used in handle_raw_profile.
The fact that profiles don't have a custom retention config is out of scope of this PR IMO.
TL;DR: A hard-coded constant is fine for the fallback, but I would define it in a custom constant and set it to 90 days to match the default event retention.
Co-authored-by: Sebastian Zivota <loewenheim@users.noreply.github.com>
…tto-profiling-support-pipeline-storage-mechanism' into feat/markushi/perfetto-profiling-support-pipeline-storage-mechanism
There was a problem hiding this comment.
Cursor Bugbot has reviewed your changes and found 2 potential issues.
❌ Bugbot Autofix is OFF. To automatically fix reported issues with cloud agents, enable autofix in the Cursor dashboard.
Reviewed by Cursor Bugbot for commit dc7c95b. Configure here.
| object_store_key = profile["raw_profile_object_store_key"] | ||
| assert object_store_key, "expected raw_profile_object_store_key in Kafka message" | ||
|
|
||
| stored = objectstore("profiles_raw", project_id).get(object_store_key) |
There was a problem hiding this comment.
Wrong objectstore usecase in test
Medium Severity
The new integration test fetches the uploaded Perfetto blob with objectstore usecase profiles_raw, but Relay uploads via profile_raw. Objectstore keys are scoped by usecase (same pattern as attachments), so the GET likely misses the object the test claims to verify.
Reviewed by Cursor Bugbot for commit dc7c95b. Configure here.
| payload: chunk.payload, | ||
| quantities: chunk.quantities, | ||
| raw_profile_object_store_key: None, | ||
| raw_profile_content_type: None, |
There was a problem hiding this comment.
Content type omitted after failures
Medium Severity
When a chunk has a raw profile, StoreProfileChunk is built with raw_profile_content_type set to None, and objectstore only fills it after a successful upload. Previously, Kafka always included the content type whenever raw_profile was present. Failed uploads, session errors, objectstore fallback, and load-shed paths now emit messages without that field despite an ingested Perfetto trace.
Additional Locations (1)
Reviewed by Cursor Bugbot for commit dc7c95b. Configure here.


Summary
StoreRawProfileinto the objectstore service to upload raw profile blobs (e.g. Perfetto traces) before forwarding to KafkaStoreProfileChunkto the Store service, even if the upload fails, ensuring the Kafka message is producedTest plan
pytest tests/integration/test_profile_chunks_perfetto.py🤖 Generated with Claude Code
This PR is part of a stack: