bones-rpc 0.0.2 → 0.0.3

data/lib/bones/rpc/read_preference/selectable.rb

@@ -6,7 +6,7 @@ module Bones
       # Provides the shared behaviour for read preferences that can filter by a
       # tag set or add query options.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       module Selectable
 
         # @!attribute tags
@@ -20,7 +20,7 @@ module Bones
         #
         # @param [ Array<Hash> ] tags The tag sets.
         #
-        # @since 2.0.0
+        # @since 0.0.1
         def initialize(tags = nil)
           @tags = tags
         end
@@ -34,7 +34,7 @@ module Bones
         #
         # @return [ Hash ] The options plus additional query options.
         #
-        # @since 2.0.0
+        # @since 0.0.1
         def query_options(options)
           options[:flags] ||= []
           options[:flags] |= [ :slave_ok ]
@@ -60,7 +60,7 @@ module Bones
         #
         # @return [ Object ] The result of the block.
         #
-        # @since 2.0.0
+        # @since 0.0.1
         def with_retry(cluster, retries = cluster.max_retries, &block)
           begin
             block.call

data/lib/bones/rpc/readable.rb

@@ -4,7 +4,7 @@ module Bones
 
     # Provides behaviour around readable objects.
     #
-    # @since 2.0.0
+    # @since 0.0.1
     module Readable
 
       private
@@ -18,7 +18,7 @@ module Bones
       #
       # @return [ Cluster ] The cluster.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       def cluster
         session.cluster
       end
@@ -32,7 +32,7 @@ module Bones
       #
       # @return [ Object ] The session's read preference.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       def read_preference
         session.read_preference
       end
@@ -48,7 +48,7 @@ module Bones
       #
       # @return [ Hash ] The new query options.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       def query_options(options = {})
         read_preference.query_options(options)
       end

data/lib/bones/rpc/session.rb

@@ -28,7 +28,7 @@ module Bones
     #   session = Bones::RPC::Session.new %w[127.0.0.1:27017],
     #   session.with(database: "admin").login("admin", "s3cr3t")
     #
-    # @since 1.0.0
+    # @since 0.0.1
     class Session
       include Optionable
 
@@ -38,13 +38,17 @@ module Bones
       #   @return [ Hash ] The configuration options.
       attr_reader :cluster, :options
 
+      def backend
+        @backend ||= Backend.get(options[:backend] || :synchronous).tap(&:setup)
+      end
+
       # Run +command+ on the current database.
       #
       # @param (see Bones::RPC::Database#command)
       #
       # @return (see Bones::RPC::Database#command)
       #
-      # @since 1.0.0
+      # @since 0.0.1
       def command(op)
         current_database.command(op)
       end
@@ -59,7 +63,7 @@ module Bones
       #
       # @return [ true ] True if the disconnect succeeded.
       #
-      # @since 1.2.0
+      # @since 0.0.1
       def disconnect
         cluster.disconnect
       end
@@ -75,51 +79,56 @@ module Bones
       #
       # @return [ String ] The string inspection.
       #
-      # @since 1.4.0
+      # @since 0.0.1
       def inspect
         "<#{self.class.name} seeds=#{cluster.seeds}>"
       end
 
       # Setup validation of allowed read preference options.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:read).allow(
         :nearest
       )
 
-      # Setup validation of allowed database options. (Any string or symbol)
+      # Setup validation of allowed adapter options. (Any string or symbol)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:adapter).allow(Optionable.any(String), Optionable.any(Symbol), Optionable.any(Module))
 
+      # Setup validation of allowed backend options, (Any string or symbol)
+      #
+      # @since 0.0.1
+      option(:backend).allow(Optionable.any(String), Optionable.any(Symbol), Optionable.any(Module))
+
       # Setup validation of allowed max retry options. (Any integer)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:max_retries).allow(Optionable.any(Integer))
 
       # Setup validation of allowed pool size options. (Any integer)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:pool_size).allow(Optionable.any(Integer))
 
       # Setup validation of allowed retry interval options. (Any numeric)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:retry_interval).allow(Optionable.any(Numeric))
 
       # Setup validation of allowed reap interval options. (Any numeric)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:reap_interval).allow(Optionable.any(Numeric))
 
       # Setup validation of allowed ssl options. (Any boolean)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:ssl).allow(true, false)
 
       # Setup validation of allowed timeout options. (Any numeric)
       #
