How can i log the output segments of a task?

I'm trying to debug a hanging job, and want to inspect the output of a task

(defn log-batch-after [event lifecycle]
  (let [task-map (:onyx.core/task-map event)
        task-name (:onyx/name task-map)]
    (log/info
             "After batch:" task-name
             "Output size:" (count (:onyx.core/results event))))
  {})

I'm using this function in a lifecycle for all tasks, containing :lifecycle/after-read-batch log-after-read-batch

but the output size is always 4

@chrisblom (count (:segments (:onyx.core/results event)))

results record includes four keys, :tree, :segments, :acks, and :retries

ah, thanks!