tigerbeetle 0.0.40 → 0.17.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (293) hide show
  1. checksums.yaml +4 -4
  2. data/LICENSE +0 -25
  3. data/README.md +670 -80
  4. data/docs/migration.md +201 -0
  5. data/sig/tigerbeetle.rbs +271 -0
  6. data/src/ext/tigerbeetle/extconf.rb +47 -0
  7. data/src/ext/tigerbeetle/lib/aarch64-linux-gnu.2.27/libtb_client.so +0 -0
  8. data/src/ext/tigerbeetle/lib/aarch64-linux-musl/libtb_client.so +0 -0
  9. data/src/ext/tigerbeetle/lib/aarch64-macos/libtb_client.dylib +0 -0
  10. data/src/ext/tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so +0 -0
  11. data/src/ext/tigerbeetle/lib/x86_64-linux-musl/libtb_client.so +0 -0
  12. data/src/ext/tigerbeetle/lib/x86_64-macos/libtb_client.dylib +0 -0
  13. data/src/ext/tigerbeetle/lib/x86_64-windows/tb_client.dll +0 -0
  14. data/src/ext/tigerbeetle/rb_tb_gen.h +458 -0
  15. data/{ext/tb_client/tigerbeetle/src/clients/rust/assets → src/ext/tigerbeetle}/tb_client.h +18 -16
  16. data/src/ext/tigerbeetle/tigerbeetle.c +310 -0
  17. data/src/tigerbeetle/bindings.rb +347 -0
  18. data/src/tigerbeetle/client.rb +129 -0
  19. data/src/tigerbeetle/completion_dispatcher.rb +108 -0
  20. data/src/tigerbeetle/id.rb +40 -0
  21. data/src/tigerbeetle/tb.rb +3 -0
  22. data/src/tigerbeetle/version.rb +3 -0
  23. data/src/tigerbeetle.rb +39 -0
  24. metadata +33 -350
  25. data/CHANGELOG.md +0 -162
  26. data/ext/tb_client/extconf.rb +0 -41
  27. data/ext/tb_client/tigerbeetle/LICENSE +0 -177
  28. data/ext/tb_client/tigerbeetle/build.zig +0 -2296
  29. data/ext/tb_client/tigerbeetle/src/aof.zig +0 -1000
  30. data/ext/tb_client/tigerbeetle/src/build/fetch.zig +0 -112
  31. data/ext/tb_client/tigerbeetle/src/build_multiversion.zig +0 -808
  32. data/ext/tb_client/tigerbeetle/src/cdc/amqp/protocol.zig +0 -1283
  33. data/ext/tb_client/tigerbeetle/src/cdc/amqp/spec.zig +0 -1704
  34. data/ext/tb_client/tigerbeetle/src/cdc/amqp/types.zig +0 -341
  35. data/ext/tb_client/tigerbeetle/src/cdc/amqp.zig +0 -1450
  36. data/ext/tb_client/tigerbeetle/src/cdc/runner.zig +0 -1659
  37. data/ext/tb_client/tigerbeetle/src/clients/c/samples/main.c +0 -406
  38. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client/context.zig +0 -1092
  39. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client/echo_client.zig +0 -286
  40. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client/packet.zig +0 -158
  41. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client/signal.zig +0 -229
  42. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client/signal_fuzz.zig +0 -110
  43. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client.h +0 -386
  44. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client.zig +0 -34
  45. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client_exports.zig +0 -281
  46. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client_header.zig +0 -312
  47. data/ext/tb_client/tigerbeetle/src/clients/c/tb_client_header_test.zig +0 -138
  48. data/ext/tb_client/tigerbeetle/src/clients/c/test.zig +0 -466
  49. data/ext/tb_client/tigerbeetle/src/clients/docs_samples.zig +0 -157
  50. data/ext/tb_client/tigerbeetle/src/clients/docs_types.zig +0 -90
  51. data/ext/tb_client/tigerbeetle/src/clients/dotnet/ci.zig +0 -203
  52. data/ext/tb_client/tigerbeetle/src/clients/dotnet/docs.zig +0 -79
  53. data/ext/tb_client/tigerbeetle/src/clients/dotnet/dotnet_bindings.zig +0 -542
  54. data/ext/tb_client/tigerbeetle/src/clients/go/ci.zig +0 -109
  55. data/ext/tb_client/tigerbeetle/src/clients/go/docs.zig +0 -86
  56. data/ext/tb_client/tigerbeetle/src/clients/go/go_bindings.zig +0 -370
  57. data/ext/tb_client/tigerbeetle/src/clients/go/pkg/native/tb_client.h +0 -386
  58. data/ext/tb_client/tigerbeetle/src/clients/java/ci.zig +0 -167
  59. data/ext/tb_client/tigerbeetle/src/clients/java/docs.zig +0 -126
  60. data/ext/tb_client/tigerbeetle/src/clients/java/java_bindings.zig +0 -996
  61. data/ext/tb_client/tigerbeetle/src/clients/java/src/client.zig +0 -748
  62. data/ext/tb_client/tigerbeetle/src/clients/java/src/jni.zig +0 -3238
  63. data/ext/tb_client/tigerbeetle/src/clients/java/src/jni_tests.zig +0 -1718
  64. data/ext/tb_client/tigerbeetle/src/clients/java/src/jni_thread_cleaner.zig +0 -190
  65. data/ext/tb_client/tigerbeetle/src/clients/node/ci.zig +0 -104
  66. data/ext/tb_client/tigerbeetle/src/clients/node/docs.zig +0 -75
  67. data/ext/tb_client/tigerbeetle/src/clients/node/node.zig +0 -522
  68. data/ext/tb_client/tigerbeetle/src/clients/node/node_bindings.zig +0 -267
  69. data/ext/tb_client/tigerbeetle/src/clients/node/src/c.zig +0 -3
  70. data/ext/tb_client/tigerbeetle/src/clients/node/src/translate.zig +0 -379
  71. data/ext/tb_client/tigerbeetle/src/clients/python/ci.zig +0 -131
  72. data/ext/tb_client/tigerbeetle/src/clients/python/docs.zig +0 -63
  73. data/ext/tb_client/tigerbeetle/src/clients/python/python_bindings.zig +0 -588
  74. data/ext/tb_client/tigerbeetle/src/clients/rust/ci.zig +0 -73
  75. data/ext/tb_client/tigerbeetle/src/clients/rust/docs.zig +0 -106
  76. data/ext/tb_client/tigerbeetle/src/clients/rust/rust_bindings.zig +0 -305
  77. data/ext/tb_client/tigerbeetle/src/config.zig +0 -296
  78. data/ext/tb_client/tigerbeetle/src/constants.zig +0 -790
  79. data/ext/tb_client/tigerbeetle/src/copyhound.zig +0 -202
  80. data/ext/tb_client/tigerbeetle/src/counting_allocator.zig +0 -72
  81. data/ext/tb_client/tigerbeetle/src/direction.zig +0 -120
  82. data/ext/tb_client/tigerbeetle/src/docs_website/build.zig +0 -158
  83. data/ext/tb_client/tigerbeetle/src/docs_website/src/content.zig +0 -156
  84. data/ext/tb_client/tigerbeetle/src/docs_website/src/docs.zig +0 -252
  85. data/ext/tb_client/tigerbeetle/src/docs_website/src/file_checker.zig +0 -313
  86. data/ext/tb_client/tigerbeetle/src/docs_website/src/html.zig +0 -87
  87. data/ext/tb_client/tigerbeetle/src/docs_website/src/page_writer.zig +0 -63
  88. data/ext/tb_client/tigerbeetle/src/docs_website/src/redirects.zig +0 -47
  89. data/ext/tb_client/tigerbeetle/src/docs_website/src/search_index_writer.zig +0 -28
  90. data/ext/tb_client/tigerbeetle/src/docs_website/src/service_worker_writer.zig +0 -61
  91. data/ext/tb_client/tigerbeetle/src/docs_website/src/single_page_writer.zig +0 -169
  92. data/ext/tb_client/tigerbeetle/src/docs_website/src/website.zig +0 -46
  93. data/ext/tb_client/tigerbeetle/src/ewah.zig +0 -445
  94. data/ext/tb_client/tigerbeetle/src/ewah_benchmark.zig +0 -128
  95. data/ext/tb_client/tigerbeetle/src/ewah_fuzz.zig +0 -171
  96. data/ext/tb_client/tigerbeetle/src/fuzz_tests.zig +0 -179
  97. data/ext/tb_client/tigerbeetle/src/integration_tests.zig +0 -662
  98. data/ext/tb_client/tigerbeetle/src/io/common.zig +0 -155
  99. data/ext/tb_client/tigerbeetle/src/io/darwin.zig +0 -1093
  100. data/ext/tb_client/tigerbeetle/src/io/linux.zig +0 -1880
  101. data/ext/tb_client/tigerbeetle/src/io/test.zig +0 -1005
  102. data/ext/tb_client/tigerbeetle/src/io/windows.zig +0 -1598
  103. data/ext/tb_client/tigerbeetle/src/io.zig +0 -34
  104. data/ext/tb_client/tigerbeetle/src/iops.zig +0 -134
  105. data/ext/tb_client/tigerbeetle/src/list.zig +0 -236
  106. data/ext/tb_client/tigerbeetle/src/lsm/binary_search.zig +0 -848
  107. data/ext/tb_client/tigerbeetle/src/lsm/binary_search_benchmark.zig +0 -179
  108. data/ext/tb_client/tigerbeetle/src/lsm/cache_map.zig +0 -424
  109. data/ext/tb_client/tigerbeetle/src/lsm/cache_map_fuzz.zig +0 -420
  110. data/ext/tb_client/tigerbeetle/src/lsm/compaction.zig +0 -2114
  111. data/ext/tb_client/tigerbeetle/src/lsm/composite_key.zig +0 -185
  112. data/ext/tb_client/tigerbeetle/src/lsm/forest.zig +0 -1146
  113. data/ext/tb_client/tigerbeetle/src/lsm/forest_fuzz.zig +0 -1102
  114. data/ext/tb_client/tigerbeetle/src/lsm/forest_table_iterator.zig +0 -200
  115. data/ext/tb_client/tigerbeetle/src/lsm/groove.zig +0 -1495
  116. data/ext/tb_client/tigerbeetle/src/lsm/k_way_merge.zig +0 -739
  117. data/ext/tb_client/tigerbeetle/src/lsm/k_way_merge_benchmark.zig +0 -166
  118. data/ext/tb_client/tigerbeetle/src/lsm/manifest.zig +0 -754
  119. data/ext/tb_client/tigerbeetle/src/lsm/manifest_level.zig +0 -1294
  120. data/ext/tb_client/tigerbeetle/src/lsm/manifest_level_fuzz.zig +0 -510
  121. data/ext/tb_client/tigerbeetle/src/lsm/manifest_log.zig +0 -1241
  122. data/ext/tb_client/tigerbeetle/src/lsm/manifest_log_fuzz.zig +0 -628
  123. data/ext/tb_client/tigerbeetle/src/lsm/node_pool.zig +0 -247
  124. data/ext/tb_client/tigerbeetle/src/lsm/scan_buffer.zig +0 -116
  125. data/ext/tb_client/tigerbeetle/src/lsm/scan_builder.zig +0 -543
  126. data/ext/tb_client/tigerbeetle/src/lsm/scan_fuzz.zig +0 -938
  127. data/ext/tb_client/tigerbeetle/src/lsm/scan_lookup.zig +0 -293
  128. data/ext/tb_client/tigerbeetle/src/lsm/scan_merge.zig +0 -359
  129. data/ext/tb_client/tigerbeetle/src/lsm/scan_range.zig +0 -99
  130. data/ext/tb_client/tigerbeetle/src/lsm/scan_state.zig +0 -17
  131. data/ext/tb_client/tigerbeetle/src/lsm/scan_tree.zig +0 -962
  132. data/ext/tb_client/tigerbeetle/src/lsm/schema.zig +0 -617
  133. data/ext/tb_client/tigerbeetle/src/lsm/scratch_memory.zig +0 -84
  134. data/ext/tb_client/tigerbeetle/src/lsm/segmented_array.zig +0 -1500
  135. data/ext/tb_client/tigerbeetle/src/lsm/segmented_array_benchmark.zig +0 -149
  136. data/ext/tb_client/tigerbeetle/src/lsm/segmented_array_fuzz.zig +0 -7
  137. data/ext/tb_client/tigerbeetle/src/lsm/set_associative_cache.zig +0 -865
  138. data/ext/tb_client/tigerbeetle/src/lsm/table.zig +0 -607
  139. data/ext/tb_client/tigerbeetle/src/lsm/table_memory.zig +0 -843
  140. data/ext/tb_client/tigerbeetle/src/lsm/table_value_iterator.zig +0 -90
  141. data/ext/tb_client/tigerbeetle/src/lsm/timestamp_range.zig +0 -40
  142. data/ext/tb_client/tigerbeetle/src/lsm/tree.zig +0 -629
  143. data/ext/tb_client/tigerbeetle/src/lsm/tree_fuzz.zig +0 -933
  144. data/ext/tb_client/tigerbeetle/src/lsm/zig_zag_merge.zig +0 -534
  145. data/ext/tb_client/tigerbeetle/src/message_buffer.zig +0 -469
  146. data/ext/tb_client/tigerbeetle/src/message_bus.zig +0 -1219
  147. data/ext/tb_client/tigerbeetle/src/message_bus_fuzz.zig +0 -936
  148. data/ext/tb_client/tigerbeetle/src/message_pool.zig +0 -343
  149. data/ext/tb_client/tigerbeetle/src/multiversion.zig +0 -2195
  150. data/ext/tb_client/tigerbeetle/src/queue.zig +0 -390
  151. data/ext/tb_client/tigerbeetle/src/repl/completion.zig +0 -201
  152. data/ext/tb_client/tigerbeetle/src/repl/parser.zig +0 -1356
  153. data/ext/tb_client/tigerbeetle/src/repl/terminal.zig +0 -496
  154. data/ext/tb_client/tigerbeetle/src/repl.zig +0 -1034
  155. data/ext/tb_client/tigerbeetle/src/scripts/amqp.zig +0 -973
  156. data/ext/tb_client/tigerbeetle/src/scripts/cfo.zig +0 -1866
  157. data/ext/tb_client/tigerbeetle/src/scripts/changelog.zig +0 -304
  158. data/ext/tb_client/tigerbeetle/src/scripts/ci.zig +0 -227
  159. data/ext/tb_client/tigerbeetle/src/scripts/client_readmes.zig +0 -658
  160. data/ext/tb_client/tigerbeetle/src/scripts/devhub.zig +0 -466
  161. data/ext/tb_client/tigerbeetle/src/scripts/release.zig +0 -1058
  162. data/ext/tb_client/tigerbeetle/src/scripts.zig +0 -105
  163. data/ext/tb_client/tigerbeetle/src/shell.zig +0 -1195
  164. data/ext/tb_client/tigerbeetle/src/stack.zig +0 -260
  165. data/ext/tb_client/tigerbeetle/src/state_machine/auditor.zig +0 -911
  166. data/ext/tb_client/tigerbeetle/src/state_machine/workload.zig +0 -2079
  167. data/ext/tb_client/tigerbeetle/src/state_machine.zig +0 -4872
  168. data/ext/tb_client/tigerbeetle/src/state_machine_fuzz.zig +0 -288
  169. data/ext/tb_client/tigerbeetle/src/state_machine_tests.zig +0 -3128
  170. data/ext/tb_client/tigerbeetle/src/static_allocator.zig +0 -82
  171. data/ext/tb_client/tigerbeetle/src/stdx/bit_set.zig +0 -157
  172. data/ext/tb_client/tigerbeetle/src/stdx/bounded_array.zig +0 -292
  173. data/ext/tb_client/tigerbeetle/src/stdx/debug.zig +0 -65
  174. data/ext/tb_client/tigerbeetle/src/stdx/flags.zig +0 -1414
  175. data/ext/tb_client/tigerbeetle/src/stdx/huge_page_allocator.zig +0 -115
  176. data/ext/tb_client/tigerbeetle/src/stdx/mlock.zig +0 -92
  177. data/ext/tb_client/tigerbeetle/src/stdx/prng.zig +0 -677
  178. data/ext/tb_client/tigerbeetle/src/stdx/radix.zig +0 -336
  179. data/ext/tb_client/tigerbeetle/src/stdx/ring_buffer.zig +0 -511
  180. data/ext/tb_client/tigerbeetle/src/stdx/sort_test.zig +0 -112
  181. data/ext/tb_client/tigerbeetle/src/stdx/stdx.zig +0 -1163
  182. data/ext/tb_client/tigerbeetle/src/stdx/testing/low_level_hash_vectors.zig +0 -142
  183. data/ext/tb_client/tigerbeetle/src/stdx/testing/snaptest.zig +0 -361
  184. data/ext/tb_client/tigerbeetle/src/stdx/time_units.zig +0 -275
  185. data/ext/tb_client/tigerbeetle/src/stdx/unshare.zig +0 -295
  186. data/ext/tb_client/tigerbeetle/src/stdx/vendored/aegis.zig +0 -436
  187. data/ext/tb_client/tigerbeetle/src/stdx/windows.zig +0 -48
  188. data/ext/tb_client/tigerbeetle/src/stdx/zipfian.zig +0 -402
  189. data/ext/tb_client/tigerbeetle/src/storage.zig +0 -489
  190. data/ext/tb_client/tigerbeetle/src/storage_fuzz.zig +0 -180
  191. data/ext/tb_client/tigerbeetle/src/testing/bench.zig +0 -146
  192. data/ext/tb_client/tigerbeetle/src/testing/cluster/grid_checker.zig +0 -53
  193. data/ext/tb_client/tigerbeetle/src/testing/cluster/journal_checker.zig +0 -61
  194. data/ext/tb_client/tigerbeetle/src/testing/cluster/manifest_checker.zig +0 -76
  195. data/ext/tb_client/tigerbeetle/src/testing/cluster/message_bus.zig +0 -110
  196. data/ext/tb_client/tigerbeetle/src/testing/cluster/network.zig +0 -412
  197. data/ext/tb_client/tigerbeetle/src/testing/cluster/state_checker.zig +0 -331
  198. data/ext/tb_client/tigerbeetle/src/testing/cluster/storage_checker.zig +0 -458
  199. data/ext/tb_client/tigerbeetle/src/testing/cluster.zig +0 -1198
  200. data/ext/tb_client/tigerbeetle/src/testing/exhaustigen.zig +0 -128
  201. data/ext/tb_client/tigerbeetle/src/testing/fixtures.zig +0 -181
  202. data/ext/tb_client/tigerbeetle/src/testing/fuzz.zig +0 -144
  203. data/ext/tb_client/tigerbeetle/src/testing/id.zig +0 -97
  204. data/ext/tb_client/tigerbeetle/src/testing/io.zig +0 -317
  205. data/ext/tb_client/tigerbeetle/src/testing/marks.zig +0 -126
  206. data/ext/tb_client/tigerbeetle/src/testing/packet_simulator.zig +0 -533
  207. data/ext/tb_client/tigerbeetle/src/testing/reply_sequence.zig +0 -154
  208. data/ext/tb_client/tigerbeetle/src/testing/state_machine.zig +0 -389
  209. data/ext/tb_client/tigerbeetle/src/testing/storage.zig +0 -1247
  210. data/ext/tb_client/tigerbeetle/src/testing/table.zig +0 -249
  211. data/ext/tb_client/tigerbeetle/src/testing/time.zig +0 -98
  212. data/ext/tb_client/tigerbeetle/src/testing/tmp_tigerbeetle.zig +0 -212
  213. data/ext/tb_client/tigerbeetle/src/testing/vortex/constants.zig +0 -26
  214. data/ext/tb_client/tigerbeetle/src/testing/vortex/faulty_network.zig +0 -579
  215. data/ext/tb_client/tigerbeetle/src/testing/vortex/java_driver/ci.zig +0 -39
  216. data/ext/tb_client/tigerbeetle/src/testing/vortex/logged_process.zig +0 -214
  217. data/ext/tb_client/tigerbeetle/src/testing/vortex/rust_driver/ci.zig +0 -34
  218. data/ext/tb_client/tigerbeetle/src/testing/vortex/supervisor.zig +0 -785
  219. data/ext/tb_client/tigerbeetle/src/testing/vortex/workload.zig +0 -543
  220. data/ext/tb_client/tigerbeetle/src/testing/vortex/zig_driver.zig +0 -181
  221. data/ext/tb_client/tigerbeetle/src/tidy.zig +0 -1449
  222. data/ext/tb_client/tigerbeetle/src/tigerbeetle/benchmark_driver.zig +0 -227
  223. data/ext/tb_client/tigerbeetle/src/tigerbeetle/benchmark_load.zig +0 -1069
  224. data/ext/tb_client/tigerbeetle/src/tigerbeetle/cli.zig +0 -1422
  225. data/ext/tb_client/tigerbeetle/src/tigerbeetle/inspect.zig +0 -1658
  226. data/ext/tb_client/tigerbeetle/src/tigerbeetle/inspect_integrity.zig +0 -518
  227. data/ext/tb_client/tigerbeetle/src/tigerbeetle/libtb_client.zig +0 -36
  228. data/ext/tb_client/tigerbeetle/src/tigerbeetle/main.zig +0 -646
  229. data/ext/tb_client/tigerbeetle/src/tigerbeetle.zig +0 -958
  230. data/ext/tb_client/tigerbeetle/src/time.zig +0 -236
  231. data/ext/tb_client/tigerbeetle/src/trace/event.zig +0 -745
  232. data/ext/tb_client/tigerbeetle/src/trace/statsd.zig +0 -462
  233. data/ext/tb_client/tigerbeetle/src/trace.zig +0 -556
  234. data/ext/tb_client/tigerbeetle/src/unit_tests.zig +0 -321
  235. data/ext/tb_client/tigerbeetle/src/vopr.zig +0 -1785
  236. data/ext/tb_client/tigerbeetle/src/vortex.zig +0 -101
  237. data/ext/tb_client/tigerbeetle/src/vsr/checkpoint_trailer.zig +0 -473
  238. data/ext/tb_client/tigerbeetle/src/vsr/checksum.zig +0 -208
  239. data/ext/tb_client/tigerbeetle/src/vsr/checksum_benchmark.zig +0 -43
  240. data/ext/tb_client/tigerbeetle/src/vsr/client.zig +0 -768
  241. data/ext/tb_client/tigerbeetle/src/vsr/client_replies.zig +0 -532
  242. data/ext/tb_client/tigerbeetle/src/vsr/client_sessions.zig +0 -338
  243. data/ext/tb_client/tigerbeetle/src/vsr/clock.zig +0 -1019
  244. data/ext/tb_client/tigerbeetle/src/vsr/fault_detector.zig +0 -279
  245. data/ext/tb_client/tigerbeetle/src/vsr/free_set.zig +0 -1381
  246. data/ext/tb_client/tigerbeetle/src/vsr/free_set_fuzz.zig +0 -315
  247. data/ext/tb_client/tigerbeetle/src/vsr/grid.zig +0 -1460
  248. data/ext/tb_client/tigerbeetle/src/vsr/grid_blocks_missing.zig +0 -757
  249. data/ext/tb_client/tigerbeetle/src/vsr/grid_scrubber.zig +0 -797
  250. data/ext/tb_client/tigerbeetle/src/vsr/journal.zig +0 -2586
  251. data/ext/tb_client/tigerbeetle/src/vsr/marzullo.zig +0 -308
  252. data/ext/tb_client/tigerbeetle/src/vsr/message_header.zig +0 -1777
  253. data/ext/tb_client/tigerbeetle/src/vsr/multi_batch.zig +0 -715
  254. data/ext/tb_client/tigerbeetle/src/vsr/multi_batch_fuzz.zig +0 -185
  255. data/ext/tb_client/tigerbeetle/src/vsr/repair_budget.zig +0 -333
  256. data/ext/tb_client/tigerbeetle/src/vsr/replica.zig +0 -12356
  257. data/ext/tb_client/tigerbeetle/src/vsr/replica_format.zig +0 -416
  258. data/ext/tb_client/tigerbeetle/src/vsr/replica_reformat.zig +0 -165
  259. data/ext/tb_client/tigerbeetle/src/vsr/replica_test.zig +0 -2928
  260. data/ext/tb_client/tigerbeetle/src/vsr/routing.zig +0 -1075
  261. data/ext/tb_client/tigerbeetle/src/vsr/superblock.zig +0 -1603
  262. data/ext/tb_client/tigerbeetle/src/vsr/superblock_fuzz.zig +0 -484
  263. data/ext/tb_client/tigerbeetle/src/vsr/superblock_quorums.zig +0 -405
  264. data/ext/tb_client/tigerbeetle/src/vsr/superblock_quorums_fuzz.zig +0 -355
  265. data/ext/tb_client/tigerbeetle/src/vsr/sync.zig +0 -29
  266. data/ext/tb_client/tigerbeetle/src/vsr.zig +0 -1727
  267. data/lib/tb_client/shared_lib.rb +0 -66
  268. data/lib/tb_client.rb +0 -282
  269. data/lib/tigerbeetle/account.rb +0 -38
  270. data/lib/tigerbeetle/account_balance.rb +0 -23
  271. data/lib/tigerbeetle/account_filter.rb +0 -31
  272. data/lib/tigerbeetle/atomic_counter.rb +0 -14
  273. data/lib/tigerbeetle/client.rb +0 -214
  274. data/lib/tigerbeetle/converters/account.rb +0 -63
  275. data/lib/tigerbeetle/converters/account_balance.rb +0 -31
  276. data/lib/tigerbeetle/converters/account_filter.rb +0 -32
  277. data/lib/tigerbeetle/converters/base.rb +0 -35
  278. data/lib/tigerbeetle/converters/create_accounts_result.rb +0 -21
  279. data/lib/tigerbeetle/converters/create_transfers_result.rb +0 -21
  280. data/lib/tigerbeetle/converters/query_filter.rb +0 -33
  281. data/lib/tigerbeetle/converters/time.rb +0 -23
  282. data/lib/tigerbeetle/converters/transfer.rb +0 -64
  283. data/lib/tigerbeetle/converters/uint_128.rb +0 -24
  284. data/lib/tigerbeetle/converters.rb +0 -12
  285. data/lib/tigerbeetle/error.rb +0 -4
  286. data/lib/tigerbeetle/id.rb +0 -30
  287. data/lib/tigerbeetle/platforms.rb +0 -9
  288. data/lib/tigerbeetle/query_filter.rb +0 -31
  289. data/lib/tigerbeetle/request.rb +0 -7
  290. data/lib/tigerbeetle/transfer.rb +0 -40
  291. data/lib/tigerbeetle/version.rb +0 -4
  292. data/lib/tigerbeetle.rb +0 -13
  293. data/tigerbeetle.gemspec +0 -60
