statelydb 0.18.1 → 0.18.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0cbd4c47b321b9ed394f2ff41a958e2a6f8deb95d2a5bef24618e1bf5e89a99f
-  data.tar.gz: 6ff3e1fed96b217284ddf4068352ec08c9987de921722bfbf3f9a9be579a48cf
+  metadata.gz: e07b296b0a2f8dda95a010bd074b91c653f8a5c45cf2a6fc0418d7b158644ca8
+  data.tar.gz: cf3895a85a5090c62c87bda2d2bf75d0d6ae7a668c1f8dfb5bb2e4068399fdeb
 SHA512:
-  metadata.gz: 3b68f6d2a0b7548eaa42bca2507bfff33d5d11cef62b8fd2ee213dcd7e59f58915ed1fd2b4a4c84fb98a28f4eb9a57f57239eeb207d1fd1963cf0828b276f140
-  data.tar.gz: a1cb23976d9e049021484714ff356824bc5ff70610607057049456d9979a7230b40cb0b9a377a91ddd59e9480be283a2490bd260ea35a5b71eb006a5afdfc93c
+  metadata.gz: 0644d032470017c1a692713fd1a9f06633e836af51c59a20bf4859c21d94e67031c58db6fa70e72ef5055594eb305cad2b961cd4437c280d8147d7ed9e8aefea
+  data.tar.gz: 64d33d0c37809b0a819014ea38cba6ec506d313a163e09286dbd20944291d9d49f34cfc21d042f80a85aeb2050bccc72d8b843061bbae5dc63e882b09802ca41

data/lib/common/error_interceptor.rb

@@ -7,33 +7,26 @@ module StatelyDB
   module Common
     # GRPC interceptor to convert errors to StatelyDB::Error
     class ErrorInterceptor < GRPC::ClientInterceptor
-      # client unary interceptor
-      def request_response(request:, call:, method:, metadata:) # rubocop:disable Lint/UnusedMethodArgument
-        yield
-      rescue Exception => e
-        raise StatelyDB::Error.from(e)
-      end
-
-      # client streaming interceptor
-      def client_streamer(requests:, call:, method:, metadata:) # rubocop:disable Lint/UnusedMethodArgument
-        yield
-      rescue Exception => e
-        raise StatelyDB::Error.from(e)
+      # Client and Server stream handling logic
+      %i[request_response client_streamer server_streamer bidi_streamer].each do |method_name|
+        define_method(method_name) do |*args, &block|
+          safe_yield(*args, &block)
+        end
       end
+    end
+  end
+end
 
-      # server streaming interceptor
-      def server_streamer(request:, call:, method:, metadata:) # rubocop:disable Lint/UnusedMethodArgument
-        yield
-      rescue Exception => e
-        raise StatelyDB::Error.from(e)
-      end
+def safe_yield(*, &block)
+  block.call(*)
+rescue Exception => e
+  raise StatelyDB::Error.from(e)
+end
 
-      # bidirectional streaming interceptor
-      def bidi_streamer(requests:, call:, method:, metadata:) # rubocop:disable Lint/UnusedMethodArgument
-        yield
-      rescue Exception => e
-        raise StatelyDB::Error.from(e)
-      end
-    end
+def safe_yield_stream(stream, &block)
+  stream.each do |msg|
+    safe_yield(msg, &block)
   end
+rescue Exception => e
+  raise StatelyDB::Error.from(e)
 end

data/lib/statelydb.rb

@@ -32,7 +32,7 @@ module StatelyDB
     #     an auth token. This is used when talking to the Stately BYOC Data Plane on localhost.
     def initialize(store_id:,
                    schema:,
-                   token_provider: Common::Auth::AuthTokenProvider.new,
+                   token_provider: nil,
                    endpoint: nil,
                    region: nil,
                    no_auth: false)
@@ -53,7 +53,7 @@ module StatelyDB
       @token_provider = token_provider || Common::Auth::AuthTokenProvider.new(endpoint:)
 
       interceptors = [Common::ErrorInterceptor.new]
-      interceptors << Common::Auth::Interceptor.new(token_provider:) unless no_auth
+      interceptors << Common::Auth::Interceptor.new(token_provider: @token_provider) unless no_auth
 
       @stub = Stately::Db::DatabaseService::Stub.new(nil, nil,
                                                      channel_override: @channel, interceptors:)
