grpc_kit 0.1.0 → 0.1.1

data/lib/grpc_kit/streams/packable.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+module GrpcKit
+  module Streams
+    module Packable
+      # @params data [String]
+      # @params compress [Boolean]
+      def pack(data, compress = false)
+        c = compress ? 1 : 0
+        [c, data.size, data].pack('CNa*')
+      end
+
+      # @params data [String]
+      def unpack(data)
+        if data
+          unpacker.feed(data)
+        end
+
+        if unpacker.readable?
+          return unpacker.read
+        end
+
+        nil
+      end
+
+      def unpacker
+        @unpacker ||= GrpcKit::Streams::Packable::Unpacker.new
+      end
+
+      class Unpacker
+        # Compressed bytes(1 Byte) + length bytes(4 Bytes)
+        METADATA_SIZE = 5
+
+        def initialize
+          @i = 0
+          @data = nil
+        end
+
+        def readable?
+          @data && !@data.empty?
+        end
+
+        def feed(data)
+          if @data
+            @data << data
+          else
+            @data = data
+          end
+        end
+
+        def read
+          metadata = @data.slice!(0, METADATA_SIZE)
+          c, size = metadata.unpack('CN')
+          data = @data.slice!(0, size)
+          [c != 0, size, data]
+        end
+      end
+    end
+  end
+end

data/lib/grpc_kit/streams/send_buffer.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module GrpcKit
+  module Streams
+    class SendBuffer
+      def initialize
+        @buffer = nil
+        @pos = 0
+        @end_write = false
+      end
+
+      def write(data, last: false)
+        end_write if last
+
+        if @buffer
+          @buffer << data
+        else
+          @buffer = data
+        end
+
+        data.size
+      end
+
+      def end_write
+        @end_write = true
+      end
+
+      def end_write?
+        @end_write
+      end
+
+      def read(size)
+        if @buffer.nil?
+          return false
+        end
+
+        data = @buffer.slice!(0, size)
+        if !data.empty?
+          data
+        elsif end_write?
+          nil # EOF
+        else
+          false # deferred
+        end
+      end
+    end
+  end
+end

data/lib/grpc_kit/streams/server.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'forwardable'
+
+require 'grpc_kit/stream'
+
+module GrpcKit
+  module Streams
+    class Server
+      extend Forwardable
+
+      delegate %i[each recv] => :@stream
+
+      def initialize(protobuf:, session:, stream:)
+        @protobuf = protobuf
+        @session = session
+        @stream = GrpcKit::Stream.new(protobuf: @protobuf, session: @session, stream: stream)
+        @sent_first_msg = false
+      end
+
+      def send_msg(data, last: false)
+        @stream.send(data, last: last)
+
+        return if @sent_first_msg
+
+        @session.submit_response(
+          @stream.stream_id,
+          ':status' => '200',
+          'content-type' => 'application/grpc',
+          'accept-encoding' => 'identity',
+        )
+        @sent_first_msg = true
+      end
+    end
+  end
+end

data/lib/grpc_kit/streams/stream.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module GrpcKit
+  module Streams
+    class Stream
+      include GrpcKit::Rpcs::Packable
+
+      def initialize(protobuf:, session:, stream:)
+        @protobuf = protobuf
+        @session = session
+        @stream = stream
+      end
+
+      def send(last: false)
+        req = @protobuf.encode(data)
+        @stream.write_send_data(pack(req), last: last)
+      end
+
+      def recv(last: false)
+        data = unpack(read(last: last))
+
+        unless data
+          raise StopIteration
+        end
+
+        compressed, size, buf = *data
+
+        unless size == buf.size
+          raise "inconsistent data: #{buf}"
+        end
+
+        if compressed
+          raise 'compress option is unsupported'
+        end
+
+        @protobuf.decode(buf)
+      end
+
+      private
+
+      def read(last: false)
+        loop do
+          data = @stream.read_recv_data(last: last)
+          if data.empty?
+            if @stream.end_read?
+              return nil
+            end
+
+            @session.run_once
+            redo
+          end
+
+          return data
+        end
+      end
+    end
+  end
+end

