diff --git a/lib/rpm_contrib/instrumentation/aws.rb b/lib/rpm_contrib/instrumentation/aws.rb index eaa271c..2ed393b 100644 --- a/lib/rpm_contrib/instrumentation/aws.rb +++ b/lib/rpm_contrib/instrumentation/aws.rb @@ -13,8 +13,14 @@ executes do # Instrument connections to the AWS-S3 service - ::AWS::S3::Connection::Management::ClassMethods.module_eval do - add_method_tracer :establish_connection!, 'AWS-S3/establish_connection!' + if defined?(::AWS::S3::Connection) + ::AWS::S3::Connection::Management::ClassMethods.module_eval do + add_method_tracer :establish_connection!, 'AWS-S3/establish_connection!' + end + else #using aws-sdk + ::AWS::S3.class_eval do + add_method_tracer :initialize, 'AWS-S3/establish_connection!' + end end # Instrument methods on Bucket