• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

plausible / ch / d24fada4e6eda58b2cb2a390d8e5ac42c2b8bdbd-PR-397

03 Aug 2026 01:06PM UTC coverage: 97.573% (-0.5%) from 98.062%
d24fada4e6eda58b2cb2a390d8e5ac42c2b8bdbd-PR-397

Pull #397

github

ruslandoga
Generate decoder state transitions
Pull Request #397: Use active sequences for container decoding

96 of 98 new or added lines in 1 file covered. (97.96%)

2 existing lines in 1 file now uncovered.

764 of 783 relevant lines covered (97.57%)

16043.1 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

99.01
/lib/ch/row_binary.ex
1
defmodule Ch.RowBinary do
2
  @moduledoc "Helpers for working with ClickHouse [RowBinary](https://clickhouse.com/docs/en/interfaces/formats/RowBinary) format."
3

4
  # @compile {:bin_opt_info, true}
5
  @dialyzer :no_improper_lists
6

7
  import Bitwise
8

9
  @epoch_gregorian_seconds 62_167_219_200
10
  @epoch_gregorian_days 719_528
11

12
  @doc false
13
  def encode_names_and_types(names, types) do
5✔
14
    [encode(:varint, length(names)), encode_many(names, :string), encode_types(types)]
15
  end
16

17
  defp encode_types([type | types]) do
12✔
18
    encoded =
12✔
19
      case type do
20
        _ when is_binary(type) -> type
11✔
21
        _ -> Ch.Types.encode(type)
1✔
22
      end
23

24
    [encode(:string, encoded) | encode_types(types)]
25
  end
26

27
  defp encode_types([] = done), do: done
5✔
28

29
  @doc """
30
  Encodes a single row to [RowBinary](https://clickhouse.com/docs/en/interfaces/formats/RowBinary) as iodata.
31

32
  Examples:
33

34
      iex> encode_row([], [])
35
      []
36

37
      iex> encode_row([1], ["UInt8"])
38
      [1]
39

40
      iex> encode_row([3, "hello"], ["UInt8", "String"])
41
      [3, [5 | "hello"]]
42

43
  """
44
  def encode_row(row, types) do
45
    _encode_row(row, encoding_types(types))
22✔
46
  end
47

48
  defp _encode_row([el | els], [type | types]), do: [encode(type, el) | _encode_row(els, types)]
106✔
49
  defp _encode_row([] = done, []), do: done
22✔
50

51
  @doc """
52
  Encodes multiple rows to [RowBinary](https://clickhouse.com/docs/en/interfaces/formats/RowBinary) as iodata.
53

54
  Examples:
55

56
      iex> encode_rows([], [])
57
      []
58

59
      iex> encode_rows([[1]], ["UInt8"])
60
      [1]
61

62
      iex> encode_rows([[3, "hello"], [4, "hi"]], ["UInt8", "String"])
63
      [3, [5 | "hello"], 4, [2 | "hi"]]
64

65
  """
66
  def encode_rows(rows, types) do
67
    _encode_rows(rows, encoding_types(types))
854✔
68
  end
69

70
  @doc false
71
  def _encode_rows([row | rows], types), do: _encode_rows(row, types, rows, types)
4,796✔
72
  def _encode_rows([] = done, _types), do: done
854✔
73

74
  defp _encode_rows([el | els], [t | ts], rows, types) do
12,304✔
75
    [encode(t, el) | _encode_rows(els, ts, rows, types)]
76
  end
77

78
  defp _encode_rows([], [], rows, types), do: _encode_rows(rows, types)
4,796✔
79

80
  @doc false
81
  def encoding_types([type | types]) do
2,262✔
82
    [encoding_type(type) | encoding_types(types)]
83
  end
84

85
  def encoding_types([] = done), do: done
884✔
86

87
  defp encoding_type(type) when is_binary(type) do
88
    encoding_type(Ch.Types.decode(type))
2,162✔
89
  end
90

91
  defp encoding_type(t)
92
       when t in [
93
              :string,
94
              :json,
95
              :dynamic,
96
              :boolean,
97
              :uuid,
98
              :date,
99
              :datetime,
100
              :date32,
101
              :time,
102
              :ipv4,
103
              :ipv6,
104
              :point,
105
              :nothing
106
            ],
107
       do: t
921✔
108

109
  defp encoding_type({:datetime = d, "UTC"}), do: d
2✔
110

111
  defp encoding_type({:datetime, tz}) do
112
    raise ArgumentError, "can't encode DateTime with non-UTC timezone: #{inspect(tz)}"
1✔
113
  end
114

115
  defp encoding_type({:fixed_string, _len} = t), do: t
311✔
116

117
  for size <- [8, 16, 32, 64, 128, 256] do
118
    defp encoding_type(unquote(:"u#{size}") = u), do: u
655✔
119
    defp encoding_type(unquote(:"i#{size}") = i), do: i
40✔
120
  end
121

122
  for size <- [32, 64] do
123
    defp encoding_type(unquote(:"f#{size}") = f), do: f
222✔
124
  end
125

126
  defp encoding_type({:array = a, t}), do: {a, encoding_type(t)}
548✔
127

128
  defp encoding_type({:tuple = t, ts}) do
5✔
129
    {t, Enum.map(ts, &encoding_type/1)}
130
  end
131

132
  defp encoding_type({:variant = v, ts}) do
3✔
133
    {v, Enum.map(ts, &encoding_type/1)}
134
  end
135

136
  defp encoding_type({:map = m, kt, vt}) do
137
    {m, encoding_type(kt), encoding_type(vt)}
8✔
138
  end
139

140
  defp encoding_type({:nullable = n, t}), do: {n, encoding_type(t)}
113✔
141
  defp encoding_type({:low_cardinality, t}), do: encoding_type(t)
202✔
142

143
  defp encoding_type({:decimal, p, s}) do
144
    case decimal_size(p) do
5✔
145
      32 -> {:decimal32, s}
1✔
146
      64 -> {:decimal64, s}
2✔
147
      128 -> {:decimal128, s}
1✔
148
      256 -> {:decimal256, s}
1✔
149
    end
150
  end
151

152
  defp encoding_type({d, _scale} = t)
153
       when d in [:decimal32, :decimal64, :decimal128, :decimal256],
154
       do: t
5✔
155

156
  defp encoding_type({:datetime64 = t, p}), do: {t, time_unit(p)}
1✔
157

158
  defp encoding_type({:datetime64 = t, p, "UTC"}), do: {t, time_unit(p)}
2✔
159

160
  defp encoding_type({:datetime64, _, tz}) do
161
    raise ArgumentError, "can't encode DateTime64 with non-UTC timezone: #{inspect(tz)}"
1✔
162
  end
163

164
  defp encoding_type({:time64 = t, p}), do: {t, time_unit(p)}
105✔
165

166
  defp encoding_type({e, mappings}) when e in [:enum8, :enum16] do
4✔
167
    {e, Map.new(mappings)}
168
  end
169

170
  defp encoding_type({:simple_aggregate_function, _f, t}), do: encoding_type(t)
1✔
171

172
  defp encoding_type(:ring), do: {:array, :point}
1✔
173
  defp encoding_type(:polygon), do: {:array, {:array, :point}}
1✔
174
  defp encoding_type(:multipolygon), do: {:array, {:array, {:array, :point}}}
1✔
175

176
  defp encoding_type(type) do
177
    raise ArgumentError, "unsupported type for encoding: #{inspect(type)}"
1✔
178
  end
179

180
  @doc false
181
  def encode(type, value)
182

183
  def encode(:varint, i) when is_integer(i) and i >= 0 and i < 128, do: i
7,695✔
184
  def encode(:varint, i) when is_integer(i) and i >= 0, do: encode_varint_cont(i)
13✔
185

186
  def encode(:varint, i) when is_integer(i) do
187
    raise ArgumentError, "invalid varint: #{inspect(i)}"
1✔
188
  end
189

190
  def encode(:string, str) do
191
    case str do
5,735✔
192
      _ when is_binary(str) -> [encode(:varint, byte_size(str)) | str]
5,701✔
193
      _ when is_list(str) -> [encode(:varint, IO.iodata_length(str)) | str]
25✔
194
      nil -> 0
3✔
195
    end
196
  end
197

198
  def encode(:json, json) do
199
    # assuming it can be sent as text and not "native" binary JSON
200
    # i.e. assumes `settings: [input_format_binary_read_json_as_string: 1]`
201
    # TODO
202
    encode(:string, JSON.encode_to_iodata!(json))
5✔
203
  end
204

205
  def encode({:fixed_string, size}, str) when byte_size(str) == size do
206
    str
760✔
207
  end
208

209
  def encode({:fixed_string, size}, str) when byte_size(str) < size do
3,406✔
210
    to_pad = size - byte_size(str)
3,406✔
211
    [str | <<0::size(to_pad * 8)>>]
212
  end
213

214
  def encode({:fixed_string, size}, nil), do: <<0::size(size * 8)>>
2✔
215

216
  # UInt8 — [0 : 255]
217
  def encode(:u8, u) when is_integer(u) and u >= 0 and u <= 255, do: u
4,544✔
218
  def encode(:u8, nil), do: 0
4✔
219

220
  def encode(:u8, term) do
