newrelic_rpm 6.15.0 → 8.13.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (427) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +7 -1
  3. data/.rubocop.yml +1817 -0
  4. data/.rubocop_todo.yml +59 -0
  5. data/.simplecov +15 -0
  6. data/.snyk +11 -0
  7. data/.yardopts +1 -0
  8. data/Brewfile +12 -0
  9. data/CHANGELOG.md +754 -40
  10. data/CONTRIBUTING.md +32 -5
  11. data/DOCKER.md +167 -0
  12. data/Dockerfile +10 -0
  13. data/Gemfile +0 -1
  14. data/Guardfile +9 -8
  15. data/LICENSE +0 -6
  16. data/README.md +21 -20
  17. data/Rakefile +27 -29
  18. data/THIRD_PARTY_NOTICES.md +14 -199
  19. data/Thorfile +5 -0
  20. data/bin/newrelic +3 -2
  21. data/bin/newrelic_cmd +1 -0
  22. data/bin/nrdebug +76 -53
  23. data/config.dot +5 -5
  24. data/docker-compose.yml +107 -0
  25. data/init.rb +4 -6
  26. data/install.rb +2 -2
  27. data/lefthook.yml +9 -0
  28. data/lib/new_relic/agent/adaptive_sampler.rb +13 -9
  29. data/lib/new_relic/agent/agent/shutdown.rb +35 -0
  30. data/lib/new_relic/agent/agent/special_startup.rb +72 -0
  31. data/lib/new_relic/agent/agent/start_worker_thread.rb +163 -0
  32. data/lib/new_relic/agent/agent/startup.rb +197 -0
  33. data/lib/new_relic/agent/agent.rb +178 -555
  34. data/lib/new_relic/agent/agent_logger.rb +23 -19
  35. data/lib/new_relic/agent/attribute_filter.rb +66 -47
  36. data/lib/new_relic/agent/attribute_processing.rb +7 -7
  37. data/lib/new_relic/agent/attributes.rb +5 -4
  38. data/lib/new_relic/agent/audit_logger.rb +8 -3
  39. data/lib/new_relic/agent/autostart.rb +20 -19
  40. data/lib/new_relic/agent/chained_call.rb +1 -1
  41. data/lib/new_relic/agent/commands/agent_command.rb +3 -3
  42. data/lib/new_relic/agent/commands/agent_command_router.rb +12 -11
  43. data/lib/new_relic/agent/commands/thread_profiler_session.rb +9 -7
  44. data/lib/new_relic/agent/configuration/default_source.rb +1728 -1194
  45. data/lib/new_relic/agent/configuration/dotted_hash.rb +6 -5
  46. data/lib/new_relic/agent/configuration/environment_source.rb +11 -9
  47. data/lib/new_relic/agent/configuration/event_harvest_config.rb +40 -17
  48. data/lib/new_relic/agent/configuration/high_security_source.rb +9 -8
  49. data/lib/new_relic/agent/configuration/manager.rb +82 -69
  50. data/lib/new_relic/agent/configuration/manual_source.rb +1 -1
  51. data/lib/new_relic/agent/configuration/mask_defaults.rb +3 -3
  52. data/lib/new_relic/agent/configuration/security_policy_source.rb +90 -77
  53. data/lib/new_relic/agent/configuration/server_source.rb +21 -19
  54. data/lib/new_relic/agent/configuration/yaml_source.rb +31 -7
  55. data/lib/new_relic/agent/configuration.rb +1 -1
  56. data/lib/new_relic/agent/connect/request_builder.rb +18 -18
  57. data/lib/new_relic/agent/connect/response_handler.rb +5 -8
  58. data/lib/new_relic/agent/custom_event_aggregator.rb +14 -14
  59. data/lib/new_relic/agent/database/explain_plan_helpers.rb +4 -5
  60. data/lib/new_relic/agent/database/obfuscation_helpers.rb +14 -13
  61. data/lib/new_relic/agent/database/obfuscator.rb +2 -2
  62. data/lib/new_relic/agent/database/postgres_explain_obfuscator.rb +3 -3
  63. data/lib/new_relic/agent/database.rb +53 -47
  64. data/lib/new_relic/agent/database_adapter.rb +3 -1
  65. data/lib/new_relic/agent/datastores/metric_helper.rb +17 -18
  66. data/lib/new_relic/agent/datastores/mongo/event_formatter.rb +7 -6
  67. data/lib/new_relic/agent/datastores/mongo/metric_translator.rb +6 -10
  68. data/lib/new_relic/agent/datastores/mongo.rb +6 -11
  69. data/lib/new_relic/agent/datastores/nosql_obfuscator.rb +41 -0
  70. data/lib/new_relic/agent/datastores/redis.rb +5 -11
  71. data/lib/new_relic/agent/datastores.rb +11 -13
  72. data/lib/new_relic/agent/deprecator.rb +1 -1
  73. data/lib/new_relic/agent/distributed_tracing/cross_app_payload.rb +10 -10
  74. data/lib/new_relic/agent/distributed_tracing/cross_app_tracing.rb +39 -32
  75. data/lib/new_relic/agent/distributed_tracing/distributed_trace_attributes.rb +22 -22
  76. data/lib/new_relic/agent/distributed_tracing/distributed_trace_metrics.rb +19 -19
  77. data/lib/new_relic/agent/distributed_tracing/distributed_trace_payload.rb +50 -52
  78. data/lib/new_relic/agent/distributed_tracing/distributed_trace_transport_type.rb +5 -6
  79. data/lib/new_relic/agent/distributed_tracing/trace_context.rb +76 -77
  80. data/lib/new_relic/agent/distributed_tracing/trace_context_payload.rb +39 -38
  81. data/lib/new_relic/agent/distributed_tracing.rb +32 -100
  82. data/lib/new_relic/agent/encoding_normalizer.rb +4 -2
  83. data/lib/new_relic/agent/error_collector.rb +73 -54
  84. data/lib/new_relic/agent/error_event_aggregator.rb +4 -5
  85. data/lib/new_relic/agent/error_filter.rb +174 -0
  86. data/lib/new_relic/agent/error_trace_aggregator.rb +4 -3
  87. data/lib/new_relic/agent/event_aggregator.rb +22 -21
  88. data/lib/new_relic/agent/event_buffer.rb +7 -8
  89. data/lib/new_relic/agent/event_listener.rb +1 -2
  90. data/lib/new_relic/agent/event_loop.rb +25 -23
  91. data/lib/new_relic/agent/external.rb +11 -44
  92. data/lib/new_relic/agent/guid_generator.rb +4 -11
  93. data/lib/new_relic/agent/harvester.rb +4 -5
  94. data/lib/new_relic/agent/heap.rb +6 -8
  95. data/lib/new_relic/agent/hostname.rb +20 -13
  96. data/lib/new_relic/agent/http_clients/abstract.rb +10 -11
  97. data/lib/new_relic/agent/http_clients/curb_wrappers.rb +6 -11
  98. data/lib/new_relic/agent/http_clients/excon_wrappers.rb +8 -9
  99. data/lib/new_relic/agent/http_clients/http_rb_wrappers.rb +4 -5
  100. data/lib/new_relic/agent/http_clients/httpclient_wrappers.rb +2 -4
  101. data/lib/new_relic/agent/http_clients/net_http_wrappers.rb +8 -4
  102. data/lib/new_relic/agent/http_clients/typhoeus_wrappers.rb +4 -5
  103. data/lib/new_relic/agent/http_clients/uri_util.rb +1 -3
  104. data/lib/new_relic/agent/instrumentation/action_cable_subscriber.rb +8 -8
  105. data/lib/new_relic/agent/instrumentation/action_controller_subscriber.rb +38 -31
  106. data/lib/new_relic/agent/instrumentation/action_view_subscriber.rb +12 -13
  107. data/lib/new_relic/agent/instrumentation/active_job.rb +14 -7
  108. data/lib/new_relic/agent/instrumentation/active_merchant.rb +20 -6
  109. data/lib/new_relic/agent/instrumentation/active_record.rb +29 -27
  110. data/lib/new_relic/agent/instrumentation/active_record_helper.rb +71 -61
  111. data/lib/new_relic/agent/instrumentation/active_record_notifications.rb +31 -44
  112. data/lib/new_relic/agent/instrumentation/active_record_prepend.rb +4 -4
  113. data/lib/new_relic/agent/instrumentation/active_record_subscriber.rb +27 -20
  114. data/lib/new_relic/agent/instrumentation/active_storage.rb +2 -2
  115. data/lib/new_relic/agent/instrumentation/active_storage_subscriber.rb +18 -15
  116. data/lib/new_relic/agent/instrumentation/active_support_logger/chain.rb +23 -0
  117. data/lib/new_relic/agent/instrumentation/active_support_logger/instrumentation.rb +20 -0
  118. data/lib/new_relic/agent/instrumentation/active_support_logger/prepend.rb +12 -0
  119. data/lib/new_relic/agent/instrumentation/active_support_logger.rb +24 -0
  120. data/lib/new_relic/agent/instrumentation/acts_as_solr.rb +18 -10
  121. data/lib/new_relic/agent/instrumentation/authlogic.rb +10 -2
  122. data/lib/new_relic/agent/instrumentation/bunny/chain.rb +45 -0
  123. data/lib/new_relic/agent/instrumentation/bunny/instrumentation.rb +150 -0
  124. data/lib/new_relic/agent/instrumentation/bunny/prepend.rb +35 -0
  125. data/lib/new_relic/agent/instrumentation/bunny.rb +11 -154
  126. data/lib/new_relic/agent/instrumentation/controller_instrumentation.rb +68 -58
  127. data/lib/new_relic/agent/instrumentation/curb/chain.rb +91 -0
  128. data/lib/new_relic/agent/instrumentation/curb/instrumentation.rb +221 -0
  129. data/lib/new_relic/agent/instrumentation/curb/prepend.rb +61 -0
  130. data/lib/new_relic/agent/instrumentation/curb.rb +11 -243
  131. data/lib/new_relic/agent/instrumentation/data_mapper.rb +65 -54
  132. data/lib/new_relic/agent/instrumentation/delayed_job/chain.rb +35 -0
  133. data/lib/new_relic/agent/instrumentation/delayed_job/instrumentation.rb +48 -0
  134. data/lib/new_relic/agent/instrumentation/delayed_job/prepend.rb +33 -0
  135. data/lib/new_relic/agent/instrumentation/delayed_job_instrumentation.rb +29 -53
  136. data/lib/new_relic/agent/instrumentation/elasticsearch/chain.rb +29 -0
  137. data/lib/new_relic/agent/instrumentation/elasticsearch/instrumentation.rb +66 -0
  138. data/lib/new_relic/agent/instrumentation/elasticsearch/prepend.rb +13 -0
  139. data/lib/new_relic/agent/instrumentation/elasticsearch.rb +31 -0
  140. data/lib/new_relic/agent/instrumentation/excon/middleware.rb +5 -5
  141. data/lib/new_relic/agent/instrumentation/excon.rb +22 -24
  142. data/lib/new_relic/agent/instrumentation/grape/chain.rb +24 -0
  143. data/lib/new_relic/agent/instrumentation/grape/instrumentation.rb +100 -0
  144. data/lib/new_relic/agent/instrumentation/grape/prepend.rb +17 -0
  145. data/lib/new_relic/agent/instrumentation/grape.rb +11 -112
  146. data/lib/new_relic/agent/instrumentation/grpc/client/chain.rb +97 -0
  147. data/lib/new_relic/agent/instrumentation/grpc/client/instrumentation.rb +89 -0
  148. data/lib/new_relic/agent/instrumentation/grpc/client/prepend.rb +111 -0
  149. data/lib/new_relic/agent/instrumentation/grpc/client/request_wrapper.rb +30 -0
  150. data/lib/new_relic/agent/instrumentation/grpc/helper.rb +32 -0
  151. data/lib/new_relic/agent/instrumentation/grpc/server/chain.rb +69 -0
  152. data/lib/new_relic/agent/instrumentation/grpc/server/instrumentation.rb +134 -0
  153. data/lib/new_relic/agent/instrumentation/grpc/server/rpc_desc_prepend.rb +35 -0
  154. data/lib/new_relic/agent/instrumentation/grpc/server/rpc_server_prepend.rb +26 -0
  155. data/lib/new_relic/agent/instrumentation/grpc_client.rb +23 -0
  156. data/lib/new_relic/agent/instrumentation/grpc_server.rb +25 -0
  157. data/lib/new_relic/agent/instrumentation/httpclient/chain.rb +24 -0
  158. data/lib/new_relic/agent/instrumentation/httpclient/instrumentation.rb +37 -0
  159. data/lib/new_relic/agent/instrumentation/httpclient/prepend.rb +15 -0
  160. data/lib/new_relic/agent/instrumentation/httpclient.rb +10 -32
  161. data/lib/new_relic/agent/instrumentation/httprb/chain.rb +22 -0
  162. data/lib/new_relic/agent/instrumentation/httprb/instrumentation.rb +30 -0
  163. data/lib/new_relic/agent/instrumentation/httprb/prepend.rb +15 -0
  164. data/lib/new_relic/agent/instrumentation/httprb.rb +29 -0
  165. data/lib/new_relic/agent/instrumentation/ignore_actions.rb +4 -5
  166. data/lib/new_relic/agent/instrumentation/logger/chain.rb +21 -0
  167. data/lib/new_relic/agent/instrumentation/logger/instrumentation.rb +66 -0
  168. data/lib/new_relic/agent/instrumentation/logger/prepend.rb +13 -0
  169. data/lib/new_relic/agent/instrumentation/logger.rb +26 -0
  170. data/lib/new_relic/agent/instrumentation/memcache/chain.rb +15 -0
  171. data/lib/new_relic/agent/instrumentation/memcache/dalli.rb +57 -128
  172. data/lib/new_relic/agent/instrumentation/memcache/helper.rb +59 -0
  173. data/lib/new_relic/agent/instrumentation/memcache/instrumentation.rb +90 -0
  174. data/lib/new_relic/agent/instrumentation/memcache/prepend.rb +101 -0
  175. data/lib/new_relic/agent/instrumentation/memcache.rb +56 -72
  176. data/lib/new_relic/agent/instrumentation/middleware_proxy.rb +13 -12
  177. data/lib/new_relic/agent/instrumentation/middleware_tracing.rb +5 -7
  178. data/lib/new_relic/agent/instrumentation/mongo.rb +5 -136
  179. data/lib/new_relic/agent/instrumentation/mongodb_command_subscriber.rb +21 -18
  180. data/lib/new_relic/agent/instrumentation/net_http/chain.rb +24 -0
  181. data/lib/new_relic/agent/instrumentation/{net_prepend.rb → net_http/instrumentation.rb} +13 -13
  182. data/lib/new_relic/agent/instrumentation/net_http/prepend.rb +21 -0
  183. data/lib/new_relic/agent/instrumentation/net_http.rb +44 -0
  184. data/lib/new_relic/agent/instrumentation/notifications_subscriber.rb +10 -9
  185. data/lib/new_relic/agent/instrumentation/padrino/chain.rb +38 -0
  186. data/lib/new_relic/agent/instrumentation/padrino/instrumentation.rb +28 -0
  187. data/lib/new_relic/agent/instrumentation/padrino/prepend.rb +20 -0
  188. data/lib/new_relic/agent/instrumentation/padrino.rb +20 -58
  189. data/lib/new_relic/agent/instrumentation/passenger_instrumentation.rb +3 -3
  190. data/lib/new_relic/agent/instrumentation/queue_time.rb +8 -9
  191. data/lib/new_relic/agent/instrumentation/rack/chain.rb +66 -0
  192. data/lib/new_relic/agent/instrumentation/rack/helpers.rb +33 -0
  193. data/lib/new_relic/agent/instrumentation/rack/instrumentation.rb +75 -0
  194. data/lib/new_relic/agent/instrumentation/rack/prepend.rb +43 -0
  195. data/lib/new_relic/agent/instrumentation/rack.rb +30 -140
  196. data/lib/new_relic/agent/instrumentation/rails3/action_controller.rb +16 -52
  197. data/lib/new_relic/agent/instrumentation/rails_middleware.rb +1 -1
  198. data/lib/new_relic/agent/instrumentation/rails_notifications/action_cable.rb +5 -4
  199. data/lib/new_relic/agent/instrumentation/rails_notifications/action_controller.rb +3 -3
  200. data/lib/new_relic/agent/instrumentation/rails_notifications/action_view.rb +3 -2
  201. data/lib/new_relic/agent/instrumentation/rainbows_instrumentation.rb +13 -3
  202. data/lib/new_relic/agent/instrumentation/rake/chain.rb +20 -0
  203. data/lib/new_relic/agent/instrumentation/rake/instrumentation.rb +149 -0
  204. data/lib/new_relic/agent/instrumentation/rake/prepend.rb +14 -0
  205. data/lib/new_relic/agent/instrumentation/rake.rb +16 -157
  206. data/lib/new_relic/agent/instrumentation/redis/chain.rb +45 -0
  207. data/lib/new_relic/agent/instrumentation/redis/constants.rb +17 -0
  208. data/lib/new_relic/agent/instrumentation/redis/instrumentation.rb +72 -0
  209. data/lib/new_relic/agent/instrumentation/redis/middleware.rb +16 -0
  210. data/lib/new_relic/agent/instrumentation/redis/prepend.rb +29 -0
  211. data/lib/new_relic/agent/instrumentation/redis.rb +18 -107
  212. data/lib/new_relic/agent/instrumentation/resque/chain.rb +21 -0
  213. data/lib/new_relic/agent/instrumentation/resque/helper.rb +19 -0
  214. data/lib/new_relic/agent/instrumentation/resque/instrumentation.rb +34 -0
  215. data/lib/new_relic/agent/instrumentation/resque/prepend.rb +15 -0
  216. data/lib/new_relic/agent/instrumentation/resque.rb +28 -39
  217. data/lib/new_relic/agent/instrumentation/sequel.rb +14 -16
  218. data/lib/new_relic/agent/instrumentation/sequel_helper.rb +2 -2
  219. data/lib/new_relic/agent/instrumentation/sidekiq/client.rb +20 -0
  220. data/lib/new_relic/agent/instrumentation/sidekiq/extensions/delayed_class.rb +30 -0
  221. data/lib/new_relic/agent/instrumentation/sidekiq/server.rb +37 -0
  222. data/lib/new_relic/agent/instrumentation/sidekiq.rb +23 -64
  223. data/lib/new_relic/agent/instrumentation/sinatra/chain.rb +55 -0
  224. data/lib/new_relic/agent/instrumentation/sinatra/ignorer.rb +30 -36
  225. data/lib/new_relic/agent/instrumentation/sinatra/instrumentation.rb +125 -0
  226. data/lib/new_relic/agent/instrumentation/sinatra/prepend.rb +33 -0
  227. data/lib/new_relic/agent/instrumentation/sinatra/transaction_namer.rb +2 -2
  228. data/lib/new_relic/agent/instrumentation/sinatra.rb +34 -163
  229. data/lib/new_relic/agent/instrumentation/sunspot.rb +11 -3
  230. data/lib/new_relic/agent/instrumentation/thread/chain.rb +24 -0
  231. data/lib/new_relic/agent/instrumentation/thread/instrumentation.rb +28 -0
  232. data/lib/new_relic/agent/instrumentation/thread/prepend.rb +22 -0
  233. data/lib/new_relic/agent/instrumentation/thread.rb +20 -0
  234. data/lib/new_relic/agent/instrumentation/tilt/chain.rb +24 -0
  235. data/lib/new_relic/agent/instrumentation/tilt/instrumentation.rb +41 -0
  236. data/lib/new_relic/agent/instrumentation/tilt/prepend.rb +13 -0
  237. data/lib/new_relic/agent/instrumentation/tilt.rb +25 -0
  238. data/lib/new_relic/agent/instrumentation/typhoeus/chain.rb +22 -0
  239. data/lib/new_relic/agent/instrumentation/typhoeus/instrumentation.rb +80 -0
  240. data/lib/new_relic/agent/instrumentation/typhoeus/prepend.rb +14 -0
  241. data/lib/new_relic/agent/instrumentation/typhoeus.rb +12 -91
  242. data/lib/new_relic/agent/instrumentation.rb +1 -1
  243. data/lib/new_relic/agent/internal_agent_error.rb +2 -2
  244. data/lib/new_relic/agent/javascript_instrumentor.rb +47 -41
  245. data/lib/new_relic/agent/linking_metadata.rb +44 -0
  246. data/lib/new_relic/agent/local_log_decorator.rb +37 -0
  247. data/lib/new_relic/agent/log_event_aggregator.rb +235 -0
  248. data/lib/new_relic/agent/log_once.rb +1 -1
  249. data/lib/new_relic/agent/log_priority.rb +20 -0
  250. data/lib/new_relic/agent/logging.rb +24 -21
  251. data/lib/new_relic/agent/memory_logger.rb +2 -2
  252. data/lib/new_relic/agent/messaging.rb +65 -81
  253. data/lib/new_relic/agent/method_tracer.rb +154 -149
  254. data/lib/new_relic/agent/method_tracer_helpers.rb +85 -3
  255. data/lib/new_relic/agent/monitors/cross_app_monitor.rb +25 -17
  256. data/lib/new_relic/agent/monitors/distributed_tracing_monitor.rb +5 -4
  257. data/lib/new_relic/agent/monitors/inbound_request_monitor.rb +1 -2
  258. data/lib/new_relic/agent/monitors/synthetics_monitor.rb +6 -9
  259. data/lib/new_relic/agent/monitors.rb +6 -7
  260. data/lib/new_relic/agent/new_relic_service/encoders.rb +6 -6
  261. data/lib/new_relic/agent/new_relic_service/json_marshaller.rb +5 -5
  262. data/lib/new_relic/agent/new_relic_service/marshaller.rb +2 -2
  263. data/lib/new_relic/agent/new_relic_service/security_policy_settings.rb +5 -4
  264. data/lib/new_relic/agent/new_relic_service.rb +248 -196
  265. data/lib/new_relic/agent/{noticible_error.rb → noticeable_error.rb} +2 -5
  266. data/lib/new_relic/agent/null_logger.rb +7 -3
  267. data/lib/new_relic/agent/obfuscator.rb +8 -10
  268. data/lib/new_relic/agent/parameter_filtering.rb +24 -10
  269. data/lib/new_relic/agent/payload_metric_mapping.rb +9 -10
  270. data/lib/new_relic/agent/pipe_channel_manager.rb +27 -17
  271. data/lib/new_relic/agent/pipe_service.rb +8 -5
  272. data/lib/new_relic/agent/prepend_supportability.rb +2 -2
  273. data/lib/new_relic/agent/priority_sampled_buffer.rb +8 -11
  274. data/lib/new_relic/agent/range_extensions.rb +8 -28
  275. data/lib/new_relic/agent/rules_engine/replacement_rule.rb +11 -11
  276. data/lib/new_relic/agent/rules_engine/segment_terms_rule.rb +13 -13
  277. data/lib/new_relic/agent/rules_engine.rb +5 -4
  278. data/lib/new_relic/agent/sampler.rb +3 -4
  279. data/lib/new_relic/agent/sampler_collection.rb +3 -4
  280. data/lib/new_relic/agent/samplers/cpu_sampler.rb +3 -2
  281. data/lib/new_relic/agent/samplers/delayed_job_sampler.rb +12 -9
  282. data/lib/new_relic/agent/samplers/memory_sampler.rb +24 -13
  283. data/lib/new_relic/agent/samplers/object_sampler.rb +1 -1
  284. data/lib/new_relic/agent/samplers/vm_sampler.rb +21 -19
  285. data/lib/new_relic/agent/span_event_aggregator.rb +12 -12
  286. data/lib/new_relic/agent/span_event_primitive.rb +62 -56
  287. data/lib/new_relic/agent/sql_sampler.rb +18 -18
  288. data/lib/new_relic/agent/stats.rb +78 -41
  289. data/lib/new_relic/agent/stats_engine/gc_profiler.rb +9 -11
  290. data/lib/new_relic/agent/stats_engine/stats_hash.rb +12 -13
  291. data/lib/new_relic/agent/stats_engine.rb +8 -8
  292. data/lib/new_relic/agent/synthetics_event_aggregator.rb +7 -8
  293. data/lib/new_relic/agent/system_info.rb +85 -65
  294. data/lib/new_relic/agent/threading/agent_thread.rb +18 -14
  295. data/lib/new_relic/agent/threading/backtrace_node.rb +12 -13
  296. data/lib/new_relic/agent/threading/backtrace_service.rb +14 -14
  297. data/lib/new_relic/agent/threading/thread_profile.rb +23 -23
  298. data/lib/new_relic/agent/timestamp_sampled_buffer.rb +1 -1
  299. data/lib/new_relic/agent/tracer.rb +76 -85
  300. data/lib/new_relic/agent/transaction/abstract_segment.rb +103 -40
  301. data/lib/new_relic/agent/transaction/datastore_segment.rb +21 -17
  302. data/lib/new_relic/agent/transaction/distributed_tracer.rb +63 -56
  303. data/lib/new_relic/agent/transaction/distributed_tracing.rb +43 -44
  304. data/lib/new_relic/agent/transaction/external_request_segment.rb +44 -34
  305. data/lib/new_relic/agent/transaction/message_broker_segment.rb +31 -36
  306. data/lib/new_relic/agent/transaction/request_attributes.rb +35 -35
  307. data/lib/new_relic/agent/transaction/segment.rb +14 -10
  308. data/lib/new_relic/agent/transaction/slowest_sample_buffer.rb +1 -3
  309. data/lib/new_relic/agent/transaction/synthetics_sample_buffer.rb +1 -1
  310. data/lib/new_relic/agent/transaction/trace.rb +15 -11
  311. data/lib/new_relic/agent/transaction/trace_builder.rb +10 -10
  312. data/lib/new_relic/agent/transaction/trace_context.rb +35 -35
  313. data/lib/new_relic/agent/transaction/trace_node.rb +25 -24
  314. data/lib/new_relic/agent/transaction/tracing.rb +14 -11
  315. data/lib/new_relic/agent/transaction/transaction_sample_buffer.rb +4 -4
  316. data/lib/new_relic/agent/transaction.rb +164 -145
  317. data/lib/new_relic/agent/transaction_error_primitive.rb +27 -25
  318. data/lib/new_relic/agent/transaction_event_aggregator.rb +12 -12
  319. data/lib/new_relic/agent/transaction_event_primitive.rb +32 -33
  320. data/lib/new_relic/agent/transaction_event_recorder.rb +16 -15
  321. data/lib/new_relic/agent/transaction_metrics.rb +10 -9
  322. data/lib/new_relic/agent/transaction_sampler.rb +5 -6
  323. data/lib/new_relic/agent/transaction_time_aggregator.rb +26 -25
  324. data/lib/new_relic/agent/utilization/aws.rb +33 -3
  325. data/lib/new_relic/agent/utilization/azure.rb +3 -3
  326. data/lib/new_relic/agent/utilization/gcp.rb +7 -7
  327. data/lib/new_relic/agent/utilization/pcf.rb +5 -4
  328. data/lib/new_relic/agent/utilization/vendor.rb +43 -28
  329. data/lib/new_relic/agent/utilization_data.rb +7 -5
  330. data/lib/new_relic/agent/vm/jruby_vm.rb +1 -1
  331. data/lib/new_relic/agent/vm/monotonic_gc_profiler.rb +2 -2
  332. data/lib/new_relic/agent/vm/mri_vm.rb +40 -15
  333. data/lib/new_relic/agent/vm/snapshot.rb +5 -5
  334. data/lib/new_relic/agent/vm.rb +1 -1
  335. data/lib/new_relic/agent/worker_loop.rb +10 -12
  336. data/lib/new_relic/agent.rb +64 -75
  337. data/lib/new_relic/cli/command.rb +20 -22
  338. data/lib/new_relic/cli/commands/deployments.rb +92 -42
  339. data/lib/new_relic/cli/commands/install.rb +12 -15
  340. data/lib/new_relic/coerce.rb +15 -13
  341. data/lib/new_relic/collection_helper.rb +50 -48
  342. data/lib/new_relic/constants.rb +1 -4
  343. data/lib/new_relic/control/class_methods.rb +3 -3
  344. data/lib/new_relic/control/frameworks/external.rb +2 -2
  345. data/lib/new_relic/control/frameworks/rails.rb +23 -17
  346. data/lib/new_relic/control/frameworks/rails3.rb +3 -4
  347. data/lib/new_relic/control/frameworks/rails4.rb +1 -1
  348. data/lib/new_relic/control/frameworks/rails_notifications.rb +1 -1
  349. data/lib/new_relic/control/frameworks/ruby.rb +3 -3
  350. data/lib/new_relic/control/frameworks/sinatra.rb +7 -1
  351. data/lib/new_relic/control/frameworks.rb +1 -1
  352. data/lib/new_relic/control/instance_methods.rb +23 -45
  353. data/lib/new_relic/control/instrumentation.rb +39 -11
  354. data/lib/new_relic/control/private_instance_methods.rb +48 -0
  355. data/lib/new_relic/control/server_methods.rb +3 -4
  356. data/lib/new_relic/control.rb +1 -2
  357. data/lib/new_relic/delayed_job_injection.rb +1 -1
  358. data/lib/new_relic/dependency_detection.rb +126 -19
  359. data/lib/new_relic/environment_report.rb +40 -32
  360. data/lib/new_relic/helper.rb +48 -7
  361. data/lib/new_relic/language_support.rb +29 -5
  362. data/lib/new_relic/latest_changes.rb +7 -6
  363. data/lib/new_relic/local_environment.rb +24 -18
  364. data/lib/new_relic/metric_data.rb +31 -26
  365. data/lib/new_relic/metric_spec.rb +8 -6
  366. data/lib/new_relic/noticed_error.rb +22 -30
  367. data/lib/new_relic/rack/agent_hooks.rb +1 -1
  368. data/lib/new_relic/rack/agent_middleware.rb +5 -3
  369. data/lib/new_relic/rack/browser_monitoring.rb +133 -123
  370. data/lib/new_relic/rack.rb +1 -1
  371. data/lib/new_relic/recipes/capistrano3.rb +3 -61
  372. data/lib/new_relic/recipes/capistrano_legacy.rb +23 -26
  373. data/lib/new_relic/recipes/helpers/send_deployment.rb +69 -0
  374. data/lib/new_relic/recipes.rb +1 -1
  375. data/lib/new_relic/supportability_helper.rb +9 -10
  376. data/lib/new_relic/traced_thread.rb +35 -0
  377. data/lib/new_relic/version.rb +6 -17
  378. data/lib/newrelic_rpm.rb +11 -35
  379. data/lib/sequel/extensions/newrelic_instrumentation.rb +12 -15
  380. data/lib/sequel/plugins/newrelic_instrumentation.rb +4 -10
  381. data/lib/tasks/all.rb +3 -3
  382. data/lib/tasks/config.rake +22 -117
  383. data/lib/tasks/coverage_report.rake +28 -0
  384. data/lib/tasks/helpers/config.html.erb +21 -0
  385. data/lib/tasks/{config.text.erb → helpers/config.text.erb} +0 -0
  386. data/lib/tasks/helpers/format.rb +123 -0
  387. data/lib/tasks/helpers/matches.rb +12 -0
  388. data/lib/tasks/helpers/prompt.rb +24 -0
  389. data/lib/tasks/helpers/removers.rb +33 -0
  390. data/lib/tasks/install.rake +4 -0
  391. data/lib/tasks/instrumentation_generator/README.md +63 -0
  392. data/lib/tasks/instrumentation_generator/TODO.md +33 -0
  393. data/lib/tasks/instrumentation_generator/instrumentation.thor +124 -0
  394. data/lib/tasks/instrumentation_generator/templates/Envfile.tt +9 -0
  395. data/lib/tasks/instrumentation_generator/templates/chain.tt +22 -0
  396. data/lib/tasks/instrumentation_generator/templates/chain_method.tt +8 -0
  397. data/lib/tasks/instrumentation_generator/templates/dependency_detection.tt +29 -0
  398. data/lib/tasks/instrumentation_generator/templates/instrumentation.tt +13 -0
  399. data/lib/tasks/instrumentation_generator/templates/instrumentation_method.tt +3 -0
  400. data/lib/tasks/instrumentation_generator/templates/newrelic.yml.tt +19 -0
  401. data/lib/tasks/instrumentation_generator/templates/prepend.tt +13 -0
  402. data/lib/tasks/instrumentation_generator/templates/prepend_method.tt +3 -0
  403. data/lib/tasks/instrumentation_generator/templates/test.tt +15 -0
  404. data/lib/tasks/multiverse.rake +4 -0
  405. data/lib/tasks/multiverse.rb +5 -32
  406. data/lib/tasks/newrelic.rb +1 -1
  407. data/lib/tasks/tests.rake +10 -15
  408. data/newrelic.yml +648 -3
  409. data/newrelic_rpm.gemspec +25 -25
  410. data/recipes/newrelic.rb +2 -2
  411. data/test/agent_helper.rb +219 -178
  412. metadata +164 -122
  413. data/ROADMAP.md +0 -24
  414. data/bin/mongrel_rpm +0 -33
  415. data/cert/cacert.pem +0 -1177
  416. data/lib/new_relic/agent/datastores/mongo/obfuscator.rb +0 -44
  417. data/lib/new_relic/agent/datastores/mongo/statement_formatter.rb +0 -53
  418. data/lib/new_relic/agent/instrumentation/excon/connection.rb +0 -49
  419. data/lib/new_relic/agent/instrumentation/http.rb +0 -49
  420. data/lib/new_relic/agent/instrumentation/merb/controller.rb +0 -44
  421. data/lib/new_relic/agent/instrumentation/merb/errors.rb +0 -33
  422. data/lib/new_relic/agent/instrumentation/net.rb +0 -70
  423. data/lib/new_relic/agent/instrumentation/rails/action_controller.rb +0 -125
  424. data/lib/new_relic/agent/instrumentation/rails/action_web_service.rb +0 -46
  425. data/lib/new_relic/agent/supported_versions.rb +0 -275
  426. data/lib/new_relic/control/frameworks/merb.rb +0 -29
  427. data/lib/tasks/config.html.erb +0 -32