-      # @since 2.0.0
+      # @since 0.0.1
       option(:timeout).allow(Optionable.any(Numeric))
 
       # Initialize a new database session.
@@ -132,7 +141,7 @@ module Bones
       #
       # @see Above options validations for allowed values in the options hash.
       #
-      # @since 1.0.0
+      # @since 0.0.1
       def initialize(seeds, options = {}, &callback)
         validate_strict(options)
         @options = options
@@ -152,7 +161,7 @@ module Bones
       #
       # @return [ Object ] The read preference.
       #
-      # @since 2.0.0
+      # @since 0.0.1
       def read_preference
         @read_preference ||= ReadPreference.get(options[:read] || :nearest)
       end
@@ -176,7 +185,7 @@ module Bones
         #
         # @return [ Session ] The new session.
         #
-        # @since 3.0.0
+        # @since 0.0.1
         def connect(uri, &block)
           uri = Uri.new(uri)
           session = new(*uri.bones_rpc_arguments, &block)

data/lib/bones/rpc/synchronous.rb

@@ -0,0 +1,2 @@
+# encoding: utf-8
+require 'bones/rpc/synchronous/node'

data/lib/bones/rpc/synchronous/connection.rb

@@ -0,0 +1,36 @@
+# encoding: utf-8
+require 'bones/rpc/connection'
+require 'monitor'
+
+module Bones
+  module RPC
+    module Synchronous
+
+      # This class contains behaviour of Bones::RPC socket connections.
+      #
+      # @since 0.0.1
+      class Connection < ::Bones::RPC::Connection
+
+        require 'bones/rpc/synchronous/connection/reader'
+        require 'bones/rpc/synchronous/connection/socket'
+        require 'bones/rpc/synchronous/connection/writer'
+
+        writer_class ::Bones::RPC::Synchronous::Connection::Writer
+
+        def write(operations)
+          with_connection do |socket|
+            proxy = writer.write(operations)
+            if proxy
+              Timeout::timeout(timeout) do
+                while not proxy.registry_empty?
+                  writer.reader.read(proxy)
+                end
+              end
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/bones/rpc/synchronous/connection/reader.rb

@@ -0,0 +1,59 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Synchronous
+      class Connection
+        class Reader
+          def initialize(connection, socket, adapter, writer)
+            @connection = connection
+            @socket = socket
+            @adapter = adapter
+            @writer = writer
+            @alive = true
+            @buffer = ""
+          end
+
+          def alive?
+            !!@alive
+          end
+
+          def parse(data, proxy)
+            @buffer << data
+            if @buffer.empty?
+              read(proxy)
+            else
+              parser = Bones::RPC::Parser.new(@buffer, @adapter)
+              begin
+                loop { send parser.read, proxy }
+              rescue EOFError
+                @buffer.replace(parser.buffer.to_str)
+              end
+              return if @buffer.empty?
+              read(proxy)
+            end
+          end
+
+          def read(proxy)
+            parse @socket.readpartial(4096), proxy
+          rescue EOFError, Errors::ConnectionFailure => e
+            Loggable.warn("  BONES-RPC:", "#{@connection.node.address.resolved} Reader terminating: #{e.message}", "n/a")
+            terminate
+            raise e
+          end
+
+          def send(message, proxy)
+            proxy.handle_message(message)
+          end
+
+          def terminate
+            return if not alive?
+            @alive = false
+            @buffer.clear
+            @writer.terminate
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/bones/rpc/synchronous/connection/socket.rb

@@ -0,0 +1,4 @@
+# encoding: utf-8
+require 'bones/rpc/connection/socket'
+require 'bones/rpc/synchronous/connection/socket/ssl'
+require 'bones/rpc/synchronous/connection/socket/tcp'

data/lib/bones/rpc/synchronous/connection/socket/ssl.rb

