Class: Pindo::TaskSystem::CustomIO

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

Overview

自定义 IO 对象拦截所有 puts/print 输出，将其路由到输出管理器

Instance Method Summary collapse

Constructor Details

#initialize(task_id, output_manager, stream_type) ⇒ CustomIO

初始化自定义 IO

Parameters:

  • task_id (String) —

    任务 ID

  • output_manager (OutputSink) —

    输出管理器

  • stream_type (Symbol) —

    流类型（:stdout 或 :stderr）



36
37
38
39
40
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 36

def initialize(task_id, output_manager, stream_type)
  @task_id = task_id
  @output_manager = output_manager
  @stream_type = stream_type
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(method, *args, &block) ⇒ Object

委托其他方法给 STDOUT



100
101
102
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 100

def method_missing(method, *args, &block)
  STDOUT.send(method, *args, &block)
end

Instance Method Details

#ensure_utf8(str) ⇒ String

确保字符串使用 UTF-8 编码

Parameters:

  • str (String) —

    输入字符串

Returns:

  • (String) —

    UTF-8 编码的字符串



63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 63

def ensure_utf8(str)
  return str if str.nil?
  return str if str.encoding == Encoding::UTF_8 && str.valid_encoding?
  
  # 尝试转换为 UTF-8
  begin
    str.encode(Encoding::UTF_8)
  rescue Encoding::InvalidByteSequenceError, Encoding::UndefinedConversionError
    # 如果转换失败，尝试使用替换字符
    str.encode(Encoding::UTF_8, invalid: :replace, undef: :replace)
  end
end

#flush ⇒ Object

flush 方法（无操作）



95
96
97
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 95

def flush
  # 不需要处理
end

#print(*args) ⇒ Object

print 方法

Parameters:

  • args (Array) —

    参数列表



89
90
91
92
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 89

def print(*args)
  write(ensure_utf8(args.join))
  nil
end

#puts(*args) ⇒ Object

puts 方法

Parameters:

  • args (Array) —

    参数列表



78
79
80
81
82
83
84
85
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 78

def puts(*args)
  if args.empty?
    write("\n")
  else
    args.each { |arg| write("#{ensure_utf8(arg.to_s)}\n") }
  end
  nil
end

#respond_to_missing?(method, include_private = false) ⇒ Boolean

响应方法检查

Parameters:

  • method (Symbol) —

    方法名

  • include_private (Boolean) (defaults to: false) —

    是否包含私有方法

Returns:

  • (Boolean) —

    是否响应该方法



108
109
110
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 108

def respond_to_missing?(method, include_private = false)
  STDOUT.respond_to?(method, include_private) || super
end

#write(message) ⇒ Integer

写入数据

Parameters:

  • message (String) —

    消息内容

Returns:

  • (Integer) —

    写入的字节数



45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/pindo/module/task/output/stdout_redirector.rb', line 45

def write(message)
  # 捕获输出，写入日志文件（跳过空行）
  unless message.strip.empty?
    # 确保消息使用 UTF-8 编码，防止编码错误导致任务中断
    safe_message = ensure_utf8(message.chomp)
    # 根据流类型选择日志级别
    if @stream_type == :stderr
      @output_manager.log_error(@task_id, safe_message)
    else
      @output_manager.log_detail(@task_id, safe_message)
    end
  end
  message.length
end