Contents

require 'set'
require 'uri'
require 'yaml'

module Einhorn
  class Client
    # Keep this in this file so client can be loaded entirely
    # standalone by user code.
    module Transport
      def self.send_message(socket, message)
        line = serialize_message(message)
        socket.write(line)
      end

      def self.receive_message(socket)
        line = socket.readline
        deserialize_message(line)
      end

      def self.serialize_message(message)
        serialized = YAML.dump(message)
        escaped = URI.escape(serialized, "%\n")
        escaped + "\n"
      end

      def self.deserialize_message(line)
        serialized = URI.unescape(line)
        YAML.load(serialized)
      end
    end

    @@responseless_commands = Set.new(['worker:ack'])

    def self.for_path(path_to_socket)
      socket = UNIXSocket.open(path_to_socket)
      self.new(socket)
    end

    def self.for_fd(fileno)
      socket = UNIXSocket.for_fd(fileno)
      self.new(socket)
    end

    def initialize(socket)
      @socket = socket
    end

    def command(command_hash)
      Transport.send_message(@socket, command_hash)
      Transport.receive_message(@socket) if expect_response?(command_hash)
    end

    def expect_response?(command_hash)
      !@@responseless_commands.include?(command_hash['command'])
    end

    def close
      @socket.close
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version	Path
einhorn-0.4.9	lib/einhorn/client.rb
einhorn-0.4.8	lib/einhorn/client.rb
einhorn-0.4.7	lib/einhorn/client.rb
einhorn-0.4.6	lib/einhorn/client.rb
einhorn-0.4.5	lib/einhorn/client.rb
einhorn-0.4.4	lib/einhorn/client.rb
einhorn-0.4.3	lib/einhorn/client.rb
einhorn-0.4.2	lib/einhorn/client.rb
einhorn-0.4.1	lib/einhorn/client.rb
einhorn-0.4.0	lib/einhorn/client.rb