summaryrefslogtreecommitdiff
path: root/lib/chef/event_dispatch/dispatcher.rb
blob: d69a7c76e89a0fef52f9d66604ac1b64aefa0aef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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
require "chef/event_dispatch/base"

class Chef
  module EventDispatch

    # == EventDispatch::Dispatcher
    # The Dispatcher handles receiving event data from the sources
    # (Chef::Client, Resources and Providers, etc.) and publishing the data to
    # the registered subscribers.
    class Dispatcher < Base

      attr_reader :subscribers
      attr_reader :event_list

      def initialize(*subscribers)
        @subscribers = subscribers
        @event_list = []
      end

      # Add a new subscriber to the list of registered subscribers
      def register(subscriber)
        subscribers << subscriber
      end

      def unregister(subscriber)
        subscribers.reject! { |x| x == subscriber }
      end

      def enqueue(method_name, *args)
        event_list << [ method_name, *args ]
        process_events_until_done unless @in_call
      end

      (Base.instance_methods - Object.instance_methods).each do |method_name|
        class_eval <<-EOM
          def #{method_name}(*args)
            enqueue(#{method_name.inspect}, *args)
          end
        EOM
      end

      # Special case deprecation, since it needs to know its caller
      def deprecation(message, location = caller(2..2)[0])
        enqueue(:deprecation, message, location)
      end

      # Check to see if we are dispatching to a formatter
      # @api private
      def formatter?
        subscribers.any? { |s| s.respond_to?(:is_formatter?) && s.is_formatter? }
      end

      ####
      # All messages are unconditionally forwarded to all subscribers, so just
      # define the forwarding in one go:
      #

      # @api private
      def call_subscribers(method_name, *args)
        @in_call = true
        subscribers.each do |s|
          # Skip new/unsupported event names
          next if !s.respond_to?(method_name)
          mth = s.method(method_name)
          # Trim arguments to match what the subscriber expects to allow
          # adding new arguments without breaking compat.
          if mth.arity < args.size && mth.arity >= 0
            mth.call(*args.take(mth.arity))
          else
            mth.call(*args)
          end
        end
      ensure
        @in_call = false
      end

      private

      # events are allowed to enqueue chained events, so pop them off until
      # empty, rather than iterating over the list.
      #
      def process_events_until_done
        call_subscribers(*event_list.shift) until event_list.empty?
      end

    end
  end
end