Class: TraceViz::Extractors::Diagram::Processors::MessageProcessor

Inherits:
Object
  • Object
show all
Defined in:
lib/trace_viz/extractors/diagram/processors/message_processor.rb

Instance Method Summary collapse

Constructor Details

#initialize(participants) ⇒ MessageProcessor

Returns a new instance of MessageProcessor.



11
12
13
14
 
# File 'lib/trace_viz/extractors/diagram/processors/message_processor.rb', line 11

def initialize(participants)
  @formatter = Formatters::Diagram::Sequence::MessageFormatter.new
  @message_builder = Builders::Diagram::MessageBuilder.new(@formatter, participants)
end

Instance Method Details

#process_node(node, caller_node = nil) ⇒ Object 



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
 
# File 'lib/trace_viz/extractors/diagram/processors/message_processor.rb', line 16

def process_node(node, caller_node = nil)
  trace = node.data
  caller_trace = caller_node&.data

  [].tap do |messages|
    # Handle participant transitions
    messages << @message_builder.build_call_message(
      caller_trace,
      trace,
    ) if caller_trace

    # Process the current node
    messages << handle_loop_start(trace)
    messages << handle_internal_message(trace)
    messages << handle_note(trace)
    messages << handle_activation(trace)
    messages.concat(process_children(node))
    messages << handle_deactivation(trace)
    messages << handle_loop_end(trace)

    # Update the current node after processing
    messages << @message_builder.build_return_message(
      trace,
      caller_trace,
    ) if caller_trace
  end.compact
end