221
    raise ArgumentError, "invalid UInt8: #{inspect(term)}"
7✔
222
  end
223

224
  # Int8 — [-128 : 127]
225
  def encode(:i8, i) when is_integer(i) and i >= 0 and i <= 127, do: i
23✔
226
  def encode(:i8, i) when is_integer(i) and i < 0 and i >= -128, do: <<i::signed>>
11✔
227
  def encode(:i8, nil), do: 0
1✔
228

229
  def encode(:i8, term) do
230
    raise ArgumentError, "invalid Int8: #{inspect(term)}"
6✔
231
  end
232

233
  for size <- [16, 32, 64, 128, 256] do
234
    unsigned_max = (1 <<< size) - 1
235
    signed_min = -(1 <<< (size - 1))
236
    signed_max = (1 <<< (size - 1)) - 1
237
    uint = :"u#{size}"
238
    int = :"i#{size}"
239

240
    def encode(unquote(uint), u) when is_integer(u) and u >= 0 and u <= unquote(unsigned_max) do
241
      <<u::unquote(size)-little>>
146✔
242
    end
243

244
    def encode(unquote(int), i)
245
        when is_integer(i) and i >= unquote(signed_min) and i <= unquote(signed_max) do
246
      <<i::unquote(size)-little-signed>>
152✔
247
    end
248

249
    def encode(unquote(uint), nil), do: <<0::unquote(size)>>
3✔
250
    def encode(unquote(int), nil), do: <<0::unquote(size)>>
3✔
251

252
    def encode(unquote(uint), term) do
253
      raise ArgumentError, "invalid UInt#{unquote(size)}: #{inspect(term)}"
15✔
254
    end
255

256
    def encode(unquote(int), term) do
257
      raise ArgumentError, "invalid Int#{unquote(size)}: #{inspect(term)}"
15✔
258
    end
259
  end
260

261
  for size <- [32, 64] do
262
    type = :"f#{size}"
263

264
    def encode(unquote(type), f) when is_number(f) do
265
      <<f::unquote(size)-little-signed-float>>
1,210✔
266
    end
267

268
    def encode(unquote(type), nil), do: <<0::unquote(size)>>
4✔
269
  end
270

271
  def encode({:decimal, precision, scale}, decimal) do
272
    type =
4✔
273
      case decimal_size(precision) do
274
        32 -> :decimal32
1✔
275
        64 -> :decimal64
1✔
276
        128 -> :decimal128
1✔
277
        256 -> :decimal256
1✔
278
      end
279

280
    encode({type, scale}, decimal)
4✔
281
  end
282

283
  for size <- [32, 64, 128, 256] do
284
    type = :"decimal#{size}"
285

286
    def encode({unquote(type), scale} = t, %Decimal{sign: sign, coef: coef, exp: exp} = d) do
287
      cond do
29✔
288
        scale == -exp ->
289
          i = sign * coef
20✔
290
          <<i::unquote(size)-little>>
20✔
291

292
        exp >= 0 ->
9✔
293
          i = sign * coef * Integer.pow(10, exp + scale)
1✔
294
          <<i::unquote(size)-little>>
1✔
295

296
        true ->
8✔
297
          encode(t, Decimal.round(d, scale))
8✔
298
      end
299
    end
300

301
    def encode({unquote(type), _scale}, nil), do: <<0::unquote(size)>>
4✔
302
  end
303

304
  def encode(:boolean, true), do: 1
934✔
305
  def encode(:boolean, false), do: 0
968✔
306
  def encode(:boolean, nil), do: 0
1✔
307

308
  def encode({:array, type}, [_ | _] = l) do
1,961✔
309
    [encode(:varint, length(l)) | encode_many(l, type)]
310
  end
311

312
  def encode({:array, _type}, []), do: 0
270✔
313
  def encode({:array, _type}, nil), do: 0
4✔
314

315
  def encode({:map, k, v}, [_ | _] = m) do
1✔
316
    [encode(:varint, length(m)) | encode_many_kv(m, k, v)]
317
  end
318

319
  def encode({:map, k, v}, m) when is_map(m) do
14✔
320
    [
321
      encode(:varint, map_size(m))
322
      | :maps.fold(fn key, value, acc -> [encode(k, key), encode(v, value) | acc] end, [], m)
15✔
323
    ]
324
  end
325

326
  def encode({:map, _k, _v}, []), do: 0
1✔
327
  def encode({:map, _k, _v}, nil), do: 0
1✔
328

329
  def encode({:tuple, _types} = t, v) when is_tuple(v) do
330
    encode(t, Tuple.to_list(v))
11✔
331
  end
332

333
  def encode({:tuple, types}, values) when is_list(types) and is_list(values) do
334
    encode_row(values, types)
11✔
335
  end
336

337
  def encode({:tuple, types}, nil) when is_list(types) do
338
    Enum.map(types, fn type -> encode(type, nil) end)
1✔
339
  end
340

341
  def encode({:variant, _types}, nil), do: 255
3✔
342

343
  def encode({:variant, types}, value) do
344
    try_encode_variant(types, 0, value)
8✔
345
  end
346

347
  def encode(:datetime, %NaiveDateTime{} = datetime) do
348
    {seconds, _micros} = NaiveDateTime.to_gregorian_seconds(datetime)
17✔
349
    <<seconds - @epoch_gregorian_seconds::32-little>>
17✔
350
  end
351

352
  def encode(:datetime, %DateTime{} = datetime) do
353
    <<DateTime.to_unix(datetime, :second)::32-little>>
5✔
354
  end
355

356
  def encode(:datetime, nil), do: <<0::32>>
1✔
357

358
  def encode({:datetime64, time_unit}, %NaiveDateTime{} = datetime) do
359
    {seconds, micros} = NaiveDateTime.to_gregorian_seconds(datetime)
4✔
360

361
    <<(seconds - @epoch_gregorian_seconds) * time_unit + div(micros * time_unit, 1_000_000)::64-little-signed>>
4✔
362
  end
363

364
  def encode({:datetime64, time_unit}, %DateTime{} = datetime) do
365
    <<DateTime.to_unix(datetime, time_unit)::64-little-signed>>
5✔
366
  end
367

368
  def encode({:datetime64, _time_unit}, nil), do: <<0::64>>
1✔
369

370
  def encode(:date, %Date{} = date) do
371
    <<Date.to_gregorian_days(date) - @epoch_gregorian_days::16-little>>
14✔
372
  end
373

374
  def encode(:date, nil), do: <<0::16>>
1✔
375

376
  def encode(:date32, %Date{} = date) do
377
    <<Date.to_gregorian_days(date) - @epoch_gregorian_days::32-little-signed>>
8✔
378
  end
379

380
  def encode(:date32, nil), do: <<0::32>>
1✔
381

382
  def encode(:time, %Time{} = time) do
383
    {s, _micros} = Time.to_seconds_after_midnight(time)
107✔
384
    <<s::32-little-signed>>
107✔
385
  end
386

387
  def encode(:time, nil), do: <<0::32>>
1✔
388

389
  def encode({:time64, time_unit}, %Time{} = time) do
390
    {s, micros} = Time.to_seconds_after_midnight(time)
117✔
391

392
    micros_as_ticks =
117✔
393
      cond do
394
        time_unit < 1_000_000 -> div(micros, div(1_000_000, time_unit))
74✔
395
        time_unit == 1_000_000 -> micros
43✔
396
        true -> micros * div(time_unit, 1_000_000)
24✔
397
      end
398

399
    ticks = s * time_unit + micros_as_ticks
117✔
400
    <<ticks::64-little-signed>>
117✔
401
  end
402

403
  def encode({:time64, _time_unit}, nil), do: <<0::64>>
1✔
404

405
  def encode(:uuid, <<u1::64, u2::64>>), do: <<u1::64-little, u2::64-little>>
12✔
406

407
  def encode(
408
        :uuid,
409
        <<a1, a2, a3, a4, a5, a6, a7, a8, ?-, b1, b2, b3, b4, ?-, c1, c2, c3, c4, ?-, d1, d2, d3,
410
          d4, ?-, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12>>
411
      ) do
412
    raw =
2✔
413
      <<d(a1)::4, d(a2)::4, d(a3)::4, d(a4)::4, d(a5)::4, d(a6)::4, d(a7)::4, d(a8)::4, d(b1)::4,
414
        d(b2)::4, d(b3)::4, d(b4)::4, d(c1)::4, d(c2)::4, d(c3)::4, d(c4)::4, d(d1)::4, d(d2)::4,
415
        d(d3)::4, d(d4)::4, d(e1)::4, d(e2)::4, d(e3)::4, d(e4)::4, d(e5)::4, d(e6)::4, d(e7)::4,
416
        d(e8)::4, d(e9)::4, d(e10)::4, d(e11)::4, d(e12)::4>>
417

418
    encode(:uuid, raw)
2✔
419
  end
420

421
  def encode(:uuid, nil), do: <<0::128>>
1✔
422

423
  def encode(:ipv4, {a, b, c, d}), do: [d, c, b, a]
6✔
424
  def encode(:ipv4, nil), do: <<0::32>>
1✔
425

426
  def encode(:ipv6, {b1, b2, b3, b4, b5, b6, b7, b8}) do
