bones-rpc 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 4f7529f76c8f4902d50e0c9f3c300c7fd5a94c67
+  data.tar.gz: b5b201099c32bb8641955f0b90ce11681153df4c
+SHA512:
+  metadata.gz: d41c7548cda95b8bdf7f385099e7adb267f4fb9cc7aa17287aceb00c0c5c3dfd8116436967ce121364b692dd2b642831bac693b6d9239d6158dbb2ebc508ef31
+  data.tar.gz: 056db3be7bbd457c18844d0ebf643b19caad4ed4e821ab216763ff1b8b15fb0090e296ea4f5e18b16324264165eed5fd55a514d895459de08e1c318cab33f3f3

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.ruby-gemset

@@ -0,0 +1 @@
+bones

data/.ruby-version

@@ -0,0 +1 @@
+ruby-2.0.0

data/Gemfile

@@ -0,0 +1,8 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in bones-rpc-ruby.gemspec
+gemspec
+
+group :development do
+  gem 'pry'
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Andrew Bennett
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# Bones::RPC
+
+Bones::RPC client for ruby
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'bones-rpc'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install bones-rpc
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bones-rpc.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'bones/rpc/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "bones-rpc"
+  spec.version       = Bones::RPC::VERSION
+  spec.authors       = ["Andrew Bennett"]
+  spec.email         = ["andrew@delorum.com"]
+  spec.description   = %q{Bones::RPC client for ruby}
+  spec.summary       = %q{Bones::RPC client for ruby}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "celluloid"
+  spec.add_dependency "celluloid-io"
+  spec.add_dependency "erlang-etf"
+  spec.add_dependency "msgpack"
+  spec.add_dependency "optionable"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+end

data/lib/bones-rpc.rb

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

data/lib/bones/rpc.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+require 'logger'
+require 'stringio'
+require 'monitor'
+require 'timeout'
+require 'celluloid/io'
+require 'optionable'
+require 'bones/rpc/errors'
+require 'bones/rpc/loggable'
+require 'bones/rpc/uri'
+require 'bones/rpc/adapter'
+require 'bones/rpc/parser'
+require 'bones/rpc/protocol'
+require 'bones/rpc/session'
+require 'bones/rpc/version'
+
+module Bones
+  module RPC
+    extend Loggable
+
+    FutureValue = Struct.new(:value)
+  end
+end

data/lib/bones/rpc/adapter.rb

@@ -0,0 +1,49 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Adapter
+      extend self
+
+      def get(adapter_name)
+        adapters[adapter_name] || raise(Errors::InvalidAdapter, "Unknown adapter #{adapter_name.inspect}")
+      end
+
+      def get_by_ext_head(head)
+        ext_heads[head] || raise(Errors::InvalidExtMessage, "Unknown adapter for ext head #{head.inspect}")
+      end
+
+      def register(adapter)
+        adapter.send(:attr_reader, :adapter_name)
+        adapter.send(:include, Adapter::Base)
+        adapter.send(:extend, adapter)
+        adapters[adapter] ||= adapter
+        adapters[adapter.adapter_name] ||= adapter
+        adapters[adapter.adapter_name.to_s] ||= adapter
+        return adapter
+      end
+
+      def register_ext_head(adapter, head)
+        ext_heads[head] ||= adapter
+        return adapter
+      end
+
+      private
+
+      def adapters
+        @adapters ||= {}
+      end
+
+      def ext_heads
+        @ext_heads ||= {}
+      end
+    end
+  end
+end
+
+require 'bones/rpc/adapter/parser'
+
+require 'bones/rpc/adapter/base'
+
+require 'bones/rpc/adapter/erlang'
+require 'bones/rpc/adapter/json'
+require 'bones/rpc/adapter/msgpack'

data/lib/bones/rpc/adapter/base.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Adapter
+      module Base
+
+        def adapter_name
+          raise NotImplementedError, "Adapter #{self.name} does not implement #adapter_name"
+        end
+
+        def deserialize(buffer = "")
+          if buffer.is_a?(String)
+            buffer = StringIO.new(buffer)
+          end
+          unpack(buffer)
+        end
+
+        def pack(message, buffer = "")
+          raise NotImplementedError, "Adapter #{self.name} does not implement #pack"
+        end
+
+        def packer(buffer)
+          raise NotImplementedError, "Adapter #{self.name} does not implement #packer"
+        end
+
+        def serialize(message, buffer = "")
+          pack(message, buffer)
+        end
+
+        def unpack(buffer)
+          raise NotImplementedError, "Adapter #{self.name} does not implement #unpack"
+        end
+
+        def unpacker(buffer)
+          raise NotImplementedError, "Adapter #{self.name} does not implement #unpacker"
+        end
+
+      end
+    end
+  end
+end