@@ -0,0 +1,66 @@
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 Logger
9
+ def skip_instrumenting?
10
+ defined?(@skip_instrumenting) && @skip_instrumenting
11
+ end
12
+
13
+ # We support setting this on loggers which might not have
14
+ # instrumentation installed yet. This lets us disable in AgentLogger
15
+ # and AuditLogger without them having to know the inner details.
16
+ def self.mark_skip_instrumenting(logger)
17
+ return if logger.frozen?
18
+
19
+ logger.instance_variable_set(:@skip_instrumenting, true)
20
+ end
21
+
22
+ def self.clear_skip_instrumenting(logger)
23
+ return if logger.frozen?
24
+
25
+ logger.instance_variable_set(:@skip_instrumenting, false)
26
+ end
27
+
28
+ def mark_skip_instrumenting
29
+ return if frozen?
30
+
31
+ @skip_instrumenting = true
32
+ end
33
+
34
+ def clear_skip_instrumenting
35
+ return if frozen?
36
+
37
+ @skip_instrumenting = false
38
+ end
39
+
40
+ def self.enabled?
41
+ NewRelic::Agent.config[:'instrumentation.logger'] != 'disabled'
42
+ end
43
+
44
+ def format_message_with_tracing(severity, datetime, progname, msg)
45
+ formatted_message = yield
46
+ return formatted_message if skip_instrumenting?
47
+
48
+ begin
49
+ # It's critical we don't instrument logging from metric recording
50
+ # methods within NewRelic::Agent, or we'll stack overflow!!
51
+ mark_skip_instrumenting
52
+
53
+ unless ::NewRelic::Agent.agent.nil?
54
+ ::NewRelic::Agent.agent.log_event_aggregator.record(formatted_message, severity)
55
+ formatted_message = LocalLogDecorator.decorate(formatted_message)
56
+ end
57
+
58
+ formatted_message
59
+ ensure
60
+ clear_skip_instrumenting
61
+ end
62
+ end
63
+ end
64
+ end
65
+ end
66
+ end
@@ -0,0 +1,13 @@
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 Logger::Prepend
7
+ include NewRelic::Agent::Instrumentation::Logger
8
+
9
+ def format_message(severity, datetime, progname, msg)
10
+ format_message_with_tracing(severity, datetime, progname, msg) { super }
11
+ end
12
+ end
13
+ end
@@ -0,0 +1,26 @@
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 'logger/instrumentation'
6
+ require_relative 'logger/chain'
7
+ require_relative 'logger/prepend'
8
+
9
+ DependencyDetection.defer do
10
+ named :logger
11
+
12
+ depends_on do
13
+ defined?(::Logger) &&
14
+ NewRelic::Agent.config[:'application_logging.enabled']
15
+ end
16
+
17
+ executes do
18
+ ::NewRelic::Agent.logger.info("Installing Logger instrumentation")
19
+
20
+ if use_prepend?
21
+ prepend_instrument ::Logger, NewRelic::Agent::Instrumentation::Logger::Prepend
22
+ else
23
+ chain_instrument NewRelic::Agent::Instrumentation::Logger
24
+ end
25
+ end
26
+ end
@@ -0,0 +1,15 @@
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 Memcache
7
+ module Chain
8
+ extend Helper
9
+
10
+ def self.instrument!(target_class)
11
+ instrument_methods(target_class, client_methods)
12
+ end
13
+ end
14
+ end
15
+ end
@@ -1,166 +1,95 @@
1
- # encoding: utf-8
2
1
  # This file is distributed under New Relic's license terms.