@@ -331,7 +331,7 @@ module StatelyDB
     def process_list_response(resp)
       items = []
       token = nil
-      resp.each do |r|
+      safe_yield_stream(resp) do |r|
         case r.response
         when :result
           r.result.items.map do |result|
@@ -360,7 +360,7 @@ module StatelyDB
       updated_outside_list_window_paths = []
       token = nil
       is_reset = false
-      resp.each do |r|
+      safe_yield_stream(resp) do |r|
         case r.response
         when :result
           r.result.changed_items.each do |item|

data/sig/statelydb.rbi

@@ -210,13 +210,13 @@ module StatelyDB
       params(
         store_id: Integer,
         schema: Module,
-        token_provider: Common::Auth::TokenProvider,
+        token_provider: T.nilable(Common::Auth::TokenProvider),
         endpoint: T.nilable(String),
         region: T.nilable(String),
         no_auth: T::Boolean
       ).void
     end
-    def initialize(store_id:, schema:, token_provider: Common::Auth::AuthTokenProvider.new, endpoint: nil, region: nil, no_auth: false); end
+    def initialize(store_id:, schema:, token_provider: nil, endpoint: nil, region: nil, no_auth: false); end
 
     # _@return_ — nil
     sig { void }
@@ -757,49 +757,6 @@ module StatelyDB
 
     # GRPC interceptor to convert errors to StatelyDB::Error
     class ErrorInterceptor < GRPC::ClientInterceptor
-      # client unary interceptor
-      sig do
-        params(
-          request: T.untyped,
-          call: T.untyped,
-          method: T.untyped,
-          metadata: T.untyped
-        ).returns(T.untyped)
-      end
-      def request_response(request:, call:, method:, metadata:); end
-
-      # client streaming interceptor
-      sig do
-        params(
-          requests: T.untyped,
-          call: T.untyped,
-          method: T.untyped,
-          metadata: T.untyped
-        ).returns(T.untyped)
-      end
-      def client_streamer(requests:, call:, method:, metadata:); end
-
-      # server streaming interceptor
-      sig do
-        params(
-          request: T.untyped,
-          call: T.untyped,
-          method: T.untyped,
-          metadata: T.untyped
-        ).returns(T.untyped)
-      end
-      def server_streamer(request:, call:, method:, metadata:); end
-
-      # bidirectional streaming interceptor
-      sig do
-        params(
-          requests: T.untyped,
-          call: T.untyped,
-          method: T.untyped,
-          metadata: T.untyped
-        ).returns(T.untyped)
-      end
-      def bidi_streamer(requests:, call:, method:, metadata:); end
     end
   end
 

data/sig/statelydb.rbs

@@ -175,7 +175,7 @@ module StatelyDB
     def initialize: (
                       store_id: Integer,
                       schema: Module,
-                      ?token_provider: Common::Auth::TokenProvider,
+                      ?token_provider: Common::Auth::TokenProvider?,
                       ?endpoint: String?,
                       ?region: String?,
                       ?no_auth: bool
@@ -651,37 +651,6 @@ module StatelyDB
 
     # GRPC interceptor to convert errors to StatelyDB::Error
     class ErrorInterceptor < GRPC::ClientInterceptor
-      # client unary interceptor
-      def request_response: (
-                              request: untyped,
-                              call: untyped,
-                              method: untyped,
-                              metadata: untyped
-                            ) -> untyped
-
-      # client streaming interceptor
-      def client_streamer: (
-                             requests: untyped,
-                             call: untyped,
-                             method: untyped,
-                             metadata: untyped
-                           ) -> untyped
-
-      # server streaming interceptor
-      def server_streamer: (
-                             request: untyped,
-                             call: untyped,
-                             method: untyped,
-                             metadata: untyped
-                           ) -> untyped
-
-      # bidirectional streaming interceptor
-      def bidi_streamer: (
-                           requests: untyped,
-                           call: untyped,
-                           method: untyped,
-                           metadata: untyped
-                         ) -> untyped
     end
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: statelydb
 version: !ruby/object:Gem::Version
-  version: 0.18.1
+  version: 0.18.3
 platform: ruby
 authors:
 - Stately Cloud, Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-04 00:00:00.000000000 Z
+date: 2025-03-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async