427
    <<b1::16, b2::16, b3::16, b4::16, b5::16, b6::16, b7::16, b8::16>>
6✔
428
  end
429

430
  def encode(:ipv6, <<_::128>> = encoded), do: encoded
1✔
431
  def encode(:ipv6, nil), do: <<0::128>>
1✔
432

433
  def encode(:point, {x, y}), do: [encode(:f64, x) | encode(:f64, y)]
22✔
434
  def encode(:point, nil), do: <<0::128>>
1✔
435
  def encode(:ring, points), do: encode({:array, :point}, points)
1✔
436
  def encode(:polygon, rings), do: encode({:array, :ring}, rings)
1✔
437
  def encode(:multipolygon, polygons), do: encode({:array, :polygon}, polygons)
1✔
438

439
  # TODO
440
  def encode(:dynamic, value) do
441
    case value do
12✔
442
      _ when is_binary(value) -> [0x15 | encode(:string, value)]
2✔
443
      _ when is_integer(value) and value >= 0 -> [0x04 | encode(:u64, value)]
3✔
444
      _ when is_integer(value) -> [0x0A | encode(:i64, value)]
1✔
445
      _ when is_float(value) -> [0x0E | encode(:f64, value)]
2✔
446
      %Date{} -> [0x0F | encode(:date, value)]
2✔
447
      %NaiveDateTime{} -> [0x11 | encode(:datetime, value)]
1✔
448
      [] -> [0x1E, 0x00]
1✔
449
    end
450
  end
451

452
  # TODO enum8 and enum16 nil
453
  for size <- [8, 16] do
454
    enum_t = :"enum#{size}"
455
    int_t = :"i#{size}"
456

457
    def encode({unquote(enum_t), mapping}, e) do
458
      i =
12✔
459
        case e do
460
          _ when is_integer(e) ->
461
            e
2✔
462

463
          _ when is_binary(e) ->
464
            case Map.fetch(mapping, e) do
10✔
465
              {:ok, res} ->
466
                res
9✔
467

468
              :error ->
469
                raise ArgumentError,
1✔
470
                      "enum value #{inspect(e)} not found in mapping: #{inspect(mapping)}"
471
            end
472
        end
473

474
      encode(unquote(int_t), i)
11✔
475
    end
476
  end
477

478
  def encode({:nullable, _type}, nil), do: 1
852✔
479

480
  def encode({:nullable, type}, value) do
481
    case encode(type, value) do
805✔
482
      e when is_list(e) or is_binary(e) -> [0 | e]
804✔
483
      e -> [0, e]
1✔
484
    end
485
  end
486

487
  defp encode_varint_cont(i) when i < 128, do: <<i>>
13✔
488

489
  defp encode_varint_cont(i) do
18✔
490
    [(i &&& 0b0111_1111) ||| 0b1000_0000 | encode_varint_cont(i >>> 7)]
491
  end
492

493
  defp encode_many([el | rest], type), do: [encode(type, el) | encode_many(rest, type)]
8,429✔
494
  defp encode_many([] = done, _type), do: done
1,966✔
495

496
  defp encode_many_kv([{key, value} | rest], key_type, value_type) do
1✔
497
    [
498
      encode(key_type, key),
499
      encode(value_type, value)
500
      | encode_many_kv(rest, key_type, value_type)
501
    ]
502
  end
503

504
  defp encode_many_kv([] = done, _key_type, _value_type), do: done
1✔
505

506
  # TODO find a better way than try/rescue
507
  defp try_encode_variant([type | types], idx, value) do
508
    try do
13✔
509
      encode(type, value)
13✔
510
    else
511
      encoded -> [idx | encoded]
7✔
512
    rescue
513
      _e -> try_encode_variant(types, idx + 1, value)
6✔
514
    end
515
  end
516

517
  defp try_encode_variant([], _idx, value) do
518
    raise ArgumentError, "no matching type found for encoding #{inspect(value)} as Variant"
1✔
519
  end
520

521
  @compile {:inline, d: 1}
522

523
  defp d(?0), do: 0
1✔
524
  defp d(?1), do: 1
1✔
525
  defp d(?2), do: 2
3✔
526
  defp d(?3), do: 3
1✔
527
  defp d(?4), do: 4
1✔
528
  defp d(?5), do: 5
3✔
529
  defp d(?6), do: 6
1✔
530
  defp d(?7), do: 7
1✔
531
  defp d(?8), do: 8
1✔
532
  defp d(?9), do: 9
1✔
533
  defp d(?A), do: 10
1✔
534
  defp d(?B), do: 11
1✔
535
  defp d(?C), do: 12
1✔
536
  defp d(?D), do: 13
1✔
537
  defp d(?E), do: 14
1✔
538
  defp d(?F), do: 15
1✔
539
  defp d(?a), do: 10
1✔
540
  defp d(?b), do: 11
1✔
541
  defp d(?c), do: 12
1✔
542
  defp d(?d), do: 13
1✔
543
  defp d(?e), do: 14
1✔
544
  defp d(?f), do: 15
1✔
545

546
  varints = [
547
    {_pattern = quote(do: <<0::1, v1::7>>), _value = quote(do: v1)},
548
    {quote(do: <<1::1, v1::7, 0::1, v2::7>>), quote(do: (v2 <<< 7) + v1)},
549
    {quote(do: <<1::1, v1::7, 1::1, v2::7, 0::1, v3::7>>),
550
     quote(do: (v3 <<< 14) + (v2 <<< 7) + v1)},
551
    {quote(do: <<1::1, v1::7, 1::1, v2::7, 1::1, v3::7, 0::1, v4::7>>),
552
     quote(do: (v4 <<< 21) + (v3 <<< 14) + (v2 <<< 7) + v1)},
553
    {quote(do: <<1::1, v1::7, 1::1, v2::7, 1::1, v3::7, 1::1, v4::7, 0::1, v5::7>>),
554
     quote(do: (v5 <<< 28) + (v4 <<< 21) + (v3 <<< 14) + (v2 <<< 7) + v1)},
555
    {quote(do: <<1::1, v1::7, 1::1, v2::7, 1::1, v3::7, 1::1, v4::7, 1::1, v5::7, 0::1, v6::7>>),
556
     quote(do: (v6 <<< 35) + (v5 <<< 28) + (v4 <<< 21) + (v3 <<< 14) + (v2 <<< 7) + v1)},
557
    {quote do
558
       <<1::1, v1::7, 1::1, v2::7, 1::1, v3::7, 1::1, v4::7, 1::1, v5::7, 1::1, v6::7, 0::1,
559
         v7::7>>
560
     end,
561
     quote do
562
       (v7 <<< 42) + (v6 <<< 35) + (v5 <<< 28) + (v4 <<< 21) + (v3 <<< 14) + (v2 <<< 7) + v1
563
     end},
564
    {quote do
565
       <<1::1, v1::7, 1::1, v2::7, 1::1, v3::7, 1::1, v4::7, 1::1, v5::7, 1::1, v6::7, 1::1,
566
         v7::7, 0::1, v8::7>>
567
     end,
568
     quote do
569
       (v8 <<< 49) + (v7 <<< 42) + (v6 <<< 35) + (v5 <<< 28) + (v4 <<< 21) + (v3 <<< 14) +
570
         (v2 <<< 7) + v1
571
     end}
572
  ]
573

574
  @doc false
575
  @spec decode_header(binary()) ::
576
          {:ok, names :: [String.t()], types :: [term], rest :: binary} | :more
577
  def decode_header(row_binary_with_names_and_types)
578

579
  for {pattern, value} <- varints do
580
    def decode_header(<<unquote(pattern), rest::bytes>>) do
581
      decode_header_names(rest, unquote(value), unquote(value), _acc = [])
36✔
582
    end
583
  end
584

585
  def decode_header(<<_bin::bytes>>) do
1✔
586
    :more
587
  end
588

589
  defp decode_header_names(<<rest::bytes>>, 0, count, names) do
590
    decode_header_types(rest, count, _acc = [], :lists.reverse(names))
21✔
591
  end
592

593
  for {pattern, value} <- varints do
594
    defp decode_header_names(
595
           <<unquote(pattern), name::size(unquote(value))-bytes, rest::bytes>>,
596
           left,
597
           count,
598
           acc
599
         ) do
600
      decode_header_names(rest, left - 1, count, [name | acc])
78✔
601
    end
602
  end
603

604
  defp decode_header_names(<<_bin::bytes>>, _left, _count, _acc) do
15✔
605
    :more
606
  end
607

608
  defp decode_header_types(<<rest::bytes>>, 0, types, names) do
609
    {:ok, names, decoding_types_reverse(types), rest}
1✔
610
  end
611

612
  for {pattern, value} <- varints do
613
    defp decode_header_types(
614
           <<unquote(pattern), type::size(unquote(value))-bytes, rest::bytes>>,
615
           count,
616
           acc,
617
           names
618
         ) do
619
      decode_header_types(rest, count - 1, [type | acc], names)
24✔
620
    end
621
  end
622

623
  defp decode_header_types(<<_bin::bytes>>, _count, _acc, _names) do
20✔
624
    :more
625
  end
626

