lib/new_relic/agent/instrumentation/redis.rb in newrelic_rpm-4.4.0.336 vs lib/new_relic/agent/instrumentation/redis.rb in newrelic_rpm-4.5.0.337

@@ -66,11 +66,11 @@
           operation, nil, hostname, port_path_or_id, db)
         begin
           segment.notice_nosql_statement(statement) if statement
           call_without_new_relic(*args, &block)
         ensure
-          segment.finish
+          segment.finish if segment
         end
       end
 
       alias_method :call_pipeline_without_new_relic, :call_pipeline
 
@@ -86,11 +86,11 @@
           operation, nil, hostname, port_path_or_id, db)
         begin
           segment.notice_nosql_statement(statement)
           call_pipeline_without_new_relic(*args, &block)
         ensure
-          segment.finish
+          segment.finish if segment
         end
       end
 
       alias_method :connect_without_new_relic, :connect
 
@@ -102,10 +102,10 @@
           NewRelic::Agent::Datastores::Redis::CONNECT, nil, hostname, port_path_or_id, db)
 
         begin
           connect_without_new_relic(*args, &block)
         ensure
-          segment.finish
+          segment.finish if segment
         end
       end
     end
   end
 end