t-tech-investments 0.1.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e68074e95141d63fe02be0f024fdba50ec74b935b61f95cdc35754dd368484f
-  data.tar.gz: adcdccd09aed1a6bcc83e274f7c02cf954a5f05645740aad949027af7fe8352d
+  metadata.gz: 4bf8867c825875c3ae7f360a619639d0a4625a9f32f3a068e1c3636f649fbff5
+  data.tar.gz: c062acc25e8df6ffe8503331b054a481390e051c0be0cb99a895d7dbf8bd2be8
 SHA512:
-  metadata.gz: 83bb6a2f390c88557029f0b349670f393faed7da128e368bec3829154490ee9921e0af9c7532278ddd34b72f800444408380ce373bdfb57827f253cb11871fb3
-  data.tar.gz: 832db0d5502ac1d24562d9703ada3f224248778857bc551e261a88351620f8977cbb16c6e95d5b1279349209cc827cf5500f5e8edf162a241d1c32ce84aaf89d
+  metadata.gz: b9e479135ca5169147c694d1c0e10369a5dda50d81e783b1a0ddb18c5cf385734e40ff1ffe31b3df12db2d9033e23ef2ebd80389d1733161b6028ff93d04d08d
+  data.tar.gz: d808c48241f5af9c48ce21709443c44555d9efec9719947c32a6aded19cb862b9611559ceaa4ed354dc83e7e9db433f661c9f932aa2f8ece215cb93cbb7de48f

data/.cursor/rules/less-code.mdc

@@ -0,0 +1,6 @@
+---
+description: Пожалуйста, решите эту задачу с минимально возможным количеством строк кода.
+alwaysApply: false
+---
+Пожалуйста, решите эту задачу с минимально возможным количеством строк кода.
+Убедитесь, что результат функционально эквивалентен исходной цели и проходит существующие тесты или сценарии использовани

data/.cursor/rules/only-important-steps.mdc

@@ -0,0 +1,5 @@
+---
+alwaysApply: false
+---
+Разбей это только на действительно необходимые шаги. Исключи любые необязательные или умозрительные функции.
+Убедись, что каждый шаг напрямую соответствует текущей цели. Предоставь объяснение, почему каждый шаг необходим.

data/.cursor/rules/sdk-architecture.mdc

@@ -7,14 +7,15 @@ alwaysApply: true
 
 ## Слои
 
-- `Client` (facade) → `Services` (adapters) → `Transport` (gRPC)
+- `Client` (facade) → `Services` → `Transport` (gRPC)
+- `Services` подслои: `contracts` (proto-обвязка), `mappers`, `stores`, `stream`
 - Вспомогательные: `Coercers/Mappers`, `Errors`, `Proto` (сгенерёнка)
 
 ## Зависимости (жёсткие запреты)
 
 - `Services` не вызывают gRPC stub напрямую — только через `Transport`.
 - Публичный API не возвращает protobuf по умолчанию (только `raw:`/debug, если нужен).
-- `Transport` без бизнес-логики; только: metadata, deadlines, retries-policy, error mapping.
+- `Transport` без бизнес-логики; только: metadata, deadlines, error mapping. Retry и управление состоянием соединения — **не** зона ответственности библиотеки; пользователь сам реализует повторные попытки при необходимости.
 - `Coercers/Mappers` — чистые и детерминированные (без IO/сети).
 
 ## Контракты (invest-contracts)

data/CHANGELOG.md

@@ -1,5 +1,24 @@
 ## [Unreleased]
 