627
  @doc """
628
  Decodes [RowBinaryWithNamesAndTypes](https://clickhouse.com/docs/en/interfaces/formats/RowBinaryWithNamesAndTypes) into rows.
629

630
  Example:
631

632
      iex> decode_rows(<<1, 3, "1+1"::bytes, 5, "UInt8"::bytes, 2>>)
633
      [[2]]
634

635
  """
636
  def decode_rows(row_binary_with_names_and_types)
637
  def decode_rows(<<>>), do: []
1✔
638

639
  for {pattern, value} <- varints do
640
    def decode_rows(<<unquote(pattern), rest::bytes>>) do
641
      skip_names(rest, unquote(value), unquote(value))
5✔
642
    end
643
  end
644

645
  @doc """
646
  Same as `decode_rows/1` but the first element is a list of column names.
647

648
  Example:
649

650
      iex> decode_names_and_rows(<<1, 3, "1+1"::bytes, 5, "UInt8"::bytes, 2>>)
651
      [["1+1"], [2]]
652

653
  """
654
  def decode_names_and_rows(row_binary_with_names_and_types)
655

656
  for {pattern, value} <- varints do
657
    def decode_names_and_rows(<<unquote(pattern), rest::bytes>>) do
658
      decode_names(rest, unquote(value), unquote(value), _acc = [])
3,030✔
659
    end
660
  end
661

662
  @doc """
663
  Decodes [RowBinary](https://clickhouse.com/docs/en/interfaces/formats/RowBinary) into rows.
664

665
  Example:
666

667
      iex> decode_rows(<<1>>, ["UInt8"])
668
      [[1]]
669

670
  """
671
  def decode_rows(row_binary, types)
672
  def decode_rows(<<>>, _types), do: []
1✔
673

674
  def decode_rows(<<data::bytes>>, types) do
675
    decode_rows!(data, decoding_types(types))
436✔
676
  end
677

678
  defp decode_rows!(data, types) do
679
    {rows, remaining_data, state} =
3,445✔
680
      decode_rows(types, data, [], [], types, :row, types, 1, [])
681

682
    case state do
3,427✔
683
      nil ->
684
        rows
3,425✔
685

686
      {:cont, types_rest, row} ->
687
        raise ArgumentError, """
2✔
688
        incomplete RowBinary data: ran out of bytes while decoding
689

690
        Expected to decode: #{inspect(types_rest)}
691
        Remaining bytes: #{byte_size(remaining_data)} bytes
2✔
692
        Partial row: #{inspect(row)}
693
        Completed rows: #{length(rows)}
2✔
694
        """
695

696
      {:cont, types_rest, row, _kind, _schema, _left, _outer} ->
NEW
697
        raise ArgumentError, """
×
698
        incomplete RowBinary data: ran out of bytes while decoding
699

700
        Expected to decode: #{inspect(types_rest)}
UNCOV
701
        Remaining bytes: #{byte_size(remaining_data)} bytes
×
702
        Partial row: #{inspect(row)}
UNCOV
703
        Completed rows: #{length(rows)}
×
704
        """
705
    end
706
  end
707

708
  @doc false
709
  def decode_rows_continue(<<data::bytes>>, types, state) do
710
    case state do
201,104✔
711
      {:cont, types_rest, row} ->
712
        decode_rows(types_rest, data, row, [], types, :row, types, 1, [])
200,907✔
713

714
      {:cont, types_rest, row, kind, schema, left, outer} ->
715
        decode_rows(types_rest, data, row, [], types, kind, schema, left, outer)
139✔
716

717
      nil ->
718
        decode_rows(types, data, [], [], types, :row, types, 1, [])
58✔
719
    end
720
  end
721

722
  @doc false
723
  def decoding_types([type | types]) do
553✔
724
    [decoding_type(type) | decoding_types(types)]
725
  end
726

727
  def decoding_types([] = done), do: done
508✔
728

729
  defp decoding_types_reverse(types), do: decoding_types_reverse(types, [])
3,011✔
730

731
  defp decoding_types_reverse([type | types], acc) do
732
    decoding_types_reverse(types, [decoding_type(type) | acc])
16,459✔
733
  end
734

735
  defp decoding_types_reverse([], acc), do: acc
3,011✔
736

737
  defp decoding_type(t) when is_binary(t) do
738
    decoding_type(Ch.Types.decode(t))
16,776✔
739
  end
740

741
  defp decoding_type(t)
742
       when t in [
743
              :string,
744
              :json,
745
              :dynamic,
746
              :boolean,
747
              :uuid,
748
              :date,
749
              :date32,
750
              :time,
751
              :time64,
752
              :ipv4,
753
              :ipv6,
754
              :point,
755
              :nothing
756
            ],
757
       do: t
3,211✔
758

759
  defp decoding_type({:datetime, _tz} = t), do: t
14✔
760
  defp decoding_type({:fixed_string, _len} = t), do: t
427✔
761

762
  for size <- [8, 16, 32, 64, 128, 256] do
763
    defp decoding_type(unquote(:"u#{size}") = u), do: u
12,185✔
764
    defp decoding_type(unquote(:"i#{size}") = i), do: i
420✔
765
  end
766

767
  for size <- [32, 64] do
768
    defp decoding_type(unquote(:"f#{size}") = f), do: f
456✔
769
  end
770

771
  defp decoding_type(:datetime = t), do: {t, _tz = nil}
13✔
772

773
  defp decoding_type({:array = a, t}), do: {a, decoding_type(t)}
1,337✔
774

775
  defp decoding_type({:tuple = t, ts}) do
318✔
776
    {t, Enum.map(ts, &decoding_type/1)}
777
  end
778

779
  defp decoding_type({:variant = v, ts}) do
17✔
780
    {v, ts |> Enum.map(&decoding_type/1) |> List.to_tuple()}
781
  end
782

783
  defp decoding_type({:map = m, kt, vt}) do
784
    {m, decoding_type(kt), decoding_type(vt)}
323✔
785
  end
786

787
  defp decoding_type({:nullable = n, t}), do: {n, decoding_type(t)}
545✔
788
  defp decoding_type({:low_cardinality, t}), do: decoding_type(t)
277✔
789

790
  defp decoding_type({:decimal = t, p, s}), do: {t, decimal_size(p), s}
357✔
791
  defp decoding_type({:decimal32, s}), do: {:decimal, 32, s}
1✔
792
  defp decoding_type({:decimal64, s}), do: {:decimal, 64, s}
1✔
793
  defp decoding_type({:decimal128, s}), do: {:decimal, 128, s}
1✔
794
  defp decoding_type({:decimal256, s}), do: {:decimal, 256, s}
1✔
795

796
  defp decoding_type({:datetime64 = t, p}), do: {t, time_unit(p), _tz = nil}
6✔
797
  defp decoding_type({:datetime64 = t, p, tz}), do: {t, time_unit(p), tz}
317✔
798

799
  defp decoding_type({:time64 = t, p}), do: {t, time_unit(p)}
262✔
800

801
  defp decoding_type({e, mappings}) when e in [:enum8, :enum16] do
15✔
802
    {e, Map.new(mappings, fn {k, v} -> {v, k} end)}
30✔
803
  end
804

805
  defp decoding_type({:simple_aggregate_function, _f, t}), do: decoding_type(t)
6✔
806

807
  defp decoding_type(:ring), do: {:array, :point}
1✔
808
  defp decoding_type(:polygon), do: {:array, {:array, :point}}
1✔
809
  defp decoding_type(:multipolygon), do: {:array, {:array, {:array, :point}}}
1✔
810

811
  defp decoding_type(type) do
812
    raise ArgumentError, "unsupported type for decoding: #{inspect(type)}"
1✔
813
  end
814

815
  defp skip_names(<<rest::bytes>>, 0, count), do: decode_types(rest, count, _acc = [])
5✔
816

817
  for {pattern, value} <- varints do
818
    defp skip_names(<<unquote(pattern), _::size(unquote(value))-bytes, rest::bytes>>, left, count) do
819
      skip_names(rest, left - 1, count)
75✔
820
    end
821
  end
822

823
  defp decode_names(<<rest::bytes>>, 0, count, names) do
3,030✔
824
    [:lists.reverse(names) | decode_types(rest, count, _acc = [])]
825
  end
826

827
  for {pattern, value} <- varints do
828
    defp decode_names(
829
           <<unquote(pattern), name::size(unquote(value))-bytes, rest::bytes>>,
830
           left,
831
           count,
832
           acc
833
         ) do
834
      decode_names(rest, left - 1, count, [name | acc])
16,447✔
835
    end
836
  end
837

838
  defp decode_types(<<>>, 0, _types), do: []
25✔
839

840
  defp decode_types(<<rest::bytes>>, 0, types) do
841
    decode_rows!(rest, decoding_types_reverse(types))
3,010✔
842
  end
843

844
  for {pattern, value} <- varints do
845
    defp decode_types(
846
           <<unquote(pattern), type::size(unquote(value))-bytes, rest::bytes>>,
847
           count,
848
           acc
849
         ) do
850
      decode_types(rest, count - 1, [type | acc])
16,522✔
851
    end
852
  end
853

854
  # Container state stays as separate function arguments at runtime. These macros only keep
855
  # decoder clauses from repeating that plumbing for every successful or partial value.
856
  defmacrop decode_next(types_rest, bin, row) do
857
    quote do
