cw-datadog 2.23.0.2

data/lib/datadog/tracing/contrib/aws/patcher.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require_relative '../patcher'
+require_relative 'ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        # Patcher enables patching of 'aws' module.
+        module Patcher
+          include Contrib::Patcher
+
+          module_function
+
+          def target_version
+            Integration.version
+          end
+
+          def patch
+            require_relative 'parsed_context'
+            require_relative 'instrumentation'
+            require_relative 'services'
+
+            add_plugin(Seahorse::Client::Base, *loaded_constants)
+
+            # Special handling for S3 URL Presigning.
+            # @see Datadog::Tracing::Contrib::Aws::S3Presigner
+            ::Aws::S3::Presigner.prepend(S3Presigner) if defined?(::Aws::S3::Presigner)
+          end
+
+          def add_plugin(*targets)
+            targets.each { |klass| klass.add_plugin(Instrumentation) }
+          end
+
+          def loaded_constants
+            # Cross-check services against loaded AWS constants
+            # Module#const_get can return a constant from ancestors when there's a miss.
+            # If this conincidentally matches another constant, it will attempt to patch
+            # the wrong constant, resulting in patch failure.
+            available_services = ::Aws.constants & SERVICES.map(&:to_sym)
+
+            available_services.each_with_object([]) do |service, constants|
+              next if ::Aws.autoload?(service)
+
+              begin
+                constants << ::Aws.const_get(service, false).const_get(:Client, false)
+              rescue
+                next
+              end
+            end
+          end
+
+          def get_option(option)
+            Datadog.configuration.tracing[:aws].get_option(option)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/base.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # Base class for all AWS service-specific tag handlers.
+          class Base
+            def add_tags(span, params)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/dynamodb.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # DynamoDB tag handlers.
+          class DynamoDB < Base
+            def add_tags(span, params)
+              table_name = params[:table_name]
+              span.set_tag(Aws::Ext::TAG_TABLE_NAME, table_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/eventbridge.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # EventBridge tag handlers.
+          class EventBridge < Base
+            def add_tags(span, params)
+              rule_name = params[:name] || params[:rule]
+              span.set_tag(Aws::Ext::TAG_RULE_NAME, rule_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/kinesis.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # Kinesis tag handlers.
+          class Kinesis < Base
+            def add_tags(span, params)
+              stream_arn = params[:stream_arn]
+              stream_name = params[:stream_name]
+
+              if stream_arn
+                # example stream_arn: arn:aws:kinesis:us-east-1:123456789012:stream/my-stream
+                parts = stream_arn.split(':')
+                stream_name = parts[-1].split('/')[-1]
+                aws_account = parts[-2]
+                span.set_tag(Aws::Ext::TAG_AWS_ACCOUNT, aws_account)
+              end
+
+              span.set_tag(Aws::Ext::TAG_STREAM_NAME, stream_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/s3.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # S3 tag handlers.
+          class S3 < Base
+            def add_tags(span, params)
+              bucket_name = params[:bucket]
+              span.set_tag(Aws::Ext::TAG_BUCKET_NAME, bucket_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/sns.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # SNS tag handlers.
+          class SNS < Base
+            def add_tags(span, params)
+              topic_arn = params[:topic_arn]
+              topic_name = params[:name]
+              if topic_arn
+                # example topic_arn: arn:aws:sns:us-west-2:123456789012:my-topic-name
+                parts = topic_arn.split(':')
+                topic_name = parts[-1]
+                aws_account = parts[-2]
+                span.set_tag(Aws::Ext::TAG_AWS_ACCOUNT, aws_account)
+              end
+              span.set_tag(Aws::Ext::TAG_TOPIC_NAME, topic_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/sqs.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # SQS tag handlers.
+          class SQS < Base
+            def add_tags(span, params)
+              queue_url = params[:queue_url]
+              queue_name = params[:queue_name]
+              if queue_url
+                _, _, _, aws_account, queue_name = queue_url.split('/')
+                span.set_tag(Aws::Ext::TAG_AWS_ACCOUNT, aws_account)
+              end
+              span.set_tag(Aws::Ext::TAG_QUEUE_NAME, queue_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/service/states.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative '../ext'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        module Service
+          # States tag handlers.
+          class States < Base
+            def add_tags(span, params)
+              state_machine_name = params[:name]
+              state_machine_arn = params[:state_machine_arn]
+              execution_arn = params[:execution_arn]
+              state_machine_account_id = ''
+
+              if execution_arn
+                # 'arn:aws:states:us-east-1:123456789012:execution:example-state-machine:example-execution'
+                parts = execution_arn.split(':')
+                state_machine_name = parts[-2]
+                state_machine_account_id = parts[4]
+              end
+
+              if state_machine_arn
+                # example statemachinearn: arn:aws:states:us-east-1:123456789012:stateMachine:MyStateMachine
+                parts = state_machine_arn.split(':')
+                state_machine_name ||= parts[-1]
+                state_machine_account_id = parts[-3]
+              end
+              span.set_tag(Aws::Ext::TAG_AWS_ACCOUNT, state_machine_account_id)
+              span.set_tag(Aws::Ext::TAG_STATE_MACHINE_NAME, state_machine_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/datadog/tracing/contrib/aws/services.rb

@@ -0,0 +1,139 @@
+# frozen_string_literal: true
+
+require_relative 'service/sqs'
+require_relative 'service/sns'
+require_relative 'service/dynamodb'
+require_relative 'service/kinesis'
+require_relative 'service/eventbridge'
+require_relative 'service/states'
+require_relative 'service/s3'
+
+module Datadog
+  module Tracing
+    module Contrib
+      module Aws
+        SERVICES = %w[
+          ACM
+          APIGateway
+          AppStream
+          ApplicationAutoScaling
+          ApplicationDiscoveryService
+          Athena
+          AutoScaling
+          Batch
+          Budgets
+          CloudDirectory
+          CloudFormation
+          CloudFront
+          CloudHSM
+          CloudHSMV2
+          CloudSearch
+          CloudSearchDomain
+          CloudTrail
+          CloudWatch
+          CloudWatchEvents
+          CloudWatchLogs
+          CodeBuild
+          CodeCommit
+          CodeDeploy
+          CodePipeline
+          CodeStar
+          CognitoIdentity
+          CognitoIdentityProvider
+          CognitoSync
+          ConfigService
+          CostandUsageReportService
+          DAX
+          DataPipeline
+          DatabaseMigrationService
+          DeviceFarm
+          DirectConnect
+          DirectoryService
+          DynamoDB
+          DynamoDBStreams
+          EC2
+          ECR
+          ECS
+          EFS
+          EMR
+          ElastiCache
+          ElasticBeanstalk
+          ElasticLoadBalancing
+          ElasticLoadBalancingV2
+          ElasticTranscoder
+          ElasticsearchService
+          EventBridge
+          Firehose
+          GameLift
+          Glacier
+          Glue
+          Greengrass
+          Health
+          IAM
+          ImportExport
+          Inspector
+          IoT
+          IoTDataPlane
+          KMS
+          Kinesis
+          KinesisAnalytics
+          Lambda
+          LambdaPreview
+          Lex
+          LexModelBuildingService
+          Lightsail
+          MTurk
+          MachineLearning
+          MarketplaceCommerceAnalytics
+          MarketplaceEntitlementService
+          MarketplaceMetering
+          MigrationHub
+          Mobile
+          OpsWorks
+          OpsWorksCM
+          Organizations
+          Pinpoint
+          Polly
+          RDS
+          Redshift
+          Rekognition
+          ResourceGroupsTaggingAPI
+          Route53
+          Route53Domains
+          S3
+          SES
+          SMS
+          SNS
+          SQS
+          SSM
+          STS
+          SWF
+          ServiceCatalog
+          Schemas
+          Shield
+          SimpleDB
+          Snowball
+          States
+          StorageGateway
+          Support
+          Textract
+          WAF
+          WAFRegional
+          WorkDocs
+          WorkSpaces
+          XRay
+        ].freeze
+
+        SERVICE_HANDLERS = {
+          'sqs' => Service::SQS.new,
+          'sns' => Service::SNS.new,
+          'dynamodb' => Service::DynamoDB.new,
+          'kinesis' => Service::Kinesis.new,
+          'eventbridge' => Service::EventBridge.new,
+          'states' => Service::States.new,
+          's3' => Service::S3.new
+        }.freeze
+      end
+    end
+  end
+end