Contents

require 'socket'
require 'json'

module Dazeus
  class Connection
    def initialize(address)
      @socket = create_socket(address)
      @cache = []
    end

    def send(message)
      if !@socket.closed?
        @socket.sendmsg(dazeusify message)
      else
        false
      end
    end

    def receive
      if @cache.length == 0
        message = @socket.recvmsg[0]
        message = message.force_encoding 'UTF-8'
        message = message.strip

        while message.length > 0
          message = dezeusify(message)
        end
      end
      @cache.shift
    end

    def dezeusify(message)
      digits = ""
      while message[0] =~ /\d/
        digits += message.slice! 0
      end
      current = message.slice!(0, digits.to_i)
      # puts "<- " + current
      @cache.push JSON.parse(current)
      message.strip
    end

    def dazeusify(message)
      msg = JSON.dump(message)
      # puts "-> " + msg
      msg.bytesize.to_s + msg
    end

    def close
      @socket.close
    end

    def closed?
      @socket.closed?
    end

    private
      def create_socket(address)
        if address.start_with? 'tcp://'
          address = address[6..-1]
          if address.count ':' != 1
            raise 'Invalid TCP socket format, specify both host and port'
          end
          host, port = address.split ':'
          TCPSocket.new host, port
        elsif address.start_with? 'unix://'
          UNIXSocket.new address[7..-1]
        else
          raise 'Invalid socket format'
        end
      end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
dazeus-0.0.1.201306211031	lib/dazeus/connection.rb
dazeus-0.0.1.201306161801	lib/dazeus/connection.rb
dazeus-0.0.1.201306161750	lib/dazeus/connection.rb