858
      decode_rows(
859
        unquote(types_rest),
860
        unquote(bin),
861
        unquote(row),
862
        var!(rows),
863
        var!(types),
864
        var!(kind),
865
        var!(schema),
866
        var!(left),
867
        var!(outer)
868
      )
869
    end
870
  end
871

872
  defmacrop decode_more(bin, types_rest, row) do
873
    quote do
874
      to_be_continued(
875
        var!(rows),
876
        unquote(bin),
877
        unquote(types_rest),
878
        unquote(row),
879
        var!(types),
880
        var!(kind),
881
        var!(schema),
882
        var!(left),
883
        var!(outer)
884
      )
885
    end
886
  end
887

888
  defmacrop dynamic_next(rest, dynamic, types_rest, row) do
889
    quote do
890
      decode_dynamic_continue(
891
        unquote(rest),
892
        unquote(dynamic),
893
        unquote(types_rest),
894
        unquote(row),
895
        var!(rows),
896
        var!(types),
897
        var!(kind),
898
        var!(schema),
899
        var!(left),
900
        var!(outer)
901
      )
902
    end
903
  end
904

905
  defmacrop dynamic_decode(rest, dynamic, types_rest, row) do
906
    quote do
907
      decode_dynamic(
908
        unquote(rest),
909
        unquote(dynamic),
910
        unquote(types_rest),
911
        unquote(row),
912
        var!(rows),
913
        var!(types),
914
        var!(kind),
915
        var!(schema),
916
        var!(left),
917
        var!(outer)
918
      )
919
    end
920
  end
921

922
  @compile inline: [decode_string_decode_rows: 9]
923

924
  for {pattern, size} <- varints do
925
    defp decode_string_decode_rows(
926
           <<unquote(pattern), s::size(unquote(size))-bytes, bin::bytes>>,
927
           types_rest,
928
           row,
929
           rows,
930
           types,
931
           kind,
932
           schema,
933
           left,
934
           outer
935
         ) do
936
      decode_next(types_rest, bin, [s | row])
8,987✔
937
    end
938
  end
939

940
  defp decode_string_decode_rows(
941
         <<bin::bytes>>,
942
         types_rest,
943
         row,
944
         rows,
945
         types,
946
         kind,
947
         schema,
948
         left,
949
         outer
950
       ) do
951
    decode_more(bin, [:string | types_rest], row)
200,144✔
952
  end
953

954
  @compile inline: [decode_string_json_decode_rows: 9]
955

956
  for {pattern, size} <- varints do
957
    defp decode_string_json_decode_rows(
958
           <<unquote(pattern), s::size(unquote(size))-bytes, bin::bytes>>,
959
           types_rest,
960
           row,
961
           rows,
962
           types,
963
           kind,
964
           schema,
965
           left,
966
           outer
967
         ) do
968
      decode_next(types_rest, bin, [JSON.decode!(s) | row])
43✔
969
    end
970
  end
971

972
  defp decode_string_json_decode_rows(
973
         <<bin::bytes>>,
974
         types_rest,
975
         row,
976
         rows,
977
         types,
978
         kind,
979
         schema,
980
         left,
981
         outer
982
       ) do
983
    decode_more(bin, [:json | types_rest], row)
46✔
984
  end
985

986
  @compile inline: [decode_array_decode_rows: 10]
987
  defp decode_array_decode_rows(
988
         <<0, bin::bytes>>,
989
         _type,
990
         types_rest,
991
         row,
992
         rows,
993
         types,
994
         kind,
995
         schema,
996
         left,
997
         outer
998
       ) do
999
    decode_next(types_rest, bin, [[] | row])
404✔
1000
  end
1001

1002
  for {pattern, size} <- varints do
1003
    defp decode_array_decode_rows(
1004
           <<unquote(pattern), bin::bytes>>,
1005
           type,
1006
           types_rest,
1007
           row,
1008
           rows,
1009
           types,
1010
           kind,
1011
           schema,
1012
           left,
1013
           outer
1014
         ) do
1015
      array_schema = [type]
2,676✔
1016

1017
      decode_rows(
2,676✔
1018
        array_schema,
1019
        bin,
1020
        [],
1021
        rows,
1022
        types,
1023
        :array,
1024
        array_schema,
1025
        unquote(size),
1026
        [{types_rest, row, kind, schema, left} | outer]
1027
      )
1028
    end
1029
  end
1030

1031
  defp decode_array_decode_rows(
1032
         <<bin::bytes>>,
1033
         type,
1034
         types_rest,
1035
         row,
1036
         rows,
1037
         types,
1038
         kind,
1039
         schema,
1040
         left,
1041
         outer
1042
       ) do
1043
    decode_more(bin, [{:array, type} | types_rest], row)
12✔
1044
  end
1045

1046
  @compile inline: [decode_map_decode_rows: 11]
1047
  defp decode_map_decode_rows(
1048
         <<0, bin::bytes>>,
1049
         _key_type,
1050
         _value_type,
1051
         types_rest,
1052
         row,
1053
         rows,
1054
         types,
1055
         kind,
1056
         schema,
1057
         left,
1058
         outer
1059
       ) do
1060
    decode_next(types_rest, bin, [%{} | row])
36✔
1061
  end
1062

1063
  for {pattern, size} <- varints do
1064
    defp decode_map_decode_rows(
1065
           <<unquote(pattern), bin::bytes>>,
1066
           key_type,
1067
           value_type,
1068
           types_rest,
1069
           row,
1070
           rows,
1071
           types,
1072
           kind,
1073
           schema,
1074
           left,
1075
           outer
1076
         ) do
1077
      map_schema = [key_type, value_type]
292✔
1078

1079
      decode_rows(
292✔
1080
        map_schema,
1081
        bin,
1082
        [],
1083
        rows,
1084
        types,
1085
        {:map, %{}},
1086
        map_schema,
1087
        unquote(size),
1088
        [{types_rest, row, kind, schema, left} | outer]
1089
      )
1090
    end
1091
  end
1092

1093
  defp decode_map_decode_rows(
1094
         <<bin::bytes>>,
1095
         key_type,
1096
         value_type,
1097
         types_rest,
1098
         row,
1099
         rows,
1100
         types,
1101
         kind,
1102
         schema,
1103
         left,
1104
         outer
1105
       ) do
1106
    decode_more(bin, [{:map, key_type, value_type} | types_rest], row)
6✔
1107
  end
1108

1109
  # https://clickhouse.com/docs/sql-reference/data-types/data-types-binary-encoding
1110
  dynamic_types = [
1111
    nothing: 0x00,
1112
    u8: 0x01,
1113
    u16: 0x02,
1114
    u32: 0x03,
1115
    u64: 0x04,
1116
    u128: 0x05,
1117
    u256: 0x06,
1118
    i8: 0x07,
1119
    i16: 0x08,
1120
    i32: 0x09,
1121
    i64: 0x0A,
1122
    i128: 0x0B,
1123
    i256: 0x0C,
1124
    f32: 0x0D,
1125
    f64: 0x0E,
1126
    date: 0x0F,
1127
    date32: 0x10,
1128
    string: 0x15,
1129
    uuid: 0x1D,
1130
    ipv4: 0x28,
1131
    ipv6: 0x29,
1132
    boolean: 0x2D
1133
  ]
1134

1135
  # TODO compile inline?
1136

1137
  for {type, code} <- dynamic_types do
1138
    defp decode_dynamic(
1139
           <<unquote(code), rest::bytes>>,
1140
           dynamic,
1141
           types_rest,
1142
           row,
1143
           rows,
1144
           types,
1145
           kind,
1146
           schema,
1147
           left,
1148
           outer
1149
         ) do
1150
      dynamic_next(rest, [unquote(type) | dynamic], types_rest, row)
120✔
1151
    end
1152
  end
1153

1154
  # DateTime 0x11
1155
  defp decode_dynamic(
1156
         <<0x11, rest::bytes>>,
1157
         dynamic,
1158
         types_rest,
1159
         row,
1160
         rows,
1161
         types,
1162
         kind,
1163
         schema,
1164
         left,
1165
         outer
1166
       ) do
1167
    dynamic_next(rest, [{:datetime, nil} | dynamic], types_rest, row)
2✔
1168
  end
1169

1170
  # DateTime(time_zone) 0x12 <var_uint_time_zone_name_size><time_zone_name_data>
1171
  for {pattern, size} <- varints do
1172
    defp decode_dynamic(
1173
           <<0x12, unquote(pattern), tz::size(unquote(size))-bytes, rest::bytes>>,
1174
           dynamic,
1175
           types_rest,
1176
           row,
1177
           rows,
1178
           types,
1179
           kind,
1180
           schema,
1181
           left,
1182
           outer
1183
         ) do
1184
      dynamic_next(rest, [{:datetime, tz} | dynamic], types_rest, row)
1✔
1185
    end
1186
  end
1187

1188
  # DateTime64(P) 0x13 <uint8_precision>
1189
  defp decode_dynamic(
1190
         <<0x13, precision, rest::bytes>>,
1191
         dynamic,
1192
         types_rest,
1193
         row,
1194
         rows,
1195
         types,
1196
         kind,
1197
         schema,
1198
         left,
1199
         outer
1200
       ) do