3
2
  # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
3
+ # frozen_string_literal: true
4
4
 
5
5
  module NewRelic
6
6
  module Agent
7
7
  module Instrumentation
8
8
  module Memcache
9
9
  module Dalli
10
- extend self
11
-
12
- MEMCACHED = "Memcached".freeze
13
- METHODS = [:get, :set, :add, :incr, :decr, :delete, :replace, :append, :prepend, :cas]
14
- SEND_MULTIGET_METRIC_NAME = "get_multi_request".freeze
15
- DATASTORE_INSTANCES_SUPPORTED_VERSION = Gem::Version.new '2.6.4'
16
- SLASH = '/'.freeze
17
- UNKNOWN = 'unknown'.freeze
18
- LOCALHOST = 'localhost'.freeze
19
-
20
- def supports_datastore_instances?
21
- DATASTORE_INSTANCES_SUPPORTED_VERSION <= Gem::Version.new(::Dalli::VERSION)
22
- end
10
+ extend Helper
11
+
12
+ module_function
23
13
 
24
- def instrument_methods
14
+ def instrument!
25
15
  if supports_datastore_instances?
26
- ::NewRelic::Agent::Instrumentation::Memcache.instrument_methods(::Dalli::Client, METHODS)
27
- instrument_multi_method :get_multi
16
+ instrument_methods(::Dalli::Client, dalli_methods)
17
+ instrument_multi_method(:get_multi)
28
18
  instrument_send_multiget