+## [0.2.2] - Unreleased
+
+- Version bump for next development cycle
+
+## [0.2.1] - 2026-02-06
+
+- ProtoLoader: harden loading order and market data signatures; add specs
+- Docs: clarify retry responsibility and client cleanup; remove stray gem artifact
+
+## [0.2.0] - 2026-02-06
+
+- Client: services exposed via Registry (method_missing), no hardcoded list; facades (e.g. market_data) discovered dynamically
+- ProtoLoader: load proto files dynamically from directory; common_pb first, then rest + services
+- Market data stream: session stop/close, reconciliation (GetMySubscriptions → restart + resubscribe on drift), max_restarts limit and StreamRestartLimitError
+- Market data: contract/mapper/store/reconciler split; subscription kinds single source of truth; candle interval validation
+- Transport: allow custom CA certs for gRPC
+- Dev: SimpleCov coverage (COVERAGE=1), codecov badge in README
+- Docs: README stream stopping, reconciliation example, installation and usage
+
 ## [0.1.1] - 2026-02-05
 
 - Docs: update README with correct links and env options

data/README.md

@@ -1,12 +1,35 @@
 # T::Tech::Investments
 
-Ruby-обёртка над gRPC API [T-Bank Invest API](https://developer.tbank.ru/invest/intro/intro/). Контракты (proto-файлы) берутся из репозитория [investAPI](https://github.com/RussianInvestments/investAPI) и пинятся по SHA в `contracts.lock`; сгенерированный protobuf/gRPC код коммитится в `lib/**/proto/**`. Гем даёт удобный Ruby-клиент для работы с брокерским счётом, инструментами, заявками и стримами маркет-данных.
+[![Gem Version](https://img.shields.io/gem/v/t-tech-investments.svg)](https://rubygems.org/gems/t-tech-investments)
+[![Ruby](https://img.shields.io/badge/ruby-%3E%3D%203.2-blue)](https://www.ruby-lang.org/)
+[![CI](https://img.shields.io/github/actions/workflow/status/naveroot/t-tech-investments/main.yml?branch=main&label=rspec)](https://github.com/naveroot/t-tech-investments/actions/workflows/main.yml)
+[![codecov](https://codecov.io/gh/naveroot/t-tech-investments/graph/badge.svg)](https://codecov.io/gh/naveroot/t-tech-investments)
 
-Гем предназначен для интеграции инвестиционных сервисов T-Bank в Ruby-приложения: типизированные вызовы вместо ручной работы с gRPC, единый стиль API и опора на официальные контракты.
+Ruby wrapper for the gRPC API of [T-Bank Invest](https://developer.tbank.ru/invest/intro/intro/). Contracts (proto files) are sourced from [investAPI](https://github.com/RussianInvestments/investAPI) and pinned by SHA in `contracts.lock`; generated protobuf/gRPC code is committed to `lib/**/proto/**`. The gem provides a convenient Ruby client for accounts, instruments, orders, and market data streams. Available `Client` methods depend on the loaded contract; after regenerating proto, the set of methods can change.
+
+This gem integrates T-Bank investment services into Ruby apps: typed calls instead of manual gRPC handling, consistent API style, and adherence to official contracts.
+
+## Contents
+
+- [Installation](#installation)
+- [Minimal requirements (Prod / Sandbox)](#minimal-requirements-prod--sandbox)
+- [Usage](#usage)
+- [Development](#development)
+- [Contributing](#contributing)
+- [License](#license)
+- [Code of Conduct](#code-of-conduct)
+
+## Key features
+
+- Typed unary RPC wrappers for all services
+- MarketData bidirectional stream with Ruby-friendly events
+- Hash/kwargs ↔ protobuf coercers (reflection-based)
+- Config snapshot per client (safe for background jobs)
+- Error mapping with tracking id extraction
 
 ## Installation
 
-Install the gem and add to the application's Gemfile by executing:
+Install the gem and add it to the application's Gemfile by executing:
 
 ```bash
 bundle add t-tech-investments
@@ -20,51 +43,51 @@ gem install t-tech-investments
 
 ## Minimal requirements (Prod / Sandbox)
 
-Минимум, который нужен пользователю, чтобы начать работать с T-Invest API:
+Minimum required to start using T-Invest API:
 
 - **Ruby**: >= 3.2
-- **Токен доступа**: получите токен в личном кабинете T-Bank Dev Portal (см. “Начало работы”) — [документация](https://developer.tbank.ru/invest/intro/intro/)
-- **Выбор контура**: используйте токен **для прода** или **для песочницы** (sandbox)
-- **gRPC endpoint** (из документации):
+- **Access token**: obtain a token in T-Bank Dev Portal (“Getting started”) — [docs](https://developer.tbank.ru/invest/intro/intro/)
+- **Environment**: use a **prod** or **sandbox** token
+- **gRPC endpoint** (from docs):
   - **Prod**: `invest-public-api.tbank.ru:443`
   - **Sandbox**: `sandbox-invest-public-api.tbank.ru:443`
 
-Рекомендуемая форма конфигурации через переменные окружения (названия переменных — соглашение этого гема):
+Recommended environment-variable configuration (names follow this gem's convention):
 
 ```bash
 # Prod
 export TINVEST_TOKEN="..."
 export TINVEST_ENDPOINT="invest-public-api.tbank.ru:443"
 
-# или Sandbox
+# or Sandbox
 export TINVEST_TOKEN="..."
 export TINVEST_ENDPOINT="sandbox-invest-public-api.tbank.ru:443"
 ```
 
-Опционально (ENV-first конфигурация):
+Optional (ENV-first configuration):
 
 ```bash
-# Идентификатор приложения (x-app-name)
+# Application identifier (x-app-name)
 export TINVEST_APP_NAME="my-app"
 
-# Таймаут по умолчанию (секунды)
+# Default timeout (seconds)
 export TINVEST_TIMEOUT="10"
 
-# Путь к PEM с CA-сертификатами (для кастомных окружений)
+# Path to CA certs PEM (custom environments)
 export TINVEST_SSL_CA_CERTS="/path/to/ca.pem"
 ```
 
 ## Usage
 
-Гем предлагает “глобальную” конфигурацию по умолчанию + возможность переопределить параметры при создании клиента.
+The gem provides a global default configuration plus per-client overrides.
 
 ### Configuration
 
-#### ENV-first (быстрый старт)
+#### ENV-first (quick start)
 
-**Given** у вас есть токен и выбран контур (prod/sandbox)  
-**When** вы задаёте `TINVEST_TOKEN` и `TINVEST_ENDPOINT`  
-**Then** вы можете создать клиента без дополнительной настройки
+**Given** you have a token and selected environment (prod/sandbox)  
+**When** you set `TINVEST_TOKEN` and `TINVEST_ENDPOINT`  
+**Then** you can create a client without additional setup
 
 ```ruby
 require "t/tech/investments"
@@ -72,19 +95,25 @@ require "t/tech/investments"
 client = T::Tech::Investments.client
 ```
 
-#### Явная конфигурация (Rails / monolith)
+If you installed the gem via `gem install` and see `uninitialized constant T`, make sure your script requires the gem:
 
-**Given** вы хотите централизованную настройку  
-**When** вы вызываете `T::Tech::Investments.configure`  
-**Then** новые клиенты используют эти значения по умолчанию
+```ruby
+require "t/tech/investments"
+```
+
+#### Explicit configuration (Rails / monolith)
+
+**Given** you want centralized configuration  
+**When** you call `T::Tech::Investments.configure`  
+**Then** new clients use these values by default
 
 ```ruby
 T::Tech::Investments.configure do |c|
   c.token    = ENV.fetch("TINVEST_TOKEN")
-  c.endpoint = ENV.fetch("TINVEST_ENDPOINT", "invest-public-api.tbank.ru:443") # или sandbox
+  c.endpoint = ENV.fetch("TINVEST_ENDPOINT", "invest-public-api.tbank.ru:443") # or sandbox
   c.timeout  = 10
 
-  # опционально
+  # optional
   c.app_name = "my-app"
   c.logger   = Logger.new($stdout)
 end
@@ -92,11 +121,11 @@ end
 client = T::Tech::Investments.client
 ```
 
-#### Override для одного клиента
+#### Override for a single client
 
-**Given** вам нужно изменить настройки только для одного вызова/процесса  
-**When** вы передаёте параметры в `client(...)`  
-**Then** они применяются только к этому экземпляру клиента
+**Given** you need different settings for one call/process  
+**When** you pass parameters to `client(...)`  
+**Then** they apply only to that instance
 
 ```ruby
 client = T::Tech::Investments.client(
@@ -107,44 +136,46 @@ client = T::Tech::Investments.client(
 
 ### Configuration best practices
 
-- **Валидируйте токен**: если токена нет/он пустой — поднимайте `ConfigurationError` (или аналог) с понятным сообщением.
-- **Не логируйте секреты**: токен не должен попадать в логи/исключения.
-- **Снимок конфигурации при создании клиента**: созданный клиент не должен “подхватывать” изменения глобального конфига на лету (важно для потоков/фоновых задач).
-- **Prod vs Sandbox**: различайте контуры только endpoint’ом из документации:
+- **Validate token**: if missing/empty, raise `ConfigurationError` (or similar) with a clear message.
+- **Do not log secrets**: token must not appear in logs or exceptions.
+- **Snapshot config per client**: a created client should not pick up global config changes later (important for jobs/threads).
+- **Prod vs Sandbox**: distinguish environments only by the endpoint from the docs:
   - **Prod**: `invest-public-api.tbank.ru:443`
   - **Sandbox**: `sandbox-invest-public-api.tbank.ru:443`
+- **Retries and connection state**: the library does not retry failed calls. The application is responsible for retries, reconnects, and connection state; implement retry logic in your code if needed.
+- **Long-lived processes**: when shutting down or discarding a client, call `client.transport.close` to release gRPC channel and stubs.
 
-### Unary requests (получение данных)
+### Unary requests (data fetch)
 
-Унарные запросы в T‑Invest API — это обычные gRPC-вызовы “один запрос → один ответ”.
+Unary requests in T-Invest API are standard gRPC calls: “one request → one response”.
 
-#### Авторизация и служебные заголовки
+#### Authorization and headers
 
-По документации T‑Invest API токен нужно передавать в **metadata каждого запроса** в виде заголовка:
+Per T-Invest API docs, the token is passed in **request metadata** as a header:
 
 - **`authorization: Bearer <token>`**
 
-Дополнительно (рекомендуется для SDK) можно передавать:
+Additionally (recommended for SDK) you can pass:
 
-- **`x-app-name`** — идентификатор приложения/SDK для статистики (рекомендации в доке)
+- **`x-app-name`** — app/SDK identifier for analytics (as recommended in the docs)
 
-Источник: [gRPC — Авторизация / AppName / trackingId](https://developer.tbank.ru/invest/intro/developer/protocols/grpc/).
+Source: [gRPC — Authorization / AppName / trackingId](https://developer.tbank.ru/invest/intro/developer/protocols/grpc/).
 
-#### Пример (общий паттерн)
+#### Example (common pattern)
 
-**Given** у вас настроены `TINVEST_TOKEN` и `TINVEST_ENDPOINT`  
-**When** вы вызываете унарный метод сервиса  
-**Then** вы получаете Hash (по умолчанию) или сырой protobuf при `raw: true`
+**Given** `TINVEST_TOKEN` and `TINVEST_ENDPOINT` are configured  
+**When** you call a unary service method  
+**Then** you get a Hash (default) or raw protobuf with `raw: true`
 
 ```ruby
 client = T::Tech::Investments.client
 
-# Список аккаунтов (Hash с :accounts и т.д.)
+# Accounts list (Hash with :accounts etc.)
 accounts = client.users.get_accounts
-# С параметрами и опциями
+# With params and options
 accounts = client.users.get_accounts(status: :open)
 
-# Свечи по инструменту (from/to — Time или Hash)
+# Candles for an instrument (from/to — Time or Hash)
 candles = client.market_data.get_candles(
   figi: "BBG004730N88",
   from: Time.now - 3600,
@@ -152,32 +183,82 @@ candles = client.market_data.get_candles(
   interval: :candle_interval_1_min
 )
 
-# Сырой protobuf (для отладки)
+# Raw protobuf (debugging)
 raw_response = client.users.get_accounts(raw: true)
 ```
 
-#### Таймауты (deadline)
+#### Timeouts (deadline)
 
-Для gRPC важно задавать deadline/timeout, чтобы запросы не зависали бесконечно. Вызов должен использовать значение `timeout` из конфигурации (или override при создании клиента).
+For gRPC it is important to set a deadline/timeout to avoid hanging requests. Calls use `timeout` from configuration (or an override per client).
 
-#### tracking id (x-tracking-id)
+#### Tracking id (x-tracking-id)
 
-Документация отмечает, что **`x-tracking-id` добавляется в ответы unary-методов**. Это UUID запроса, который нужно указывать при обращении в техподдержку.
+Docs note that **`x-tracking-id` is included in unary responses**. This UUID should be provided to support when troubleshooting.
 
-Рекомендуемая практика для гема:
+Recommended practice for the gem:
 
-- логировать/пробрасывать `x-tracking-id` при ошибках;
-- (опционально) предоставлять его пользователю через объект-обёртку ответа или через instrumentation hooks.
+- log/propagate `x-tracking-id` on errors;
+- optionally expose it via a response wrapper or instrumentation hooks.
 
 ### Streaming (MarketData)
 
-Bidirectional stream: в блоке задаёте подписки, затем `each_event` отдаёт события. События — Hash с `:type` и `:data`; **ping игнорируется** SDK.
+Bidirectional stream: in the block you send subscriptions, then `each_event` yields events. Events are Hashes with `:type` and `:data`; **ping is ignored** by the SDK.
+
+#### Correct usage
+
+- Treat the stream as **long-lived** and set a higher `timeout` (or override per call).
+- Enqueue subscriptions **before** calling `each_event`.
+- Keep `each_event` running; break the block when you want to stop the stream.
+
+```ruby
+client.market_data.stream(timeout: 300) do |s|
+  s.subscribe_candles(
+    subscription_action: :SUBSCRIPTION_ACTION_SUBSCRIBE,
+    instruments: [
+      {
+        figi: "BBG004730N88",
+        interval: :SUBSCRIPTION_INTERVAL_ONE_MINUTE
+      }
+    ],
+    waiting_close: true,
+    candle_source_type: :CANDLE_SOURCE_EXCHANGE
+  )
+
+  s.each_event do |event|
+    p event
+    # break if you want to stop the stream
+  end
+end
+```
+
+#### Troubleshooting
+
+If you see `SUBSCRIPTION_STATUS_SUBSCRIPTION_NOT_FOUND` or the server responds with an unsubscribe action, explicitly pass full enum values:
+
+```ruby
+client.market_data.stream(timeout: 300) do |s|
+  s.subscribe_candles(
+    subscription_action: :SUBSCRIPTION_ACTION_SUBSCRIBE,
+    instruments: [
+      {
+        figi: "BBG004730N88",
+        interval: :SUBSCRIPTION_INTERVAL_ONE_MINUTE
+      }
+    ],
+    waiting_close: true,
+    candle_source_type: :CANDLE_SOURCE_EXCHANGE
+  )
+  s.each_event { |event| p event }
+end
+```
+
+Also ensure your **token and endpoint match the same environment** (prod vs sandbox).
 
 ```ruby
 client = T::Tech::Investments.client
 
 client.market_data.stream do |s|
-  # 1) Подписки (subscription_action: :subscribe или :unsubscribe)
+  # 1) Subscriptions (subscription_action: :subscribe or :unsubscribe)
   s.subscribe_candles(
     subscription_action: :subscribe,
     instruments: [{ figi: "BBG004730N88", interval: :subscription_interval_one_minute }],
@@ -188,7 +269,7 @@ client.market_data.stream do |s|
     instruments: [{ figi: "BBG004730N88", depth: 10 }]
   )
 
-  # 2) Чтение событий (event[:type], event[:data])
+  # 2) Read events (event[:type], event[:data])
   s.each_event do |event|
     case event[:type]
     when :candle
@@ -196,26 +277,69 @@ client.market_data.stream do |s|
     when :orderbook
       puts "orderbook: #{event[:data].inspect}"
     when :subscribe_candles_response, :subscribe_order_book_response
-      # ответы подписки (tracking_id и статусы в event[:data])
+      # subscription responses (tracking_id and statuses in event[:data])
       warn "subscription: #{event[:data].inspect}"
     when :trading_status, :last_price, :trade
-      # остальные типы по контракту
+      # other types by contract
     end
   end
 end
 ```
 
-Reconciliation: при рассинхроне подписок вызовите `s.get_my_subscriptions` и при необходимости переподпишитесь или перезапустите стрим.
+Reconciliation: if subscriptions drift, call `s.get_my_subscriptions` and resubscribe or restart the stream as needed.
+
+#### Stopping a stream (graceful)
+
+If you need to stop a long-running stream from another thread or from a signal handler, call `stop`/`close` on the session to terminate the request stream cleanly:
+
+```ruby
+session = nil
+stream_thread = Thread.new do
+  client.market_data.stream(timeout: 300) do |s|
+    session = s
+    s.subscribe_candles(
+      subscription_action: :subscribe,
+      instruments: [{ figi: "BBG004730N88", interval: :subscription_interval_one_minute }]
+    )
+    s.each_event { |event| p event }
+  end
+end
+
+# later...
+session&.stop
+stream_thread.join
+```
+
+#### Reconciliation example (server vs local subscriptions)
+
+To detect drift, request server-side subscriptions and let the session reconcile:
+
+```ruby
+client.market_data.stream(timeout: 300) do |s|
+  s.subscribe_candles(
+    subscription_action: :subscribe,
+    instruments: [{ figi: "BBG004730N88", interval: :subscription_interval_one_minute }]
+  )
+  s.get_my_subscriptions
+
+  s.each_event do |event|
+    case event[:type]
+    when :subscribe_candles_response
+      # local subscription responses
+    end
+  end
+end
+```
 
-Примечания:
+Notes:
 
-- Авторизация для стрима — через metadata `Authorization: Bearer <token>` (как и для unary). Подробнее: [gRPC — Авторизация / AppName / trackingId](https://developer.tbank.ru/invest/intro/developer/protocols/grpc/).
-- В одном стриме можно отправить несколько подписок (свечи, стакан, сделки и т.д.) — каждая своим запросом в очередь.
-- Подробнее: [Описание сервиса котировок](https://developer.tbank.ru/invest/services/quotes/head-marketdata/).
+- Stream authorization uses metadata `authorization: Bearer <token>` (same as unary). See [gRPC — Authorization / AppName / trackingId](https://developer.tbank.ru/invest/intro/developer/protocols/grpc/).
+- A single stream can send multiple subscription requests (candles, order book, trades, etc.) — each is enqueued separately.
+- See [Market data service docs](https://developer.tbank.ru/invest/services/quotes/head-marketdata/).
 
 ## Development
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. To generate a coverage report (development only), run `COVERAGE=1 bundle exec rspec`; the report is written to `coverage/`. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 

data/lib/t/tech/investments/client.rb

@@ -18,46 +18,30 @@ module T
           @transport ||= Transport.new(config)
         end
 
-        def users
-          service(:users)
-        end
-
-        def instruments
-          service(:instruments)
-        end
-
-        def market_data
-          @market_data ||= begin
-            ProtoLoader.load!
-            unary = Services::UnaryAdapter.new(Services::Registry.service_class(:market_data), transport)
-            Services::MarketDataFacade.new(unary, transport)
+        def method_missing(method_name, *args, **kwargs, &block)
+          if Services::Registry.public_service_names.include?(method_name) &&
+             args.empty? && kwargs.empty? && block.nil?
+            return service(method_name)
           end
-        end
 
-        def operations
-          service(:operations)
+          super
         end
 
-        def orders
-          service(:orders)
-        end
-
-        def sandbox
-          service(:sandbox)
-        end
-
-        def stop_orders
-          service(:stop_orders)
-        end
-
-        def signals
-          service(:signals)
+        def respond_to_missing?(method_name, include_private = false)
+          Services::Registry.public_service_names.include?(method_name) || super
         end
 
         private
 
         def service(name)
-          return market_data if name == :market_data
+          if (facade_class = Services::Registry.facade_class(name))
+            @facades ||= {}
+            return @facades[name] ||= begin
+              ProtoLoader.load!
+              unary = Services::UnaryAdapter.new(Services::Registry.service_class(name), transport)
+              facade_class.new(unary, transport)
+            end
+          end
 
           @services[name] ||= begin
             ProtoLoader.load!

data/lib/t/tech/investments/errors.rb

@@ -18,6 +18,9 @@ module T
             @tracking_id = tracking_id
           end
         end
+
+        # Raised when the market data stream exceeds the configured max restarts (reconnect loop).
+        class StreamRestartLimitError < Base; end
       end
     end
   end

data/lib/t/tech/investments/proto_loader.rb

@@ -8,40 +8,65 @@ module T
       module ProtoLoader
         PROTO_DIR = File.expand_path("proto", __dir__)
 
-        # Contract _pb.rb (messages); order respects dependencies.
-        PB_FILES = %w[
-          common_pb
-          instruments_pb
-          marketdata_pb
-          operations_pb
-          orders_pb
-          sandbox_pb
-          signals_pb
-          stoporders_pb
-          users_pb
-        ].freeze
-
-        # Service definitions (depend on _pb).
-        SERVICES_FILES = %w[
-          users_services_pb
-          instruments_services_pb
-          marketdata_services_pb
-          operations_services_pb
-          orders_services_pb
-          sandbox_services_pb
-          signals_services_pb
-          stoporders_services_pb
-        ].freeze
-
-        module_function
-
-        def load!
-          return if @loaded
-
-          $LOAD_PATH.unshift(PROTO_DIR) unless $LOAD_PATH.include?(PROTO_DIR)
-          PB_FILES.each { |f| require f }
-          SERVICES_FILES.each { |f| require f }
-          @loaded = true
+        @load_mutex = Mutex.new
+        @load_cv = ConditionVariable.new
+        @loaded = false
+        @loading = false
+        @loading_thread = nil
+        @load_error = nil
+
+        class << self
+          def load!
+            @load_mutex.synchronize do
+              return if @loaded
+
+              if @loading
+                if @loading_thread == Thread.current
+                  raise RuntimeError, "ProtoLoader.load! re-entrant call while loading"
+                end
+                @load_cv.wait(@load_mutex) while @loading
+                return if @loaded
+                raise @load_error if @load_error
+              end
+
+              @load_error = nil
+              @loading = true
+              @loading_thread = Thread.current
+            end
+
+            begin
+              load_files
+              @load_mutex.synchronize do
+                @loaded = true
+                @loading = false
+                @loading_thread = nil
+                @load_cv.broadcast
+              end
+            rescue StandardError => e
+              @load_mutex.synchronize do
+                @load_error = e
+                @loading = false
+                @loading_thread = nil
+                @load_cv.broadcast
+              end
+              raise
+            end
+          end
+
+          private
+
+          def load_files
+            $LOAD_PATH.unshift(PROTO_DIR) unless $LOAD_PATH.include?(PROTO_DIR)
+            pb_files = Dir[File.join(PROTO_DIR, "*_pb.rb")]
+                       .reject { |f| f.end_with?("_services_pb.rb") }
+                       .map { |f| File.basename(f, ".rb") }
+                       .sort
+            services_files = Dir[File.join(PROTO_DIR, "*_services_pb.rb")]
+                             .map { |f| File.basename(f, ".rb") }
+                             .sort
+            pb_files = pb_files.partition { |f| f == "common_pb" }.flatten
+            (pb_files + services_files).each { |f| require f }
+          end
         end
       end
     end