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

初始化输出管理器

Parameters:

  • log_dir (String) (defaults to: './pindo_logs') —

    日志目录路径

  • max_lines_per_task (Integer) (defaults to: 5) —

    每个任务显示的最大行数

  • max_recent_completed (Integer) (defaults to: 3) —

    显示的最近完成任务数

  • auto_adjust (Boolean) (defaults to: true) —

    是否自动适应终端大小



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

任务失败

Parameters:

  • task_id (String) —

    任务 ID

  • error_message (String) —

    错误消息



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

记录详细日志

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    详细日志



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

记录错误

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    错误消息



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

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

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    日志消息



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

记录警告

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    警告消息



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

注册任务到输出管理器

Parameters:

  • task (Object) —

    任务对象（需要有 id, name, task_key 属性）



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

任务开始

Parameters:

  • task_id (String) —

    任务 ID



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

任务成功

Parameters:

  • task_id (String) —

    任务 ID

  • message (String, nil) (defaults to: nil) —

    成功消息

  • execution_time (Float, nil) (defaults to: nil) —

    执行时间（秒）



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

更新当前行（不滚动）

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    更新消息



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

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

Parameters:

  • task_id (String) —

    任务 ID

  • message (String) —

    进度消息



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