itsi 0.1.9 → 0.1.12

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 (294) hide show
  1. checksums.yaml +4 -4
  2. data/Cargo.lock +1542 -43
  3. data/Itsi.rb +125 -0
  4. data/Rakefile +8 -4
  5. data/crates/itsi_error/src/lib.rs +9 -0
  6. data/crates/itsi_rb_helpers/Cargo.toml +1 -0
  7. data/crates/itsi_rb_helpers/src/heap_value.rs +18 -0
  8. data/crates/itsi_rb_helpers/src/lib.rs +59 -9
  9. data/crates/itsi_server/Cargo.toml +70 -28
  10. data/crates/itsi_server/src/lib.rs +80 -80
  11. data/crates/itsi_server/src/{body_proxy → ruby_types/itsi_body_proxy}/big_bytes.rs +10 -5
  12. data/{gems/server/ext/itsi_server/src/body_proxy/itsi_body_proxy.rs → crates/itsi_server/src/ruby_types/itsi_body_proxy/mod.rs} +22 -3
  13. data/crates/itsi_server/src/ruby_types/itsi_grpc_request.rs +147 -0
  14. data/crates/itsi_server/src/ruby_types/itsi_grpc_response.rs +19 -0
  15. data/crates/itsi_server/src/ruby_types/itsi_grpc_stream/mod.rs +216 -0
  16. data/crates/itsi_server/src/ruby_types/itsi_http_request.rs +282 -0
  17. data/crates/itsi_server/src/{response/itsi_response.rs → ruby_types/itsi_http_response.rs} +72 -41
  18. data/crates/itsi_server/src/ruby_types/itsi_server/file_watcher.rs +225 -0
  19. data/crates/itsi_server/src/ruby_types/itsi_server/itsi_server_config.rs +355 -0
  20. data/crates/itsi_server/src/ruby_types/itsi_server.rs +82 -0
  21. data/crates/itsi_server/src/ruby_types/mod.rs +55 -0
  22. data/crates/itsi_server/src/server/bind.rs +29 -17
  23. data/crates/itsi_server/src/server/byte_frame.rs +32 -0
  24. data/crates/itsi_server/src/server/cache_store.rs +74 -0
  25. data/crates/itsi_server/src/server/itsi_service.rs +172 -0
  26. data/crates/itsi_server/src/server/lifecycle_event.rs +3 -0
  27. data/crates/itsi_server/src/server/listener.rs +111 -11
  28. data/crates/itsi_server/src/server/middleware_stack/middleware.rs +153 -0
  29. data/crates/itsi_server/src/server/middleware_stack/middlewares/allow_list.rs +47 -0
  30. data/crates/itsi_server/src/server/middleware_stack/middlewares/auth_api_key.rs +58 -0
  31. data/crates/itsi_server/src/server/middleware_stack/middlewares/auth_basic.rs +82 -0
  32. data/crates/itsi_server/src/server/middleware_stack/middlewares/auth_jwt.rs +321 -0
  33. data/crates/itsi_server/src/server/middleware_stack/middlewares/cache_control.rs +139 -0
  34. data/crates/itsi_server/src/server/middleware_stack/middlewares/compression.rs +300 -0
  35. data/crates/itsi_server/src/server/middleware_stack/middlewares/cors.rs +287 -0
  36. data/crates/itsi_server/src/server/middleware_stack/middlewares/deny_list.rs +48 -0
  37. data/crates/itsi_server/src/server/middleware_stack/middlewares/error_response.rs +127 -0
  38. data/crates/itsi_server/src/server/middleware_stack/middlewares/etag.rs +191 -0
  39. data/crates/itsi_server/src/server/middleware_stack/middlewares/grpc_service.rs +72 -0
  40. data/crates/itsi_server/src/server/middleware_stack/middlewares/header_interpretation.rs +85 -0
  41. data/crates/itsi_server/src/server/middleware_stack/middlewares/intrusion_protection.rs +195 -0
  42. data/crates/itsi_server/src/server/middleware_stack/middlewares/log_requests.rs +82 -0
  43. data/crates/itsi_server/src/server/middleware_stack/middlewares/mod.rs +82 -0
  44. data/crates/itsi_server/src/server/middleware_stack/middlewares/proxy.rs +216 -0
  45. data/crates/itsi_server/src/server/middleware_stack/middlewares/rate_limit.rs +124 -0
  46. data/crates/itsi_server/src/server/middleware_stack/middlewares/redirect.rs +76 -0
  47. data/crates/itsi_server/src/server/middleware_stack/middlewares/request_headers.rs +43 -0
  48. data/crates/itsi_server/src/server/middleware_stack/middlewares/response_headers.rs +34 -0
  49. data/crates/itsi_server/src/server/middleware_stack/middlewares/ruby_app.rs +93 -0
  50. data/crates/itsi_server/src/server/middleware_stack/middlewares/static_assets.rs +162 -0
  51. data/crates/itsi_server/src/server/middleware_stack/middlewares/string_rewrite.rs +158 -0
  52. data/crates/itsi_server/src/server/middleware_stack/middlewares/token_source.rs +12 -0
  53. data/crates/itsi_server/src/server/middleware_stack/mod.rs +315 -0
  54. data/crates/itsi_server/src/server/mod.rs +8 -1
  55. data/crates/itsi_server/src/server/process_worker.rs +44 -11
  56. data/crates/itsi_server/src/server/rate_limiter.rs +565 -0
  57. data/crates/itsi_server/src/server/request_job.rs +11 -0
  58. data/crates/itsi_server/src/server/serve_strategy/cluster_mode.rs +129 -46
  59. data/crates/itsi_server/src/server/serve_strategy/mod.rs +9 -6
  60. data/crates/itsi_server/src/server/serve_strategy/single_mode.rs +325 -167
  61. data/crates/itsi_server/src/server/signal.rs +20 -4
  62. data/crates/itsi_server/src/server/static_file_server.rs +984 -0
  63. data/crates/itsi_server/src/server/thread_worker.rs +165 -88
  64. data/crates/itsi_server/src/server/tls.rs +1 -1
  65. data/crates/itsi_server/src/server/types.rs +43 -0
  66. data/crates/itsi_server/test.md +14 -0
  67. data/crates/itsi_tracing/Cargo.toml +1 -0
  68. data/crates/itsi_tracing/src/lib.rs +216 -45
  69. data/docs/.gitignore +7 -0
  70. data/docs/.gitpod.yml +15 -0
  71. data/docs/Itsi.rb +17 -0
  72. data/docs/content/_index.md +17 -0
  73. data/docs/content/about.md +6 -0
  74. data/docs/content/docs/_index.md +18 -0
  75. data/docs/content/docs/first-page.md +9 -0
  76. data/docs/content/docs/folder/_index.md +10 -0
  77. data/docs/content/docs/folder/leaf.md +7 -0
  78. data/docs/go.mod +5 -0
  79. data/docs/go.sum +2 -0
  80. data/docs/hugo.yaml +77 -0
  81. data/examples/static_assets_example.rb +83 -0
  82. data/gems/_index.md +18 -0
  83. data/gems/scheduler/CODE_OF_CONDUCT.md +7 -0
  84. data/gems/scheduler/Cargo.lock +75 -14
  85. data/gems/scheduler/README.md +5 -0
  86. data/gems/scheduler/_index.md +7 -0
  87. data/gems/scheduler/itsi-scheduler.gemspec +4 -1
  88. data/gems/scheduler/lib/itsi/scheduler/version.rb +1 -1
  89. data/gems/scheduler/lib/itsi/scheduler.rb +2 -2
  90. data/gems/scheduler/test/test_file_io.rb +0 -1
  91. data/gems/scheduler/test/test_itsi_scheduler.rb +1 -1
  92. data/gems/server/CHANGELOG.md +5 -0
  93. data/gems/server/CODE_OF_CONDUCT.md +7 -0
  94. data/gems/server/Cargo.lock +1543 -43
  95. data/gems/server/README.md +4 -0
  96. data/gems/server/_index.md +6 -0
  97. data/gems/server/exe/itsi +46 -57
  98. data/gems/server/itsi-server.gemspec +3 -2
  99. data/gems/server/lib/itsi/{request.rb → http_request.rb} +29 -5
  100. data/gems/server/lib/itsi/http_response.rb +39 -0
  101. data/gems/server/lib/itsi/server/Itsi.rb +119 -0
  102. data/gems/server/lib/itsi/server/config/dsl.rb +506 -0
  103. data/gems/server/lib/itsi/server/config.rb +131 -0
  104. data/gems/server/lib/itsi/server/default_app/default_app.rb +38 -0
  105. data/gems/server/lib/itsi/server/grpc_interface.rb +213 -0
  106. data/gems/server/lib/itsi/server/rack/handler/itsi.rb +9 -6
  107. data/gems/server/lib/itsi/server/rack_interface.rb +24 -9
  108. data/gems/server/lib/itsi/server/scheduler_interface.rb +1 -1
  109. data/gems/server/lib/itsi/server/scheduler_mode.rb +4 -0
  110. data/gems/server/lib/itsi/server/signal_trap.rb +6 -1
  111. data/gems/server/lib/itsi/server/version.rb +1 -1
  112. data/gems/server/lib/itsi/server.rb +75 -60
  113. data/gems/server/lib/itsi/standard_headers.rb +86 -0
  114. data/gems/server/test/helpers/test_helper.rb +14 -12
  115. data/gems/server/test/test_itsi_server.rb +21 -2
  116. data/lib/itsi/version.rb +1 -1
  117. data/sandbox/itsi_file/Gemfile +11 -0
  118. data/sandbox/itsi_file/Gemfile.lock +69 -0
  119. data/sandbox/itsi_file/Itsi.rb +276 -0
  120. data/sandbox/itsi_file/error.html +2 -0
  121. data/sandbox/itsi_file/organisations_controller.rb +20 -0
  122. data/sandbox/itsi_file/public/assets/image.png +0 -0
  123. data/sandbox/itsi_file/public/assets/index.html +1 -0
  124. data/sandbox/itsi_sandbox_async/Gemfile +1 -1
  125. data/sandbox/itsi_sandbox_hanami/Gemfile.lock +2 -2
  126. data/sandbox/itsi_sandbox_rack/Gemfile.lock +2 -2
  127. data/sandbox/itsi_sandbox_rack/config.ru +2 -15
  128. data/sandbox/itsi_sandbox_rails/.dockerignore +2 -5
  129. data/sandbox/itsi_sandbox_rails/.github/workflows/ci.yml +1 -1
  130. data/sandbox/itsi_sandbox_rails/.gitignore +2 -1
  131. data/sandbox/itsi_sandbox_rails/Dockerfile +6 -9
  132. data/sandbox/itsi_sandbox_rails/Gemfile +16 -22
  133. data/sandbox/itsi_sandbox_rails/Gemfile.lock +100 -225
  134. data/sandbox/itsi_sandbox_rails/app/assets/config/manifest.js +4 -0
  135. data/sandbox/itsi_sandbox_rails/app/assets/stylesheets/application.css +11 -6
  136. data/sandbox/itsi_sandbox_rails/app/channels/application_cable/channel.rb +4 -0
  137. data/sandbox/itsi_sandbox_rails/app/channels/application_cable/connection.rb +4 -0
  138. data/sandbox/itsi_sandbox_rails/app/controllers/live_controller.rb +7 -8
  139. data/sandbox/itsi_sandbox_rails/app/controllers/uploads_controller.rb +0 -3
  140. data/sandbox/itsi_sandbox_rails/app/views/layouts/application.html.erb +2 -7
  141. data/sandbox/itsi_sandbox_rails/bin/docker-entrypoint +3 -4
  142. data/sandbox/itsi_sandbox_rails/bin/setup +8 -5
  143. data/sandbox/itsi_sandbox_rails/config/application.rb +1 -35
  144. data/sandbox/itsi_sandbox_rails/config/cable.yml +3 -10
  145. data/sandbox/itsi_sandbox_rails/config/credentials.yml.enc +1 -1
  146. data/sandbox/itsi_sandbox_rails/config/database.yml +9 -19
  147. data/sandbox/itsi_sandbox_rails/config/environment.rb +1 -1
  148. data/sandbox/itsi_sandbox_rails/config/environments/development.rb +21 -12
  149. data/sandbox/itsi_sandbox_rails/config/environments/production.rb +49 -34
  150. data/sandbox/itsi_sandbox_rails/config/environments/test.rb +19 -5
  151. data/sandbox/itsi_sandbox_rails/config/initializers/assets.rb +5 -0
  152. data/sandbox/itsi_sandbox_rails/config/initializers/filter_parameter_logging.rb +1 -1
  153. data/sandbox/itsi_sandbox_rails/config/initializers/permissions_policy.rb +13 -0
  154. data/sandbox/itsi_sandbox_rails/config/puma.rb +2 -9
  155. data/sandbox/itsi_sandbox_rails/config.ru +0 -1
  156. data/sandbox/itsi_sandbox_rails/db/migrate/20250301041554_create_posts.rb +1 -1
  157. data/sandbox/itsi_sandbox_rails/db/schema.rb +2 -2
  158. data/sandbox/itsi_sandbox_rails/lib/assets/.keep +0 -0
  159. data/sandbox/itsi_sandbox_rails/public/404.html +66 -113
  160. data/sandbox/itsi_sandbox_rails/public/406-unsupported-browser.html +65 -113
  161. data/sandbox/itsi_sandbox_rails/public/422.html +66 -113
  162. data/sandbox/itsi_sandbox_rails/public/500.html +65 -113
  163. data/sandbox/itsi_sandbox_rails/public/icon.png +0 -0
  164. data/sandbox/itsi_sandbox_rails/public/icon.svg +2 -2
  165. data/sandbox/itsi_sandbox_rails/test/channels/application_cable/connection_test.rb +13 -0
  166. data/sandbox/itsi_sandbox_roda/Gemfile.lock +3 -10
  167. data/tasks.txt +72 -12
  168. metadata +94 -139
  169. data/crates/itsi_server/src/body_proxy/itsi_body_proxy.rs +0 -122
  170. data/crates/itsi_server/src/body_proxy/mod.rs +0 -2
  171. data/crates/itsi_server/src/request/itsi_request.rs +0 -305
  172. data/crates/itsi_server/src/request/mod.rs +0 -1
  173. data/crates/itsi_server/src/response/mod.rs +0 -1
  174. data/crates/itsi_server/src/server/itsi_server.rs +0 -294
  175. data/gems/scheduler/ext/itsi_error/Cargo.lock +0 -368
  176. data/gems/scheduler/ext/itsi_error/Cargo.toml +0 -11
  177. data/gems/scheduler/ext/itsi_error/src/from.rs +0 -68
  178. data/gems/scheduler/ext/itsi_error/src/lib.rs +0 -24
  179. data/gems/scheduler/ext/itsi_instrument_entry/Cargo.toml +0 -15
  180. data/gems/scheduler/ext/itsi_instrument_entry/src/lib.rs +0 -31
  181. data/gems/scheduler/ext/itsi_rb_helpers/Cargo.lock +0 -355
  182. data/gems/scheduler/ext/itsi_rb_helpers/Cargo.toml +0 -10
  183. data/gems/scheduler/ext/itsi_rb_helpers/src/heap_value.rs +0 -121
  184. data/gems/scheduler/ext/itsi_rb_helpers/src/lib.rs +0 -178
  185. data/gems/scheduler/ext/itsi_scheduler/Cargo.toml +0 -24
  186. data/gems/scheduler/ext/itsi_scheduler/extconf.rb +0 -6
  187. data/gems/scheduler/ext/itsi_scheduler/src/itsi_scheduler/io_helpers.rs +0 -56
  188. data/gems/scheduler/ext/itsi_scheduler/src/itsi_scheduler/io_waiter.rs +0 -44
  189. data/gems/scheduler/ext/itsi_scheduler/src/itsi_scheduler/timer.rs +0 -44
  190. data/gems/scheduler/ext/itsi_scheduler/src/itsi_scheduler.rs +0 -308
  191. data/gems/scheduler/ext/itsi_scheduler/src/lib.rs +0 -38
  192. data/gems/scheduler/ext/itsi_server/Cargo.lock +0 -2956
  193. data/gems/scheduler/ext/itsi_server/Cargo.toml +0 -47
  194. data/gems/scheduler/ext/itsi_server/extconf.rb +0 -6
  195. data/gems/scheduler/ext/itsi_server/src/body_proxy/big_bytes.rs +0 -104
  196. data/gems/scheduler/ext/itsi_server/src/body_proxy/itsi_body_proxy.rs +0 -122
  197. data/gems/scheduler/ext/itsi_server/src/body_proxy/mod.rs +0 -2
  198. data/gems/scheduler/ext/itsi_server/src/env.rs +0 -43
  199. data/gems/scheduler/ext/itsi_server/src/lib.rs +0 -112
  200. data/gems/scheduler/ext/itsi_server/src/request/itsi_request.rs +0 -305
  201. data/gems/scheduler/ext/itsi_server/src/request/mod.rs +0 -1
  202. data/gems/scheduler/ext/itsi_server/src/response/itsi_response.rs +0 -357
  203. data/gems/scheduler/ext/itsi_server/src/response/mod.rs +0 -1
  204. data/gems/scheduler/ext/itsi_server/src/server/bind.rs +0 -170
  205. data/gems/scheduler/ext/itsi_server/src/server/bind_protocol.rs +0 -37
  206. data/gems/scheduler/ext/itsi_server/src/server/io_stream.rs +0 -104
  207. data/gems/scheduler/ext/itsi_server/src/server/itsi_server.rs +0 -294
  208. data/gems/scheduler/ext/itsi_server/src/server/lifecycle_event.rs +0 -9
  209. data/gems/scheduler/ext/itsi_server/src/server/listener.rs +0 -318
  210. data/gems/scheduler/ext/itsi_server/src/server/mod.rs +0 -11
  211. data/gems/scheduler/ext/itsi_server/src/server/process_worker.rs +0 -196
  212. data/gems/scheduler/ext/itsi_server/src/server/serve_strategy/cluster_mode.rs +0 -254
  213. data/gems/scheduler/ext/itsi_server/src/server/serve_strategy/mod.rs +0 -27
  214. data/gems/scheduler/ext/itsi_server/src/server/serve_strategy/single_mode.rs +0 -263
  215. data/gems/scheduler/ext/itsi_server/src/server/signal.rs +0 -77
  216. data/gems/scheduler/ext/itsi_server/src/server/thread_worker.rs +0 -367
  217. data/gems/scheduler/ext/itsi_server/src/server/tls/locked_dir_cache.rs +0 -132
  218. data/gems/scheduler/ext/itsi_server/src/server/tls.rs +0 -265
  219. data/gems/scheduler/ext/itsi_tracing/Cargo.lock +0 -274
  220. data/gems/scheduler/ext/itsi_tracing/Cargo.toml +0 -16
  221. data/gems/scheduler/ext/itsi_tracing/src/lib.rs +0 -58
  222. data/gems/server/ext/itsi_error/Cargo.lock +0 -368
  223. data/gems/server/ext/itsi_error/Cargo.toml +0 -11
  224. data/gems/server/ext/itsi_error/src/from.rs +0 -68
  225. data/gems/server/ext/itsi_error/src/lib.rs +0 -24
  226. data/gems/server/ext/itsi_instrument_entry/Cargo.toml +0 -15
  227. data/gems/server/ext/itsi_instrument_entry/src/lib.rs +0 -31
  228. data/gems/server/ext/itsi_rb_helpers/Cargo.lock +0 -355
  229. data/gems/server/ext/itsi_rb_helpers/Cargo.toml +0 -10
  230. data/gems/server/ext/itsi_rb_helpers/src/heap_value.rs +0 -121
  231. data/gems/server/ext/itsi_rb_helpers/src/lib.rs +0 -178
  232. data/gems/server/ext/itsi_scheduler/Cargo.toml +0 -24
  233. data/gems/server/ext/itsi_scheduler/extconf.rb +0 -6
  234. data/gems/server/ext/itsi_scheduler/src/itsi_scheduler/io_helpers.rs +0 -56
  235. data/gems/server/ext/itsi_scheduler/src/itsi_scheduler/io_waiter.rs +0 -44
  236. data/gems/server/ext/itsi_scheduler/src/itsi_scheduler/timer.rs +0 -44
  237. data/gems/server/ext/itsi_scheduler/src/itsi_scheduler.rs +0 -308
  238. data/gems/server/ext/itsi_scheduler/src/lib.rs +0 -38
  239. data/gems/server/ext/itsi_server/Cargo.lock +0 -2956
  240. data/gems/server/ext/itsi_server/Cargo.toml +0 -47
  241. data/gems/server/ext/itsi_server/extconf.rb +0 -6
  242. data/gems/server/ext/itsi_server/src/body_proxy/big_bytes.rs +0 -104
  243. data/gems/server/ext/itsi_server/src/body_proxy/mod.rs +0 -2
  244. data/gems/server/ext/itsi_server/src/env.rs +0 -43
  245. data/gems/server/ext/itsi_server/src/lib.rs +0 -112
  246. data/gems/server/ext/itsi_server/src/request/itsi_request.rs +0 -305
  247. data/gems/server/ext/itsi_server/src/request/mod.rs +0 -1
  248. data/gems/server/ext/itsi_server/src/response/itsi_response.rs +0 -357
  249. data/gems/server/ext/itsi_server/src/response/mod.rs +0 -1
  250. data/gems/server/ext/itsi_server/src/server/bind.rs +0 -170
  251. data/gems/server/ext/itsi_server/src/server/bind_protocol.rs +0 -37
  252. data/gems/server/ext/itsi_server/src/server/io_stream.rs +0 -104
  253. data/gems/server/ext/itsi_server/src/server/itsi_server.rs +0 -294
  254. data/gems/server/ext/itsi_server/src/server/lifecycle_event.rs +0 -9
  255. data/gems/server/ext/itsi_server/src/server/listener.rs +0 -318
  256. data/gems/server/ext/itsi_server/src/server/mod.rs +0 -11
  257. data/gems/server/ext/itsi_server/src/server/process_worker.rs +0 -196
  258. data/gems/server/ext/itsi_server/src/server/serve_strategy/cluster_mode.rs +0 -254
  259. data/gems/server/ext/itsi_server/src/server/serve_strategy/mod.rs +0 -27
  260. data/gems/server/ext/itsi_server/src/server/serve_strategy/single_mode.rs +0 -263
  261. data/gems/server/ext/itsi_server/src/server/signal.rs +0 -77
  262. data/gems/server/ext/itsi_server/src/server/thread_worker.rs +0 -367
  263. data/gems/server/ext/itsi_server/src/server/tls/locked_dir_cache.rs +0 -132
  264. data/gems/server/ext/itsi_server/src/server/tls.rs +0 -265
  265. data/gems/server/ext/itsi_tracing/Cargo.lock +0 -274
  266. data/gems/server/ext/itsi_tracing/Cargo.toml +0 -16
  267. data/gems/server/ext/itsi_tracing/src/lib.rs +0 -58
  268. data/gems/server/lib/itsi/stream_io.rb +0 -38
  269. data/sandbox/itsi_sandbox_rails/.kamal/hooks/docker-setup.sample +0 -3
  270. data/sandbox/itsi_sandbox_rails/.kamal/hooks/post-app-boot.sample +0 -3
  271. data/sandbox/itsi_sandbox_rails/.kamal/hooks/post-deploy.sample +0 -14
  272. data/sandbox/itsi_sandbox_rails/.kamal/hooks/post-proxy-reboot.sample +0 -3
  273. data/sandbox/itsi_sandbox_rails/.kamal/hooks/pre-app-boot.sample +0 -3
  274. data/sandbox/itsi_sandbox_rails/.kamal/hooks/pre-build.sample +0 -51
  275. data/sandbox/itsi_sandbox_rails/.kamal/hooks/pre-connect.sample +0 -47
  276. data/sandbox/itsi_sandbox_rails/.kamal/hooks/pre-deploy.sample +0 -109
  277. data/sandbox/itsi_sandbox_rails/.kamal/hooks/pre-proxy-reboot.sample +0 -3
  278. data/sandbox/itsi_sandbox_rails/.kamal/secrets +0 -17
  279. data/sandbox/itsi_sandbox_rails/bin/dev +0 -2
  280. data/sandbox/itsi_sandbox_rails/bin/jobs +0 -6
  281. data/sandbox/itsi_sandbox_rails/bin/kamal +0 -27
  282. data/sandbox/itsi_sandbox_rails/bin/thrust +0 -5
  283. data/sandbox/itsi_sandbox_rails/config/cache.yml +0 -16
  284. data/sandbox/itsi_sandbox_rails/config/deploy.yml +0 -116
  285. data/sandbox/itsi_sandbox_rails/config/queue.yml +0 -18
  286. data/sandbox/itsi_sandbox_rails/config/recurring.yml +0 -10
  287. data/sandbox/itsi_sandbox_rails/db/cable_schema.rb +0 -11
  288. data/sandbox/itsi_sandbox_rails/db/cache_schema.rb +0 -14
  289. data/sandbox/itsi_sandbox_rails/db/queue_schema.rb +0 -129
  290. data/sandbox/itsi_sandbox_rails/public/400.html +0 -114
  291. data/sandbox/itsi_sandbox_rails/test/fixtures/posts.yml +0 -9
  292. data/sandbox/itsi_sandbox_rails/test/models/post_test.rb +0 -7
  293. /data/{sandbox/itsi_sandbox_rails/script/.keep → crates/_index.md} +0 -0
  294. /data/gems/server/lib/itsi/{index.html → server/default_app/index.html} +0 -0
