add size limit check

DataDog · Oct 8, 2024 · 0b69656 · 0b69656
1 parent 413b82e
commit 0b69656
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/contrib/aws/aws-sdk-go-v2/aws/eventbridge/eventbridge.go b/contrib/aws/aws-sdk-go-v2/aws/eventbridge/eventbridge.go
@@ -24,6 +24,7 @@ const (
  datadogKey = "_datadog"
  startTimeKey = "x-datadog-start-time"
  resourceNameKey = "x-datadog-resource-name"
+ maxSizeBytes = 256 * 1024 // 256 KB
 )
 
 func init() {
@@ -106,5 +107,11 @@ func injectTraceContext(ctx context.Context, entryPtr *types.PutEventsRequestEnt
  return
  }
 
+ // Check new detail size
+ if len(updatedDetail) > maxSizeBytes {
+ log.Info("Payload size too large to pass context")
+ return
+ }
+
  entryPtr.Detail = aws.String(string(updatedDetail))
 }
diff --git a/contrib/aws/aws-sdk-go-v2/aws/sns/sns.go b/contrib/aws/aws-sdk-go-v2/aws/sns/sns.go
@@ -84,7 +84,7 @@ func injectTraceContext(ctx context.Context, messageAttributes map[string]types.
  // https://docs.aws.amazon.com/sns/latest/dg/sns-message-attributes.html
  // Only inject if there's room.
  if len(messageAttributes) >= maxMessageAttributes {
- log.Debug("Cannot inject trace context: message already has maximum allowed attributes")
+ log.Info("Cannot inject trace context: message already has maximum allowed attributes")
  return
  }
 

diff --git a/contrib/aws/aws-sdk-go-v2/aws/sqs/sqs.go b/contrib/aws/aws-sdk-go-v2/aws/sqs/sqs.go
@@ -84,7 +84,7 @@ func injectTraceContext(ctx context.Context, messageAttributes map[string]types.
  // https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-message-metadata.html#sqs-message-attributes
  // Only inject if there's room.
  if len(messageAttributes) >= maxMessageAttributes {
- log.Debug("Cannot inject trace context: message already has maximum allowed attributes")
+ log.Info("Cannot inject trace context: message already has maximum allowed attributes")
  return
  }