newrelic_rpm 5.7.0.350 → 9.0.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (476) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +15 -1
  3. data/.rubocop.yml +1919 -0
  4. data/.rubocop_todo.yml +100 -0
  5. data/.simplecov +15 -0
  6. data/.snyk +11 -0
  7. data/.yardopts +2 -0
  8. data/Brewfile +12 -0
  9. data/CHANGELOG.md +4056 -2339
  10. data/CONTRIBUTING.md +132 -19
  11. data/DOCKER.md +167 -0
  12. data/Dockerfile +10 -0
  13. data/Gemfile +5 -2
  14. data/Guardfile +22 -4
  15. data/LICENSE +202 -38
  16. data/README.md +87 -87
  17. data/Rakefile +27 -27
  18. data/THIRD_PARTY_NOTICES.md +28 -0
  19. data/Thorfile +5 -0
  20. data/bin/newrelic +3 -2
  21. data/bin/newrelic_cmd +1 -0
  22. data/bin/nrdebug +77 -54
  23. data/config.dot +5 -5
  24. data/docker-compose.yml +107 -0
  25. data/init.rb +5 -7
  26. data/install.rb +3 -3
  27. data/lefthook.yml +9 -0
  28. data/lib/new_relic/agent/adaptive_sampler.rb +14 -10
  29. data/lib/new_relic/agent/agent.rb +125 -969
  30. data/lib/new_relic/agent/agent_helpers/connect.rb +227 -0
  31. data/lib/new_relic/agent/agent_helpers/harvest.rb +153 -0
  32. data/lib/new_relic/agent/agent_helpers/shutdown.rb +72 -0
  33. data/lib/new_relic/agent/agent_helpers/special_startup.rb +74 -0
  34. data/lib/new_relic/agent/agent_helpers/start_worker_thread.rb +167 -0
  35. data/lib/new_relic/agent/agent_helpers/startup.rb +202 -0
  36. data/lib/new_relic/agent/agent_helpers/transmit.rb +76 -0
  37. data/lib/new_relic/agent/agent_logger.rb +26 -18
  38. data/lib/new_relic/agent/attribute_filter.rb +69 -52
  39. data/lib/new_relic/agent/attribute_processing.rb +8 -8
  40. data/lib/new_relic/agent/attributes.rb +153 -0
  41. data/lib/new_relic/agent/audit_logger.rb +19 -4
  42. data/lib/new_relic/agent/autostart.rb +34 -28
  43. data/lib/new_relic/agent/chained_call.rb +2 -2
  44. data/lib/new_relic/agent/commands/agent_command.rb +4 -4
  45. data/lib/new_relic/agent/commands/agent_command_router.rb +15 -33
  46. data/lib/new_relic/agent/commands/thread_profiler_session.rb +13 -11
  47. data/lib/new_relic/agent/configuration/default_source.rb +1480 -1053
  48. data/lib/new_relic/agent/configuration/dotted_hash.rb +7 -6
  49. data/lib/new_relic/agent/configuration/environment_source.rb +15 -11
  50. data/lib/new_relic/agent/configuration/event_harvest_config.rb +68 -0
  51. data/lib/new_relic/agent/configuration/high_security_source.rb +9 -9
  52. data/lib/new_relic/agent/configuration/manager.rb +96 -79
  53. data/lib/new_relic/agent/configuration/manual_source.rb +2 -2
  54. data/lib/new_relic/agent/configuration/mask_defaults.rb +4 -4
  55. data/lib/new_relic/agent/configuration/security_policy_source.rb +83 -86
  56. data/lib/new_relic/agent/configuration/server_source.rb +49 -12
  57. data/lib/new_relic/agent/configuration/yaml_source.rb +42 -13
  58. data/lib/new_relic/agent/configuration.rb +2 -2
  59. data/lib/new_relic/agent/connect/request_builder.rb +61 -0
  60. data/lib/new_relic/agent/connect/response_handler.rb +58 -0
  61. data/lib/new_relic/agent/custom_event_aggregator.rb +15 -15
  62. data/lib/new_relic/agent/database/explain_plan_helpers.rb +5 -6
  63. data/lib/new_relic/agent/database/obfuscation_helpers.rb +16 -15
  64. data/lib/new_relic/agent/database/obfuscator.rb +3 -3
  65. data/lib/new_relic/agent/database/postgres_explain_obfuscator.rb +4 -4
  66. data/lib/new_relic/agent/database.rb +44 -53
  67. data/lib/new_relic/agent/database_adapter.rb +35 -0
  68. data/lib/new_relic/agent/datastores/metric_helper.rb +18 -20
  69. data/lib/new_relic/agent/datastores/mongo/event_formatter.rb +9 -8
  70. data/lib/new_relic/agent/datastores/mongo/metric_translator.rb +7 -11
  71. data/lib/new_relic/agent/datastores/mongo.rb +7 -12
  72. data/lib/new_relic/agent/datastores/nosql_obfuscator.rb +41 -0
  73. data/lib/new_relic/agent/datastores/redis.rb +6 -12
  74. data/lib/new_relic/agent/datastores.rb +19 -23
  75. data/lib/new_relic/agent/deprecator.rb +2 -2
  76. data/lib/new_relic/agent/{cross_app_payload.rb → distributed_tracing/cross_app_payload.rb} +13 -12
  77. data/lib/new_relic/agent/{cross_app_tracing.rb → distributed_tracing/cross_app_tracing.rb} +87 -66
  78. data/lib/new_relic/agent/distributed_tracing/distributed_trace_attributes.rb +84 -0
  79. data/lib/new_relic/agent/distributed_tracing/distributed_trace_metrics.rb +75 -0
  80. data/lib/new_relic/agent/distributed_tracing/distributed_trace_payload.rb +163 -0
  81. data/lib/new_relic/agent/distributed_tracing/distributed_trace_transport_type.rb +38 -0
  82. data/lib/new_relic/agent/distributed_tracing/trace_context.rb +245 -0
  83. data/lib/new_relic/agent/distributed_tracing/trace_context_payload.rb +127 -0
  84. data/lib/new_relic/agent/distributed_tracing.rb +113 -32
  85. data/lib/new_relic/agent/encoding_normalizer.rb +5 -3
  86. data/lib/new_relic/agent/error_collector.rb +99 -63
  87. data/lib/new_relic/agent/error_event_aggregator.rb +10 -8
  88. data/lib/new_relic/agent/error_filter.rb +174 -0
  89. data/lib/new_relic/agent/error_trace_aggregator.rb +6 -4
  90. data/lib/new_relic/agent/event_aggregator.rb +43 -48
  91. data/lib/new_relic/agent/event_buffer.rb +8 -9
  92. data/lib/new_relic/agent/event_listener.rb +2 -3
  93. data/lib/new_relic/agent/event_loop.rb +27 -25
  94. data/lib/new_relic/agent/external.rb +20 -51
  95. data/lib/new_relic/agent/guid_generator.rb +30 -0
  96. data/lib/new_relic/agent/harvester.rb +5 -6
  97. data/lib/new_relic/agent/heap.rb +8 -10
  98. data/lib/new_relic/agent/hostname.rb +26 -5
  99. data/lib/new_relic/agent/http_clients/abstract.rb +81 -0
  100. data/lib/new_relic/agent/http_clients/curb_wrappers.rb +26 -26
  101. data/lib/new_relic/agent/http_clients/excon_wrappers.rb +31 -17
  102. data/lib/new_relic/agent/http_clients/http_rb_wrappers.rb +18 -23
  103. data/lib/new_relic/agent/http_clients/httpclient_wrappers.rb +12 -15
  104. data/lib/new_relic/agent/http_clients/net_http_wrappers.rb +24 -8
  105. data/lib/new_relic/agent/http_clients/typhoeus_wrappers.rb +9 -12
  106. data/lib/new_relic/agent/http_clients/uri_util.rb +12 -13
  107. data/lib/new_relic/agent/instrumentation/action_cable_subscriber.rb +22 -52
  108. data/lib/new_relic/agent/instrumentation/action_controller_other_subscriber.rb +39 -0
  109. data/lib/new_relic/agent/instrumentation/action_controller_subscriber.rb +59 -72
  110. data/lib/new_relic/agent/instrumentation/action_dispatch.rb +31 -0
  111. data/lib/new_relic/agent/instrumentation/action_dispatch_subscriber.rb +64 -0
  112. data/lib/new_relic/agent/instrumentation/action_mailbox.rb +30 -0
  113. data/lib/new_relic/agent/instrumentation/action_mailbox_subscriber.rb +33 -0
  114. data/lib/new_relic/agent/instrumentation/action_mailer.rb +30 -0
  115. data/lib/new_relic/agent/instrumentation/action_mailer_subscriber.rb +85 -0
  116. data/lib/new_relic/agent/instrumentation/action_view_subscriber.rb +86 -62
  117. data/lib/new_relic/agent/instrumentation/active_job.rb +38 -19
  118. data/lib/new_relic/agent/instrumentation/active_job_subscriber.rb +41 -0
  119. data/lib/new_relic/agent/instrumentation/active_merchant.rb +21 -7
  120. data/lib/new_relic/agent/instrumentation/active_record.rb +95 -46
  121. data/lib/new_relic/agent/instrumentation/active_record_helper.rb +82 -61
  122. data/lib/new_relic/agent/instrumentation/active_record_notifications.rb +155 -0
  123. data/lib/new_relic/agent/instrumentation/active_record_prepend.rb +36 -12
  124. data/lib/new_relic/agent/instrumentation/active_record_subscriber.rb +69 -64
  125. data/lib/new_relic/agent/instrumentation/active_storage.rb +8 -4
  126. data/lib/new_relic/agent/instrumentation/active_storage_subscriber.rb +11 -32
  127. data/lib/new_relic/agent/instrumentation/active_support.rb +27 -0
  128. data/lib/new_relic/agent/instrumentation/active_support_logger/chain.rb +23 -0
  129. data/lib/new_relic/agent/instrumentation/active_support_logger/instrumentation.rb +20 -0
  130. data/lib/new_relic/agent/instrumentation/active_support_logger/prepend.rb +12 -0
  131. data/lib/new_relic/agent/instrumentation/active_support_logger.rb +24 -0
  132. data/lib/new_relic/agent/instrumentation/active_support_subscriber.rb +41 -0
  133. data/lib/new_relic/agent/instrumentation/bunny/chain.rb +45 -0
  134. data/lib/new_relic/agent/instrumentation/bunny/instrumentation.rb +150 -0
  135. data/lib/new_relic/agent/instrumentation/bunny/prepend.rb +35 -0
  136. data/lib/new_relic/agent/instrumentation/bunny.rb +14 -134
  137. data/lib/new_relic/agent/instrumentation/concurrent_ruby/chain.rb +36 -0
  138. data/lib/new_relic/agent/instrumentation/concurrent_ruby/instrumentation.rb +21 -0
  139. data/lib/new_relic/agent/instrumentation/concurrent_ruby/prepend.rb +27 -0
  140. data/lib/new_relic/agent/instrumentation/concurrent_ruby.rb +31 -0
  141. data/lib/new_relic/agent/instrumentation/controller_instrumentation.rb +77 -61
  142. data/lib/new_relic/agent/instrumentation/curb/chain.rb +91 -0
  143. data/lib/new_relic/agent/instrumentation/curb/instrumentation.rb +221 -0
  144. data/lib/new_relic/agent/instrumentation/curb/prepend.rb +61 -0
  145. data/lib/new_relic/agent/instrumentation/curb.rb +15 -187
  146. data/lib/new_relic/agent/instrumentation/custom_events.rb +12 -0
  147. data/lib/new_relic/agent/instrumentation/custom_events_subscriber.rb +37 -0
  148. data/lib/new_relic/agent/instrumentation/delayed_job/chain.rb +35 -0
  149. data/lib/new_relic/agent/instrumentation/delayed_job/instrumentation.rb +48 -0
  150. data/lib/new_relic/agent/instrumentation/delayed_job/prepend.rb +33 -0
  151. data/lib/new_relic/agent/instrumentation/delayed_job_instrumentation.rb +30 -52
  152. data/lib/new_relic/agent/instrumentation/elasticsearch/chain.rb +29 -0
  153. data/lib/new_relic/agent/instrumentation/elasticsearch/instrumentation.rb +66 -0
  154. data/lib/new_relic/agent/instrumentation/elasticsearch/prepend.rb +13 -0
  155. data/lib/new_relic/agent/instrumentation/elasticsearch.rb +31 -0
  156. data/lib/new_relic/agent/instrumentation/excon/middleware.rb +8 -7
  157. data/lib/new_relic/agent/instrumentation/excon.rb +29 -31
  158. data/lib/new_relic/agent/instrumentation/fiber/chain.rb +20 -0
  159. data/lib/new_relic/agent/instrumentation/fiber/instrumentation.rb +24 -0
  160. data/lib/new_relic/agent/instrumentation/fiber/prepend.rb +18 -0
  161. data/lib/new_relic/agent/instrumentation/fiber.rb +25 -0
  162. data/lib/new_relic/agent/instrumentation/grape/chain.rb +24 -0
  163. data/lib/new_relic/agent/instrumentation/grape/instrumentation.rb +100 -0
  164. data/lib/new_relic/agent/instrumentation/grape/prepend.rb +17 -0
  165. data/lib/new_relic/agent/instrumentation/grape.rb +16 -121
  166. data/lib/new_relic/agent/instrumentation/grpc/client/chain.rb +97 -0
  167. data/lib/new_relic/agent/instrumentation/grpc/client/instrumentation.rb +89 -0
  168. data/lib/new_relic/agent/instrumentation/grpc/client/prepend.rb +111 -0
  169. data/lib/new_relic/agent/instrumentation/grpc/client/request_wrapper.rb +30 -0
  170. data/lib/new_relic/agent/instrumentation/grpc/helper.rb +32 -0
  171. data/lib/new_relic/agent/instrumentation/grpc/server/chain.rb +69 -0
  172. data/lib/new_relic/agent/instrumentation/grpc/server/instrumentation.rb +134 -0
  173. data/lib/new_relic/agent/instrumentation/grpc/server/rpc_desc_prepend.rb +35 -0
  174. data/lib/new_relic/agent/instrumentation/grpc/server/rpc_server_prepend.rb +26 -0
  175. data/lib/new_relic/agent/instrumentation/grpc_client.rb +23 -0
  176. data/lib/new_relic/agent/instrumentation/grpc_server.rb +25 -0
  177. data/lib/new_relic/agent/instrumentation/httpclient/chain.rb +24 -0
  178. data/lib/new_relic/agent/instrumentation/httpclient/instrumentation.rb +37 -0
  179. data/lib/new_relic/agent/instrumentation/httpclient/prepend.rb +15 -0
  180. data/lib/new_relic/agent/instrumentation/httpclient.rb +12 -32
  181. data/lib/new_relic/agent/instrumentation/httprb/chain.rb +22 -0
  182. data/lib/new_relic/agent/instrumentation/httprb/instrumentation.rb +30 -0
  183. data/lib/new_relic/agent/instrumentation/httprb/prepend.rb +15 -0
  184. data/lib/new_relic/agent/instrumentation/httprb.rb +29 -0
  185. data/lib/new_relic/agent/instrumentation/ignore_actions.rb +5 -6
  186. data/lib/new_relic/agent/instrumentation/logger/chain.rb +21 -0
  187. data/lib/new_relic/agent/instrumentation/logger/instrumentation.rb +66 -0
  188. data/lib/new_relic/agent/instrumentation/logger/prepend.rb +13 -0
  189. data/lib/new_relic/agent/instrumentation/logger.rb +26 -0
  190. data/lib/new_relic/agent/instrumentation/memcache/chain.rb +15 -0
  191. data/lib/new_relic/agent/instrumentation/memcache/dalli.rb +58 -125
  192. data/lib/new_relic/agent/instrumentation/memcache/helper.rb +59 -0
  193. data/lib/new_relic/agent/instrumentation/memcache/instrumentation.rb +90 -0
  194. data/lib/new_relic/agent/instrumentation/memcache/prepend.rb +101 -0
  195. data/lib/new_relic/agent/instrumentation/memcache.rb +57 -71
  196. data/lib/new_relic/agent/instrumentation/middleware_proxy.rb +15 -14
  197. data/lib/new_relic/agent/instrumentation/middleware_tracing.rb +21 -14
  198. data/lib/new_relic/agent/instrumentation/mongo.rb +7 -132
  199. data/lib/new_relic/agent/instrumentation/mongodb_command_subscriber.rb +49 -13
  200. data/lib/new_relic/agent/instrumentation/net_http/chain.rb +24 -0
  201. data/lib/new_relic/agent/instrumentation/net_http/instrumentation.rb +40 -0
  202. data/lib/new_relic/agent/instrumentation/net_http/prepend.rb +21 -0
  203. data/lib/new_relic/agent/instrumentation/net_http.rb +44 -0
  204. data/lib/new_relic/agent/instrumentation/notifications_subscriber.rb +142 -0
  205. data/lib/new_relic/agent/instrumentation/padrino/chain.rb +38 -0
  206. data/lib/new_relic/agent/instrumentation/padrino/instrumentation.rb +28 -0
  207. data/lib/new_relic/agent/instrumentation/padrino/prepend.rb +20 -0
  208. data/lib/new_relic/agent/instrumentation/padrino.rb +22 -58
  209. data/lib/new_relic/agent/instrumentation/passenger_instrumentation.rb +7 -7
  210. data/lib/new_relic/agent/instrumentation/queue_time.rb +9 -10
  211. data/lib/new_relic/agent/instrumentation/rack/chain.rb +66 -0
  212. data/lib/new_relic/agent/instrumentation/rack/helpers.rb +33 -0
  213. data/lib/new_relic/agent/instrumentation/rack/instrumentation.rb +75 -0
  214. data/lib/new_relic/agent/instrumentation/rack/prepend.rb +43 -0
  215. data/lib/new_relic/agent/instrumentation/rack.rb +33 -141
  216. data/lib/new_relic/agent/instrumentation/rails3/action_controller.rb +19 -55
  217. data/lib/new_relic/agent/instrumentation/rails_middleware.rb +5 -5
  218. data/lib/new_relic/agent/instrumentation/rails_notifications/action_cable.rb +36 -0
  219. data/lib/new_relic/agent/instrumentation/rails_notifications/action_controller.rb +45 -0
  220. data/lib/new_relic/agent/instrumentation/rails_notifications/action_view.rb +30 -0
  221. data/lib/new_relic/agent/instrumentation/rails_notifications/custom_events.rb +30 -0
  222. data/lib/new_relic/agent/instrumentation/rake/chain.rb +20 -0
  223. data/lib/new_relic/agent/instrumentation/rake/instrumentation.rb +142 -0
  224. data/lib/new_relic/agent/instrumentation/rake/prepend.rb +14 -0
  225. data/lib/new_relic/agent/instrumentation/rake.rb +18 -159
  226. data/lib/new_relic/agent/instrumentation/redis/chain.rb +45 -0
  227. data/lib/new_relic/agent/instrumentation/redis/constants.rb +17 -0
  228. data/lib/new_relic/agent/instrumentation/redis/instrumentation.rb +93 -0
  229. data/lib/new_relic/agent/instrumentation/redis/middleware.rb +16 -0
  230. data/lib/new_relic/agent/instrumentation/redis/prepend.rb +29 -0
  231. data/lib/new_relic/agent/instrumentation/redis.rb +20 -103
  232. data/lib/new_relic/agent/instrumentation/resque/chain.rb +21 -0
  233. data/lib/new_relic/agent/instrumentation/resque/helper.rb +19 -0
  234. data/lib/new_relic/agent/instrumentation/resque/instrumentation.rb +34 -0
  235. data/lib/new_relic/agent/instrumentation/resque/prepend.rb +15 -0
  236. data/lib/new_relic/agent/instrumentation/resque.rb +33 -41
  237. data/lib/new_relic/agent/instrumentation/sequel.rb +17 -20
  238. data/lib/new_relic/agent/instrumentation/sequel_helper.rb +3 -3
  239. data/lib/new_relic/agent/instrumentation/sidekiq/client.rb +20 -0
  240. data/lib/new_relic/agent/instrumentation/sidekiq/extensions/delayed_class.rb +30 -0
  241. data/lib/new_relic/agent/instrumentation/sidekiq/server.rb +37 -0
  242. data/lib/new_relic/agent/instrumentation/sidekiq.rb +29 -46
  243. data/lib/new_relic/agent/instrumentation/sinatra/chain.rb +55 -0
  244. data/lib/new_relic/agent/instrumentation/sinatra/ignorer.rb +31 -37
  245. data/lib/new_relic/agent/instrumentation/sinatra/instrumentation.rb +125 -0
  246. data/lib/new_relic/agent/instrumentation/sinatra/prepend.rb +33 -0
  247. data/lib/new_relic/agent/instrumentation/sinatra/transaction_namer.rb +3 -3
  248. data/lib/new_relic/agent/instrumentation/sinatra.rb +35 -165
  249. data/lib/new_relic/agent/instrumentation/thread/chain.rb +24 -0
  250. data/lib/new_relic/agent/instrumentation/thread/instrumentation.rb +28 -0
  251. data/lib/new_relic/agent/instrumentation/thread/prepend.rb +22 -0
  252. data/lib/new_relic/agent/instrumentation/thread.rb +20 -0
  253. data/lib/new_relic/agent/instrumentation/tilt/chain.rb +24 -0
  254. data/lib/new_relic/agent/instrumentation/tilt/instrumentation.rb +41 -0
  255. data/lib/new_relic/agent/instrumentation/tilt/prepend.rb +13 -0
  256. data/lib/new_relic/agent/instrumentation/tilt.rb +25 -0
  257. data/lib/new_relic/agent/instrumentation/typhoeus/chain.rb +22 -0
  258. data/lib/new_relic/agent/instrumentation/typhoeus/instrumentation.rb +80 -0
  259. data/lib/new_relic/agent/instrumentation/typhoeus/prepend.rb +14 -0
  260. data/lib/new_relic/agent/instrumentation/typhoeus.rb +14 -76
  261. data/lib/new_relic/agent/instrumentation.rb +2 -2
  262. data/lib/new_relic/agent/internal_agent_error.rb +3 -3
  263. data/lib/new_relic/agent/javascript_instrumentor.rb +51 -45
  264. data/lib/new_relic/agent/linking_metadata.rb +44 -0
  265. data/lib/new_relic/agent/local_log_decorator.rb +37 -0
  266. data/lib/new_relic/agent/log_event_aggregator.rb +235 -0
  267. data/lib/new_relic/agent/log_once.rb +2 -2
  268. data/lib/new_relic/agent/log_priority.rb +20 -0
  269. data/lib/new_relic/agent/logging.rb +142 -0
  270. data/lib/new_relic/agent/memory_logger.rb +3 -3
  271. data/lib/new_relic/agent/messaging.rb +81 -164
  272. data/lib/new_relic/agent/method_tracer.rb +152 -145
  273. data/lib/new_relic/agent/method_tracer_helpers.rb +90 -13
  274. data/lib/new_relic/agent/monitors/cross_app_monitor.rb +117 -0
  275. data/lib/new_relic/agent/monitors/distributed_tracing_monitor.rb +28 -0
  276. data/lib/new_relic/agent/{inbound_request_monitor.rb → monitors/inbound_request_monitor.rb} +5 -6
  277. data/lib/new_relic/agent/{synthetics_monitor.rb → monitors/synthetics_monitor.rb} +9 -15
  278. data/lib/new_relic/agent/monitors.rb +26 -0
  279. data/lib/new_relic/agent/new_relic_service/encoders.rb +7 -7
  280. data/lib/new_relic/agent/new_relic_service/json_marshaller.rb +6 -7
  281. data/lib/new_relic/agent/new_relic_service/marshaller.rb +8 -29
  282. data/lib/new_relic/agent/new_relic_service/security_policy_settings.rb +5 -5
  283. data/lib/new_relic/agent/new_relic_service.rb +282 -166
  284. data/lib/new_relic/agent/noticeable_error.rb +19 -0
  285. data/lib/new_relic/agent/null_logger.rb +8 -4
  286. data/lib/new_relic/agent/obfuscator.rb +9 -11
  287. data/lib/new_relic/agent/parameter_filtering.rb +35 -8
  288. data/lib/new_relic/agent/payload_metric_mapping.rb +10 -11
  289. data/lib/new_relic/agent/pipe_channel_manager.rb +28 -18
  290. data/lib/new_relic/agent/pipe_service.rb +9 -6
  291. data/lib/new_relic/agent/prepend_supportability.rb +3 -3
  292. data/lib/new_relic/agent/priority_sampled_buffer.rb +16 -14
  293. data/lib/new_relic/agent/range_extensions.rb +9 -29
  294. data/lib/new_relic/agent/rules_engine/replacement_rule.rb +12 -12
  295. data/lib/new_relic/agent/rules_engine/segment_terms_rule.rb +13 -14
  296. data/lib/new_relic/agent/rules_engine.rb +6 -5
  297. data/lib/new_relic/agent/sampler.rb +4 -5
  298. data/lib/new_relic/agent/sampler_collection.rb +4 -5
  299. data/lib/new_relic/agent/samplers/cpu_sampler.rb +4 -3
  300. data/lib/new_relic/agent/samplers/delayed_job_sampler.rb +14 -11
  301. data/lib/new_relic/agent/samplers/memory_sampler.rb +26 -15
  302. data/lib/new_relic/agent/samplers/object_sampler.rb +2 -2
  303. data/lib/new_relic/agent/samplers/vm_sampler.rb +22 -20
  304. data/lib/new_relic/agent/span_event_aggregator.rb +14 -16
  305. data/lib/new_relic/agent/span_event_primitive.rb +118 -58
  306. data/lib/new_relic/agent/sql_sampler.rb +25 -25
  307. data/lib/new_relic/agent/stats.rb +79 -42
  308. data/lib/new_relic/agent/stats_engine/gc_profiler.rb +11 -13
  309. data/lib/new_relic/agent/stats_engine/stats_hash.rb +13 -14
  310. data/lib/new_relic/agent/stats_engine.rb +11 -11
  311. data/lib/new_relic/agent/synthetics_event_aggregator.rb +8 -9
  312. data/lib/new_relic/agent/system_info.rb +100 -66
  313. data/lib/new_relic/agent/threading/agent_thread.rb +20 -16
  314. data/lib/new_relic/agent/threading/backtrace_node.rb +13 -14
  315. data/lib/new_relic/agent/threading/backtrace_service.rb +18 -18
  316. data/lib/new_relic/agent/threading/thread_profile.rb +31 -45
  317. data/lib/new_relic/agent/timestamp_sampled_buffer.rb +2 -2
  318. data/lib/new_relic/agent/tracer.rb +513 -0
  319. data/lib/new_relic/agent/transaction/abstract_segment.rb +131 -41
  320. data/lib/new_relic/agent/transaction/datastore_segment.rb +22 -18
  321. data/lib/new_relic/agent/transaction/distributed_tracer.rb +184 -0
  322. data/lib/new_relic/agent/transaction/distributed_tracing.rb +72 -163
  323. data/lib/new_relic/agent/transaction/external_request_segment.rb +66 -63
  324. data/lib/new_relic/agent/transaction/message_broker_segment.rb +34 -46
  325. data/lib/new_relic/agent/transaction/request_attributes.rb +36 -36
  326. data/lib/new_relic/agent/transaction/segment.rb +46 -10
  327. data/lib/new_relic/agent/transaction/slowest_sample_buffer.rb +2 -4
  328. data/lib/new_relic/agent/transaction/synthetics_sample_buffer.rb +2 -2
  329. data/lib/new_relic/agent/transaction/trace.rb +21 -24
  330. data/lib/new_relic/agent/transaction/trace_builder.rb +11 -12
  331. data/lib/new_relic/agent/transaction/trace_context.rb +168 -0
  332. data/lib/new_relic/agent/transaction/trace_node.rb +31 -28
  333. data/lib/new_relic/agent/transaction/tracing.rb +15 -111
  334. data/lib/new_relic/agent/transaction/transaction_sample_buffer.rb +6 -6
  335. data/lib/new_relic/agent/transaction.rb +252 -259
  336. data/lib/new_relic/agent/transaction_error_primitive.rb +34 -37
  337. data/lib/new_relic/agent/transaction_event_aggregator.rb +13 -13
  338. data/lib/new_relic/agent/transaction_event_primitive.rb +44 -56
  339. data/lib/new_relic/agent/transaction_event_recorder.rb +17 -16
  340. data/lib/new_relic/agent/transaction_metrics.rb +11 -10
  341. data/lib/new_relic/agent/transaction_sampler.rb +7 -12
  342. data/lib/new_relic/agent/transaction_time_aggregator.rb +41 -26
  343. data/lib/new_relic/agent/utilization/aws.rb +34 -4
  344. data/lib/new_relic/agent/utilization/azure.rb +4 -4
  345. data/lib/new_relic/agent/utilization/gcp.rb +8 -8
  346. data/lib/new_relic/agent/utilization/pcf.rb +6 -5
  347. data/lib/new_relic/agent/utilization/vendor.rb +44 -29
  348. data/lib/new_relic/agent/utilization_data.rb +43 -6
  349. data/lib/new_relic/agent/vm/jruby_vm.rb +2 -2
  350. data/lib/new_relic/agent/vm/monotonic_gc_profiler.rb +3 -3
  351. data/lib/new_relic/agent/vm/mri_vm.rb +46 -19
  352. data/lib/new_relic/agent/vm/snapshot.rb +6 -6
  353. data/lib/new_relic/agent/vm.rb +2 -2
  354. data/lib/new_relic/agent/worker_loop.rb +11 -13
  355. data/lib/new_relic/agent.rb +151 -79
  356. data/lib/new_relic/cli/command.rb +21 -23
  357. data/lib/new_relic/cli/commands/deployments.rb +94 -45
  358. data/lib/new_relic/cli/commands/install.rb +24 -26
  359. data/lib/new_relic/coerce.rb +42 -15
  360. data/lib/new_relic/collection_helper.rb +51 -49
  361. data/lib/new_relic/constants.rb +39 -0
  362. data/lib/new_relic/control/class_methods.rb +11 -5
  363. data/lib/new_relic/control/frameworks/external.rb +3 -3
  364. data/lib/new_relic/control/frameworks/rails.rb +24 -18
  365. data/lib/new_relic/control/frameworks/rails3.rb +4 -5
  366. data/lib/new_relic/control/frameworks/rails4.rb +2 -2
  367. data/lib/new_relic/control/frameworks/rails_notifications.rb +14 -0
  368. data/lib/new_relic/control/frameworks/ruby.rb +4 -4
  369. data/lib/new_relic/control/frameworks/sinatra.rb +8 -2
  370. data/lib/new_relic/control/frameworks.rb +2 -2
  371. data/lib/new_relic/control/instance_methods.rb +33 -42
  372. data/lib/new_relic/control/instrumentation.rb +40 -12
  373. data/lib/new_relic/control/private_instance_methods.rb +48 -0
  374. data/lib/new_relic/control/server_methods.rb +4 -5
  375. data/lib/new_relic/control.rb +2 -3
  376. data/lib/new_relic/delayed_job_injection.rb +2 -2
  377. data/lib/new_relic/dependency_detection.rb +129 -18
  378. data/lib/new_relic/environment_report.rb +41 -35
  379. data/lib/new_relic/helper.rb +49 -8
  380. data/lib/new_relic/language_support.rb +30 -6
  381. data/lib/new_relic/latest_changes.rb +9 -8
  382. data/lib/new_relic/local_environment.rb +23 -27
  383. data/lib/new_relic/metric_data.rb +32 -27
  384. data/lib/new_relic/metric_spec.rb +9 -7
  385. data/lib/new_relic/noticed_error.rb +46 -33
  386. data/lib/new_relic/rack/agent_hooks.rb +2 -2
  387. data/lib/new_relic/rack/agent_middleware.rb +7 -5
  388. data/lib/new_relic/rack/browser_monitoring.rb +134 -117
  389. data/lib/new_relic/rack.rb +2 -2
  390. data/lib/new_relic/recipes/capistrano3.rb +4 -62
  391. data/lib/new_relic/recipes/capistrano_legacy.rb +24 -27
  392. data/lib/new_relic/recipes/helpers/send_deployment.rb +70 -0
  393. data/lib/new_relic/recipes.rb +2 -2
  394. data/lib/new_relic/supportability_helper.rb +21 -7
  395. data/lib/new_relic/traced_thread.rb +39 -0
  396. data/lib/new_relic/version.rb +7 -18
  397. data/lib/newrelic_rpm.rb +20 -33
  398. data/lib/sequel/extensions/{newrelic_instrumentation.rb → new_relic_instrumentation.rb} +16 -19
  399. data/lib/sequel/plugins/{newrelic_instrumentation.rb → new_relic_instrumentation.rb} +9 -15
  400. data/lib/tasks/all.rb +4 -4
  401. data/lib/tasks/config.rake +22 -118
  402. data/lib/tasks/coverage_report.rake +28 -0
  403. data/lib/tasks/helpers/config.html.erb +21 -0
  404. data/lib/tasks/helpers/format.rb +123 -0
  405. data/lib/tasks/helpers/matches.rb +12 -0
  406. data/lib/tasks/helpers/prompt.rb +24 -0
  407. data/lib/tasks/helpers/removers.rb +33 -0
  408. data/lib/tasks/install.rake +4 -0
  409. data/lib/tasks/instrumentation_generator/README.md +63 -0
  410. data/lib/tasks/instrumentation_generator/TODO.md +33 -0
  411. data/lib/tasks/instrumentation_generator/instrumentation.thor +121 -0
  412. data/lib/tasks/instrumentation_generator/templates/Envfile.tt +9 -0
  413. data/lib/tasks/instrumentation_generator/templates/chain.tt +22 -0
  414. data/lib/tasks/instrumentation_generator/templates/chain_method.tt +8 -0
  415. data/lib/tasks/instrumentation_generator/templates/dependency_detection.tt +29 -0
  416. data/lib/tasks/instrumentation_generator/templates/instrumentation.tt +13 -0
  417. data/lib/tasks/instrumentation_generator/templates/instrumentation_method.tt +3 -0
  418. data/lib/tasks/instrumentation_generator/templates/newrelic.yml.tt +19 -0
  419. data/lib/tasks/instrumentation_generator/templates/prepend.tt +13 -0
  420. data/lib/tasks/instrumentation_generator/templates/prepend_method.tt +3 -0
  421. data/lib/tasks/instrumentation_generator/templates/test.tt +15 -0
  422. data/lib/tasks/multiverse.rake +4 -0
  423. data/lib/tasks/multiverse.rb +12 -5
  424. data/lib/tasks/newrelic.rb +2 -2
  425. data/lib/tasks/tests.rake +14 -14
  426. data/newrelic.yml +672 -3
  427. data/newrelic_rpm.gemspec +40 -31
  428. data/recipes/newrelic.rb +3 -3
  429. data/test/agent_helper.rb +419 -98
  430. metadata +238 -127
  431. data/.travis.yml +0 -228
  432. data/bin/mongrel_rpm +0 -33
  433. data/cert/cacert.pem +0 -1177
  434. data/lib/new_relic/agent/commands/xray_session.rb +0 -55
  435. data/lib/new_relic/agent/commands/xray_session_collection.rb +0 -161
  436. data/lib/new_relic/agent/cross_app_monitor.rb +0 -110
  437. data/lib/new_relic/agent/datastores/mongo/obfuscator.rb +0 -44
  438. data/lib/new_relic/agent/datastores/mongo/statement_formatter.rb +0 -53
  439. data/lib/new_relic/agent/distributed_trace_monitor.rb +0 -41
  440. data/lib/new_relic/agent/distributed_trace_payload.rb +0 -246
  441. data/lib/new_relic/agent/http_clients/abstract_request.rb +0 -31
  442. data/lib/new_relic/agent/instrumentation/active_record_4.rb +0 -42
  443. data/lib/new_relic/agent/instrumentation/active_record_5.rb +0 -41
  444. data/lib/new_relic/agent/instrumentation/acts_as_solr.rb +0 -74
  445. data/lib/new_relic/agent/instrumentation/authlogic.rb +0 -25
  446. data/lib/new_relic/agent/instrumentation/data_mapper.rb +0 -202
  447. data/lib/new_relic/agent/instrumentation/evented_subscriber.rb +0 -104
  448. data/lib/new_relic/agent/instrumentation/excon/connection.rb +0 -46
  449. data/lib/new_relic/agent/instrumentation/http.rb +0 -46
  450. data/lib/new_relic/agent/instrumentation/merb/controller.rb +0 -44
  451. data/lib/new_relic/agent/instrumentation/merb/errors.rb +0 -33
  452. data/lib/new_relic/agent/instrumentation/net.rb +0 -50
  453. data/lib/new_relic/agent/instrumentation/rails/action_controller.rb +0 -125
  454. data/lib/new_relic/agent/instrumentation/rails/action_web_service.rb +0 -46
  455. data/lib/new_relic/agent/instrumentation/rails4/action_controller.rb +0 -32
  456. data/lib/new_relic/agent/instrumentation/rails4/action_view.rb +0 -27
  457. data/lib/new_relic/agent/instrumentation/rails5/action_cable.rb +0 -36
  458. data/lib/new_relic/agent/instrumentation/rails5/action_controller.rb +0 -33
  459. data/lib/new_relic/agent/instrumentation/rails5/action_view.rb +0 -27
  460. data/lib/new_relic/agent/instrumentation/rainbows_instrumentation.rb +0 -26
  461. data/lib/new_relic/agent/instrumentation/sunspot.rb +0 -33
  462. data/lib/new_relic/agent/supported_versions.rb +0 -275
  463. data/lib/new_relic/agent/transaction/attributes.rb +0 -154
  464. data/lib/new_relic/agent/transaction/xray_sample_buffer.rb +0 -64
  465. data/lib/new_relic/agent/transaction_state.rb +0 -186
  466. data/lib/new_relic/build.rb +0 -2
  467. data/lib/new_relic/control/frameworks/merb.rb +0 -29
  468. data/lib/new_relic/control/frameworks/rails5.rb +0 -14
  469. data/lib/new_relic/metrics.rb +0 -13
  470. data/lib/tasks/config.html.erb +0 -32
  471. data/lib/tasks/versions.html.erb +0 -28
  472. data/lib/tasks/versions.postface.html +0 -8
  473. data/lib/tasks/versions.preface.html +0 -9
  474. data/lib/tasks/versions.rake +0 -65
  475. data/lib/tasks/versions.txt.erb +0 -14
  476. /data/lib/tasks/{config.text.erb → helpers/config.text.erb} +0 -0
