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
89
90
91
|
require_relative "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
def initialize(*subscribers)
@subscribers = subscribers
end
def event_list
Thread.current[:chef_client_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 unless 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
|
