task_queue.h

#pragma once

#include <condition_variable>
#include <mutex>
#include <queue>

#include "mmap_reader.h"

// A bounded-free, thread-safe queue of Chunks for the producer-consumer
// pipeline: one producer pushes chunks, many workers pop them. Uses a mutex +
// condition_variable (the spec's preferred MVP; a lock-free queue is a stretch
// goal). `close()` lets workers drain and then exit cleanly.
class TaskQueue {
 public:
  void push(const Chunk& chunk);

  // Block until a chunk is available or the queue is closed-and-empty. Returns
  // true and fills `out` on success; returns false once drained after close().
  bool pop(Chunk& out);

  // Signal that no more chunks will be pushed; wakes all waiting workers.
  void close();

 private:
  std::mutex mu_;
  std::condition_variable cv_;
  std::queue<Chunk> q_;
  bool closed_ = false;
};