@@ -0,0 +1,45 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ require 'new_relic/agent/instrumentation/action_controller_subscriber'
6
+ require 'new_relic/agent/instrumentation/action_controller_other_subscriber'
7
+ require 'new_relic/agent/prepend_supportability'
8
+
9
+ DependencyDetection.defer do
10
+ @name = :action_controller_notifications
11
+
12
+ depends_on do
13
+ defined?(Rails::VERSION::MAJOR) && Rails::VERSION::MAJOR.to_i >= 4
14
+ end
15
+
16
+ depends_on do
17
+ !NewRelic::Agent.config[:disable_action_controller] &&
18
+ defined?(ActionController) && (defined?(ActionController::Base) || defined?(ActionController::API))
19
+ end
20
+
21
+ executes do
22
+ NewRelic::Agent.logger.info('Installing notifications based Action Controller instrumentation')
23
+ end
24
+
25
+ executes do
26
+ ActiveSupport.on_load(:action_controller) do
27
+ include NewRelic::Agent::Instrumentation::ControllerInstrumentation
28
+
29
+ NewRelic::Agent::PrependSupportability.record_metrics_for(self)
30
+ end
31
+
32
+ NewRelic::Agent::Instrumentation::ActionControllerSubscriber \
33
+ .subscribe(/^process_action.action_controller$/)
34
+
35
+ subs = %w[send_file
36
+ send_data
37
+ redirect_to
38
+ halted_callback
39
+ unpermitted_parameters]
40
+
41
+ # have to double escape period because its going from string -> regex
42
+ NewRelic::Agent::Instrumentation::ActionControllerOtherSubscriber \
43
+ .subscribe(Regexp.new("^(#{subs.join('|')})\\.action_controller$"))
44
+ end
45
+ end
@@ -0,0 +1,30 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ require 'new_relic/agent/instrumentation/action_view_subscriber'
6
+ require 'new_relic/agent/prepend_supportability'
7
+
8
+ DependencyDetection.defer do
9
+ @name = :action_view_notifications
10
+
11
+ depends_on do
12
+ defined?(Rails::VERSION::MAJOR) && Rails::VERSION::MAJOR.to_i >= 4
13
+ end
14
+
15
+ depends_on do
16
+ !NewRelic::Agent.config[:disable_view_instrumentation] &&
17
+ !NewRelic::Agent::Instrumentation::ActionViewSubscriber.subscribed?
18
+ end
19
+
20
+ executes do
21
+ NewRelic::Agent.logger.info('Installing notification based Action View instrumentation')
22
+ end
23
+
24
+ executes do
25
+ ActiveSupport.on_load(:action_view) do
26
+ NewRelic::Agent::Instrumentation::ActionViewSubscriber.subscribe(/render_.+\.action_view$/)
27
+ NewRelic::Agent::PrependSupportability.record_metrics_for(ActionView::Base, ActionView::Template, ActionView::Renderer)
28
+ end
29
+ end
30
+ end
@@ -0,0 +1,30 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ require 'new_relic/agent/instrumentation/custom_events_subscriber'
6
+ require 'new_relic/agent/prepend_supportability'
7
+
8
+ DependencyDetection.defer do
9
+ @name = :custom_event_notifications
10
+
11
+ depends_on do
12
+ defined?(ActiveSupport::Notifications) &&
13
+ defined?(ActiveSupport::IsolatedExecutionState)
14
+ end
15
+
16
+ depends_on do
17
+ !NewRelic::Agent.config[:active_support_custom_events_names].empty? &&
18
+ !NewRelic::Agent::Instrumentation::CustomEventsSubscriber.subscribed?
19
+ end
20
+
21
+ executes do
22
+ NewRelic::Agent.logger.info('Installing notifications based ActiveSupport custom events instrumentation')
23
+ end
24
+
25
+ executes do
26
+ NewRelic::Agent.config[:active_support_custom_events_names].each do |name|
27
+ ActiveSupport::Notifications.subscribe(name, NewRelic::Agent::Instrumentation::CustomEventsSubscriber.new)
28
+ end
29
+ end
30
+ end
@@ -0,0 +1,20 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation
6
+ module Rake
7
+ module Chain
8
+ def self.instrument!
9
+ ::Rake::Task.class_eval do
10
+ include NewRelic::Agent::Instrumentation::Rake::Tracer
11
+ alias_method(:invoke_without_newrelic, :invoke)
12
+
13
+ def invoke(*args)
14
+ invoke_with_newrelic_tracing(*args) { invoke_without_newrelic(*args) }
15
+ end
16
+ end
17
+ end
18
+ end
19
+ end
20
+ end
@@ -0,0 +1,142 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic
6
+ module Agent
7
+ module Instrumentation
8
+ module Rake
9
+ module Tracer
10
+ def invoke_with_newrelic_tracing(*args)
11
+ unless NewRelic::Agent::Instrumentation::Rake.should_trace?(name)
12
+ return yield
13
+ end
14
+
15
+ begin
16
+ timeout = NewRelic::Agent.config[:'rake.connect_timeout']
17
+ NewRelic::Agent.instance.wait_on_connect(timeout)
18
+ rescue => e
19
+ NewRelic::Agent.logger.error("Exception in wait_on_connect", e)
20
+ return yield
21
+ end
22
+
23
+ NewRelic::Agent::Instrumentation::Rake.before_invoke_transaction(self)
24
+
25
+ NewRelic::Agent::Tracer.in_transaction(name: "OtherTransaction/Rake/invoke/#{name}", category: :rake) do
26
+ NewRelic::Agent::Instrumentation::Rake.record_attributes(args, self)
27
+ yield
28
+ end
29
+ end
30
+ end
31
+
32
+ module_function
33
+
34
+ def should_install?
35
+ safe_from_third_party_gem?
36
+ end
37
+
38
+ def safe_from_third_party_gem?
39
+ return true unless NewRelic::LanguageSupport.bundled_gem?("newrelic-rake")
40
+
41
+ ::NewRelic::Agent.logger.info("Not installing New Relic supported Rake instrumentation because the third party newrelic-rake gem is present")
42
+ false
43
+ end
44
+
45
+ def should_trace?(name)
46
+ NewRelic::Agent.config[:'rake.tasks'].any? do |regex|
47
+ regex.match(name)
48
+ end
49
+ end
50
+
51
+ def instrument_execute_on_prereqs(task)
52
+ task.prerequisite_tasks.each do |child_task|
53
+ instrument_execute(child_task)
54
+ end
55
+ end
56
+
57
+ def instrument_execute(task)
58
+ return if task.instance_variable_get(:@__newrelic_instrumented_execute)
59
+
60
+ task.instance_variable_set(:@__newrelic_instrumented_execute, true)
61
+ task.instance_eval do
62
+ def execute(*args, &block)
63
+ NewRelic::Agent::MethodTracer.trace_execution_scoped("Rake/execute/#{self.name}") do
64
+ super
65
+ end
66
+ end
67
+ end
68
+
69
+ instrument_execute_on_prereqs(task)
70
+ end
71
+
72
+ def instrument_invoke_prerequisites_concurrently(task)
73
+ task.instance_eval do
74
+ def invoke_prerequisites_concurrently(*_)
75
+ NewRelic::Agent::MethodTracer.trace_execution_scoped("Rake/execute/multitask") do
76
+ super
77
+ end
78
+ end
79
+ end
80
+ end
81
+
82
+ def before_invoke_transaction(task)
83
+ ensure_at_exit
84
+
85
+ instrument_execute_on_prereqs(task)
86
+ if task.application.options.always_multitask
87
+ instrument_invoke_prerequisites_concurrently(task)
88
+ end
89
+ rescue => e
90
+ NewRelic::Agent.logger.error("Error during Rake task invoke", e)
91
+ end
92
+
93
+ def record_attributes(args, task)
94
+ command_line = task.application.top_level_tasks.join(" ")
95
+ NewRelic::Agent::Transaction.merge_untrusted_agent_attributes({:command => command_line},
96
+ :'job.rake',
97
+ NewRelic::Agent::AttributeFilter::DST_NONE)
98
+ named_args = name_the_args(args, task.arg_names)
99
+ unless named_args.empty?
100
+ NewRelic::Agent::Transaction.merge_untrusted_agent_attributes(named_args,
101
+ :'job.rake.args',
102
+ NewRelic::Agent::AttributeFilter::DST_NONE)
103
+ end
104
+ rescue => e
105
+ NewRelic::Agent.logger.error("Error during Rake task attribute recording.", e)
106
+ end
107
+
108
+ # Expects literal args passed to the task and array of task names
109
+ # If names are present without matching args, still sets them with nils
110
+ def name_the_args(args, names)
111
+ unfulfilled_names_length = names.length - args.length
112
+ if unfulfilled_names_length > 0
113
+ args.concat(Array.new(unfulfilled_names_length))
114
+ end
115
+
116
+ result = {}
117
+ args.zip(names).each_with_index do |(value, key), index|
118
+ result[key || index.to_s] = value
119
+ end
120
+ result
121
+ end
122
+
123
+ def ensure_at_exit
124
+ return if @installed_at_exit
125
+
126
+ at_exit do
127
+ # The agent's default at_exit might not default to installing, but
128
+ # if we are running an instrumented rake task, we always want it.
129
+ # No code coverage, as the strategy for mocking Kernel.exit prevents
130
+ # the at_exit block from being executed, and not mocking results in
131
+ # an early unwanted exit of MiniTest.
132
+ # :nocov:
133
+ NewRelic::Agent.shutdown
134
+ # :nocov:
135
+ end
136
+
137
+ @installed_at_exit = true
138
+ end
139
+ end
140
+ end
141
+ end
142
+ end
@@ -0,0 +1,14 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation
6
+ module Rake
7
+ module Prepend
8
+ include NewRelic::Agent::Instrumentation::Rake::Tracer
9
+ def invoke(*args)
10
+ invoke_with_newrelic_tracing(*args) { super }
11
+ end
12
+ end
13
+ end
14
+ end
@@ -1,172 +1,31 @@
1
- # encoding: utf-8
2
1
  # This file is distributed under New Relic's license terms.
