mirror of
https://github.com/penpot/penpot.git
synced 2026-05-12 03:23:46 +00:00
9eaa55b711
They are not necessary and they are pretty common, because the user can interrupt the connection at any time.
71 lines
2.4 KiB
Clojure
71 lines
2.4 KiB
Clojure
;; This Source Code Form is subject to the terms of the Mozilla Public
|
|
;; License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
;; file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
;;
|
|
;; Copyright (c) KALEIDOS INC
|
|
|
|
(ns app.http.sse
|
|
"SSE (server sent events) helpers"
|
|
(:refer-clojure :exclude [tap])
|
|
(:require
|
|
[app.common.data :as d]
|
|
[app.common.exceptions :as ex]
|
|
[app.common.logging :as l]
|
|
[app.common.transit :as t]
|
|
[app.http.errors :as errors]
|
|
[app.util.events :as events]
|
|
[promesa.exec :as px]
|
|
[promesa.exec.csp :as sp]
|
|
[promesa.util :as pu]
|
|
[yetti.response :as yres])
|
|
(:import
|
|
java.io.OutputStream))
|
|
|
|
(defn- write!
|
|
[^OutputStream output ^bytes data]
|
|
(l/trc :hint "writting data" :data data :length (alength data))
|
|
(.write output data)
|
|
(.flush output))
|
|
|
|
(defn- encode
|
|
[[name data]]
|
|
(try
|
|
(let [data (with-out-str
|
|
(println "event:" (d/name name))
|
|
(println "data:" (t/encode-str data {:type :json-verbose}))
|
|
(println))]
|
|
(.getBytes data "UTF-8"))
|
|
(catch Throwable cause
|
|
(l/err :hint "unexpected error on encoding value on sse stream"
|
|
:cause cause)
|
|
nil)))
|
|
|
|
;; ---- PUBLIC API
|
|
|
|
(def default-headers
|
|
{"Content-Type" "text/event-stream;charset=UTF-8"
|
|
"Cache-Control" "no-cache, no-store, max-age=0, must-revalidate"
|
|
"Pragma" "no-cache"})
|
|
|
|
(defn response
|
|
[handler & {:keys [buf] :or {buf 32} :as opts}]
|
|
(fn [request]
|
|
{::yres/headers default-headers
|
|
::yres/status 200
|
|
::yres/body (yres/stream-body
|
|
(fn [_ output]
|
|
(binding [events/*channel* (sp/chan :buf buf :xf (keep encode))]
|
|
(let [listener (events/start-listener
|
|
(partial write! output)
|
|
(partial pu/close! output))]
|
|
(try
|
|
(let [result (handler)]
|
|
(events/tap :end result))
|
|
(catch Throwable cause
|
|
(events/tap :error (errors/handle' cause request))
|
|
(when-not (ex/instance? java.io.EOFException cause)
|
|
(l/err :hint "unexpected error on processing sse response" :cause cause)))
|
|
(finally
|
|
(sp/close! events/*channel*)
|
|
(px/await! listener)))))))}))
|