data/lib/grpc_kit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GrpcKit
-  VERSION = '0.1.0'
+  VERSION = '0.1.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: grpc_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - ganmacs
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-10-08 00:00:00.000000000 Z
+date: 2018-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ds9
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.1.1
+        version: 1.2.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.1.1
+        version: 1.2.0
 - !ruby/object:Gem::Dependency
   name: google-protobuf
   requirement: !ruby/object:Gem::Requirement
@@ -159,6 +159,8 @@ files:
 - examples/helloworld/helloworld_services_pb.rb
 - examples/helloworld_client.rb
 - examples/helloworld_server.rb
+- examples/interceptors/client_logging_interceptor.rb
+- examples/interceptors/server_logging_interceptor.rb
 - examples/routeguide/routeguide.json
 - examples/routeguide/routeguide.proto
 - examples/routeguide/routeguide_pb.rb
@@ -172,14 +174,39 @@ files:
 - lib/grpc_kit/errors.rb
 - lib/grpc_kit/grpc/dsl.rb
 - lib/grpc_kit/grpc/generic_service.rb
+- lib/grpc_kit/grpc/interceptor.rb
+- lib/grpc_kit/grpc/logger.rb
 - lib/grpc_kit/grpc/stream.rb
-- lib/grpc_kit/io/basic.rb
+- lib/grpc_kit/interceptors.rb
+- lib/grpc_kit/interceptors/client_streamer.rb
+- lib/grpc_kit/interceptors/request_response.rb
+- lib/grpc_kit/interceptors/server_streamer.rb
+- lib/grpc_kit/interceptors/streaming.rb
+- lib/grpc_kit/method_config.rb
+- lib/grpc_kit/protobuffer.rb
 - lib/grpc_kit/rpc_desc.rb
+- lib/grpc_kit/rpcs.rb
+- lib/grpc_kit/rpcs/base.rb
+- lib/grpc_kit/rpcs/bidi_streamer.rb
+- lib/grpc_kit/rpcs/call.rb
+- lib/grpc_kit/rpcs/client_streamer.rb
+- lib/grpc_kit/rpcs/request_response.rb
+- lib/grpc_kit/rpcs/server_streamer.rb
 - lib/grpc_kit/server.rb
+- lib/grpc_kit/session/buffer.rb
 - lib/grpc_kit/session/client.rb
+- lib/grpc_kit/session/duration.rb
+- lib/grpc_kit/session/headers.rb
+- lib/grpc_kit/session/io.rb
 - lib/grpc_kit/session/server.rb
 - lib/grpc_kit/session/stream.rb
 - lib/grpc_kit/status_codes.rb
+- lib/grpc_kit/stream.rb
+- lib/grpc_kit/streams/client.rb
+- lib/grpc_kit/streams/packable.rb
+- lib/grpc_kit/streams/send_buffer.rb
+- lib/grpc_kit/streams/server.rb
+- lib/grpc_kit/streams/stream.rb
 - lib/grpc_kit/version.rb
 homepage: https://github.com/ganmacs/grpc_kit
 licenses:

data/lib/grpc_kit/io/basic.rb

@@ -1,35 +0,0 @@
-# frozen_string_literal: true
-
-require 'ds9'
-
-module GrpcKit
-  module IO
-    class Basic
-      def initialize(reader, writer)
-        @reader = reader
-        @writer = writer
-      end
-
-      def read(length)
-        data = @reader.read_nonblock(length, nil, exception: false)
-
-        case data
-        when :wait_readable
-          DS9::ERR_WOULDBLOCK
-        when nil
-          ''
-        else
-          data
-        end
-      end
-
-      def write(data)
-        @writer.write(data)
-      end
-
-      def wait_readable
-        @reader.wait_writable
-      end
-    end
-  end
-end