Class: Pindo::TaskSystem::MultiLineOutputManager

Inherits:
OutputSink
  • Object
show all
Defined in:
lib/pindo/module/task/output/multi_line_output_manager.rb

Overview

多行输出管理器管理所有任务的输出（终端显示 + 日志文件）实现 OutputSink 接口

Instance Method Summary collapse

Constructor Details

#initialize(log_dir: './pindo_logs', max_lines_per_task: 5, max_recent_completed: 3, auto_adjust: true) ⇒ MultiLineOutputManager

初始化输出管理器



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
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 17

def initialize(
  log_dir: './pindo_logs',
  max_lines_per_task: 5,
  max_recent_completed: 3,
  auto_adjust: true
)
  @mutex = Mutex.new
  @log_dir = log_dir
  @max_lines_per_task = max_lines_per_task
  @max_recent_completed = max_recent_completed
  @auto_adjust = auto_adjust

  # 任务显示器映射
  @task_displays = {}  # task_id => MultiLineTaskDisplay

  # 日志文件映射
  @task_loggers = {}   # task_id => File

  # 终端状态
  @terminal_lines = 0  # 当前占用的终端行数
  @last_render_time = Time.now
  @min_render_interval = 0.1  # 最小刷新间隔（秒）

  # 创建日志目录
  FileUtils.mkdir_p(@log_dir)

  # 隐藏光标（更流畅的动画）
  STDOUT.print "\e[?25l" if supports_ansi?

  # 注册退出时恢复光标
  at_exit { STDOUT.print "\e[?25h" if supports_ansi? }
end

Instance Method Details

#error_task(task_id, error_message) ⇒ Object

任务失败



122
123
124
125
126
127
128
129
130
131
132
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 122

def error_task(task_id, error_message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.mark_error(error_message)

    write_log(task_id, "✗ 任务失败: #{error_message}")

    close_log(task_id)
    render_terminal
  end
end

#log_detail(task_id, message) ⇒ Object

记录详细日志



206
207
208
209
210
211
212
213
214
215
216
217
218
219
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 206

def log_detail(task_id, message)
  @mutex.synchronize do
    # 只有非空消息才更新显示
    unless message.strip.empty?
      display = @task_displays[task_id]
      # 为了避免刷屏，普通日志使用 update_last_line 还是 add_line?
      # 考虑到 log_detail 可能是一行行输出，使用 add_line 更像滚动的日志
      display&.add_line(message)
    end

    write_log(task_id, message)
    render_terminal
  end
end

#log_error(task_id, message) ⇒ Object

记录错误



193
194
195
196
197
198
199
200
201
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 193

def log_error(task_id, message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.add_line("\e[31m#{message}\e[0m") # 红色高亮

    write_log(task_id, "[ERROR] #{message}")
    render_terminal
  end
end

#log_info(task_id, message) ⇒ Object

记录日志（更新终端 + 写文件）



167
168
169
170
171
172
173
174
175
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 167

def log_info(task_id, message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.add_line(message)
    
    write_log(task_id, "[INFO] #{message}")
    render_terminal
  end
end

#log_warning(task_id, message) ⇒ Object

记录警告



180
181
182
183
184
185
186
187
188
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 180

def log_warning(task_id, message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.add_line("\e[33m#{message}\e[0m") # 黄色高亮

    write_log(task_id, "[WARN] #{message}")
    render_terminal
  end
end

#register_task(task) ⇒ Object

注册任务到输出管理器



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 56

def register_task(task)
  @mutex.synchronize do
    task_id = task.id

    # 创建多行显示器
    display = MultiLineTaskDisplay.new(
      task_id,
      task.name,
      max_lines: @max_lines_per_task
    )
    @task_displays[task_id] = display

    # 创建日志文件
    log_file_name = "#{task.task_key}_#{task_id}.log"
    log_file_path = File.join(@log_dir, log_file_name)
    @task_loggers[task_id] = File.open(log_file_path, 'w')

    # 渲染终端
    render_terminal
  end
end

#start_task(task_id) ⇒ Object

任务开始



84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 84

def start_task(task_id)
  @mutex.synchronize do
    display = @task_displays[task_id]

    # 不在终端显示泛泛的"任务开始"，任务名称已经足够清晰
    # 只在日志文件中记录
    write_log(task_id, "开始执行")

    render_terminal
  end
end

#success_task(task_id, message = nil, execution_time: nil) ⇒ Object

任务成功



100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 100

def success_task(task_id, message = nil, execution_time: nil)
  @mutex.synchronize do
    display = @task_displays[task_id]

    msg = if execution_time
            message ? "#{message} (耗时 #{execution_time.round(2)}s)" : "完成 (耗时 #{execution_time.round(2)}s)"
          else
            message || "完成"
          end

    display&.mark_success(msg)

    write_log(task_id, "✓ 任务成功: #{msg}")

    close_log(task_id)
    render_terminal
  end
end

#update_current_line(task_id, message) ⇒ Object

更新当前行（不滚动）



154
155
156
157
158
159
160
161
162
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 154

def update_current_line(task_id, message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.update_last_line(message)

    write_log(task_id, "[UPDATE] #{message}")
    render_terminal
  end
end

#update_progress(task_id, message) ⇒ Object

更新进度（终端可见 + 日志记录）



141
142
143
144
145
146
147
148
149
 
# File 'lib/pindo/module/task/output/multi_line_output_manager.rb', line 141

def update_progress(task_id, message)
  @mutex.synchronize do
    display = @task_displays[task_id]
    display&.add_line(message)

    write_log(task_id, "[PROGRESS] #{message}")
    render_terminal
  end
end