@@ -1,1659 +0,0 @@
1
- const std = @import("std");
2
- const log = std.log.scoped(.amqp);
3
-
4
- const vsr = @import("../vsr.zig");
5
- const assert = std.debug.assert;
6
- const maybe = vsr.stdx.maybe;
7
- const fatal = @import("amqp/protocol.zig").fatal;
8
-
9
- const stdx = vsr.stdx;
10
- const tb = vsr.tigerbeetle;
11
- const IO = vsr.io.IO;
12
- const Time = vsr.time.Time;
13
- const MessagePool = vsr.message_pool.MessagePool;
14
- const MessageBus = vsr.message_bus.MessageBusType(IO);
15
- const Operation = vsr.tigerbeetle.Operation;
16
- const Client = vsr.ClientType(Operation, MessageBus);
17
- const TimestampRange = vsr.lsm.TimestampRange;
18
-
19
- pub const amqp = @import("amqp.zig");
20
-
21
- /// CDC processor targeting an AMQP 0.9.1 compliant server (e.g., RabbitMQ).
22
- /// Producer: TigerBeetle `get_change_events` operation.
23
- /// Consumer: AMQP publisher.
24
- /// Both consumer and producer run concurrently using `io_uring`.
25
- /// See `DualBuffer` for more details.
26
- pub const Runner = struct {
27
- const StateRecoveryMode = union(enum) {
28
- recover,
29
- override: u64,
30
- };
31
-
32
- const constants = struct {
33
- const tick_ms = vsr.constants.tick_ms;
34
- const idle_interval_ns: u63 = 1 * std.time.ns_per_s;
35
- const reply_timeout_ticks = @divExact(
36
- 30 * std.time.ms_per_s,
37
- constants.tick_ms,
38
- );
39
- const app_id = "tigerbeetle";
40
- const progress_tracker_queue = "tigerbeetle.internal.progress";
41
- const locker_queue = "tigerbeetle.internal.locker";
42
- const event_count_max: u32 = Operation.get_change_events.result_max(
43
- vsr.constants.message_body_size_max,
44
- );
45
- };
46
-
47
- io: IO,
48
- idle_completion: IO.Completion = undefined,
49
- idle_interval_ns: u63,
50
- event_count_max: u32,
51
-
52
- message_pool: MessagePool,
53
- vsr_client: Client,
54
- buffer: DualBuffer,
55
-
56
- amqp_client: amqp.Client,
57
- publish_exchange: []const u8,
58
- publish_routing_key: []const u8,
59
- progress_tracker_queue: []const u8,
60
- locker_queue: []const u8,
61
-
62
- connected: struct {
63
- /// VSR client registered.
64
- vsr: bool = false,
65
- /// AMQP client connected and ready to publish.
66
- amqp: bool = false,
67
- },
68
- /// The producer is responsible for reading events from TigerBeetle.
69
- producer: enum {
70
- idle,
71
- // Waiting for the rate limit to allow more requests.
72
- rate_limit,
73
- // Calling the VSR client.
74
- request,
75
- /// No events to publish.
76
- /// Waiting for the idle timeout to check for new events.
77
- waiting,
78
- },
79
-
80
- /// The consumer is responsible to publish events on the AMQP server.
81
- consumer: enum {
82
- idle,
83
- publish,
84
- progress_update,
85
- },
86
-
87
- rate_limit: RateLimit,
88
- metrics: Metrics,
89
-
90
- state: union(enum) {
91
- unknown: StateRecoveryMode,
92
- recovering: struct {
93
- timestamp_last: ?u64,
94
- phase: union(enum) {
95
- validate_exchange,
96
- declare_locker_queue,
97
- declare_progress_queue,
98
- get_progress_message,
99
- nack_progress_message: struct {
100
- delivery_tag: u64,
101
- },
102
- },
103
- },
104
- last: struct {
105
- /// Last event read from TigerBeetle.
106
- producer_timestamp: u64,
107
- /// Last event published.
108
- consumer_timestamp: u64,
109
- },
110
- },
111
-
112
- pub fn init(
113
- self: *Runner,
114
- allocator: std.mem.Allocator,
115
- time: Time,
116
- options: struct {
117
- /// TigerBeetle cluster ID.
118
- cluster_id: u128,
119
- /// TigerBeetle cluster addresses.
120
- addresses: []const std.net.Address,
121
- /// AMQP host address.
122
- host: std.net.Address,
123
- /// AMQP User name for PLAIN authentication.
124
- user: []const u8,
125
- /// AMQP Password for PLAIN authentication.
126
- password: []const u8,
127
- /// AMQP vhost.
128
- vhost: []const u8,
129
- /// AMQP exchange name for publishing messages.
130
- publish_exchange: ?[]const u8,
131
- /// AMQP routing key for publishing messages.
132
- publish_routing_key: ?[]const u8,
133
- /// Overrides the number max of events produced/consumed each time.
134
- event_count_max: ?u32,
135
- /// Overrides the number of milliseconds to query again if there's no new events to
136
- /// process.
137
- /// Must be greater than zero.
138
- idle_interval_ms: ?u32,
139
- /// Limits the number of requests per second.
140
- /// Must be greater than zero.
141
- requests_per_second_limit: ?u32,
142
- /// Indicates whether to recover the last timestamp published on the state
143
- /// tracker queue, or override it with a user-defined value.
144
- recovery_mode: StateRecoveryMode,
145
- },
146
- ) !void {
147
- assert(options.addresses.len > 0);
148
-
149
- const idle_interval_ns: u63 = if (options.idle_interval_ms) |value|
150
- @intCast(@as(u64, value) * std.time.ns_per_ms)
151
- else
152
- constants.idle_interval_ns;
153
- assert(idle_interval_ns > 0);
154
-
155
- const event_count_max: u32 = if (options.event_count_max) |event_count_max|
156
- @min(event_count_max, constants.event_count_max)
157
- else
158
- constants.event_count_max;
159
- assert(event_count_max > 0);
160
-
161
- const publish_exchange: []const u8 = options.publish_exchange orelse "";
162
- const publish_routing_key: []const u8 = options.publish_routing_key orelse "";
163
- assert(publish_exchange.len > 0 or publish_routing_key.len > 0);
164
-
165
- const progress_tracker_queue_owned: []const u8 = try std.fmt.allocPrint(
166
- allocator,
167
- "{s}.{}",
168
- .{
169
- constants.progress_tracker_queue,
170
- options.cluster_id,
171
- },
172
- );
173
- errdefer allocator.free(progress_tracker_queue_owned);
174
- assert(progress_tracker_queue_owned.len <= 255);
175
-
176
- const locker_queue_owned: []const u8 = try std.fmt.allocPrint(
177
- allocator,
178
- "{s}.{}",
179
- .{
180
- constants.locker_queue,
181
- options.cluster_id,
182
- },
183
- );
184
- errdefer allocator.free(locker_queue_owned);
185
- assert(locker_queue_owned.len <= 255);
186
-
187
- const dual_buffer = try DualBuffer.init(allocator, event_count_max);
188
- errdefer self.buffer.deinit(allocator);
189
-
190
- self.* = .{
191
- .idle_interval_ns = idle_interval_ns,
192
- .event_count_max = event_count_max,
193
- .publish_exchange = publish_exchange,
194
- .publish_routing_key = publish_routing_key,
195
- .progress_tracker_queue = progress_tracker_queue_owned,
196
- .locker_queue = locker_queue_owned,
197
- .connected = .{},
198
- .io = undefined,
199
- .producer = .idle,
200
- .consumer = .idle,
201
- .rate_limit = undefined,
202
- .metrics = undefined,
203
- .state = .{ .unknown = options.recovery_mode },
204
- .buffer = dual_buffer,
205
- .message_pool = undefined,
206
- .vsr_client = undefined,
207
- .amqp_client = undefined,
208
- };
209
-
210
- self.rate_limit = RateLimit.init(time, .{
211
- .limit = options.requests_per_second_limit orelse std.math.maxInt(u32),
212
- // The rate limit is expressed in "requests per second".
213
- .period = .seconds(1),
214
- });
215
-
216
- self.metrics = .{
217
- .producer = .{
218
- .timer = .init(time),
219
- },
220
- .consumer = .{
221
- .timer = .init(time),
222
- },
223
- .flush_ticks = 0,
224
- .flush_timeout_ticks = @divExact(30 * std.time.ms_per_s, constants.tick_ms),
225
- };
226
-
227
- self.io = try IO.init(32, 0);
228
- errdefer self.io.deinit();
229
-
230
- self.message_pool = try MessagePool.init(allocator, .client);
231
- errdefer self.message_pool.deinit(allocator);
232
-
233
- self.vsr_client = try Client.init(
234
- allocator,
235
- time,
236
- &self.message_pool,
237
- .{
238
- .id = stdx.unique_u128(),
239
- .cluster = options.cluster_id,
240
- .replica_count = @intCast(options.addresses.len),
241
- .aof_recovery = false,
242
- .message_bus_options = .{
243
- .configuration = options.addresses,
244
- .io = &self.io,
245
- .trace = null,
246
- },
247
- },
248
- );
249
- errdefer self.vsr_client.deinit(allocator);
250
-
251
- self.amqp_client = try amqp.Client.init(allocator, .{
252
- .io = &self.io,
253
- .message_count_max = self.event_count_max,
254
- .message_body_size_max = Message.json_string_size_max,
255
- .reply_timeout_ticks = constants.reply_timeout_ticks,
256
- });
257
- errdefer self.amqp_client.deinit(allocator);
258
-
259
- // Starting both the VSR and the AMQP clients:
260
-
261
- try self.amqp_client.connect(
262
- &struct {
263
- fn callback(context: *amqp.Client) void {
264
- const runner: *Runner = @alignCast(@fieldParentPtr("amqp_client", context));
265
- assert(!runner.connected.amqp);
266
- maybe(runner.connected.vsr);
267
- log.info("AMQP connected.", .{});
268
- runner.connected.amqp = true;
269
- runner.recover();
270
- }
271
- }.callback,
272
- .{
273
- .host = options.host,
274
- .user_name = options.user,
275
- .password = options.password,
276
- .vhost = options.vhost,
277
- },
278
- );
279
- }
280
-
281
- pub fn deinit(self: *Runner, allocator: std.mem.Allocator) void {
282
- self.amqp_client.deinit(allocator);
283
- self.vsr_client.deinit(allocator);
284
- self.message_pool.deinit(allocator);
285
- self.io.deinit();
286
- self.buffer.deinit(allocator);
287
- allocator.free(self.locker_queue);
288
- allocator.free(self.progress_tracker_queue);
289
- }
290
-
291
- /// To make the CDC stateless, internal queues are used to store the state:
292
- ///
293
- /// - Progress tracking queue:
294
- /// A persistent queue with a maximum size of 1 message and "drop head" behavior on overflow.
295
- /// During publishing, a message containing the last timestamp is pushed into this queue at
296
- /// the end of each published batch.
297
- /// On restart, the presence of a message indicates the `timestamp_min` from which to resume
298
- /// processing events. Otherwise, processing starts from the beginning.
299
- /// The queue name is generated to be unique based on the `cluster_id`.
300
- /// The initial timestamp can be overridden via the command line.
301
- ///
302
- /// - Locker queue:
303
- /// A temporary, exclusive queue used to ensure that only a single CDC process is publishing
304
- /// at any given time. This queue is not used for publishing or consuming messages.
305
- /// The queue name is generated to be unique based on the `cluster_id`.
306
- fn recover(self: *Runner) void {
307
- assert(self.connected.amqp);
308
- assert(self.state == .unknown);
309
- const recovery_mode = self.state.unknown;
310
- const timestamp_override: ?u64 = switch (recovery_mode) {
311
- .recover => null,
312
- .override => |timestamp| timestamp,
313
- };
314
-
315
- const is_default_exchange = self.publish_exchange.len == 0;
316
- self.state = .{
317
- .recovering = .{
318
- .timestamp_last = timestamp_override,
319
- .phase = if (is_default_exchange)
320
- // No need to validate the default exchange, skipping `validate_exchange`.
321
- .declare_locker_queue
322
- else
323
- .validate_exchange,
324
- },
325
- };
326
- self.recover_dispatch();
327
- }
328
-
329
- fn recover_dispatch(self: *Runner) void {
330
- assert(self.connected.amqp);
331
- assert(self.state == .recovering);
332
- switch (self.state.recovering.phase) {
333
- // Check whether the exchange exists.
334
- // Declaring the exchange with `passive==true` only asserts if it already exists.
335
- .validate_exchange => {
336
- assert(self.publish_exchange.len > 0);
337
- maybe(self.state.recovering.timestamp_last == null);
338
-
339
- self.amqp_client.exchange_declare(
340
- &struct {
341
- fn callback(context: *amqp.Client) void {
342
- const runner: *Runner = @alignCast(@fieldParentPtr(
343
- "amqp_client",
344
- context,
345
- ));
346
- assert(runner.state == .recovering);
347
- const recovering = &runner.state.recovering;
348
- assert(recovering.phase == .validate_exchange);
349
- maybe(recovering.timestamp_last == null);
350
-
351
- recovering.phase = .declare_locker_queue;
352
- runner.recover_dispatch();
353
- }
354
- }.callback,
355
- .{
356
- .exchange = self.publish_exchange,
357
- .type = "",
358
- .passive = true,
359
- .durable = false,
360
- .internal = false,
361
- .auto_delete = false,
362
- },
363
- );
364
- },
365
- // Declaring the locker queue.
366
- // With `durable=false`, a temporary queue is created that exists only for the
367
- // duration of the current connection, and with `exclusive=true`, no other connection
368
- // can declare the same queue while this one is active.
369
- // This effectively acts as a distributed lock to prevent multiple CDC
370
- // instances from running simultaneously.
371
- .declare_locker_queue => {
372
- maybe(self.state.recovering.timestamp_last == null);
373
-
374
- self.amqp_client.queue_declare(
375
- &struct {
376
- fn callback(context: *amqp.Client) void {
377
- const runner: *Runner = @alignCast(@fieldParentPtr(
378
- "amqp_client",
379
- context,
380
- ));
381
- switch (runner.state) {
382
- .recovering => |*recovering| {
383
- assert(recovering.phase == .declare_locker_queue);
384
- maybe(recovering.timestamp_last == null);
385
-
386
- recovering.phase = .declare_progress_queue;
387
- runner.recover_dispatch();
388
- },
389
- else => unreachable,
390
- }
391
- }
392
- }.callback,
393
- .{
394
- .queue = self.locker_queue,
395
- .passive = false,
396
- .durable = false,
397
- .exclusive = true,
398
- .auto_delete = true,
399
- .arguments = .{
400
- .overflow = .drop_head,
401
- .max_length = 0,
402
- .max_length_bytes = 0,
403
- .single_active_consumer = true,
404
- },
405
- },
406
- );
407
- },
408
- // Declaring the progress tracking queue.
409
- // It's a no-op if the queue already exists.
410
- .declare_progress_queue => {
411
- maybe(self.state.recovering.timestamp_last == null);
412
-
413
- self.amqp_client.queue_declare(
414
- &struct {
415
- fn callback(context: *amqp.Client) void {
416
- const runner: *Runner = @alignCast(@fieldParentPtr(
417
- "amqp_client",
418
- context,
419
- ));
420
- switch (runner.state) {
421
- .recovering => |*recovering| {
422
- assert(recovering.phase == .declare_progress_queue);
423
-
424
- // Overriding the progress-tracking timestamp.
425
- if (recovering.timestamp_last) |timestamp_override| {
426
- runner.state = .{
427
- .last = .{
428
- .consumer_timestamp = timestamp_override,
429
- .producer_timestamp = timestamp_override + 1,
430
- },
431
- };
432
- return runner.vsr_register();
433
- }
434
- assert(recovering.timestamp_last == null);
435
-
436
- recovering.phase = .get_progress_message;
437
- runner.recover_dispatch();
438
- },
439
- else => unreachable,
440
- }
441
- }
442
- }.callback,
443
- .{
444
- .queue = self.progress_tracker_queue,
445
- .passive = false,
446
- .durable = true,
447
- .exclusive = false,
448
- .auto_delete = false,
449
- .arguments = .{
450
- .overflow = .drop_head,
451
- .max_length = 1,
452
- .max_length_bytes = 0,
453
- .single_active_consumer = true,
454
- },
455
- },
456
- );
457
- },
458
- // Getting the message header from the progress tracking queue.
459
- .get_progress_message => {
460
- assert(self.state.recovering.timestamp_last == null);
461
- self.amqp_client.get_message(
462
- &struct {
463
- fn callback(
464
- context: *amqp.Client,
465
- found: ?amqp.GetMessagePropertiesResult,
466
- ) amqp.Decoder.Error!void {
467
- const runner: *Runner = @alignCast(@fieldParentPtr(
468
- "amqp_client",
469
- context,
470
- ));
471
- switch (runner.state) {
472
- .recovering => |*recovering| {
473
- assert(recovering.phase == .get_progress_message);
474
- assert(recovering.timestamp_last == null);
475
-
476
- if (found) |result| {
477
- // Since this queue is declared with `limit == 1`,
478
- // we don't expect more than one message.
479
- if (result.message_count > 0) fatal(
480
- "Unexpected message_count={} in the progress queue.",
481
- .{result.message_count},
482
- );
483
- assert(!result.has_body);
484
- assert(result.delivery_tag > 0);
485
- // Recovering from a valid timestamp is crucial,
486
- // otherwise `get_change_events` may return empty results
487
- // due to invalid filters.
488
- const progress_tracker = try ProgressTrackerMessage.parse(
489
- result.properties.headers,
490
- );
491
- assert(TimestampRange.valid(progress_tracker.timestamp));
492
-
493
- // Downgrading the CDC job is not allowed.
494
- if (vsr.constants.config.process.release.value <
495
- progress_tracker.release.value)
496
- {
497
- fatal("The last event was published using a newer " ++
498
- "release (event={} current={}).", .{
499
- progress_tracker.release,
500
- vsr.constants.config.process.release,
501
- });
502
- }
503
-
504
- recovering.timestamp_last = progress_tracker.timestamp;
505
- recovering.phase = .{
506
- .nack_progress_message = .{
507
- .delivery_tag = result.delivery_tag,
508
- },
509
- };
510
-
511
- return runner.recover_dispatch();
512
- }
513
-
514
- // No previous progress record found,
515
- // starting from the beginning.
516
- assert(found == null);
517
- runner.state = .{ .last = .{
518
- .consumer_timestamp = 0,
519
- .producer_timestamp = TimestampRange.timestamp_min,
520
- } };
521
- runner.vsr_register();
522
- },
523
- else => unreachable,
524
- }
525
- }
526
- }.callback,
527
- .{
528
- .queue = self.progress_tracker_queue,
529
- .no_ack = false,
530
- },
531
- );
532
- },
533
- // Sending a `nack` with `requeue=true`, so the message remains in the progress
534
- // tracking queue in case we restart and need to recover again.
535
- .nack_progress_message => |message| {
536
- assert(self.state.recovering.timestamp_last != null);
537
- assert(TimestampRange.valid(self.state.recovering.timestamp_last.?));
538
- assert(message.delivery_tag > 0);
539
-
540
- self.amqp_client.nack(&struct {
541
- fn callback(context: *amqp.Client) void {
542
- const runner: *Runner = @alignCast(@fieldParentPtr(
543
- "amqp_client",
544
- context,
545
- ));
546
- switch (runner.state) {
547
- .recovering => |*recovering| {
548
- assert(recovering.phase == .nack_progress_message);
549
- assert(recovering.timestamp_last != null);
550
- assert(TimestampRange.valid(recovering.timestamp_last.?));
551
-
552
- const nack = recovering.phase.nack_progress_message;
553
- assert(nack.delivery_tag > 0);
554
-
555
- runner.state = .{ .last = .{
556
- .consumer_timestamp = recovering.timestamp_last.?,
557
- .producer_timestamp = recovering.timestamp_last.? + 1,
558
- } };
559
- runner.vsr_register();
560
- },
561
- else => unreachable,
562
- }
563
- }
564
- }.callback, .{
565
- .delivery_tag = message.delivery_tag,
566
- .requeue = true,
567
- .multiple = false,
568
- });
569
- },
570
- }
571
- }
572
-
573
- fn vsr_register(self: *Runner) void {
574
- assert(self.connected.amqp);
575
- assert(!self.connected.vsr);
576
- assert(self.producer == .idle);
577
- assert(self.consumer == .idle);
578
- assert(self.state == .last);
579
-
580
- // Register the VSR client as the last step to avoid unnecessarily joining the cluster
581
- // in case the CDC fails due to connectivity or configuration issues with the AMQP server.
582
- self.vsr_client.register(
583
- &struct {
584
- fn callback(
585
- user_data: u128,
586
- result: *const vsr.RegisterResult,
587
- ) void {
588
- const runner: *Runner = @ptrFromInt(@as(usize, @intCast(user_data)));
589
- assert(runner.connected.amqp);
590
- assert(!runner.connected.vsr);
591
- log.info("VSR client registered.", .{});
592
- runner.vsr_client.batch_size_limit = result.batch_size_limit;
593
- runner.connected.vsr = true;
594
-
595
- log.info("Starting CDC.", .{});
596
- runner.produce();
597
- }
598
- }.callback,
599
- @as(u128, @intCast(@intFromPtr(self))),
600
- );
601
- }
602
-
603
- /// The "Producer" fetches events from TigerBeetle (`get_change_events` operation) into a buffer
604
- /// to be consumed by the "Consumer".
605
- fn produce(self: *Runner) void {
606
- assert(self.connected.vsr);
607
- assert(self.connected.amqp);
608
- assert(self.state == .last);
609
- assert(TimestampRange.valid(self.state.last.producer_timestamp));
610
- assert(self.state.last.consumer_timestamp == 0 or
611
- TimestampRange.valid(self.state.last.consumer_timestamp));
612
- assert(self.state.last.producer_timestamp > self.state.last.consumer_timestamp);
613
- switch (self.producer) {
614
- .idle => {
615
- if (!self.buffer.producer_begin()) {
616
- // No free buffers (they must be `ready` and `consuming`).
617
- // The running consumer will resume the producer once it finishes.
618
- assert(self.consumer != .idle);
619
- assert(self.buffer.find(.ready) != null);
620
- assert(self.buffer.find(.consuming) != null);
621
- return;
622
- }
623
- self.producer = .rate_limit;
624
- self.metrics.producer.timer.reset();
625
- self.produce_dispatch();
626
- },
627
- else => unreachable, // Already running.
628
- }
629
- }
630
-
631
- fn produce_dispatch(self: *Runner) void {
632
- assert(self.connected.vsr);
633
- assert(self.connected.amqp);
634
- assert(self.state == .last);
635
- assert(TimestampRange.valid(self.state.last.producer_timestamp));
636
- assert(self.state.last.consumer_timestamp == 0 or
637
- TimestampRange.valid(self.state.last.consumer_timestamp));
638
- assert(self.state.last.producer_timestamp > self.state.last.consumer_timestamp);
639
- dispatch: switch (self.producer) {
640
- .idle => unreachable,
641
- // Check the configured rate limit.
642
- .rate_limit => switch (self.rate_limit.attempt()) {
643
- .ok => {
644
- self.producer = .request;
645
- continue :dispatch self.producer;
646
- },
647
- .wait => |duration| {
648
- assert(duration.ns > 0);
649
- self.io.timeout(
650
- *Runner,
651
- self,
652
- struct {
653
- fn callback(
654
- runner: *Runner,
655
- completion: *IO.Completion,
656
- result: IO.TimeoutError!void,
657
- ) void {
658
- result catch unreachable;
659
- _ = completion;
660
- assert(runner.producer == .rate_limit);
661
- assert(runner.buffer.find(.producing) != null);
662
- maybe(runner.consumer == .idle);
663
-
664
- runner.producer = .request;
665
- runner.produce_dispatch();
666
- }
667
- }.callback,
668
- &self.idle_completion,
669
- @intCast(duration.ns),
670
- );
671
- },
672
- },
673
- // Submitting the request through the VSR client.
674
- .request => {
675
- assert(self.buffer.find(.producing) != null);
676
-
677
- const filter: tb.ChangeEventsFilter = .{
678
- .limit = self.event_count_max,
679
- .timestamp_min = self.state.last.producer_timestamp,
680
- .timestamp_max = 0,
681
- };
682
-
683
- self.vsr_client.request(
684
- &produce_request_callback,
685
- @intFromPtr(self),
686
- .get_change_events,
687
- std.mem.asBytes(&filter),
688
- );
689
- },
690
- // No running consumer and no events returned from the last query,
691
- // waiting for the timeout to resume the producer.
692
- .waiting => {
693
- self.io.timeout(
694
- *Runner,
695
- self,
696
- struct {
697
- fn callback(
698
- runner: *Runner,
699
- completion: *IO.Completion,
700
- result: IO.TimeoutError!void,
701
- ) void {
702
- result catch unreachable;
703
- _ = completion;
704
- assert(runner.buffer.all_free());
705
- assert(runner.consumer == .idle);
706
- assert(runner.producer == .waiting);
707
-
708
- const producer_begin = runner.buffer.producer_begin();
709
- assert(producer_begin);
710
- runner.producer = .rate_limit;
711
- runner.produce_dispatch();
712
- }
713
- }.callback,
714
- &self.idle_completion,
715
- self.idle_interval_ns,
716
- );
717
- },
718
- }
719
- }
720
-
721
- fn produce_request_callback(
722
- context: u128,
723
- operation_vsr: vsr.Operation,
724
- timestamp: u64,
725
- result: []u8,
726
- ) void {
727
- const operation = operation_vsr.cast(tb.Operation);
728
- assert(operation == .get_change_events);
729
- assert(timestamp != 0);
730
- const runner: *Runner = @ptrFromInt(@as(usize, @intCast(context)));
731
- assert(runner.producer == .request);
732
-
733
- const source: []const tb.ChangeEvent = stdx.bytes_as_slice(.exact, tb.ChangeEvent, result);
734
- const target: []tb.ChangeEvent = runner.buffer.get_producer_buffer();
735
- assert(source.len <= target.len);
736
-
737
- stdx.copy_disjoint(
738
- .inexact,
739
- tb.ChangeEvent,
740
- target,
741
- source,
742
- );
743
- runner.buffer.producer_finish(@intCast(source.len));
744
-
745
- if (runner.buffer.all_free()) {
746
- // No events to publish.
747
- // Going idle and will check again for new events.
748
- assert(source.len == 0);
749
- assert(runner.consumer == .idle);
750
- runner.producer = .waiting;
751
- return runner.produce_dispatch();
752
- }
753
-
754
- runner.producer = .idle;
755
- runner.metrics.producer.record(source.len);
756
- assert(source.len > 0 or runner.consumer != .idle);
757
- if (source.len > 0) {
758
- const timestamp_next = source[source.len - 1].timestamp + 1;
759
- assert(TimestampRange.valid(timestamp_next));
760
- runner.state.last.producer_timestamp = timestamp_next;
761
-
762
- // Since the buffer was populated,
763
- // resume consuming (if not already running).
764
- if (runner.consumer == .idle) runner.consume();
765
-
766
- // Resume producing (if there's a free buffer).
767
- runner.produce();
768
- }
769
- }
770
-
771
- /// The "Consumer" reads from the buffer populated by the "Producer"
772
- /// and publishes the events to the AMQP server.
773
- fn consume(self: *Runner) void {
774
- assert(self.connected.vsr);
775
- assert(self.connected.amqp);
776
- assert(self.state == .last);
777
- assert(TimestampRange.valid(self.state.last.producer_timestamp));
778
- assert(self.state.last.consumer_timestamp == 0 or
779
- TimestampRange.valid(self.state.last.consumer_timestamp));
780
- assert(self.state.last.producer_timestamp > self.state.last.consumer_timestamp);
781
- switch (self.consumer) {
782
- .idle => {
783
- if (!self.buffer.consumer_begin()) {
784
- // No buffers ready (they must be both `free` or still `producing`).
785
- // The running/waiting producer will resume the consumer once it finishes.
786
- if (self.buffer.all_free()) {
787
- assert(self.producer == .idle);
788
- } else {
789
- assert(self.producer == .rate_limit or self.producer == .request);
790
- assert(self.buffer.find(.free) != null);
791
- assert(self.buffer.find(.producing) != null);
792
- }
793
- return;
794
- }
795
- self.consumer = .publish;
796
- self.metrics.consumer.timer.reset();
797
- self.consume_dispatch();
798
- },
799
- else => unreachable, // Already running.
800
- }
801
- }
802
-
803
- fn consume_dispatch(self: *Runner) void {
804
- assert(self.connected.vsr);
805
- assert(self.connected.amqp);
806
- assert(self.state == .last);
807
- assert(TimestampRange.valid(self.state.last.producer_timestamp));
808
- assert(self.state.last.consumer_timestamp == 0 or
809
- TimestampRange.valid(self.state.last.consumer_timestamp));
810
- assert(self.state.last.producer_timestamp > self.state.last.consumer_timestamp);
811
- switch (self.consumer) {
812
- .idle => unreachable,
813
- // Publishes a batch of events and waits until the AMQP server acknowledges it.
814
- // N.B.: TigerBeetle guarantees at-least-once semantics when publishing,
815
- // and makes a best effort to prevent duplicate messages.
816
- // Publishing uses `confirm.select` instead of `tx.select`, as the former provides
817
- // better performance with equivalent delivery guarantees. However, neither can
818
- // ensure exactly-once delivery in case of crashes in the middle of the operation.
819
- // From https://www.rabbitmq.com/docs/semantics:
820
- // "RabbitMQ provides no atomicity guarantees even in case of transactions involving
821
- // just a single queue, e.g. a fault during tx.commit can result in a sub-set of the
822
- // transaction's publishes appearing in the queue after a broker restart.
823
- .publish => {
824
- const events: []const tb.ChangeEvent = self.buffer.get_consumer_buffer();
825
- assert(events.len > 0);
826
- for (events) |*event| {
827
- const message = Message.init(event);
828
- self.amqp_client.publish_enqueue(.{
829
- .exchange = self.publish_exchange,
830
- .routing_key = self.publish_routing_key,
831
- .mandatory = true,
832
- .immediate = false,
833
- .properties = .{
834
- .content_type = Message.content_type,
835
- .delivery_mode = .persistent,
836
- .app_id = constants.app_id,
837
- // AMQP timestamp in seconds.
838
- .timestamp = @divTrunc(event.timestamp, std.time.ns_per_s),
839
- .headers = message.header(),
840
- },
841
- .body = message.body(),
842
- });
843
- }
844
- self.amqp_client.publish_send(&struct {
845
- fn callback(context: *amqp.Client) void {
846
- const runner: *Runner = @alignCast(@fieldParentPtr(
847
- "amqp_client",
848
- context,
849
- ));
850
- assert(runner.consumer == .publish);
851
- runner.consumer = .progress_update;
852
- runner.consume_dispatch();
853
- }
854
- }.callback);
855
- },
856
- // Publishes the progress-tracking message containing the last timestamp
857
- // *after* the batch of events has been acknowledged by the AMQP server.
858
- .progress_update => {
859
- const progress_tracker: ProgressTrackerMessage = progress: {
860
- const events = self.buffer.get_consumer_buffer();
861
- assert(events.len > 0);
862
- break :progress .{
863
- .timestamp = events[events.len - 1].timestamp,
864
- .release = vsr.constants.config.process.release,
865
- };
866
- };
867
- self.amqp_client.publish_enqueue(.{
868
- .exchange = "", // No exchange sends directly to this queue.
869
- .routing_key = self.progress_tracker_queue,
870
- .mandatory = true,
871
- .immediate = false,
872
- .properties = .{
873
- .delivery_mode = .persistent,
874
- .timestamp = @intCast(std.time.milliTimestamp()),
875
- .headers = progress_tracker.header(),
876
- },
877
- .body = null,
878
- });
879
- self.amqp_client.publish_send(&struct {
880
- fn callback(context: *amqp.Client) void {
881
- const runner: *Runner = @alignCast(@fieldParentPtr(
882
- "amqp_client",
883
- context,
884
- ));
885
- assert(runner.consumer == .progress_update);
886
-
887
- const event_count: usize, const timestamp_last: u64 = events: {
888
- const events = runner.buffer.get_consumer_buffer();
889
- assert(events.len > 0);
890
- break :events .{ events.len, events[events.len - 1].timestamp };
891
- };
892
- runner.buffer.consumer_finish();
893
- runner.state.last.consumer_timestamp = timestamp_last;
894
-
895
- // Resume consuming (if there's a buffer ready).
896
- runner.consumer = .idle;
897
- runner.metrics.consumer.record(event_count);
898
- runner.consume();
899
-
900
- // Since the buffer was released,
901
- // resume producing (if not already running).
902
- if (runner.producer == .idle) runner.produce();
903
- }
904
- }.callback);
905
- },
906
- }
907
- }
908
-
909
- pub fn tick(self: *Runner) void {
910
- assert(!self.vsr_client.evicted);
911
- self.vsr_client.tick();
912
- self.amqp_client.tick();
913
- self.io.run_for_ns(constants.tick_ms * std.time.ns_per_ms) catch unreachable;
914
-
915
- self.metrics.tick();
916
- }
917
- };
918
-
919
- /// Rate limit to throttle the maximum number of requests to TigerBeetle within a time period.
920
- pub const RateLimit = struct {
921
- const Options = struct {
922
- /// The rate limit expressed as "requests per second".
923
- /// Must be greater than zero. Inclusive.
924
- limit: u32,
925
- /// Time interval used to enforce the request limit.
926
- period: stdx.Duration,
927
- };
928
-
929
- count: u32,
930
- timer: vsr.time.Timer,
931
- options: Options,
932
-
933
- pub fn init(time: vsr.time.Time, options: Options) RateLimit {
934
- assert(options.limit > 0);
935
- assert(options.period.ns > 0);
936
-
937
- return .{
938
- .count = 0,
939
- .timer = .init(time),
940
- .options = options,
941
- };
942
- }
943
-
944
- /// Attempt to increment the counter.
945
- /// Return `.ok` if it succeed within the configured `limit` per second,
946
- /// or `.wait` with the required amount of time to wait.
947
- pub fn attempt(self: *RateLimit) union(enum) {
948
- ok,
949
- wait: stdx.Duration,
950
- } {
951
- assert(self.options.limit > 0);
952
- assert(self.options.period.ns > 0);
953
- assert(self.count <= self.options.limit);
954
-
955
- if (self.count == 0) {
956
- self.timer.reset();
957
- self.count = 1;
958
- return .ok;
959
- }
960
- assert(self.count > 0);
961
-
962
- const duration = self.timer.read();
963
- maybe(duration.ns == 0);
964
-
965
- if (duration.ns >= self.options.period.ns) {
966
- self.timer.reset();
967
- self.count = 0;
968
- } else if (self.count == self.options.limit) {
969
- assert(duration.ns < self.options.period.ns);
970
- return .{ .wait = .{
971
- .ns = self.options.period.ns - duration.ns,
972
- } };
973
- }
974
-
975
- self.count += 1;
976
- assert(self.count <= self.options.limit);
977
-
978
- return .ok;
979
- }
980
- };
981
-
982
- /// Inspired by the StateMachine metrics,
983
- /// though the current method of shipping the metrics is a temporary solution.
984
- const Metrics = struct {
985
- const TimingSummary = struct {
986
- timer: vsr.time.Timer,
987
-
988
- duration_min: ?stdx.Duration = null,
989
- duration_max: ?stdx.Duration = null,
990
- duration_sum: stdx.Duration = .{ .ns = 0 },
991
- event_count: u64 = 0,
992
- count: u64 = 0,
993
-
994
- fn record(
995
- metrics: *TimingSummary,
996
- event_count: u64,
997
- ) void {
998
- metrics.timing(
999
- event_count,
1000
- metrics.timer.read(),
1001
- );
1002
- }
1003
-
1004
- fn timing(
1005
- metrics: *TimingSummary,
1006
- event_count: u64,
1007
- duration: stdx.Duration,
1008
- ) void {
1009
- maybe(duration.ns == 0);
1010
- maybe(event_count == 0);
1011
-
1012
- metrics.count += 1;
1013
- metrics.event_count += event_count;
1014
- metrics.duration_min = if (metrics.duration_min) |min| duration.min(min) else duration;
1015
- metrics.duration_max = if (metrics.duration_max) |max| duration.max(max) else duration;
1016
- metrics.duration_sum.ns += duration.ns;
1017
- }
1018
- };
1019
-
1020
- producer: TimingSummary,
1021
- consumer: TimingSummary,
1022
- flush_ticks: u64,
1023
- flush_timeout_ticks: u64,
1024
-
1025
- fn tick(self: *Metrics) void {
1026
- assert(self.flush_ticks < self.flush_timeout_ticks);
1027
- self.flush_ticks += 1;
1028
- if (self.flush_ticks == self.flush_timeout_ticks) {
1029
- self.flush_ticks = 0;
1030
- self.log_and_reset();
1031
- }
1032
- }
1033
-
1034
- fn log_and_reset(metrics: *Metrics) void {
1035
- const Fields = enum { producer, consumer };
1036
- const runner: *const Runner = @alignCast(@fieldParentPtr("metrics", metrics));
1037
- inline for (comptime std.enums.values(Fields)) |field| {
1038
- const summary: *TimingSummary = &@field(metrics, @tagName(field));
1039
- if (summary.count > 0 and summary.duration_sum.ns > 0) {
1040
- assert(runner.state == .last);
1041
- assert(summary.duration_min != null);
1042
- assert(summary.duration_max != null);
1043
-
1044
- const timestamp_last = switch (field) {
1045
- .consumer => runner.state.last.consumer_timestamp,
1046
- .producer => runner.state.last.producer_timestamp,
1047
- };
1048
- const event_rate = @divTrunc(
1049
- summary.event_count * std.time.ns_per_s,
1050
- summary.duration_sum.ns,
1051
- );
1052
- log.info("{s}: p0={}ms mean={}ms p100={}ms " ++
1053
- "event_count={} throughput={} op/s " ++
1054
- "last timestamp={} ({})", .{
1055
- @tagName(field),
1056
- summary.duration_min.?.to_ms(),
1057
- @divFloor(summary.duration_sum.to_ms(), summary.count),
1058
- summary.duration_max.?.to_ms(),
1059
- summary.event_count,
1060
- event_rate,
1061
- timestamp_last,
1062
- stdx.InstantUnix{ .ns = timestamp_last },
1063
- });
1064
- }
1065
- summary.* = .{
1066
- .timer = summary.timer,
1067
- };
1068
- }
1069
- }
1070
- };
1071
-
1072
- /// Buffers swapped between producer and consumer, allowing reading from TigerBeetle
1073
- /// and publishing to AMQP to happen concurrently.
1074
- const DualBuffer = struct {
1075
- const State = enum {
1076
- free,
1077
- producing,
1078
- ready,
1079
- consuming,
1080
- };
1081
-
1082
- const Buffer = struct {
1083
- buffer: []tb.ChangeEvent,
1084
- state: union(State) {
1085
- free,
1086
- producing,
1087
- ready: u32,
1088
- consuming: u32,
1089
- } = .free,
1090
- };
1091
-
1092
- buffer_1: Buffer,
1093
- buffer_2: Buffer,
1094
-
1095
- pub fn init(allocator: std.mem.Allocator, event_count: u32) !DualBuffer {
1096
- assert(event_count > 0);
1097
- assert(event_count <= Runner.constants.event_count_max);
1098
-
1099
- const buffer_1 = try allocator.alloc(tb.ChangeEvent, event_count);
1100
- errdefer allocator.free(buffer_1);
1101
-
1102
- const buffer_2 = try allocator.alloc(tb.ChangeEvent, event_count);
1103
- errdefer allocator.free(buffer_2);
1104
-
1105
- return .{
1106
- .buffer_1 = .{
1107
- .buffer = buffer_1,
1108
- .state = .free,
1109
- },
1110
- .buffer_2 = .{
1111
- .buffer = buffer_2,
1112
- .state = .free,
1113
- },
1114
- };
1115
- }
1116
-
1117
- pub fn deinit(self: *DualBuffer, allocator: std.mem.Allocator) void {
1118
- allocator.free(self.buffer_2.buffer);
1119
- allocator.free(self.buffer_1.buffer);
1120
- }
1121
-
1122
- pub fn producer_begin(self: *DualBuffer) bool {
1123
- self.assert_state();
1124
- // Already producing.
1125
- assert(self.find(.producing) == null);
1126
- const buffer = self.find(.free) orelse
1127
- // No free buffers.
1128
- return false;
1129
- buffer.state = .producing;
1130
- return true;
1131
- }
1132
-
1133
- pub fn get_producer_buffer(self: *DualBuffer) []tb.ChangeEvent {
1134
- self.assert_state();
1135
- const buffer = self.find(.producing).?;
1136
- return buffer.buffer;
1137
- }
1138
-
1139
- pub fn producer_finish(self: *DualBuffer, count: u32) void {
1140
- self.assert_state();
1141
- const buffer = self.find(.producing).?;
1142
- buffer.state = if (count == 0) .free else .{ .ready = count };
1143
- }
1144
-
1145
- pub fn consumer_begin(self: *DualBuffer) bool {
1146
- self.assert_state();
1147
- // Already consuming.
1148
- assert(self.find(.consuming) == null);
1149
- const buffer = self.find(.ready) orelse
1150
- // No buffers ready.
1151
- return false;
1152
- const count = buffer.state.ready;
1153
- buffer.state = .{ .consuming = count };
1154
- return true;
1155
- }
1156
-
1157
- pub fn get_consumer_buffer(self: *DualBuffer) []const tb.ChangeEvent {
1158
- self.assert_state();
1159
- const buffer = self.find(.consuming).?;
1160
- return buffer.buffer[0..buffer.state.consuming];
1161
- }
1162
-
1163
- pub fn consumer_finish(self: *DualBuffer) void {
1164
- self.assert_state();
1165
- const buffer = self.find(.consuming).?;
1166
- buffer.state = .free;
1167
- }
1168
-
1169
- pub fn all_free(self: *const DualBuffer) bool {
1170
- return self.buffer_1.state == .free and
1171
- self.buffer_2.state == .free;
1172
- }
1173
-
1174
- fn find(self: *DualBuffer, state: State) ?*Buffer {
1175
- self.assert_state();
1176
- if (self.buffer_1.state == state) return &self.buffer_1;
1177
- if (self.buffer_2.state == state) return &self.buffer_2;
1178
- return null;
1179
- }
1180
-
1181
- fn assert_state(self: *const DualBuffer) void {
1182
- // Two buffers: one can be producing while the other is consuming,
1183
- // but never two consumers or producers.
1184
- assert(!(self.buffer_1.state == .producing and self.buffer_2.state == .producing));
1185
- assert(!(self.buffer_1.state == .consuming and self.buffer_2.state == .consuming));
1186
- assert(!(self.buffer_1.state == .ready and self.buffer_2.state == .ready));
1187
- maybe(self.buffer_1.state == .free and self.buffer_2.state == .free);
1188
- }
1189
- };
1190
-
1191
- /// Progress tracker message with no body, containing the timestamp
1192
- /// and the release version of the last acknowledged publish.
1193
- const ProgressTrackerMessage = struct {
1194
- release: vsr.Release,
1195
- timestamp: u64,
1196
-
1197
- fn header(self: *const ProgressTrackerMessage) amqp.Encoder.Table {
1198
- const vtable: amqp.Encoder.Table.VTable = comptime .{
1199
- .write = &struct {
1200
- fn write(context: *const anyopaque, encoder: *amqp.Encoder.TableEncoder) void {
1201
- const message: *const ProgressTrackerMessage = @ptrCast(@alignCast(context));
1202
- var release_buffer: [
1203
- std.fmt.count("{}", vsr.Release.from(.{
1204
- .major = std.math.maxInt(u16),
1205
- .minor = std.math.maxInt(u8),
1206
- .patch = std.math.maxInt(u8),
1207
- }))
1208
- ]u8 = undefined;
1209
- encoder.put("release", .{ .string = std.fmt.bufPrint(
1210
- &release_buffer,
1211
- "{}",
1212
- .{message.release},
1213
- ) catch unreachable });
1214
- encoder.put("timestamp", .{ .int64 = @intCast(message.timestamp) });
1215
- }
1216
- }.write,
1217
- };
1218
- return .{ .context = self, .vtable = &vtable };
1219
- }
1220
-
1221
- fn parse(table: ?amqp.Decoder.Table) amqp.Decoder.Error!ProgressTrackerMessage {
1222
- if (table) |headers| {
1223
- var timestamp: ?u64 = null;
1224
- var release: ?vsr.Release = null;
1225
-
1226
- // Intentionally allows the presence of header fields other than `timestamp`,
1227
- // since some plugin may insert additional headers into messages.
1228
- var iterator = headers.iterator();
1229
- while (try iterator.next()) |entry| {
1230
- if (std.mem.eql(u8, entry.key, "timestamp")) {
1231
- switch (entry.value) {
1232
- .int64 => |int64| {
1233
- const value: u64 = @intCast(int64);
1234
- if (!TimestampRange.valid(value)) break;
1235
- timestamp = value;
1236
- },
1237
- else => break,
1238
- }
1239
- }
1240
- if (std.mem.eql(u8, entry.key, "release")) {
1241
- switch (entry.value) {
1242
- .string => |value| {
1243
- release = vsr.Release.parse(value) catch break;
1244
- },
1245
- else => break,
1246
- }
1247
- }
1248
-
1249
- if (timestamp != null and release != null) return .{
1250
- .timestamp = timestamp.?,
1251
- .release = release.?,
1252
- };
1253
- }
1254
- }
1255
- fatal(
1256
- \\Invalid progress tracker message.
1257
- \\Use `--timestamp-last` to restore a valid initial timestamp.
1258
- , .{});
1259
- }
1260
- };
1261
-
1262
- /// Message with the body in the JSON schema.
1263
- pub const Message = struct {
1264
- pub const content_type = "application/json";
1265
-
1266
- pub const json_string_size_max = size: {
1267
- var counting_writer = std.io.countingWriter(std.io.null_writer);
1268
- std.json.stringify(
1269
- worse_case(Message),
1270
- stringify_options,
1271
- counting_writer.writer(),
1272
- ) catch unreachable;
1273
- break :size counting_writer.bytes_written;
1274
- };
1275
-
1276
- const stringify_options = std.json.StringifyOptions{
1277
- .whitespace = .minified,
1278
- .emit_nonportable_numbers_as_strings = true,
1279
- };
1280
-
1281
- timestamp: u64,
1282
- type: tb.ChangeEventType,
1283
- ledger: u32,
1284
- transfer: struct {
1285
- id: u128,
1286
- amount: u128,
1287
- pending_id: u128,
1288
- user_data_128: u128,
1289
- user_data_64: u64,
1290
- user_data_32: u32,
1291
- timeout: u32,
1292
- code: u16,
1293
- flags: u16,
1294
- timestamp: u64,
1295
- },
1296
- debit_account: struct {
1297
- id: u128,
1298
- debits_pending: u128,
1299
- debits_posted: u128,
1300
- credits_pending: u128,
1301
- credits_posted: u128,
1302
- user_data_128: u128,
1303
- user_data_64: u64,
1304
- user_data_32: u32,
1305
- code: u16,
1306
- flags: u16,
1307
- timestamp: u64,
1308
- },
1309
- credit_account: struct {
1310
- id: u128,
1311
- debits_pending: u128,
1312
- debits_posted: u128,
1313
- credits_pending: u128,
1314
- credits_posted: u128,
1315
- user_data_128: u128,
1316
- user_data_64: u64,
1317
- user_data_32: u32,
1318
- code: u16,
1319
- flags: u16,
1320
- timestamp: u64,
1321
- },
1322
-
1323
- pub fn init(event: *const tb.ChangeEvent) Message {
1324
- return .{
1325
- .timestamp = event.timestamp,
1326
- .type = event.type,
1327
- .ledger = event.ledger,
1328
- .transfer = .{
1329
- .id = event.transfer_id,
1330
- .amount = event.transfer_amount,
1331
- .pending_id = event.transfer_pending_id,
1332
- .user_data_128 = event.transfer_user_data_128,
1333
- .user_data_64 = event.transfer_user_data_64,
1334
- .user_data_32 = event.transfer_user_data_32,
1335
- .timeout = event.transfer_timeout,
1336
- .code = event.transfer_code,
1337
- .flags = @bitCast(event.transfer_flags),
1338
- .timestamp = event.transfer_timestamp,
1339
- },
1340
- .debit_account = .{
1341
- .id = event.debit_account_id,
1342
- .debits_pending = event.debit_account_debits_pending,
1343
- .debits_posted = event.debit_account_debits_posted,
1344
- .credits_pending = event.debit_account_credits_pending,
1345
- .credits_posted = event.debit_account_credits_posted,
1346
- .user_data_128 = event.debit_account_user_data_128,
1347
- .user_data_64 = event.debit_account_user_data_64,
1348
- .user_data_32 = event.debit_account_user_data_32,
1349
- .code = event.debit_account_code,
1350
- .flags = @bitCast(event.debit_account_flags),
1351
- .timestamp = event.debit_account_timestamp,
1352
- },
1353
- .credit_account = .{
1354
- .id = event.credit_account_id,
1355
- .debits_pending = event.credit_account_debits_pending,
1356
- .debits_posted = event.credit_account_debits_posted,
1357
- .credits_pending = event.credit_account_credits_pending,
1358
- .credits_posted = event.credit_account_credits_posted,
1359
- .user_data_128 = event.credit_account_user_data_128,
1360
- .user_data_64 = event.credit_account_user_data_64,
1361
- .user_data_32 = event.credit_account_user_data_32,
1362
- .code = event.credit_account_code,
1363
- .flags = @bitCast(event.credit_account_flags),
1364
- .timestamp = event.credit_account_timestamp,
1365
- },
1366
- };
1367
- }
1368
-
1369
- fn header(self: *const Message) amqp.Encoder.Table {
1370
- const vtable: amqp.Encoder.Table.VTable = comptime .{
1371
- .write = &struct {
1372
- fn write(context: *const anyopaque, encoder: *amqp.Encoder.TableEncoder) void {
1373
- const message: *const Message = @ptrCast(@alignCast(context));
1374
- encoder.put("event_type", .{ .string = @tagName(message.type) });
1375
-
1376
- // N.B.: Unsigned integers like u32 and u16 are not universally supported by
1377
- // all RabbitMQ clients.
1378
- // To ensure compatibility, we promote them to a signed integer.
1379
- encoder.put("ledger", .{ .int64 = message.ledger });
1380
- encoder.put("transfer_code", .{ .int32 = message.transfer.code });
1381
- encoder.put("debit_account_code", .{
1382
- .int32 = message.debit_account.code,
1383
- });
1384
- encoder.put("credit_account_code", .{
1385
- .int32 = message.credit_account.code,
1386
- });
1387
- }
1388
- }.write,
1389
- };
1390
- return .{ .context = self, .vtable = &vtable };
1391
- }
1392
-
1393
- fn body(self: *const Message) amqp.Encoder.Body {
1394
- const vtable: amqp.Encoder.Body.VTable = comptime .{
1395
- .write = &struct {
1396
- fn write(context: *const anyopaque, buffer: []u8) usize {
1397
- const message: *const Message = @ptrCast(@alignCast(context));
1398
- var fbs = std.io.fixedBufferStream(buffer);
1399
- std.json.stringify(message, .{
1400
- .whitespace = .minified,
1401
- .emit_nonportable_numbers_as_strings = true,
1402
- }, fbs.writer()) catch unreachable;
1403
- return fbs.pos;
1404
- }
1405
- }.write,
1406
- };
1407
- return .{ .context = self, .vtable = &vtable };
1408
- }
1409
-
1410
- /// Fill all fields for the largest string representation.
1411
- fn worse_case(comptime T: type) T {
1412
- var value: T = undefined;
1413
- for (std.meta.fields(T)) |field| {
1414
- @field(value, field.name) = switch (@typeInfo(field.type)) {
1415
- .int => std.math.maxInt(field.type),
1416
- .@"enum" => max: {
1417
- var name: []const u8 = "";
1418
- for (std.enums.values(tb.ChangeEventType)) |tag| {
1419
- if (@tagName(tag).len > name.len) {
1420
- name = @tagName(tag);
1421
- }
1422
- }
1423
- break :max @field(field.type, name);
1424
- },
1425
- .@"struct" => worse_case(field.type),
1426
- else => unreachable,
1427
- };
1428
- }
1429
- return value;
1430
- }
1431
- };
1432
-
1433
- const testing = std.testing;
1434
- const fixtures = @import("../testing/fixtures.zig");
1435
-
1436
- test "amqp: RateLimit" {
1437
- // Simulated clock with 300ms resolution,
1438
- // to force an uneven ratio of 3.333 requests per second.
1439
- const resolution: u64 = 300 * std.time.ns_per_ms;
1440
- var time_sim = fixtures.init_time(.{ .resolution = resolution });
1441
- const time = time_sim.time();
1442
- var rate_limit = RateLimit.init(
1443
- time,
1444
- .{
1445
- .limit = 3,
1446
- .period = .seconds(1),
1447
- },
1448
- );
1449
-
1450
- try testing.expect(rate_limit.attempt() == .ok);
1451
- time.tick();
1452
-
1453
- try testing.expect(rate_limit.attempt() == .ok);
1454
- time.tick();
1455
-
1456
- try testing.expect(rate_limit.attempt() == .ok);
1457
- try switch (rate_limit.attempt()) {
1458
- .ok => testing.expect(false),
1459
- .wait => |duration| testing.expectEqual(
1460
- // 3 requests in 600ms, needs to wait 400ms.
1461
- std.time.ns_per_s - (2 * resolution),
1462
- duration.ns,
1463
- ),
1464
- };
1465
- time.tick();
1466
-
1467
- try switch (rate_limit.attempt()) {
1468
- .ok => testing.expect(false),
1469
- .wait => |duration| testing.expectEqual(
1470
- // 3 requests in 900ms, needs to wait 100ms.
1471
- std.time.ns_per_s - (3 * resolution),
1472
- duration.ns,
1473
- ),
1474
- };
1475
- time.tick();
1476
-
1477
- try testing.expect(rate_limit.attempt() == .ok);
1478
- time.tick();
1479
-
1480
- try testing.expect(rate_limit.attempt() == .ok);
1481
- time.tick();
1482
-
1483
- try testing.expect(rate_limit.attempt() == .ok);
1484
- time.tick();
1485
-
1486
- try testing.expect(rate_limit.attempt() == .wait);
1487
- }
1488
-
1489
- test "amqp: DualBuffer" {
1490
- const event_count_max = Runner.constants.event_count_max;
1491
-
1492
- var prng = stdx.PRNG.from_seed_testing();
1493
- var dual_buffer = try DualBuffer.init(testing.allocator, event_count_max);
1494
- defer dual_buffer.deinit(testing.allocator);
1495
-
1496
- for (0..4096) |_| {
1497
- try testing.expect(dual_buffer.all_free());
1498
-
1499
- // Starts a producer:
1500
- const producer_begin = dual_buffer.producer_begin();
1501
- try testing.expect(producer_begin);
1502
- try testing.expect(!dual_buffer.all_free());
1503
- // We can't consume yet.
1504
- try testing.expect(!dual_buffer.consumer_begin());
1505
-
1506
- const producer1_buffer = dual_buffer.get_producer_buffer();
1507
- try testing.expectEqual(@as(usize, event_count_max), producer1_buffer.len);
1508
-
1509
- const producer1_count = prng.range_inclusive(u32, 1, event_count_max);
1510
- prng.fill(std.mem.sliceAsBytes(producer1_buffer[0..producer1_count]));
1511
- dual_buffer.producer_finish(producer1_count);
1512
-
1513
- // Starts a consumer after the producer has finished:
1514
- const consumer_begin = dual_buffer.consumer_begin();
1515
- try testing.expect(consumer_begin);
1516
- try testing.expect(!dual_buffer.all_free());
1517
-
1518
- // Concurrently starts another producer:
1519
- const producer_begin_concurrently = dual_buffer.producer_begin();
1520
- try testing.expect(producer_begin_concurrently);
1521
- try testing.expect(!dual_buffer.all_free());
1522
-
1523
- const producer2_buffer = dual_buffer.get_producer_buffer();
1524
- try testing.expectEqual(@as(usize, event_count_max), producer2_buffer.len);
1525
-
1526
- const producer2_count = prng.range_inclusive(u32, 0, event_count_max);
1527
- maybe(producer2_count == 0); // Testing zeroed producers.
1528
- prng.fill(std.mem.sliceAsBytes(producer2_buffer[0..producer2_count]));
1529
- dual_buffer.producer_finish(producer2_count);
1530
-
1531
- // Consuming the first producer:
1532
- const consumer_buffer = dual_buffer.get_consumer_buffer();
1533
- try testing.expectEqual(producer1_buffer.ptr, consumer_buffer.ptr);
1534
- try testing.expectEqual(@as(usize, producer1_count), consumer_buffer.len);
1535
- try testing.expectEqualSlices(
1536
- u8,
1537
- std.mem.sliceAsBytes(producer1_buffer[0..producer1_count]),
1538
- std.mem.sliceAsBytes(consumer_buffer),
1539
- );
1540
- dual_buffer.consumer_finish();
1541
-
1542
- // Consuming the second producer.
1543
- // It might not have produced anything, so the buffer cannot be consumed:
1544
- const consumer_begin_again = dual_buffer.consumer_begin();
1545
- if (producer2_count == 0) {
1546
- try testing.expect(!consumer_begin_again);
1547
- try testing.expect(dual_buffer.all_free());
1548
- continue;
1549
- }
1550
-
1551
- try testing.expect(consumer_begin_again);
1552
- try testing.expect(!dual_buffer.all_free());
1553
-
1554
- const consumer2_buffer = dual_buffer.get_consumer_buffer();
1555
- try testing.expectEqual(producer2_buffer.ptr, consumer2_buffer.ptr);
1556
- try testing.expectEqual(@as(usize, producer2_count), consumer2_buffer.len);
1557
- try testing.expectEqualSlices(
1558
- u8,
1559
- std.mem.sliceAsBytes(producer2_buffer[0..producer2_count]),
1560
- std.mem.sliceAsBytes(consumer2_buffer),
1561
- );
1562
-
1563
- dual_buffer.consumer_finish();
1564
- try testing.expect(dual_buffer.all_free());
1565
- }
1566
- }
1567
-
1568
- test "amqp: ProgressTrackerMessage" {
1569
- const buffer = try testing.allocator.alloc(u8, amqp.frame_min_size);
1570
- defer testing.allocator.free(buffer);
1571
-
1572
- const values: []const u64 = &.{
1573
- TimestampRange.timestamp_min,
1574
- 1745055501942402250,
1575
- TimestampRange.timestamp_max,
1576
- };
1577
- for (values) |value| {
1578
- const message: ProgressTrackerMessage = .{
1579
- .release = vsr.Release.minimum,
1580
- .timestamp = value,
1581
- };
1582
- var encoder = amqp.Encoder.init(buffer);
1583
- encoder.write_table(message.header());
1584
-
1585
- var decoder = amqp.Decoder.init(buffer[0..encoder.index]);
1586
- const decoded_message = try ProgressTrackerMessage.parse(try decoder.read_table());
1587
- try testing.expectEqual(message.release.value, decoded_message.release.value);
1588
- try testing.expectEqual(message.timestamp, decoded_message.timestamp);
1589
- }
1590
- }
1591
-
1592
- test "amqp: JSON message" {
1593
- const Snap = stdx.Snap;
1594
- const snap = Snap.snap_fn("src");
1595
-
1596
- const buffer = try testing.allocator.alloc(u8, Message.json_string_size_max);
1597
- defer testing.allocator.free(buffer);
1598
-
1599
- {
1600
- const message: Message = std.mem.zeroInit(Message, .{});
1601
- const size = message.body().write(buffer);
1602
- try testing.expectEqual(@as(usize, 564), size);
1603
-
1604
- try snap(@src(),
1605
- \\{"timestamp":0,"type":"single_phase","ledger":0,"transfer":{"id":0,"amount":0,"pending_id":0,"user_data_128":0,"user_data_64":0,"user_data_32":0,"timeout":0,"code":0,"flags":0,"timestamp":0},"debit_account":{"id":0,"debits_pending":0,"debits_posted":0,"credits_pending":0,"credits_posted":0,"user_data_128":0,"user_data_64":0,"user_data_32":0,"code":0,"flags":0,"timestamp":0},"credit_account":{"id":0,"debits_pending":0,"debits_posted":0,"credits_pending":0,"credits_posted":0,"user_data_128":0,"user_data_64":0,"user_data_32":0,"code":0,"flags":0,"timestamp":0}}
1606
- ).diff(buffer[0..size]);
1607
- }
1608
-
1609
- {
1610
- const message = comptime Message.worse_case(Message);
1611
- const size = message.body().write(buffer);
1612
- try testing.expectEqual(@as(usize, 1425), size);
1613
- try testing.expectEqual(size, buffer.len);
1614
-
1615
- try snap(@src(),
1616
- \\{"timestamp":"18446744073709551615","type":"two_phase_pending","ledger":4294967295,"transfer":{"id":"340282366920938463463374607431768211455","amount":"340282366920938463463374607431768211455","pending_id":"340282366920938463463374607431768211455","user_data_128":"340282366920938463463374607431768211455","user_data_64":"18446744073709551615","user_data_32":4294967295,"timeout":4294967295,"code":65535,"flags":65535,"timestamp":"18446744073709551615"},"debit_account":{"id":"340282366920938463463374607431768211455","debits_pending":"340282366920938463463374607431768211455","debits_posted":"340282366920938463463374607431768211455","credits_pending":"340282366920938463463374607431768211455","credits_posted":"340282366920938463463374607431768211455","user_data_128":"340282366920938463463374607431768211455","user_data_64":"18446744073709551615","user_data_32":4294967295,"code":65535,"flags":65535,"timestamp":"18446744073709551615"},"credit_account":{"id":"340282366920938463463374607431768211455","debits_pending":"340282366920938463463374607431768211455","debits_posted":"340282366920938463463374607431768211455","credits_pending":"340282366920938463463374607431768211455","credits_posted":"340282366920938463463374607431768211455","user_data_128":"340282366920938463463374607431768211455","user_data_64":"18446744073709551615","user_data_32":4294967295,"code":65535,"flags":65535,"timestamp":"18446744073709551615"}}
1617
- ).diff(buffer);
1618
- }
1619
- }
1620
-
1621
- test "amqp: metrics" {
1622
- var time_sim = fixtures.init_time(.{});
1623
- var summary: Metrics.TimingSummary = .{
1624
- .timer = .init(time_sim.time()),
1625
- };
1626
-
1627
- try testing.expectEqual(@as(u64, 0), summary.count);
1628
- try testing.expectEqual(@as(u64, 0), summary.event_count);
1629
- try testing.expectEqual(@as(u64, 0), summary.duration_sum.ns);
1630
- try testing.expect(summary.duration_max == null);
1631
- try testing.expect(summary.duration_min == null);
1632
-
1633
- summary.timing(10, .{ .ns = 50 });
1634
- try testing.expectEqual(@as(u64, 1), summary.count);
1635
- try testing.expectEqual(@as(u64, 10), summary.event_count);
1636
- try testing.expectEqual(@as(u64, 50), summary.duration_sum.ns);
1637
- try testing.expect(summary.duration_min != null);
1638
- try testing.expect(summary.duration_max != null);
1639
- try testing.expectEqual(@as(u64, 50), summary.duration_min.?.ns);
1640
- try testing.expectEqual(@as(u64, 50), summary.duration_max.?.ns);
1641
-
1642
- summary.timing(5, .{ .ns = 100 });
1643
- try testing.expectEqual(@as(u64, 2), summary.count);
1644
- try testing.expectEqual(@as(u64, 15), summary.event_count);
1645
- try testing.expectEqual(@as(u64, 150), summary.duration_sum.ns);
1646
- try testing.expect(summary.duration_min != null);
1647
- try testing.expect(summary.duration_max != null);
1648
- try testing.expectEqual(@as(u64, 50), summary.duration_min.?.ns);
1649
- try testing.expectEqual(@as(u64, 100), summary.duration_max.?.ns);
1650
-
1651
- summary.timing(0, .{ .ns = 10 });
1652
- try testing.expectEqual(@as(u64, 3), summary.count);
1653
- try testing.expectEqual(@as(u64, 15), summary.event_count);
1654
- try testing.expectEqual(@as(u64, 160), summary.duration_sum.ns);
1655
- try testing.expect(summary.duration_min != null);
1656
- try testing.expect(summary.duration_max != null);
1657
- try testing.expectEqual(@as(u64, 10), summary.duration_min.?.ns);
1658
- try testing.expectEqual(@as(u64, 100), summary.duration_max.?.ns);
1659
- }