Module: Datadog::DI::Remote Private

Defined in:
lib/datadog/di/remote.rb

Overview

This module is part of a private API. You should avoid using this module if possible, as it may be removed or be changed in the future.

Provides an interface expected by the core Remote subsystem to receive DI-specific remote configuration.

In order to apply (i.e., act on) the configuration, we need the state stored under DI Component. Thus, this module forwards actual configuration application to the ProbeManager associated with the global DI Component.

Defined Under Namespace

Classes: ReadError

Constant Summary collapse

PRODUCT =

This constant is part of a private API. You should avoid using this constant if possible, as it may be removed or be changed in the future. 

'LIVE_DEBUGGING'

Class Method Summary collapse

Class Method Details

.capabilities ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future. 



24
25
26
 
# File 'lib/datadog/di/remote.rb', line 24

def capabilities
  []
end

.products ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future. 



20
21
22
 
# File 'lib/datadog/di/remote.rb', line 20

def products
  [PRODUCT]
end

.receiver(products = [PRODUCT], &block) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future. 



105
106
107
108
 
# File 'lib/datadog/di/remote.rb', line 105

def receiver(products = [PRODUCT], &block)
  matcher = Core::Remote::Dispatcher::Matcher::Product.new(products)
  [Core::Remote::Dispatcher::Receiver.new(matcher, &block)]
end

.receivers(telemetry) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future. 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/datadog/di/remote.rb', line 28

def receivers(telemetry)
  receiver do |repository, _changes|
    # DEV: Filter our by product. Given it will be very common
    # DEV: we can filter this out before we receive the data in this method.
    # DEV: Apply this refactor to AppSec as well if implemented.

    component = DI.component
    # We should always have a non-nil DI component here, because we
    # only add DI product to remote config request if DI is enabled.
    # Ideally, we should be injected with the DI component here
    # rather than having to retrieve it from global state.
    # If the component is nil for some reason, we also don't have a
    # logger instance to report the issue.
    if component

      probe_manager = component.probe_manager

      current_probe_ids = {}
      repository.contents.each do |content|
        case content.path.product
        when PRODUCT
          begin
            probe_spec = parse_content(content)
            probe = ProbeBuilder.build_from_remote_config(probe_spec)
            payload = component.probe_notification_builder.build_received(probe)
            component.probe_notifier_worker.add_status(payload)
            component.logger.info("Received probe from RC: #{probe.type} #{probe.location}")

            begin
              # TODO test exception capture
              probe_manager.add_probe(probe)
              content.applied
            rescue => exc
              raise if component.settings.dynamic_instrumentation.internal.propagate_all_exceptions

              component.logger.warn("Unhandled exception adding probe in DI remote receiver: #{exc.class}: #{exc}")
              component.telemetry&.report(exc, description: "Unhandled exception adding probe in DI remote receiver")

              # If a probe fails to install, we will mark the content
              # as errored. On subsequent remote configuration application
              # attemps, probe manager will raise the "previously errored"
              # exception and we'll rescue it here, again marking the
              # content as errored but with a somewhat different exception
              # message.
              # TODO stack trace must be redacted or not sent at all
              content.errored("Error applying dynamic instrumentation configuration: #{exc.class.name} #{exc.message}: #{Array(exc.backtrace).join("\n")}")
            end

            # Important: even if processing fails for this probe config,
            # we need to note it as being current so that we do not
            # try to remove instrumentation that is still supposed to be
            # active.
            current_probe_ids[probe_spec.fetch('id')] = true
          rescue => exc
            raise if component.settings.dynamic_instrumentation.internal.propagate_all_exceptions

            component.logger.warn("Unhandled exception handling probe in DI remote receiver: #{exc.class}: #{exc}")
            component.telemetry&.report(exc, description: "Unhandled exception handling probe in DI remote receiver")

            content.errored("Error applying dynamic instrumentation configuration: #{exc.class.name} #{exc.message}: #{Array(exc.backtrace).join("\n")}")
          end
        end
      end

      begin
        # TODO test exception capture
        probe_manager.remove_other_probes(current_probe_ids.keys)
      rescue => exc
        raise if component.settings.dynamic_instrumentation.internal.propagate_all_exceptions

        component.logger.warn("Unhandled exception removing probes in DI remote receiver: #{exc.class}: #{exc}")
        component.telemetry&.report(exc, description: "Unhandled exception removing probes in DI remote receiver")
      end
    end
  end
end