29
19
  instrument_server_for_key
30
20
  else
31
- ::NewRelic::Agent::Instrumentation::Memcache.instrument_methods(::Dalli::Client,
32
- ::NewRelic::Agent::Instrumentation::Memcache::METHODS)
21
+ instrument_methods(::Dalli::Client, client_methods)
22
+ end
23
+ end
24
+
25
+ def instrument_multi_method(method_name)
26
+ visibility = NewRelic::Helper.instance_method_visibility(::Dalli::Client, method_name)
27
+ method_name_without = :"#{method_name}_without_newrelic_trace"
28
+
29
+ ::Dalli::Client.class_eval do
30
+ alias_method(method_name_without, method_name)
31
+
32
+ define_method(method_name) do |*args, &block|
33
+ get_multi_with_newrelic_tracing(method_name) { __send__(method_name_without, *args, &block) }
34
+ end
35
+
36
+ __send__(visibility, method_name)
37
+ __send__(visibility, method_name_without)
33
38
  end
34
39
  end
35
40
 
36
41
  def instrument_server_for_key
37
42
  ::Dalli::Ring.class_eval do
38
- alias_method :server_for_key_without_newrelic_trace, :server_for_key
39
-
40
- def server_for_key key
41
- server = server_for_key_without_newrelic_trace key
42
- begin
43
- if txn = ::NewRelic::Agent::Tracer.current_transaction
44
- segment = txn.current_segment
45
- if ::NewRelic::Agent::Transaction::DatastoreSegment === segment
46
- ::NewRelic::Agent::Instrumentation::Memcache::Dalli.assign_instance_to(segment, server)
47
- end
48
- end
49
- rescue => e
50
- ::NewRelic::Agent.logger.warn "Unable to set instance info on datastore segment: #{e.message}"
51
- end
52
- server
43
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
44
+
45
+ alias_method(:server_for_key_without_newrelic_trace, :server_for_key)
46
+
47
+ def server_for_key(key)
48
+ server_for_key_with_newrelic_tracing { server_for_key_without_newrelic_trace(key) }
53
49
  end
