mirror of
https://github.com/discourse/discourse.git
synced 2024-12-01 01:43:40 +08:00
24 lines
485 B
Ruby
24 lines
485 B
Ruby
|
module MessageBus::DiagnosticsHelper
|
||
|
|
def publish(channel, data, opts = nil)
|
||
|
|
id = super(channel, data, opts)
|
||
|
|
if @tracking
|
||
|
|
m = MessageBus::Message.new(-1, id, channel, data)
|
||
|
|
m.user_ids = opts[:user_ids] if opts
|
||
|
|
m.group_ids = opts[:group_ids] if opts
|
||
|
|
@tracking << m
|
||
|
|
end
|
||
|
|
id
|
||
|
|
end
|
||
|
|
|
||
|
|
def track_publish
|
||
|
|
@tracking = tracking = []
|
||
|
|
yield
|
||
|
|
@tracking = nil
|
||
|
|
tracking
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
module MessageBus
|
||
|
|
extend MessageBus::DiagnosticsHelper
|
||
|
|
end
|