Teek::BackgroundThread::BackgroundWork::TaskContext Class

def check_message
  return nil if @message_queue.empty?
  msg = @message_queue.pop(true)
  handle_control_message(msg)
  msg
rescue ThreadError
  nil
end

def check_pause
  # First drain any pending messages (non-blocking)
  until @message_queue.empty?
    msg = @message_queue.pop(true)
    handle_control_message(msg)
  end

  # Then block while paused
  while @paused
    msg = @message_queue.pop  # Blocking wait
    handle_control_message(msg)
  end
end

def initialize(output_queue, message_queue)
  @output_queue = output_queue
  @message_queue = message_queue
  @paused = false
end

def send_message(msg)
  @output_queue << [:message, msg]
end

def wait_message
  msg = @message_queue.pop
  handle_control_message(msg)
  msg
end

def yield(value)
  @output_queue << [:result, value]
  Thread.pass
end