54
50
  end
55
51
  end
56
52
 
57
53
  def instrument_send_multiget
58
- ::Dalli::Server.class_eval do
59
- alias_method :send_multiget_without_newrelic_trace, :send_multiget
60
-
61
- def send_multiget(keys)
62
- segment = ::NewRelic::Agent::Tracer.start_datastore_segment(
63
- product: MEMCACHED,
64
- operation: SEND_MULTIGET_METRIC_NAME
65
- )
66
- ::NewRelic::Agent::Instrumentation::Memcache::Dalli.assign_instance_to(segment, self)
67
-
68
- begin
69
- NewRelic::Agent::Tracer.capture_segment_error segment do
70
- send_multiget_without_newrelic_trace(keys)
71
- end
72
- ensure
73
- if ::NewRelic::Agent.config[:capture_memcache_keys]
74
- segment.notice_nosql_statement "#{SEND_MULTIGET_METRIC_NAME} #{keys.inspect}"
75
- end
76
- segment.finish if segment
54
+ if supports_binary_protocol?
55
+ ::Dalli::Protocol::Binary
56
+ else
57
+ ::Dalli::Server
58
+ end.class_eval do
59
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
60
+
61
+ # TODO: Dalli - 3.1.0 renamed send_multiget to pipelined_get, but the method is otherwise the same
62
+ if Gem::Version.new(::Dalli::VERSION) >= Gem::Version.new('3.1.0')
63
+ alias_method(:pipelined_get_without_newrelic_trace, :pipelined_get)
64
+ def pipelined_get(keys)
65
+ send_multiget_with_newrelic_tracing(keys) { pipelined_get_without_newrelic_trace(keys) }
66
+ end
67
+ else
68
+ alias_method(:send_multiget_without_newrelic_trace, :send_multiget)
69
+ def send_multiget(keys)
70
+ send_multiget_with_newrelic_tracing(keys) { send_multiget_without_newrelic_trace(keys) }
77
71
  end
