gistfile1.txt

(defn my-inc [{:keys [n] :as segment}]
  (assoc segment :n (inc n)))

(defn after-batch [{:keys [onyx.core/results onyx.core/task] :as event} lifecycle]
  (let [additional-context task
        tree-new (mapv (fn [{:keys [root leaves] :as message}]
                         (assoc message
                                :leaves
                                (mapv (fn [leaf]
                                        (let [input-segment-context (:context (:message root))]
                                          (assoc-in leaf
                                                    [:message :context]
                                                    (update-in input-segment-context
                                                               [:path]
                                                               (fn [path] (conj (vec path) additional-context))))))
                                      leaves)))
                       (:tree results))
        segments-new (mapcat :leaves tree-new)]
    {:onyx.core/results (assoc results :tree tree-new :segments segments-new)}))