Merge pull request #24 from fiverr/master

Updated the dependency in AWS::S3 to AWS - SDK gem
newrelic · Mar 8, 2012 · 9d40136 · 9d40136
2 parents 9c9ccd0 + a99a5a8
commit 9d40136
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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