@@ -1,263 +0,0 @@
1
- use crate::{
2
- request::itsi_request::ItsiRequest,
3
- server::{
4
- io_stream::IoStream,
5
- itsi_server::{RequestJob, Server},
6
- lifecycle_event::LifecycleEvent,
7
- listener::{Listener, ListenerInfo},
8
- thread_worker::{build_thread_workers, ThreadWorker},
9
- },
10
- };
11
- use http::Request;
12
- use hyper::{body::Incoming, service::service_fn};
13
- use hyper_util::{
14
- rt::{TokioExecutor, TokioIo, TokioTimer},
15
- server::conn::auto::Builder,
16
- };
17
- use itsi_error::{ItsiError, Result};
18
- use itsi_tracing::{debug, error, info};
19
- use nix::unistd::Pid;
20
- use std::{
21
- num::NonZeroU8,
22
- pin::Pin,
23
- sync::Arc,
24
- time::{Duration, Instant},
25
- };
26
- use tokio::{
27
- runtime::{Builder as RuntimeBuilder, Runtime},
28
- sync::{
29
- broadcast,
30
- watch::{self, Sender},
31
- },
32
- task::JoinSet,
33
- };
34
- use tracing::instrument;
35
-
36
- pub struct SingleMode {
37
- pub executor: Builder<TokioExecutor>,
38
- pub server: Arc<Server>,
39
- pub sender: async_channel::Sender<RequestJob>,
40
- pub(crate) listeners: Arc<Vec<Arc<Listener>>>,
41
- pub(crate) thread_workers: Arc<Vec<ThreadWorker>>,
42
- pub(crate) lifecycle_channel: broadcast::Sender<LifecycleEvent>,
43
- }
44
-
45
- pub enum RunningPhase {
46
- Running,
47
- ShutdownPending,
48
- Shutdown,
49
- }
50
-
51
- impl SingleMode {
52
- #[instrument(parent=None, skip_all, fields(pid=format!("{:?}", Pid::this())))]
53
- pub(crate) fn new(
54
- server: Arc<Server>,
55
- listeners: Arc<Vec<Arc<Listener>>>,
56
- lifecycle_channel: broadcast::Sender<LifecycleEvent>,
57
- ) -> Result<Self> {
58
- let (thread_workers, sender) = build_thread_workers(
59
- Pid::this(),
60
- NonZeroU8::try_from(server.threads).unwrap(),
61
- server.app.clone(),
62
- server.scheduler_class.clone(),
63
- )?;
64
- Ok(Self {
65
- executor: Builder::new(TokioExecutor::new()),
66
- listeners,
67
- server,
68
- sender,
69
- thread_workers,
70
- lifecycle_channel,
71
- })
72
- }
73
-
74
- pub fn build_runtime(&self) -> Runtime {
75
- let mut builder: RuntimeBuilder = RuntimeBuilder::new_current_thread();
76
- builder
77
- .thread_name("itsi-server-accept-loop")
78
- .thread_stack_size(3 * 1024 * 1024)
79
- .enable_io()
80
- .enable_time()
81
- .build()
82
- .expect("Failed to build Tokio runtime")
83
- }
84
-
85
- pub fn stop(&self) -> Result<()> {
86
- self.lifecycle_channel
87
- .send(LifecycleEvent::Shutdown)
88
- .expect("Failed to send shutdown event");
89
- Ok(())
90
- }
91
-
92
- #[instrument(parent=None, skip(self), fields(pid=format!("{}", Pid::this())))]
93
- pub fn run(self: Arc<Self>) -> Result<()> {
94
- let mut listener_task_set = JoinSet::new();
95
- let self_ref = Arc::new(self);
96
- let runtime = self_ref.build_runtime();
97
-
98
- runtime.block_on(async {
99
- let tokio_listeners = self_ref
100
- .listeners
101
- .iter()
102
- .map(|list| Arc::new(list.to_tokio_listener()))
103
- .collect::<Vec<_>>();
104
- let (shutdown_sender, _) = watch::channel::<RunningPhase>(RunningPhase::Running);
105
- for listener in tokio_listeners.iter() {
106
- let mut lifecycle_rx = self_ref.lifecycle_channel.subscribe();
107
- let listener_info = Arc::new(listener.listener_info());
108
- let self_ref = self_ref.clone();
109
- let listener = listener.clone();
110
- let shutdown_sender = shutdown_sender.clone();
111
-
112
-
113
- let listener_clone = listener.clone();
114
- let mut shutdown_receiver = shutdown_sender.clone().subscribe();
115
- let shutdown_receiver_clone = shutdown_receiver.clone();
116
- listener_task_set.spawn(async move {
117
- listener_clone.spawn_state_task(shutdown_receiver_clone).await;
118
- });
119
-
120
- listener_task_set.spawn(async move {
121
- let strategy = self_ref.clone();
122
- loop {
123
- tokio::select! {
124
- accept_result = listener.accept() => match accept_result {
125
- Ok(accept_result) => {
126
- match strategy.serve_connection(accept_result, listener_info.clone(), shutdown_receiver.clone()).await {
127
- Ok(_) => {
128
- debug!("Connection accepted and served");
129
- },
130
- Err(e) => error!("Error in serve_connection {:?}", e)
131
- }
132
- },
133
- Err(e) => debug!("Listener.accept failed {:?}", e),
134
- },
135
- _ = shutdown_receiver.changed() => {
136
- break;
137
- }
138
- lifecycle_event = lifecycle_rx.recv() => match lifecycle_event{
139
- Ok(lifecycle_event) => {
140
- if let Err(e) = strategy.handle_lifecycle_event(lifecycle_event, shutdown_sender.clone()).await{
141
- match e {
142
- ItsiError::Break() => break,
143
- _ => error!("Error in handle_lifecycle_event {:?}", e)
144
- }
145
- }
146
-
147
- },
148
- Err(e) => error!("Error receiving lifecycle_event: {:?}", e),
149
- }
150
- }
151
- }
152
- });
153
-
154
- }
155
-
156
- while let Some(_res) = listener_task_set.join_next().await {}
157
-
158
- });
159
- runtime.shutdown_timeout(Duration::from_millis(100));
160
-
161
- info!("Runtime has shut down");
162
- Ok(())
163
- }
164
-
165
- pub(crate) async fn serve_connection(
166
- &self,
167
- stream: IoStream,
168
- listener: Arc<ListenerInfo>,
169
- shutdown_channel: watch::Receiver<RunningPhase>,
170
- ) -> Result<()> {
171
- let sender_clone = self.sender.clone();
172
- let addr = stream.addr();
173
- let io: TokioIo<Pin<Box<IoStream>>> = TokioIo::new(Box::pin(stream));
174
- let server = self.server.clone();
175
- let executor = self.executor.clone();
176
- let mut shutdown_channel_clone = shutdown_channel.clone();
177
- tokio::spawn(async move {
178
- let server = server.clone();
179
- let mut executor = executor.clone();
180
- let mut binding = executor.http1();
181
- let shutdown_channel = shutdown_channel_clone.clone();
182
- let mut serve = Box::pin(
183
- binding
184
- .timer(TokioTimer::new())
185
- .header_read_timeout(Duration::from_secs(1))
186
- .serve_connection_with_upgrades(
187
- io,
188
- service_fn(move |hyper_request: Request<Incoming>| {
189
- ItsiRequest::process_request(
190
- hyper_request,
191
- sender_clone.clone(),
192
- server.clone(),
193
- listener.clone(),
194
- addr.clone(),
195
- shutdown_channel.clone(),
196
- )
197
- }),
198
- ),
199
- );
200
-
201
- tokio::select! {
202
- // Await the connection finishing naturally.
203
- res = &mut serve => {
204
- match res{
205
- Ok(()) => {
206
- debug!("Connection closed normally")
207
- },
208
- Err(res) => {
209
- debug!("Connection finished with error: {:?}", res)
210
- }
211
- }
212
- serve.as_mut().graceful_shutdown();
213
- },
214
- // A lifecycle event triggers shutdown.
215
- _ = shutdown_channel_clone.changed() => {
216
- // Initiate graceful shutdown.
217
- serve.as_mut().graceful_shutdown();
218
- // Now await the connection to finish shutting down.
219
- if let Err(e) = serve.await {
220
- debug!("Connection shutdown error: {:?}", e);
221
- }
222
- }
223
- }
224
- });
225
- Ok(())
226
- }
227
-
228
- pub async fn handle_lifecycle_event(
229
- &self,
230
- lifecycle_event: LifecycleEvent,
231
- shutdown_sender: Sender<RunningPhase>,
232
- ) -> Result<()> {
233
- info!("Handling lifecycle event: {:?}", lifecycle_event);
234
- if let LifecycleEvent::Shutdown = lifecycle_event {
235
- shutdown_sender.send(RunningPhase::ShutdownPending).ok();
236
- let deadline = Instant::now() + Duration::from_secs_f64(self.server.shutdown_timeout);
237
- for worker in &*self.thread_workers {
238
- worker.request_shutdown().await;
239
- }
240
- while Instant::now() < deadline {
241
- tokio::time::sleep(Duration::from_millis(50)).await;
242
- let alive_threads = self
243
- .thread_workers
244
- .iter()
245
- .filter(|worker| worker.poll_shutdown(deadline))
246
- .count();
247
- if alive_threads == 0 {
248
- break;
249
- }
250
- tokio::time::sleep(Duration::from_millis(200)).await;
251
- }
252
-
253
- info!("Sending shutdown signal");
254
- shutdown_sender.send(RunningPhase::Shutdown).ok();
255
- self.thread_workers.iter().for_each(|worker| {
256
- worker.poll_shutdown(deadline);
257
- });
258
-
259
- return Err(ItsiError::Break());
260
- }
261
- Ok(())
262
- }
263
- }
@@ -1,77 +0,0 @@
1
- use std::sync::{atomic::AtomicI8, LazyLock};
2
-
3
- use nix::libc::{self, sighandler_t};
4
- use tokio::sync::{self, broadcast};
5
-
6
- use super::lifecycle_event::LifecycleEvent;
7
-
8
- pub static SIGNAL_HANDLER_CHANNEL: LazyLock<(
9
- broadcast::Sender<LifecycleEvent>,
10
- broadcast::Receiver<LifecycleEvent>,
11
- )> = LazyLock::new(|| sync::broadcast::channel(5));
12
-
13
- pub fn send_shutdown_event() {
14
- SIGNAL_HANDLER_CHANNEL
15
- .0
16
- .send(LifecycleEvent::Shutdown)
17
- .expect("Failed to send shutdown event");
18
- }
19
-
20
- pub static SIGINT_COUNT: AtomicI8 = AtomicI8::new(0);
21
- fn receive_signal(signum: i32, _: sighandler_t) {
22
- SIGINT_COUNT.fetch_add(-1, std::sync::atomic::Ordering::SeqCst);
23
- match signum {
24
- libc::SIGTERM | libc::SIGINT => {
25
- SIGINT_COUNT.fetch_add(2, std::sync::atomic::Ordering::SeqCst);
26
- if SIGINT_COUNT.load(std::sync::atomic::Ordering::SeqCst) < 2 {
27
- SIGNAL_HANDLER_CHANNEL.0.send(LifecycleEvent::Shutdown).ok();
28
- } else {
29
- // Not messing about. Force shutdown.
30
- SIGNAL_HANDLER_CHANNEL
31
- .0
32
- .send(LifecycleEvent::ForceShutdown)
33
- .ok();
34
- }
35
- }
36
- libc::SIGUSR1 => {
37
- SIGNAL_HANDLER_CHANNEL.0.send(LifecycleEvent::Restart).ok();
38
- }
39
- libc::SIGTTIN => {
40
- SIGNAL_HANDLER_CHANNEL
41
- .0
42
- .send(LifecycleEvent::IncreaseWorkers)
43
- .ok();
44
- }
45
- libc::SIGTTOU => {
46
- SIGNAL_HANDLER_CHANNEL
47
- .0
48
- .send(LifecycleEvent::DecreaseWorkers)
49
- .ok();
50
- }
51
- _ => {}
52
- }
53
- }
54
-
55
- pub fn reset_signal_handlers() -> bool {
56
- SIGINT_COUNT.store(0, std::sync::atomic::Ordering::SeqCst);
57
- unsafe {
58
- libc::signal(libc::SIGTERM, receive_signal as usize);
59
- libc::signal(libc::SIGINT, receive_signal as usize);
60
- libc::signal(libc::SIGUSR1, receive_signal as usize);
61
- libc::signal(libc::SIGUSR2, receive_signal as usize);
62
- libc::signal(libc::SIGTTIN, receive_signal as usize);
63
- libc::signal(libc::SIGTTOU, receive_signal as usize);
64
- }
65
- true
66
- }
67
-
68
- pub fn clear_signal_handlers() {
69
- unsafe {
70
- libc::signal(libc::SIGTERM, libc::SIG_DFL);
71
- libc::signal(libc::SIGINT, libc::SIG_DFL);
72
- libc::signal(libc::SIGUSR1, libc::SIG_DFL);
73
- libc::signal(libc::SIGUSR2, libc::SIG_DFL);
74
- libc::signal(libc::SIGTTIN, libc::SIG_DFL);
75
- libc::signal(libc::SIGTTOU, libc::SIG_DFL);
76
- }
77
- }
@@ -1,367 +0,0 @@
1
- use super::itsi_server::RequestJob;
2
- use crate::{request::itsi_request::ItsiRequest, ITSI_SERVER};
3
- use itsi_rb_helpers::{
4
- call_with_gvl, call_without_gvl, create_ruby_thread, kill_threads, HeapVal, HeapValue,
5
- };
6
- use itsi_tracing::{debug, error, info, warn};
7
- use magnus::{
8
- error::Result,
9
- value::{InnerValue, Lazy, LazyId, Opaque, ReprValue},
10
- Module, RClass, Ruby, Thread, Value,
11
- };
12
- use nix::unistd::Pid;
13
- use parking_lot::{Mutex, RwLock};
14
- use std::{
15
- num::NonZeroU8,
16
- ops::Deref,
17
- sync::{
18
- atomic::{AtomicBool, Ordering},
19
- Arc,
20
- },
21
- thread,
22
- time::{Duration, Instant},
23
- };
24
- use tokio::{runtime::Builder as RuntimeBuilder, sync::watch};
25
- use tracing::instrument;
26
- pub struct ThreadWorker {
27
- pub id: String,
28
- pub app: Opaque<Value>,
29
- pub receiver: Arc<async_channel::Receiver<RequestJob>>,
30
- pub sender: async_channel::Sender<RequestJob>,
31
- pub thread: RwLock<Option<HeapValue<Thread>>>,
32
- pub terminated: Arc<AtomicBool>,
33
- pub scheduler_class: Option<Opaque<Value>>,
34
- }
35
-
36
- static ID_CALL: LazyId = LazyId::new("call");
37
- static ID_ALIVE: LazyId = LazyId::new("alive?");
38
- static ID_SCHEDULER: LazyId = LazyId::new("scheduler");
39
- static ID_SCHEDULE: LazyId = LazyId::new("schedule");
40
- static ID_BLOCK: LazyId = LazyId::new("block");
41
- static ID_YIELD: LazyId = LazyId::new("yield");
42
- static ID_CONST_GET: LazyId = LazyId::new("const_get");
43
- static CLASS_FIBER: Lazy<RClass> = Lazy::new(|ruby| {
44
- ruby.module_kernel()
45
- .const_get::<_, RClass>("Fiber")
46
- .unwrap()
47
- });
48
-
49
- pub struct TerminateWakerSignal(bool);
50
-
51
- #[instrument(name = "Boot", parent=None, skip(threads, app, pid, scheduler_class))]
52
- pub fn build_thread_workers(
53
- pid: Pid,
54
- threads: NonZeroU8,
55
- app: HeapVal,
56
- scheduler_class: Option<String>,
57
- ) -> Result<(Arc<Vec<ThreadWorker>>, async_channel::Sender<RequestJob>)> {
58
- let (sender, receiver) = async_channel::bounded(20);
59
- let receiver_ref = Arc::new(receiver);
60
- let sender_ref = sender;
61
- let (app, scheduler_class) = load_app(app, scheduler_class)?;
62
- Ok((
63
- Arc::new(
64
- (1..=u8::from(threads))
65
- .map(|id| {
66
- info!(pid = pid.as_raw(), id, "Thread");
67
- ThreadWorker::new(
68
- format!("{:?}#{:?}", pid, id),
69
- app,
70
- receiver_ref.clone(),
71
- sender_ref.clone(),
72
- scheduler_class,
73
- )
74
- })
75
- .collect::<Result<Vec<_>>>()?,
76
- ),
77
- sender_ref,
78
- ))
79
- }
80
-
81
- pub fn load_app(
82
- app: HeapVal,
83
- scheduler_class: Option<String>,
84
- ) -> Result<(Opaque<Value>, Option<Opaque<Value>>)> {
85
- call_with_gvl(|ruby| {
86
- let app = Opaque::from(
87
- app.funcall::<_, _, Value>(*ID_CALL, ())
88
- .expect("Couldn't load app"),
89
- );
90
- let scheduler_class = if let Some(scheduler_class) = scheduler_class {
91
- Some(Opaque::from(
92
- ruby.module_kernel()
93
- .funcall::<_, _, Value>(*ID_CONST_GET, (scheduler_class,))?,
94
- ))
95
- } else {
96
- None
97
- };
98
- Ok((app, scheduler_class))
99
- })
100
- }
101
- impl ThreadWorker {
102
- pub fn new(
103
- id: String,
104
- app: Opaque<Value>,
105
- receiver: Arc<async_channel::Receiver<RequestJob>>,
106
- sender: async_channel::Sender<RequestJob>,
107
- scheduler_class: Option<Opaque<Value>>,
108
- ) -> Result<Self> {
109
- let mut worker = Self {
110
- id,
111
- app,
112
- receiver,
113
- sender,
114
- thread: RwLock::new(None),
115
- terminated: Arc::new(AtomicBool::new(false)),
116
- scheduler_class,
117
- };
118
- worker.run()?;
119
- Ok(worker)
120
- }
121
-
122
- #[instrument(skip(self), fields(id = self.id))]
123
- pub async fn request_shutdown(&self) {
124
- match self.sender.send(RequestJob::Shutdown).await {
125
- Ok(_) => {}
126
- Err(err) => error!("Failed to send shutdown request: {}", err),
127
- };
128
- info!("Requesting shutdown");
129
- }
130
-
131
- #[instrument(skip(self, deadline), fields(id = self.id))]
132
- pub fn poll_shutdown(&self, deadline: Instant) -> bool {
133
- call_with_gvl(|_ruby| {
134
- if let Some(thread) = self.thread.read().deref() {
135
- if Instant::now() > deadline {
136
- warn!("Worker shutdown timed out. Killing thread");
137
- self.terminated.store(true, Ordering::SeqCst);
138
- kill_threads(vec![thread.as_value()]);
139
- }
140
- if thread.funcall::<_, _, bool>(*ID_ALIVE, ()).unwrap_or(false) {
141
- return true;
142
- }
143
- info!("Thread has shut down");
144
- }
145
- self.thread.write().take();
146
-
147
- false
148
- })
149
- }
150
-
151
- pub fn run(&mut self) -> Result<()> {
152
- let id = self.id.clone();
153
- let app = self.app;
154
- let receiver = self.receiver.clone();
155
- let terminated = self.terminated.clone();
156
- let scheduler_class = self.scheduler_class;
157
- call_with_gvl(|_| {
158
- *self.thread.write() = Some(
159
- create_ruby_thread(move || {
160
- if let Some(scheduler_class) = scheduler_class {
161
- if let Err(err) =
162
- Self::fiber_accept_loop(id, app, receiver, scheduler_class, terminated)
163
- {
164
- error!("Error in fiber_accept_loop: {:?}", err);
165
- }
166
- } else {
167
- Self::accept_loop(id, app, receiver, terminated);
168
- }
169
- })
170
- .into(),
171
- );
172
- Ok::<(), magnus::Error>(())
173
- })?;
174
- Ok(())
175
- }
176
-
177
- pub fn build_scheduler_proc(
178
- app: Opaque<Value>,
179
- leader: &Arc<Mutex<Option<RequestJob>>>,
180
- receiver: &Arc<async_channel::Receiver<RequestJob>>,
181
- terminated: &Arc<AtomicBool>,
182
- waker_sender: &watch::Sender<TerminateWakerSignal>,
183
- ) -> magnus::block::Proc {
184
- let leader = leader.clone();
185
- let receiver = receiver.clone();
186
- let terminated = terminated.clone();
187
- let waker_sender = waker_sender.clone();
188
- Ruby::get().unwrap().proc_from_fn(move |ruby, _args, _blk| {
189
- let scheduler = ruby
190
- .get_inner(&CLASS_FIBER)
191
- .funcall::<_, _, Value>(*ID_SCHEDULER, ())
192
- .unwrap();
193
- let server = ruby.get_inner(&ITSI_SERVER);
194
- let thread_current = ruby.thread_current();
195
- let leader_clone = leader.clone();
196
- let receiver = receiver.clone();
197
- let terminated = terminated.clone();
198
- let waker_sender = waker_sender.clone();
199
- let mut batch = Vec::with_capacity(MAX_BATCH_SIZE as usize);
200
-
201
- static MAX_BATCH_SIZE: i32 = 25;
202
- call_without_gvl(move || loop {
203
- let mut idle_counter = 0;
204
- if let Some(v) = leader_clone.lock().take() {
205
- match v {
206
- RequestJob::ProcessRequest(itsi_request) => {
207
- batch.push(RequestJob::ProcessRequest(itsi_request))
208
- }
209
- RequestJob::Shutdown => {
210
- waker_sender.send(TerminateWakerSignal(true)).unwrap();
211
- break;
212
- }
213
- }
214
- }
215
- for _ in 0..MAX_BATCH_SIZE {
216
- if let Ok(req) = receiver.try_recv() {
217
- batch.push(req);
218
- } else {
219
- break;
220
- }
221
- }
222
-
223
- let shutdown_requested = call_with_gvl(|_| {
224
- for req in batch.drain(..) {
225
- match req {
226
- RequestJob::ProcessRequest(request) => {
227
- let response = request.response.clone();
228
- if let Err(err) =
229
- server.funcall::<_, _, Value>(*ID_SCHEDULE, (app, request))
230
- {
231
- ItsiRequest::internal_error(ruby, response, err)
232
- }
233
- }
234
- RequestJob::Shutdown => return true,
235
- }
236
- }
237
- false
238
- });
239
-
240
- if shutdown_requested || terminated.load(Ordering::Relaxed) {
241
- waker_sender.send(TerminateWakerSignal(true)).unwrap();
242
- break;
243
- }
244
-
245
- let yield_result = if receiver.is_empty() {
246
- waker_sender.send(TerminateWakerSignal(false)).unwrap();
247
- idle_counter = (idle_counter + 1) % 100;
248
- call_with_gvl(|ruby| {
249
- if idle_counter == 0 {
250
- ruby.gc_start();
251
- }
252
- scheduler.funcall::<_, _, Value>(*ID_BLOCK, (thread_current, None::<u8>))
253
- })
254
- } else {
255
- call_with_gvl(|_| scheduler.funcall::<_, _, Value>(*ID_YIELD, ()))
256
- };
257
-
258
- if yield_result.is_err() {
259
- break;
260
- }
261
- })
262
- })
263
- }
264
-
265
- #[instrument(skip_all, fields(thread_worker=id))]
266
- pub fn fiber_accept_loop(
267
- id: String,
268
- app: Opaque<Value>,
269
- receiver: Arc<async_channel::Receiver<RequestJob>>,
270
- scheduler_class: Opaque<Value>,
271
- terminated: Arc<AtomicBool>,
272
- ) -> Result<()> {
273
- let ruby = Ruby::get().unwrap();
274
- let (waker_sender, waker_receiver) = watch::channel(TerminateWakerSignal(false));
275
- let leader: Arc<Mutex<Option<RequestJob>>> = Arc::new(Mutex::new(None));
276
- let server = ruby.get_inner(&ITSI_SERVER);
277
- let scheduler_proc =
278
- Self::build_scheduler_proc(app, &leader, &receiver, &terminated, &waker_sender);
279
- let (scheduler, scheduler_fiber) = server.funcall::<_, _, (Value, Value)>(
280
- "start_scheduler_loop",
281
- (scheduler_class, scheduler_proc),
282
- )?;
283
- Self::start_waker_thread(
284
- scheduler.into(),
285
- scheduler_fiber.into(),
286
- leader,
287
- receiver,
288
- waker_receiver,
289
- );
290
- Ok(())
291
- }
292
-
293
- #[allow(clippy::await_holding_lock)]
294
- pub fn start_waker_thread(
295
- scheduler: Opaque<Value>,
296
- scheduler_fiber: Opaque<Value>,
297
- leader: Arc<Mutex<Option<RequestJob>>>,
298
- receiver: Arc<async_channel::Receiver<RequestJob>>,
299
- mut waker_receiver: watch::Receiver<TerminateWakerSignal>,
300
- ) {
301
- create_ruby_thread(move || {
302
- let scheduler = scheduler.get_inner_with(&Ruby::get().unwrap());
303
- let leader = leader.clone();
304
- call_without_gvl(|| {
305
- RuntimeBuilder::new_current_thread()
306
- .build()
307
- .expect("Failed to build Tokio runtime")
308
- .block_on(async {
309
- loop {
310
- waker_receiver.changed().await.ok();
311
- if waker_receiver.borrow().0 {
312
- break;
313
- }
314
- tokio::select! {
315
- _ = waker_receiver.changed() => {
316
- if waker_receiver.borrow().0 {
317
- break;
318
- }
319
- },
320
- next_msg = receiver.recv() => {
321
- *leader.lock() = next_msg.ok();
322
- call_with_gvl(|_| {
323
- scheduler
324
- .funcall::<_, _, Value>(
325
- "unblock",
326
- (None::<u8>, scheduler_fiber),
327
- )
328
- .ok();
329
- });
330
- }
331
- }
332
- }
333
- })
334
- });
335
- });
336
- }
337
-
338
- #[instrument(skip_all, fields(thread_worker=id))]
339
- pub fn accept_loop(
340
- id: String,
341
- app: Opaque<Value>,
342
- receiver: Arc<async_channel::Receiver<RequestJob>>,
343
- terminated: Arc<AtomicBool>,
344
- ) {
345
- let ruby = Ruby::get().unwrap();
346
- let server = ruby.get_inner(&ITSI_SERVER);
347
- call_without_gvl(|| loop {
348
- match receiver.recv_blocking() {
349
- Ok(RequestJob::ProcessRequest(request)) => {
350
- if terminated.load(Ordering::Relaxed) {
351
- break;
352
- }
353
- call_with_gvl(|_ruby| {
354
- request.process(&ruby, server, app).ok();
355
- })
356
- }
357
- Ok(RequestJob::Shutdown) => {
358
- debug!("Shutting down thread worker");
359
- break;
360
- }
361
- Err(_) => {
362
- thread::sleep(Duration::from_micros(1));
363
- }
364
- }
365
- });
366
- }
367
- }