@@ -0,0 +1,57 @@
+# encoding: utf-8
+require 'openssl'
+
+module Bones
+  module RPC
+    module Synchronous
+      class Connection
+        module Socket
+
+          # This is a wrapper around a tcp socket.
+          class SSL < ::OpenSSL::SSL::SSLSocket
+            include ::Bones::RPC::Connection::Socket::Connectable
+
+            attr_reader :socket
+
+            # Initialize the new TCPSocket with SSL.
+            #
+            # @example Initialize the socket.
+            #   SSL.new("127.0.0.1", 27017)
+            #
+            # @param [ String ] host The host.
+            # @param [ Integer ] port The port.
+            #
+            # @since 0.0.1
+            def initialize(remote_host, remote_port, local_host = nil, local_port = nil)
+              @host, @port = remote_host.to_s, remote_port
+              handle_socket_errors do
+                @socket = TCPSocket.new(@host, remote_port, local_host, local_port)
+                super(socket)
+                self.sync_close = true
+                connect
+              end
+            end
+
+            # Set the encoding of the underlying socket.
+            #
+            # @param [ String ] string The encoding.
+            #
+            # @since 0.0.1
+            def set_encoding(string)
+              socket.set_encoding(string)
+            end
+
+            # Set a socket option on the underlying socket.
+            #
+            # @param [ Array<Object> ] args The option arguments.
+            #
+            # @since 0.0.1
+            def setsockopt(*args)
+              socket.setsockopt(*args)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/bones/rpc/synchronous/connection/socket/tcp.rb

@@ -0,0 +1,30 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Synchronous
+      class Connection
+        module Socket
+
+          # This is a wrapper around a tcp socket.
+          class TCP < ::TCPSocket
+            include ::Bones::RPC::Connection::Socket::Connectable
+
+            # Initialize the new TCPSocket.
+            #
+            # @example Initialize the socket.
+            #   TCP.new("127.0.0.1", 27017)
+            #
+            # @param [ String ] remote_host The host.
+            # @param [ Integer ] remote_port The port.
+            #
+            # @since 0.0.1
+            def initialize(remote_host, remote_port, local_host = nil, local_port = nil)
+              @host, @port = remote_host.to_s, remote_port
+              handle_socket_errors { super(@host, port, local_host, local_port) }
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/bones/rpc/synchronous/connection/writer.rb

@@ -0,0 +1,86 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Synchronous
+      class Connection
+        class Writer
+          attr_reader :reader
+
+          def initialize(connection, socket, adapter)
+            @connection = connection
+            @socket = socket
+            @adapter = adapter
+            @resolved = @connection.node.address.resolved
+            @alive = true
+            @buffer = ""
+            @reader = Reader.new(@connection, @socket, @adapter, self)
+          end
+
+          def alive?
+            !!@alive
+          end
+
+          def async
+            self
+          end
+
+          def write(operations)
+            proxy = NodeProxy.new(@connection.node)
+            operations.each do |message, future|
+              message.serialize(@buffer, @adapter)
+              message.attach(proxy, future) if future
+            end
+            @socket.write(@buffer)
+            @buffer = ""
+            return proxy
+          rescue EOFError, Errors::ConnectionFailure => e
+            Loggable.warn("  BONES-RPC:", "#{@resolved} Writer terminating: #{e.message}", "n/a")
+            terminate
+            raise e
+          end
+
+          def terminate
+            return if not alive?
+            @alive = false
+            @reader.terminate
+            @connection.cleanup_socket(@socket)
+          end
+
+          class NodeProxy < ::BasicObject
+            def attach(channel, id, future)
+              @registry.set(channel, id, future)
+            end
+
+            def detach(channel, id)
+              @registry.get(channel, id)
+            end
+
+            def handle_message(message)
+              logging(message) do
+                if future = message.get(self)
+                  message.signal(future)
+                end
+              end
+            end
+
+            def initialize(node)
+              @node = node
+              @registry = Node::Registry.new
+            end
+
+            def registry_empty?
+              @registry.empty?
+            end
+
+            protected
+
+            def method_missing(name, *args, &block)
+              @node.__send__(name, *args, &block)
+            end
+          end
+
+        end
+      end
+    end
+  end
+end