78
72
  end
79
73
  end
80
74
  end
75
+ end
81
76
 
82
- def instrument_multi_method method_name
83
- visibility = NewRelic::Helper.instance_method_visibility ::Dalli::Client, method_name
84
- method_name_without = :"#{method_name}_without_newrelic_trace"
77
+ module DalliCAS
78
+ extend Dalli
79
+ extend Helper
85
80
 
86
- ::Dalli::Client.class_eval do
87
- alias_method method_name_without, method_name
88
-
89
- define_method method_name do |*args, &block|
90
- segment = NewRelic::Agent::Tracer.start_segment name: "Ruby/Memcached/Dalli/#{method_name}"
91
- begin
92
- NewRelic::Agent::Tracer.capture_segment_error segment do
93
- __send__ method_name_without, *args, &block
94
- end
95
- ensure
96
- segment.finish if segment
97
- end
98
- end
81
+ module_function
99
82
 
100
- __send__ visibility, method_name
101
- __send__ visibility, method_name_without
102
- end
83
+ def should_instrument?
84
+ supported_methods_for(::Dalli::Client, dalli_cas_methods).any?
103
85
  end
104
86
 
105
- def assign_instance_to segment, server
106
- host = port_path_or_id = nil
107
- if server.hostname.start_with? SLASH
108
- host = LOCALHOST
109
- port_path_or_id = server.hostname
110
- else
111
- host = server.hostname
112
- port_path_or_id = server.port
113
- end
114
- segment.set_instance_info host, port_path_or_id
115
- rescue => e
116
- ::NewRelic::Agent.logger.debug "Failed to retrieve memcached instance info: #{e.message}"
117
- segment.set_instance_info UNKNOWN, UNKNOWN
87
+ def instrument!
88
+ instrument_methods(::Dalli::Client, dalli_cas_methods)
89
+ instrument_multi_method(:get_multi_cas)
118
90
  end
