forked from metosin/muuntaja
/
transit.clj
40 lines (36 loc) · 1.26 KB
/
transit.clj
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
(ns muuntaja.format.transit
(:require [cognitect.transit :as transit]
[muuntaja.format.core :as core])
(:import (java.io ByteArrayOutputStream OutputStream)))
(defn decoder
[type options]
(reify
core/Decode
(decode [_ data _]
(let [reader (transit/reader data type options)]
(transit/read reader)))))
(defn encoder [type {:keys [verbose] :as options}]
(let [full-type (if (and (= type :json) verbose) :json-verbose type)]
(reify
core/EncodeToBytes
(encode-to-bytes [_ data _]
(let [baos (ByteArrayOutputStream.)
writer (transit/writer baos full-type options)]
(transit/write writer data)
(.toByteArray baos)))
core/EncodeToOutputStream
(encode-to-output-stream [_ data _]
(fn [^OutputStream output-stream]
(transit/write
(transit/writer output-stream full-type options) data)
(.flush output-stream))))))
(def json-format
(core/map->Format
{:name "application/transit+json"
:decoder [(partial decoder :json)]
:encoder [(partial encoder :json)]}))
(def msgpack-format
(core/map->Format
{:name "application/transit+msgpack"
:decoder [(partial decoder :msgpack)]
:encoder [(partial encoder :msgpack)]}))