newrelic_rpm 5.7.0.350 → 9.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
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