data/lib/bones/rpc/adapter/erlang.rb

@@ -0,0 +1,28 @@
+# encoding: utf-8
+require 'erlang/etf'
+
+module Bones
+  module RPC
+    module Adapter
+      module Erlang
+
+        @adapter_name = :erlang
+
+        def pack(message, buffer="")
+          data = ::Erlang.term_to_binary(message)
+          len = data.bytesize
+          buffer << [len].pack('N')
+          buffer << data
+        end
+
+        def unpack(buffer)
+          len, = buffer.read(4).unpack('N')
+          data = buffer.read(len)
+          ::Erlang.binary_to_term(data)
+        end
+
+        Adapter.register self
+      end
+    end
+  end
+end

data/lib/bones/rpc/adapter/json.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+require 'json'
+
+module Bones
+  module RPC
+    module Adapter
+      module JSON
+
+        @adapter_name = :json
+
+        def pack(message, buffer="")
+          buffer << ::JSON.dump(message)
+        end
+
+        def unpack(buffer)
+          ::JSON.load(buffer)
+        end
+
+        Adapter.register self
+      end
+    end
+  end
+end

data/lib/bones/rpc/adapter/msgpack.rb

@@ -0,0 +1,52 @@
+# encoding: utf-8
+require 'msgpack'
+
+module Bones
+  module RPC
+    module Adapter
+      module MessagePack
+
+        @adapter_name = :msgpack
+
+        def pack(message, buffer="")
+          buffer << ::MessagePack.pack(message)
+        end
+
+        def unpack(buffer)
+          ::MessagePack.unpack(buffer)
+        end
+
+        def unpack_stream(stream)
+          buffer = StringIO.new(stream)
+          ::MessagePack::Unpacker.new(buffer).read
+        end
+
+        def read(unpacker)
+          (unpacker.adapter_unpacker ||= ::MessagePack::Unpacker.new(unpacker.buffer)).read
+        end
+
+        def unpacker(data)
+          ::MessagePack::Unpacker.new(StringIO.new(data))
+        end
+
+        def parser(data)
+          Adapter::Parser.new(self, data)
+        end
+
+        def unpacker_pos(parser)
+          size = parser.unpacker.buffer.size
+          pos  = parser.unpacker.buffer.io.pos
+          (pos > size) ? (pos - size) : 0
+        end
+
+        def unpacker_seek(parser, n)
+          pos = unpacker_pos(parser)
+          parser.unpacker.buffer.skip(n - pos) if pos < n
+          return pos
+        end
+
+        Adapter.register self
+      end
+    end
+  end
+end

data/lib/bones/rpc/adapter/parser.rb

@@ -0,0 +1,37 @@
+# encoding: utf-8
+module Bones
+  module RPC
+    module Adapter
+      class Parser
+
+        attr_reader :adaper, :data
+
+        def initialize(adapter, data)
+          @adapter = Adapter.get(adapter)
+          @data = data
+        end
+
+        def packer
+          @packer ||= @adapter.packer(data)
+        end
+
+        def read
+          unpacker.read
+        end
+
+        def unpacker_seek(n)
+          @adapter.unpacker_seek(self, n)
+        end
+
+        def unpacker
+          @unpacker ||= @adapter.unpacker(data)
+        end
+
+        def unpacker_pos
+          @adapter.unpacker_pos(self)
+        end
+
+      end
+    end
+  end
+end

data/lib/bones/rpc/address.rb