1201
    dynamic_next(
1✔
1202
      rest,
1203
      [decoding_type({:datetime64, precision}) | dynamic],
1204
      types_rest,
1205
      row
1206
    )
1207
  end
1208

1209
  # DateTime64(P, time_zone) 0x14 <uint8_precision><var_uint_time_zone_name_size><time_zone_name_data>
1210
  for {pattern, size} <- varints do
1211
    defp decode_dynamic(
1212
           <<0x14, precision, unquote(pattern), tz::size(unquote(size))-bytes, rest::bytes>>,
1213
           dynamic,
1214
           types_rest,
1215
           row,
1216
           rows,
1217
           types,
1218
           kind,
1219
           schema,
1220
           left,
1221
           outer
1222
         ) do
1223
      dynamic_next(
1✔
1224
        rest,
1225
        [decoding_type({:datetime64, precision, tz}) | dynamic],
1226
        types_rest,
1227
        row
1228
      )
1229
    end
1230
  end
1231

1232
  # FixedString(N) 0x16 <var_uint_size>
1233
  for {pattern, size} <- varints do
1234
    defp decode_dynamic(
1235
           <<0x16, unquote(pattern), rest::bytes>>,
1236
           dynamic,
1237
           types_rest,
1238
           row,
1239
           rows,
1240
           types,
1241
           kind,
1242
           schema,
1243
           left,
1244
           outer
1245
         ) do
1246
      decode_dynamic_continue(
2✔
1247
        rest,
1248
        [{:fixed_string, unquote(size)} | dynamic],
1249
        types_rest,
1250
        row,
1251
        rows,
1252
        types,
1253
        kind,
1254
        schema,
1255
        left,
1256
        outer
1257
      )
1258
    end
1259
  end
1260

1261
  # Decimal32(P, S) 0x19 <uint8_precision><uint8_scale>
1262
  # Decimal64(P, S) 0x1A <uint8_precision><uint8_scale>
1263
  # Decimal128(P, S) 0x1B <uint8_precision><uint8_scale>
1264
  # Decimal256(P, S) 0x1C <uint8_precision><uint8_scale>
1265
  for {code, size} <- [{0x19, 32}, {0x1A, 64}, {0x1B, 128}, {0x1C, 256}] do
1266
    defp decode_dynamic(
1267
           <<unquote(code), _precision, scale, rest::bytes>>,
1268
           dynamic,
1269
           types_rest,
1270
           row,
1271
           rows,
1272
           types,
1273
           kind,
1274
           schema,
1275
           left,
1276
           outer
1277
         ) do
1278
      dynamic_next(rest, [{:decimal, unquote(size), scale} | dynamic], types_rest, row)
4✔
1279
    end
1280
  end
1281

1282
  # Array(T) 0x1E <nested_type_encoding>
1283
  # Nullable(T) 0x23 <nested_type_encoding>
1284
  # LowCardinality(T) 0x26 <nested_type_encoding>
1285
  for {code, wrapper} <- [{0x1E, :array}, {0x23, :nullable}, {0x26, :low_cardinality}] do
1286
    defp decode_dynamic(
1287
           <<unquote(code), rest::bytes>>,
1288
           dynamic,
1289
           types_rest,
1290
           row,
1291
           rows,
1292
           types,
1293
           kind,
1294
           schema,
1295
           left,
1296
           outer
1297
         ) do
1298
      dynamic_next(rest, [unquote(wrapper) | dynamic], types_rest, row)
36✔
1299
    end
1300
  end
1301

1302
  # TODO
1303
  # Enum8        0x17 <var_uint_number_of_elements><var_uint_name_size_1><name_data_1><int8_value_1>...<var_uint_name_size_N><name_data_N><int8_value_N>
1304
  # Enum16        0x18 <var_uint_number_of_elements><var_uint_name_size_1><name_data_1><int16_little_endian_value_1>...><var_uint_name_size_N><name_data_N><int16_little_endian_value_N>
1305
  # Tuple(T1, ..., TN)        0x1F <var_uint_number_of_elements><nested_type_encoding_1>...<nested_type_encoding_N>
1306
  # Tuple(name1 T1, ..., nameN TN)        0x20 <var_uint_number_of_elements><var_uint_name_size_1><name_data_1><nested_type_encoding_1>...<var_uint_name_size_N><name_data_N><nested_type_encoding_N>
1307
  # Set        0x21
1308
  # Interval        0x22 <interval_kind> (see interval kind binary encoding)
1309
  # Function        0x24<var_uint_number_of_arguments><argument_type_encoding_1>...<argument_type_encoding_N><return_type_encoding>
1310
  # AggregateFunction(function_name(param_1, ..., param_N), arg_T1, ..., arg_TN)        0x25<var_uint_version><var_uint_function_name_size><function_name_data><var_uint_number_of_parameters><param_1>...<param_N><var_uint_number_of_arguments><argument_type_encoding_1>...<argument_type_encoding_N> (see aggregate function parameter binary encoding)
1311
  # Map(K, V)        0x27<key_type_encoding><value_type_encoding>
1312
  # Variant(T1, ..., TN)        0x2A<var_uint_number_of_variants><variant_type_encoding_1>...<variant_type_encoding_N>
1313
  # Dynamic(max_types=N)        0x2B<uint8_max_types>
1314
  # Custom type (Ring, Polygon, etc)        0x2C<var_uint_type_name_size><type_name_data>
1315
  # SimpleAggregateFunction(function_name(param_1, ..., param_N), arg_T1, ..., arg_TN)        0x2E<var_uint_function_name_size><function_name_data><var_uint_number_of_parameters><param_1>...<param_N><var_uint_number_of_arguments><argument_type_encoding_1>...<argument_type_encoding_N> (see aggregate function parameter binary encoding)
1316
  # Nested(name1 T1, ..., nameN TN)        0x2F<var_uint_number_of_elements><var_uint_name_size_1><name_data_1><nested_type_encoding_1>...<var_uint_name_size_N><name_data_N><nested_type_encoding_N>
1317
  # JSON(max_dynamic_paths=N, max_dynamic_types=M, path Type, SKIP skip_path, SKIP REGEXP skip_path_regexp)        0x30<uint8_serialization_version><var_int_max_dynamic_paths><uint8_max_dynamic_types><var_uint_number_of_typed_paths><var_uint_path_name_size_1><path_name_data_1><encoded_type_1>...<var_uint_number_of_skip_paths><var_uint_skip_path_size_1><skip_path_data_1>...<var_uint_number_of_skip_path_regexps><var_uint_skip_path_regexp_size_1><skip_path_data_regexp_1>...
1318

1319
  unsupported_dynamic_types = %{
1320
    "Enum8" => 0x17,
1321
    "Enum16" => 0x18,
1322
    "Tuple" => 0x1F,
1323
    "TupleWithNames" => 0x20,
1324
    "Set" => 0x21,
1325
    "Interval" => 0x22,
1326
    "Function" => 0x24,
1327
    "AggregateFunction" => 0x25,
1328
    "Map" => 0x27,
1329
    "Variant" => 0x2A,
1330
    "Dynamic" => 0x2B,
1331
    "CustomType" => 0x2C,
1332
    "SimpleAggregateFunction" => 0x2E,
1333
    "Nested" => 0x2F,
1334
    "JSON" => 0x30
1335
  }
1336

1337
  for {type, code} <- unsupported_dynamic_types do
1338
    defp decode_dynamic(
1339
           <<unquote(code), _::bytes>>,
1340
           _dynamic,
1341
           _types_rest,
1342
           _row,
1343
           _rows,
1344
           _types,
1345
           _kind,
1346
           _schema,
1347
           _left,
1348
           _outer
1349
         ) do
1350
      raise ArgumentError, "unsupported dynamic type #{unquote(type)}"
9✔
1351
    end
1352
  end
1353

1354
  defp decode_dynamic(
1355
         <<bin::bytes>>,
1356
         dynamic,
1357
         types_rest,
1358
         row,
1359
         rows,
1360
         types,
1361
         kind,
1362
         schema,
1363
         left,
1364
         outer
1365
       ) do
1366
    decode_more(bin, [{:dynamic, dynamic} | types_rest], row)
2✔
1367
  end
1368

1369
  @compile inline: [decode_dynamic_continue: 10]
1370

1371
  defp decode_dynamic_continue(
1372
         <<rest::bytes>>,
1373
         dynamic,
1374
         types_rest,
1375
         row,
1376
         rows,
1377
         types,
1378
         kind,
1379
         schema,
1380
         left,
1381
         outer
1382
       ) do
1383
    continue? =
103✔
1384
      case dynamic do
1385
        [:array | _] -> true
29✔
1386
        [:nullable | _] -> true
5✔
1387
        [:low_cardinality | _] -> true
2✔
1388
        _ -> false
103✔
1389
      end
1390

1391
    if continue? do
103✔
1392
      dynamic_decode(rest, dynamic, types_rest, row)
36✔
1393
    else
1394
      type = build_dynamic_type(:lists.reverse(dynamic))
103✔
1395
      decode_next([type | types_rest], rest, row)
103✔
1396
    end
1397
  end
1398

1399
  defp build_dynamic_type([type]), do: type
131✔
1400

1401
  defp build_dynamic_type(type) do
1402
    case type do