3
- # See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ require_relative 'rake/instrumentation'
6
+ require_relative 'rake/chain'
7
+ require_relative 'rake/prepend'
4
8
 
5
9
  DependencyDetection.defer do
6
10
  # Why not :rake? newrelic-rake used that name, so avoid conflicting
7
- named :rake_instrumentation
11
+ @name = :rake_instrumentation
12
+ configure_with :rake
8
13
 
9
- depends_on do
10
- defined?(::Rake) &&
11
- ::NewRelic::Agent.config[:'disable_rake'] == false &&
12
- ::NewRelic::Agent.config[:'rake.tasks'].any? &&
13
- ::NewRelic::Agent::Instrumentation::RakeInstrumentation.should_install?
14
- end
14
+ depends_on { defined?(Rake) && defined?(Rake::VERSION) }
15
+ depends_on { Gem::Version.new(Rake::VERSION) >= Gem::Version.new("10.0.0") }
16
+ depends_on { NewRelic::Agent.config[:'rake.tasks'].any? }
17
+ depends_on { NewRelic::Agent::Instrumentation::Rake.safe_from_third_party_gem? }
15
18
 
16
19
  executes do
17
- ::NewRelic::Agent.logger.info "Installing Rake instrumentation"
18
- ::NewRelic::Agent.logger.debug "Instrumenting Rake tasks: #{::NewRelic::Agent.config[:'rake.tasks']}"
20
+ NewRelic::Agent.logger.info("Installing Rake instrumentation")
21
+ NewRelic::Agent.logger.debug("Instrumenting Rake tasks: #{NewRelic::Agent.config[:'rake.tasks']}")
19
22
  end