@@ -0,0 +1,167 @@
+# encoding: utf-8
+module Bones
+  module RPC
+
+    # Encapsulates behaviour around addresses and resolving dns.
+    #
+    # @since 2.0.0
+    class Address
+
+      # @!attribute host
+      #   @return [ String ] The host name.
+      # @!attribute ip
+      #   @return [ String ] The ip address.
+      # @!attribute original
+      #   @return [ String ] The original host name.
+      # @!attribute port
+      #   @return [ Integer ] The port.
+      # @!attribute resolved
+      #   @return [ String ] The full resolved address.
+      attr_reader :host, :ip, :original, :path, :port
+
+      # Instantiate the new address.
+      #
+      # @example Instantiate the address.
+      #   Bones::RPC::Address.new("localhost:27017")
+      #
+      # @param [ String ] address The host:port pair as a string.
+      #
+      # @since 2.0.0
+      def initialize(address, port = nil)
+        if address.is_a?(Bones::RPC::Address)
+          @host     = address.host
+          @ip       = address.ip
+          @original = address.original
+          @port     = address.port
+          @path     = address.path
+          return
+        end
+
+        if port.nil?
+          @original = address
+        else
+          @original = "#{address}:#{port}"
+        end
+      end
+
+      def inspect
+        "<#{self.class} \"#{to_s}\">"
+      end
+
+      def ipv4?
+        ip.is_a?(Resolv::IPv4)
+      end
+
+      def ipv6?
+        ip.is_a?(Resolv::IPv6)
+      end
+
+      def resolved
+        if valid?
+          if unix?
+            "unix:#{path}"
+          else
+            host_string = ipv6? ? "[#{ip}]" : ip.to_s
+            [host_string, port].join(':')
+          end
+        else
+          original
+        end
+      end
+      alias_method :to_s, :resolved
+
+      def unix?
+        path.is_a?(String)
+      end
+
+      def valid?
+        unix? or (valid_ip? and valid_port?)
+      end
+      alias_method :connectable?, :valid?
+
+      def valid_ip?
+        ipv4? or ipv6?
+      end
+
+      def valid_port?
+        !port.nil? && port != 0
+      end
+
+      # Resolve the address for the provided node. If the address cannot be
+      # resolved the node will be flagged as down.
+      #
+      # @example Resolve the address.
+      #   address.resolve(node)
+      #
+      # @param [ Node ] node The node to resolve for.
+      #
+      # @return [ String ] The resolved address.
+      #
+      # @since 2.0.0
+      def resolve(node)
+        begin
+          resolve! unless valid?
+          valid?
+        rescue Resolv::ResolvError, SocketError => e
+          node.instrument(Node::WARN, prefix: "  BONES-RPC:", message: "Could not resolve IP or UNIX path for: #{original}")
+          node.down and false
+        end
+      end
+
+      def resolve!
+        address = @original
+
+        host = nil
+        path = nil
+
+        if address.is_a?(String) && port.nil?
+          if !!(address =~ /\Aunix:/) # UNIX
+            path = address.gsub(/\Aunix:/, '')
+          elsif !!(address =~ /\A\[.+\]\:\d+\z/) # IPv6
+            host, port = address.split(']:')
+            host.gsub!(/\A\[/, '')
+          else # IPv4 (hopefully)
+            host, port = address.split(':')
+          end
+        end
+
+        if path
+          # Ensure path is valid
+          @path = ::Socket.unpack_sockaddr_un(::Socket.pack_sockaddr_un(path))
+          return
+        elsif port.nil?
+          raise ArgumentError, "wrong number of arguments (1 for 2)"
+        else
+          @host = host || address
+          @port = port.to_i
+        end
+
+        # Is it an IPv4 address?
+        if !!(Resolv::IPv4::Regex =~ @host)
+          @ip = Resolv::IPv4.create(@host)
+        end
+
+        # Guess it's not IPv4! Is it IPv6?
+        unless @ip
+          if !!(Resolv::IPv6::Regex =~ @host)
+            @original = "[#{@host}]:#{@port}"
+            @ip = Resolv::IPv6.create(@host)
+          end
+        end
+
+        # Guess it's not an IP address, so let's try DNS
+        unless @ip
+          addrs = Array(::Celluloid::IO::DNSResolver.new.resolve(@host))
+          raise Resolv::ResolvError, "DNS result has no information for #{@host}" if addrs.empty?
+
+          # Pseudorandom round-robin DNS support :/
+          @ip = addrs[rand(addrs.size)]
+        end
+
+        if !valid_ip?
+          raise ArgumentError, "unsupported address class: #{@ip.class}"
+        end
+      end
+    end
+  end
+end