32✔
1403
      [:array | rest] -> {:array, build_dynamic_type(rest)}
25✔
1404
      [:nullable | rest] -> {:nullable, build_dynamic_type(rest)}
5✔
1405
      [:low_cardinality | rest] -> build_dynamic_type(rest)
2✔
1406
    end
1407
  end
1408

1409
  simple_types = %{
1410
    u8: %{pattern: quote(do: <<u>>), value: quote(do: u)},
1411
    u16: %{pattern: quote(do: <<u::16-little>>), value: quote(do: u)},
1412
    u32: %{pattern: quote(do: <<u::32-little>>), value: quote(do: u)},
1413
    u64: %{pattern: quote(do: <<u::64-little>>), value: quote(do: u)},
1414
    u128: %{pattern: quote(do: <<u::128-little>>), value: quote(do: u)},
1415
    u256: %{pattern: quote(do: <<u::256-little>>), value: quote(do: u)},
1416
    i8: %{pattern: quote(do: <<i::signed>>), value: quote(do: i)},
1417
    i16: %{pattern: quote(do: <<i::16-little-signed>>), value: quote(do: i)},
1418
    i32: %{pattern: quote(do: <<i::32-little-signed>>), value: quote(do: i)},
1419
    i64: %{pattern: quote(do: <<i::64-little-signed>>), value: quote(do: i)},
1420
    i128: %{pattern: quote(do: <<i::128-little-signed>>), value: quote(do: i)},
1421
    i256: %{pattern: quote(do: <<i::256-little-signed>>), value: quote(do: i)},
1422
    f32: [
1423
      %{pattern: quote(do: <<f::32-little-float>>), value: quote(do: f)},
1424
      %{pattern: quote(do: <<_nan_or_inf::32>>), value: quote(do: nil)}
1425
    ],
1426
    f64: [
1427
      %{pattern: quote(do: <<f::64-little-float>>), value: quote(do: f)},
1428
      %{pattern: quote(do: <<_nan_or_inf::64>>), value: quote(do: nil)}
1429
    ],
1430
    uuid: %{
1431
      pattern: quote(do: <<u1::64-little, u2::64-little>>),
1432
      value: quote(do: <<u1::64, u2::64>>)
1433
    },
1434
    date: %{
1435
      pattern: quote(do: <<d::16-little>>),
1436
      value: quote(do: Date.from_gregorian_days(d + @epoch_gregorian_days))
1437
    },
1438
    date32: %{
1439
      pattern: quote(do: <<d::32-little-signed>>),
1440
      value: quote(do: Date.from_gregorian_days(d + @epoch_gregorian_days))
1441
    },
1442
    time: %{
1443
      pattern: quote(do: <<s::32-little-signed>>),
1444
      value: quote(do: time_after_midnight(s, 1))
1445
    },
1446
    boolean: [
1447
      %{pattern: quote(do: <<0>>), value: quote(do: false)},
1448
      %{pattern: quote(do: <<1>>), value: quote(do: true)},
1449
      %{pattern: quote(do: <<b>>), value: quote(do: raise("invalid boolean value: #{b}"))}
1450
    ],
1451
    ipv4: %{
1452
      pattern: quote(do: <<b4, b3, b2, b1>>),
1453
      value: quote(do: {b1, b2, b3, b4})
1454
    },
1455
    ipv6: %{
1456
      pattern: quote(do: <<b1::16, b2::16, b3::16, b4::16, b5::16, b6::16, b7::16, b8::16>>),
1457
      value: quote(do: {b1, b2, b3, b4, b5, b6, b7, b8})
1458
    },
1459
    point: %{
1460
      pattern: quote(do: <<x::64-little-float, y::64-little-float>>),
1461
      value: quote(do: {x, y})
1462
    }
1463
  }
1464

1465
  for {type, clauses} <- simple_types do
1466
    fun = :"decode_#{type}_decode_rows"
1467
    @compile inline: [{fun, 9}]
1468

1469
    for %{pattern: pattern, value: value} <- List.wrap(clauses) do
1470
      defp unquote(fun)(
1471
             <<unquote(pattern), rest::bytes>>,
1472
             types_rest,
1473
             row,
1474
             rows,
1475
             types,
1476
             kind,
1477
             schema,
1478
             left,
1479
             outer
1480
           ) do
1481
        decode_rows(
2,021,622✔
1482
          types_rest,
1483
          rest,
1484
          [unquote(value) | row],
1485
          rows,
1486
          types,
1487
          kind,
1488
          schema,
1489
          left,
1490
          outer
1491
        )
1492
      end
1493
    end
1494

1495
    defp unquote(fun)(
1496
           <<bin::bytes>>,
1497
           types_rest,
1498
           row,
1499
           rows,
1500
           types,
1501
           kind,
1502
           schema,
1503
           left,
1504
           outer
1505
         ) do
1506
      decode_more(bin, [unquote(type) | types_rest], row)
583✔
1507
    end
1508
  end
1509

1510
  # The active sequence is carried in kind/schema/left. Only nested parents are stored in outer,
1511
  # so decoding another collection item does not allocate a replacement continuation frame.
1512
  defp decode_rows(
1513
         [type | types_rest],
1514
         <<bin::bytes>>,
1515
         row,
1516
         rows,
1517
         types,
1518
         kind,
1519
         schema,
1520
         left,
1521
         outer
1522
       ) do
1523
    case type do
2,243,933✔
1524
      :u8 ->
1525
        decode_u8_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
5,980✔
1526

1527
      :u16 ->
1528
        decode_u16_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
10,299✔
1529

1530
      :u32 ->
1531
        decode_u32_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
112✔
1532

1533
      :u64 ->
1534
        decode_u64_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
2,000,226✔
1535

1536
      :u128 ->
1537
        decode_u128_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
70✔
1538

1539
      :u256 ->
1540
        decode_u256_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
109✔
1541

1542
      :i8 ->
1543
        decode_i8_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
177✔
1544

1545
      :i16 ->
1546
        decode_i16_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
192✔
1547

1548
      :i32 ->
1549
        decode_i32_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
78✔
1550

1551
      :i64 ->
1552
        decode_i64_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
135✔
1553

1554
      :i128 ->
1555
        decode_i128_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
89✔
1556

1557
      :i256 ->
1558
        decode_i256_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
115✔
1559

1560
      :f32 ->
1561
        decode_f32_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
861✔
1562

1563
      :f64 ->
1564
        decode_f64_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
887✔
1565

1566
      :string ->
1567
        decode_string_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
209,131✔
1568

1569
      :json ->
1570
        # assuming it arrives as text and not "native" binary JSON
1571
        # i.e. assumes `settings: [output_format_binary_write_json_as_string: 1]`
1572
        # TODO
1573
        decode_string_json_decode_rows(
89✔
1574
          bin,
1575
          types_rest,
1576
          row,
1577
          rows,
1578
          types,
1579
          kind,
1580
          schema,
1581
          left,
1582
          outer
1583
        )
1584

1585
      :dynamic ->
1586
        dynamic_decode(bin, _dynamic = [], types_rest, row)
140✔
1587

1588
      {:dynamic, dynamic} ->
1589
        dynamic_decode(bin, dynamic, types_rest, row)
2✔
1590

1591
      {:fixed_string, size} ->
1592
        case bin do
4,301✔
1593
          <<s::size(^size)-bytes, rest::bytes>> ->
1594
            decode_next(types_rest, rest, [s | row])
4,287✔
1595

1596
          _ ->
1597
            decode_more(bin, [type | types_rest], row)
14✔
1598
        end
1599

1600
      :boolean ->
1601
        decode_boolean_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
2,048✔
1602

1603
      :uuid ->
1604
        decode_uuid_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
197✔
1605

1606
      :date ->
1607
        decode_date_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
123✔
1608

1609
      :date32 ->
1610
        decode_date32_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
47✔
1611

1612
      :time ->
1613
        decode_time_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
230✔
1614

1615
      {:time64, time_unit} ->
1616
        case bin do
300✔
1617
          <<ticks::64-little-signed, bin::bytes>> ->
1618
            time = time_after_midnight(ticks, time_unit)
270✔
1619
            decode_next(types_rest, bin, [time | row])
265✔
1620

1621
          _ ->
1622
            decode_more(bin, [type | types_rest], row)
30✔
1623
        end
1624

1625
      {:datetime, timezone} ->
1626
        case bin do
53✔
1627
          <<s::32-little, bin::bytes>> ->
1628
            dt = DateTime.from_unix!(s)
31✔
1629

1630
            dt =
31✔
1631
              case timezone do
1632
                nil -> DateTime.to_naive(dt)
16✔
1633
                "UTC" -> dt
9✔
1634
                _ -> DateTime.shift_zone!(dt, timezone)
6✔
1635
              end
1636

1637
            decode_next(types_rest, bin, [dt | row])
31✔
1638

1639
          _ ->
1640
            decode_more(bin, [type | types_rest], row)
22✔
1641
        end
1642

1643
      {:decimal, size, scale} ->
1644
        case bin do
498✔
1645
          <<val::size(^size)-little-signed, bin::bytes>> ->
1646
            sign = if val < 0, do: -1, else: 1
380✔
1647
            d = Decimal.new(sign, abs(val), -scale)