119
-
120
91
  end
121
92
  end
122
93
  end
123
94
  end
124
95
  end
125
-
126
- DependencyDetection.defer do
127
- named :dalli
128
-
129
- depends_on do
130
- NewRelic::Agent::Instrumentation::Memcache.enabled?
131
- end
132
-
133
- depends_on do
134
- defined?(::Dalli::Client)
135
- end
136
-
137
- executes do
138
- ::NewRelic::Agent.logger.info 'Installing Memcache instrumentation for dalli gem'
139
- ::NewRelic::Agent::Instrumentation::Memcache::Dalli.instrument_methods
140
- end
141
- end
142
-
143
- DependencyDetection.defer do
144
- named :dalli_cas_client
145
-
146
- depends_on do
147
- NewRelic::Agent::Instrumentation::Memcache.enabled?
148
- end
149
-
150
- depends_on do
151
- # These CAS client methods are only optionally defined if users require
152
- # dalli/cas/client. Use a separate dependency block so it can potentially
153
- # re-evaluate after they've done that require.
154
- defined?(::Dalli::Client) &&
155
- ::NewRelic::Agent::Instrumentation::Memcache.supported_methods_for(::Dalli::Client,
156
- CAS_CLIENT_METHODS).any?
157
- end
158
-
159
- CAS_CLIENT_METHODS = [:get_cas, :set_cas, :replace_cas, :delete_cas]
160
-
161
- executes do
162
- ::NewRelic::Agent.logger.info 'Installing Dalli CAS Client Memcache instrumentation'
163
- ::NewRelic::Agent::Instrumentation::Memcache.instrument_methods(::Dalli::Client, CAS_CLIENT_METHODS)
164
- ::NewRelic::Agent::Instrumentation::Memcache::Dalli.instrument_multi_method(:get_multi_cas)
165
- end
166
- end
@@ -0,0 +1,59 @@
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 Memcache
7
+ module Helper
8
+ DATASTORE_INSTANCES_SUPPORTED_VERSION = Gem::Version.new('2.6.4')
9
+ BINARY_PROTOCOL_SUPPORTED_VERSION = Gem::Version.new('3.0.2')
10
+
11
+ def supports_datastore_instances?
12
+ DATASTORE_INSTANCES_SUPPORTED_VERSION <= Gem::Version.new(::Dalli::VERSION)
13
+ end
14
+
15
+ def supports_binary_protocol?
16
+ BINARY_PROTOCOL_SUPPORTED_VERSION <= Gem::Version.new(::Dalli::VERSION)
17
+ end
18
+
19
+ def client_methods
20
+ [:get, :get_multi, :set, :add, :incr, :decr, :delete, :replace, :append,
21
+ :prepend, :cas, :single_get, :multi_get, :single_cas, :multi_cas]
22
+ end
23
+
24
+ def dalli_methods
25
+ [:get, :set, :add, :incr, :decr, :delete, :replace, :append, :prepend, :cas]
26
+ end
27
+
28
+ def dalli_cas_methods
29
+ [:get_cas, :set_cas, :replace_cas, :delete_cas]
30
+ end
31
+
32
+ def supported_methods_for(client_class, methods)
33
+ methods.select do |method_name|
34
+ client_class.method_defined?(method_name) || client_class.private_method_defined?(method_name)
35
+ end
36
+ end
37
+
38
+ def instrument_methods(client_class, requested_methods = METHODS)
39
+ supported_methods_for(client_class, requested_methods).each do |method_name|
40
+ visibility = NewRelic::Helper.instance_method_visibility(client_class, method_name)
41
+ method_name_without = :"#{method_name}_without_newrelic_trace"
42
+
43
+ client_class.class_eval do
44
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
45
+
46
+ alias_method(method_name_without, method_name)
47
+
48
+ define_method(method_name) do |*args, &block|
49
+ with_newrelic_tracing(method_name, *args) { __send__(method_name_without, *args, &block) }
50
+ end
51
+
52
+ __send__(visibility, method_name)
53
+ __send__(visibility, method_name_without)
54
+ end
55
+ end
56
+ end
57
+ end
58
+ end
59
+ end
@@ -0,0 +1,90 @@
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 Memcache
7
+ module Tracer
8
+ SLASH = '/'
9
+ UNKNOWN = 'unknown'
10
+ LOCALHOST = 'localhost'
11
+ MULTIGET_METRIC_NAME = "get_multi_request"
12
+ MEMCACHED = "Memcached"
13
+
14
+ def with_newrelic_tracing(operation, *args)
15
+ segment = NewRelic::Agent::Tracer.start_datastore_segment( \
16
+ product: MEMCACHED,
17
+ operation: operation
18
+ )
19
+
20
+ begin
21
+ NewRelic::Agent::Tracer.capture_segment_error(segment) { yield }
22
+ ensure
23
+ if NewRelic::Agent.config[:capture_memcache_keys]
24
+ segment.notice_nosql_statement("#{operation} #{args.first.inspect}")
25
+ end
26
+ segment.finish if segment
27
+ end
28
+ end
29
+
30
+ def server_for_key_with_newrelic_tracing
31
+ yield.tap do |server|
32
+ begin
33
+ if txn = ::NewRelic::Agent::Tracer.current_transaction
34
+ segment = txn.current_segment
35
+ if ::NewRelic::Agent::Transaction::DatastoreSegment === segment
36
+ assign_instance_to(segment, server)
37
+ end
38
+ end
39
+ rescue => e
40
+ ::NewRelic::Agent.logger.warn("Unable to set instance info on datastore segment: #{e.message}")
41
+ end
42
+ end
43
+ end
44
+
45
+ def get_multi_with_newrelic_tracing(method_name)
46
+ segment = NewRelic::Agent::Tracer.start_segment( \
47
+ name: "Ruby/Memcached/Dalli/#{method_name}"
48
+ )
49
+
50
+ begin
51
+ NewRelic::Agent::Tracer.capture_segment_error(segment) { yield }
52
+ ensure
53
+ segment.finish if segment
54
+ end
55
+ end
56
+
57
+ def send_multiget_with_newrelic_tracing(keys)
58
+ segment = ::NewRelic::Agent::Tracer.start_datastore_segment( \
59
+ product: MEMCACHED,
60
+ operation: MULTIGET_METRIC_NAME
61
+ )
62
+
63
+ begin
64
+ assign_instance_to(segment, self)
65
+ NewRelic::Agent::Tracer.capture_segment_error(segment) { yield }
66
+ ensure
67
+ if ::NewRelic::Agent.config[:capture_memcache_keys]
68
+ segment.notice_nosql_statement("#{MULTIGET_METRIC_NAME} #{keys.inspect}")
69
+ end
70
+ segment.finish if segment
71
+ end
72
+ end
73
+
74
+ def assign_instance_to(segment, server)
75
+ host = port_path_or_id = nil
76
+ if server.hostname.start_with?(SLASH)
77
+ host = LOCALHOST
78
+ port_path_or_id = server.hostname
79
+ else
80
+ host = server.hostname
81
+ port_path_or_id = server.port
82
+ end
83
+ segment.set_instance_info(host, port_path_or_id)
84
+ rescue => e
85
+ ::NewRelic::Agent.logger.debug("Failed to retrieve memcached instance info: #{e.message}")
86
+ segment.set_instance_info(UNKNOWN, UNKNOWN)
87
+ end
88
+ end
89
+ end
90
+ end
@@ -0,0 +1,101 @@
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 Memcache
7
+ module Prepend
8
+ extend Helper
9
+
10
+ module_function
11
+
12
+ def client_prepender(client_class)
13
+ Module.new do
14
+ extend Helper
15
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
16
+
17
+ supported_methods_for(client_class, client_methods).each do |method_name|
18
+ define_method method_name do |*args, &block|
19
+ with_newrelic_tracing(method_name, *args) { super(*args, &block) }
20
+ end
21
+ end
22
+ end
23
+ end
24
+
25
+ def dalli_cas_prependers
26
+ yield(::Dalli::Client, dalli_client_prepender(dalli_cas_methods))
27
+ yield(::Dalli::Client, dalli_get_multi_prepender(:get_multi_cas))
28
+ end
29
+
30
+ def dalli_prependers
31
+ if supports_datastore_instances?
32
+ yield(::Dalli::Client, dalli_client_prepender(dalli_methods))
33
+ yield(::Dalli::Client, dalli_get_multi_prepender(:get_multi))
34
+
35
+ if supports_binary_protocol?
36
+ yield(::Dalli::Protocol::Binary, dalli_server_prepender)
37
+ else
38
+ yield(::Dalli::Server, dalli_server_prepender)
39
+ end
40
+
41
+ yield(::Dalli::Ring, dalli_ring_prepender)
42
+ else
43
+ yield(::Dalli::Client, dalli_client_prepender(client_methods))
44
+ end
45
+ end
46
+
47
+ def dalli_client_prepender(supported_methods)
48
+ Module.new do
49
+ extend Helper
50
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
51
+
52
+ supported_methods.each do |method_name|
53
+ define_method method_name do |*args, &block|
54
+ with_newrelic_tracing(method_name, *args) { super(*args, &block) }
55
+ end
56
+ end
57
+ end
58
+ end
59
+
60
+ def dalli_get_multi_prepender(method_name)
61
+ Module.new do
62
+ extend Helper
63
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
64
+
65
+ define_method method_name do |*args, &block|
66
+ get_multi_with_newrelic_tracing(method_name) { super(*args, &block) }
67
+ end
68
+ end
69
+ end
70
+
71
+ def dalli_ring_prepender
72
+ Module.new do
73
+ extend Helper
74
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
75
+
76
+ def server_for_key(key)
77
+ server_for_key_with_newrelic_tracing { super }
78
+ end
79
+ end
80
+ end
81
+
82
+ def dalli_server_prepender
83
+ Module.new do
84
+ extend Helper
85
+ include NewRelic::Agent::Instrumentation::Memcache::Tracer
86
+
87
+ # TODO: Dalli - 3.1.0 renamed send_multiget to pipelined_get, but the method is otherwise the same
88
+ if Gem::Version.new(::Dalli::VERSION) >= Gem::Version.new('3.1.0')
89
+ def pipelined_get(keys)
90
+ send_multiget_with_newrelic_tracing(keys) { super }
91
+ end
92
+ else
93
+ def send_multiget(keys)
94
+ send_multiget_with_newrelic_tracing(keys) { super }
95
+ end
96
+ end
97
+ end
98
+ end
99
+ end
100
+ end
101
+ end