20
23
 
21
24
  executes do
22
- module Rake
23
- class Task
24
- alias_method :invoke_without_newrelic, :invoke
25
-
26
- def invoke(*args)
27
- unless NewRelic::Agent::Instrumentation::RakeInstrumentation.should_trace? name
28
- return invoke_without_newrelic(*args)
29
- end
30
-
31
- begin
32
- timeout = NewRelic::Agent.config[:'rake.connect_timeout']
33
- NewRelic::Agent.instance.wait_on_connect(timeout)
34
- rescue => e
35
- NewRelic::Agent.logger.error("Exception in wait_on_connect", e)
36
- return invoke_without_newrelic(*args)
37
- end
38
-
39
- NewRelic::Agent::Instrumentation::RakeInstrumentation.before_invoke_transaction(self)
40
-
41
- state = NewRelic::Agent::TransactionState.tl_get
42
- NewRelic::Agent::Transaction.wrap(state, "OtherTransaction/Rake/invoke/#{name}", :rake) do
43
- NewRelic::Agent::Instrumentation::RakeInstrumentation.record_attributes(args, self)
44
- invoke_without_newrelic(*args)
45
- end
46
- end
47
- end
48
- end
49
- end
50
- end
51
-
52
- module NewRelic
53
- module Agent
54
- module Instrumentation
55
- module RakeInstrumentation
56
- def self.should_install?
57
- is_supported_version? && safe_from_third_party_gem?
58
- end
59
-
60
- def self.is_supported_version?
61
- Gem::Version.new(::Rake::VERSION) >= Gem::Version.new("10.0.0")
62
- end
63
-
64
- def self.safe_from_third_party_gem?
65
- if NewRelic::LanguageSupport.bundled_gem?("newrelic-rake")
66
- ::NewRelic::Agent.logger.info("Not installing New Relic supported Rake instrumentation because the third party newrelic-rake gem is present")
67
- false
68
- else
69
- true
70
- end
71
- end
72
-
73
- def self.should_trace?(name)
74
- NewRelic::Agent.config[:'rake.tasks'].any? do |regex|
75
- regex.match(name)
76
- end
77
- end
78
-
79
- def self.instrument_execute_on_prereqs(task)
80
- task.prerequisite_tasks.each do |child_task|
81
- instrument_execute(child_task)
82
- end
83
- end
84
-
85
- def self.instrument_execute(task)
86
- return if task.instance_variable_get(:@__newrelic_instrumented_execute)
87
-
88
- task.instance_variable_set(:@__newrelic_instrumented_execute, true)
89
- task.instance_eval do
90
- def execute(*args, &block)
91
- NewRelic::Agent::MethodTracer.trace_execution_scoped("Rake/execute/#{self.name}") do
92
- super
93
- end
94
- end
95
- end
96
-
97
- instrument_execute_on_prereqs(task)
98
- end
99
-
100
- def self.instrument_invoke_prerequisites_concurrently(task)
101
- task.instance_eval do
102
- def invoke_prerequisites_concurrently(*_)
103
- NewRelic::Agent::MethodTracer.trace_execution_scoped("Rake/execute/multitask") do
104
- prereqs = self.prerequisite_tasks.map(&:name).join(", ")
105
- if txn = ::NewRelic::Agent::TransactionState.tl_get.current_transaction
106
- txn.current_segment.params[:statement] = NewRelic::Agent::Database.truncate_query("Couldn't trace concurrent prereq tasks: #{prereqs}")
107
- end
108
- super
109
- end
110
- end
111
- end
112
- end
113
-
114
- def self.before_invoke_transaction(task)
115
- ensure_at_exit
116
-
117
- # We can't represent overlapping operations yet, so if multitask just
118
- # make one node and annotate with prereq task names
119
- if task.application.options.always_multitask
120
- instrument_invoke_prerequisites_concurrently(task)
121
- else
122
- instrument_execute_on_prereqs(task)
123
- end
124
- rescue => e
125
- NewRelic::Agent.logger.error("Error during Rake task invoke", e)
126
- end
127
-
128
- def self.record_attributes(args, task)
129
- command_line = task.application.top_level_tasks.join(" ")
130
- NewRelic::Agent::Transaction.merge_untrusted_agent_attributes({ :command => command_line },
131
- :'job.rake',
132
- NewRelic::Agent::AttributeFilter::DST_NONE)
133
- named_args = name_the_args(args, task.arg_names)
134
- unless named_args.empty?
135
- NewRelic::Agent::Transaction.merge_untrusted_agent_attributes(named_args,
136
- :'job.rake.args',
137
- NewRelic::Agent::AttributeFilter::DST_NONE)
138
- end
139
- rescue => e
140
- NewRelic::Agent.logger.error("Error during Rake task attribute recording.", e)
141
- end
142
-
143
- # Expects literal args passed to the task and array of task names
144
- # If names are present without matching args, still sets them with nils
145
- def self.name_the_args(args, names)
146
- unfulfilled_names_length = names.length - args.length
147
- if unfulfilled_names_length > 0
148
- args.concat(Array.new(unfulfilled_names_length))
149
- end
150
-
151
- result = {}
152
- args.zip(names).each_with_index do |(value, key), index|
153
- result[key || index.to_s] = value
154
- end
155
- result
156
- end
157
-
158
- def self.ensure_at_exit
159
- return if @installed_at_exit
160
-
161
- at_exit do
162
- # The agent's default at_exit might not default to installing, but
163
- # if we are running an instrumented rake task, we always want it.
164
- NewRelic::Agent.shutdown
165
- end
166
-
167
- @installed_at_exit = true
168
- end
169
- end
25
+ if use_prepend?
26
+ prepend_instrument Rake::Task, NewRelic::Agent::Instrumentation::Rake::Prepend
27
+ else
28
+ chain_instrument NewRelic::Agent::Instrumentation::Rake::Chain
170
29
  end
171
30
  end
172
31
  end
@@ -0,0 +1,45 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation
6
+ module Redis
7
+ module Chain
8
+ def self.instrument!
9
+ ::Redis::Client.class_eval do
10
+ include NewRelic::Agent::Instrumentation::Redis
11
+
12
+ if method_defined?(:call_v)
13
+ alias_method(:call_v_without_new_relic, :call_v)
14
+
15
+ def call_v(*args, &block)
16
+ call_with_tracing(args[0]) { call_v_without_new_relic(*args, &block) }
17
+ end
18
+ end
19
+
20
+ if method_defined?(:call)
21
+ alias_method(:call_without_new_relic, :call)
22
+
23
+ def call(*args, &block)
24
+ call_with_tracing(args[0]) { call_without_new_relic(*args, &block) }
25
+ end
26
+ end
27
+
28
+ if method_defined?(:call_pipeline)
29
+ alias_method(:call_pipeline_without_new_relic, :call_pipeline)
30
+
31
+ def call_pipeline(*args, &block)
32
+ call_pipeline_with_tracing(args[0]) { call_pipeline_without_new_relic(*args, &block) }
33
+ end
34
+ end
35
+
36
+ alias_method(:connect_without_new_relic, :connect)
37
+
38
+ def connect(*args, &block)
39
+ connect_with_tracing { connect_without_new_relic(*args, &block) }
40
+ end
41
+ end
42
+ end
43
+ end
44
+ end
45
+ end
@@ -0,0 +1,17 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation::Redis
6
+ class Constants
7
+ PRODUCT_NAME = 'Redis'
8
+ CONNECT = 'connect'
9
+ UNKNOWN = 'unknown'
10
+ LOCALHOST = 'localhost'
11
+ MULTI_OPERATION = 'multi'
12
+ PIPELINE_OPERATION = 'pipeline'
13
+ HAS_REDIS_CLIENT = defined?(::Redis) &&
14
+ Gem::Version.new(::Redis::VERSION) >= Gem::Version.new('5.0.0') &&
15
+ !!defined?(::RedisClient)
16
+ end
17
+ end
@@ -0,0 +1,93 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ require_relative 'constants'
6
+
7
+ module NewRelic::Agent::Instrumentation
8
+ module Redis
9
+ def connect_with_tracing
10
+ with_tracing(Constants::CONNECT, database: db) { yield }
11
+ end
12
+
13
+ def call_with_tracing(command, &block)
14
+ operation = command[0]
15
+ statement = ::NewRelic::Agent::Datastores::Redis.format_command(command)
16
+
17
+ with_tracing(operation, statement: statement, database: db) { yield }
18
+ end
19
+
20
+ # Used for Redis 4.x and 3.x
21
+ def call_pipeline_with_tracing(pipeline)
22
+ operation = pipeline.is_a?(::Redis::Pipeline::Multi) ? Constants::MULTI_OPERATION : Constants::PIPELINE_OPERATION
23
+ statement = ::NewRelic::Agent::Datastores::Redis.format_pipeline_commands(pipeline.commands)
24
+
25
+ with_tracing(operation, statement: statement, database: db) { yield }
26
+ end
27
+
28
+ # Used for Redis 5.x+
29
+ def call_pipelined_with_tracing(pipeline)
30
+ db = begin
31
+ _nr_redis_client_config.db
32
+ rescue StandardError => e
33
+ NewRelic::Agent.logger.error("Failed to determine configured Redis db value: #{e.class} - #{e.message}")
34
+ nil
35
+ end
36
+
37
+ operation = pipeline.flatten.include?('MULTI') ? Constants::MULTI_OPERATION : Constants::PIPELINE_OPERATION
38
+ statement = ::NewRelic::Agent::Datastores::Redis.format_pipeline_commands(pipeline)
39
+
40
+ with_tracing(operation, statement: statement, database: db) { yield }
41
+ end
42
+
43
+ private
44
+
45
+ def with_tracing(operation, statement: nil, database: nil)
46
+ segment = NewRelic::Agent::Tracer.start_datastore_segment(
47
+ product: Constants::PRODUCT_NAME,
48
+ operation: operation,
49
+ host: _nr_hostname,
50
+ port_path_or_id: _nr_port_path_or_id,
51
+ database_name: database
52
+ )
53
+ begin
54
+ segment.notice_nosql_statement(statement) if statement
55
+ NewRelic::Agent::Tracer.capture_segment_error(segment) { yield }
56
+ ensure
57
+ ::NewRelic::Agent::Transaction::Segment.finish(segment)
58
+ end
59
+ end
60
+
61
+ def _nr_hostname
62
+ _nr_redis_client_config.path ? Constants::LOCALHOST : _nr_redis_client_config.host
63
+ rescue => e
64
+ NewRelic::Agent.logger.debug("Failed to retrieve Redis host: #{e.class} - #{e.message}")
65
+ Constants::UNKNOWN
66
+ end
67
+
68
+ def _nr_port_path_or_id
69
+ _nr_redis_client_config.path || _nr_redis_client_config.port
70
+ rescue => e
71
+ NewRelic::Agent.logger.debug("Failed to retrieve Redis port_path_or_id: #{e.class} - #{e.message}")
72
+ Constants::UNKNOWN
73
+ end
74
+
75
+ def _nr_redis_client_config
76
+ @nr_config ||= begin
77
+ # redis gem
78
+ config = if defined?(::Redis::Client) && self.is_a?(::Redis::Client)
79
+ self
80
+ # redis-client gem v0.11+ (self is a RedisClient::Middlewares)
81
+ elsif respond_to?(:client)
82
+ client && client.config
83
+ # redis-client gem <0.11 (self is a RedisClient::Middlewares)
84
+ elsif defined?(::RedisClient)
85
+ ::RedisClient.config if ::RedisClient.respond_to?(:config)
86
+ end
87
+ raise 'Unable to locate the underlying Redis client configuration.' unless config
88
+
89
+ config
90
+ end
91
+ end
92
+ end
93
+ end
@@ -0,0 +1,16 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation
6
+ module RedisClient
7
+ module Middleware
8
+ # This module is used to instrument Redis 5.x+
9
+ include NewRelic::Agent::Instrumentation::Redis
10
+
11
+ def call_pipelined(*args, &block)
12
+ call_pipelined_with_tracing(args[0]) { super }
13
+ end
14
+ end
15
+ end
16
+ end
@@ -0,0 +1,29 @@
1
+ # This file is distributed under New Relic's license terms.
2
+ # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
+
5
+ module NewRelic::Agent::Instrumentation
6
+ module Redis
7
+ module Prepend
8
+ include NewRelic::Agent::Instrumentation::Redis
9
+
10
+ # Defined in version 5.x+
11
+ def call_v(*args, &block)
12
+ call_with_tracing(args[0]) { super }
13
+ end
14
+
15
+ # Defined in version 4.x, 3.x
16
+ def call(*args, &block)
17
+ call_with_tracing(args[0]) { super }
18
+ end
19
+
20
+ def call_pipeline(*args, &block)
21
+ call_pipeline_with_tracing(args[0]) { super }
22
+ end
23
+
24
+ def connect(*args, &block)
25
+ connect_with_tracing { super }
26
+ end
27
+ end
28
+ end
29
+ end