380✔
1648
            decode_next(types_rest, bin, [d | row])
380✔
1649

1650
          _ ->
1651
            decode_more(bin, [type | types_rest], row)
118✔
1652
        end
1653

1654
      {:nullable, inner_type} ->
1655
        case bin do
2,690✔
1656
          <<b, bin::bytes>> ->
1657
            case b do
2,687✔
1658
              0 ->
1659
                decode_next([inner_type | types_rest], bin, row)
1,317✔
1660

1661
              1 ->
1662
                decode_next(types_rest, bin, [nil | row])
1,370✔
1663
            end
1664

1665
          _ ->
1666
            decode_more(bin, [type | types_rest], row)
3✔
1667
        end
1668

1669
      :nothing ->
1670
        decode_next(types_rest, bin, [nil | row])
27✔
1671

1672
      {:array, inner_type} ->
1673
        decode_array_decode_rows(
3,092✔
1674
          bin,
1675
          inner_type,
1676
          types_rest,
1677
          row,
1678
          rows,
1679
          types,
1680
          kind,
1681
          schema,
1682
          left,
1683
          outer
1684
        )
1685

1686
      {:map, key_type, value_type} ->
1687
        decode_map_decode_rows(
334✔
1688
          bin,
1689
          key_type,
1690
          value_type,
1691
          types_rest,
1692
          row,
1693
          rows,
1694
          types,
1695
          kind,
1696
          schema,
1697
          left,
1698
          outer
1699
        )
1700

1701
      {:tuple, []} ->
NEW
1702
        decode_next(types_rest, bin, [{} | row])
×
1703

1704
      {:tuple, tuple_types} ->
1705
        decode_rows(
332✔
1706
          tuple_types,
1707
          bin,
1708
          [],
1709
          rows,
1710
          types,
1711
          :tuple,
1712
          tuple_types,
1713
          1,
1714
          [{types_rest, row, kind, schema, left} | outer]
1715
        )
1716

1717
      {:variant, variant_types} ->
1718
        case bin do
35✔
1719
          <<255, bin::bytes>> ->
1720
            # 255 is the variant type index for "nothing"
1721
            decode_next(types_rest, bin, [nil | row])
7✔
1722

1723
          # TODO varint?
1724
          <<variant_type_index::8, bin::bytes>>
1725
          when variant_type_index < tuple_size(variant_types) ->
1726
            variant_type = elem(variant_types, variant_type_index)
24✔
1727

1728
            decode_next([variant_type | types_rest], bin, row)
24✔
1729

1730
          <<variant_type_index::8, _bin::bytes>> ->
1731
            raise ArgumentError, "invalid Variant type index: #{variant_type_index}"
1✔
1732

1733
          _ ->
1734
            decode_more(bin, [type | types_rest], row)
3✔
1735
        end
1736

1737
      {:datetime64, time_unit, timezone} ->
1738
        case bin do
665✔
1739
          <<s::64-little-signed, bin::bytes>> ->
1740
            dt = DateTime.from_unix!(s, time_unit)
603✔
1741

1742
            dt =
603✔
1743
              case timezone do
1744
                nil -> DateTime.to_naive(dt)
7✔
1745
                "UTC" -> dt
590✔
1746
                _ -> DateTime.shift_zone!(dt, timezone)
6✔
1747
              end
1748

1749
            decode_next(types_rest, bin, [dt | row])
603✔
1750

1751
          _ ->
1752
            decode_more(bin, [type | types_rest], row)
62✔
1753
        end
1754

1755
      {:enum8, mapping} ->
1756
        case bin do
33✔
1757
          <<v::signed, bin::bytes>> ->
1758
            decode_next(types_rest, bin, [Map.fetch!(mapping, v) | row])
32✔
1759

1760
          _ ->
1761
            decode_more(bin, [type | types_rest], row)
1✔
1762
        end
1763

1764
      {:enum16, mapping} ->
1765
        case bin do
6✔
1766
          <<v::16-little-signed, bin::bytes>> ->
1767
            decode_next(types_rest, bin, [Map.fetch!(mapping, v) | row])
2✔
1768

1769
          _ ->
1770
            decode_more(bin, [type | types_rest], row)
4✔
1771
        end
1772

1773
      :ipv4 ->
1774
        decode_ipv4_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
33✔
1775

1776
      :ipv6 ->
1777
        decode_ipv6_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
89✔
1778

1779
      :point ->
1780
        decode_point_decode_rows(bin, types_rest, row, rows, types, kind, schema, left, outer)
108✔
1781
    end
1782
  end
1783

1784
  defp decode_rows([], <<bin::bytes>>, row, rows, types, :array, schema, left, outer)
1785
       when left > 1 do
1786
    decode_rows(schema, bin, row, rows, types, :array, schema, left - 1, outer)
8,351✔
1787
  end
1788

1789
  defp decode_rows([], <<bin::bytes>>, row, rows, types, {:map, map}, schema, left, outer) do
1790
    [value, key] = row
1,168✔
1791
    map = Map.put(map, key, value)
1,168✔
1792

1793
    if left > 1 do
1,168✔
1794
      decode_rows(schema, bin, [], rows, types, {:map, map}, schema, left - 1, outer)
876✔
1795
    else
1796
      restore_parent(map, bin, rows, types, outer)
292✔
1797
    end
1798
  end
1799

1800
  defp decode_rows([], <<bin::bytes>>, row, rows, types, :array, _schema, 1, outer) do
1801
    restore_parent(:lists.reverse(row), bin, rows, types, outer)
2,675✔
1802
  end
1803

1804
  defp decode_rows([], <<bin::bytes>>, row, rows, types, :tuple, _schema, 1, outer) do
1805
    tuple = row |> :lists.reverse() |> List.to_tuple()
332✔
1806
    restore_parent(tuple, bin, rows, types, outer)
332✔
1807
  end
1808

1809
  defp decode_rows([], <<>> = empty, row, rows, _types, :row, _schema, 1, []) do
1810
    rows = :lists.reverse([:lists.reverse(row) | rows])
3,481✔
1811
    {rows, empty, _no_state = nil}
3,481✔
1812
  end
1813

1814
  defp decode_rows([], <<bin::bytes>>, row, rows, types, :row, schema, 1, []) do
1815
    row = :lists.reverse(row)
2,004,011✔
1816
    decode_rows(schema, bin, [], [row | rows], types, :row, schema, 1, [])
2,004,011✔
1817
  end
1818

1819
  @compile inline: [restore_parent: 5]
1820
  defp restore_parent(
1821
         value,
1822
         bin,
1823
         rows,
1824
         types,
1825
         [{parent_types, parent_row, parent_kind, parent_schema, parent_left} | outer]
1826
       ) do
1827
    decode_rows(
3,299✔
1828
      parent_types,
1829
      bin,
1830
      [value | parent_row],
1831
      rows,
1832
      types,
1833
      parent_kind,
1834
      parent_schema,
1835
      parent_left,
1836
      outer
1837
    )
1838
  end
1839

1840
  @compile inline: [to_be_continued: 9]
1841
  defp to_be_continued(rows, bin, types_rest, row, _types, kind, schema, left, outer) do
1842
    state =
200,901✔
1843
      if kind == :row and outer == [] do
201,047✔
1844
        {:cont, types_rest, row}
200,908✔
1845
      else
1846
        {:cont, types_rest, row, kind, schema, left, outer}
139✔
1847
      end
1848

1849
    {:lists.reverse(rows), bin, state}
201,047✔
1850
  end
1851

1852
  @compile inline: [decimal_size: 1]
1853
  # https://clickhouse.com/docs/en/sql-reference/data-types/decimal/
1854
  defp decimal_size(precision) when is_integer(precision) do
1855
    cond do
366✔
1856
      precision >= 39 -> 256
207✔
1857
      precision >= 19 -> 128
158✔
1858
      precision >= 10 -> 64
152✔
1859
      true -> 32
18✔
1860
    end
1861
  end
1862

1863
  @compile inline: [time_unit: 1]
1864
  for precision <- 0..9 do
1865
    time_unit = Integer.pow(10, precision)
1866
    defp time_unit(unquote(precision)), do: unquote(time_unit)
693✔
1867
  end
1868

1869
  @compile inline: [time_after_midnight: 2]
1870
  defp time_after_midnight(ticks, time_unit) do
1871
    if ticks >= 0 and ticks < 86400 * time_unit do
486✔
1872
      ticks |> DateTime.from_unix!(time_unit) |> DateTime.to_time()
478✔
1873
    else
1874
      # since ClickHouse supports Time64 values of [-999:59:59.999999999, 999:59:59.999999999]
1875
      # and Elixir's Time supports values of [00:00:00.000000, 23:59:59.999999]
1876
      # we raise an error when ClickHouse's Time64 value is out of Elixir's Time range
1877
      raise ArgumentError,
8✔
1878
            "ClickHouse Time value #{:erlang.float_to_binary(ticks / time_unit, [:short])} (seconds) is out of Elixir's Time range (00:00:00.000000 - 23:59:59.999999)"
8✔
1879

1880
      # TODO: we could potentially decode ClickHouse's Time/Time64 values as Elixir's Duration when it's out of Elixir's Time range
1881
    end
1882
  end
1883
end
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE TRIAL · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc