skydb 0.2.0

data/lib/skydb/message/lua/map_reduce.rb

@@ -0,0 +1,59 @@
+class SkyDB
+  class Message
+    class Lua
+      class MapReduce < SkyDB::Message
+        ########################################################################
+        #
+        # Constructor
+        #
+        ########################################################################
+
+        # Initializes the 'lua::map_reduce' message.
+        #
+        # @param [String] source  the Lua source to execute.
+        def initialize(source=nil, options={})
+          super('lua::map_reduce')
+          self.source = source
+        end
+
+
+        ##########################################################################
+        #
+        # Attributes
+        #
+        ##########################################################################
+
+        ##################################
+        # Source
+        ##################################
+
+        # The Lua source code.
+        attr_reader :source
+      
+        def source=(value)
+          @source = value.to_s
+        end
+
+
+        ##########################################################################
+        #
+        # Methods
+        #
+        ##########################################################################
+
+        ####################################
+        # Encoding
+        ####################################
+
+        # Encodes the message body.
+        #
+        # @param [IO] buffer  the buffer to write the header to.
+        def encode_body(buffer)
+          buffer << {
+            :source => source
+          }.to_msgpack
+        end
+      end
+    end
+  end
+end

data/lib/skydb/message/multi.rb

@@ -0,0 +1,57 @@
+class SkyDB
+  class Message
+    class Multi < SkyDB::Message
+      ########################################################################
+      #
+      # Constructor
+      #
+      ########################################################################
+
+      # Initializes the 'multi' message.
+      def initialize(options={})
+        super('multi')
+        self.messages = []
+      end
+
+
+      ##########################################################################
+      #
+      # Attributes
+      #
+      ##########################################################################
+
+      ##################################
+      # Messages
+      ##################################
+
+      # A list of message to send to the server.
+      attr_accessor :messages
+
+
+      ##########################################################################
+      #
+      # Methods
+      #
+      ##########################################################################
+
+      ####################################
+      # Encoding
+      ####################################
+
+      # Encodes the message body.
+      #
+      # @param [IO] buffer  the buffer to write to.
+      def encode_body(buffer)
+        # Encode the message count.
+        buffer << {
+          :count => messages.length
+        }.to_msgpack
+        
+        # Encode all the messages.
+        messages.each do |message|
+          message.encode(buffer)
+        end
+      end
+    end
+  end
+end

data/lib/skydb/message/next_actions.rb

@@ -0,0 +1,55 @@
+class SkyDB
+  class Message
+    class NextActions < SkyDB::Message
+      ########################################################################
+      #
+      # Constructor
+      #
+      ########################################################################
+
+      # Initializes the 'next action' message.
+      #
+      # @param [Array] prior_action_ids  the prior action ids.
+      def initialize(prior_action_ids=nil, options={})
+        super('next_actions')
+        self.prior_action_ids = prior_action_ids
+      end
+
+
+      ##########################################################################
+      #
+      # Attributes
+      #
+      ##########################################################################
+
+      ##################################
+      # Prior Action IDs
+      ##################################
+
+      # The prior action ids.
+      attr_reader :prior_action_ids
+      
+      def prior_action_ids=(value)
+        @prior_action_ids = value.is_a?(Array) ? value : []
+      end
+
+
+      ##########################################################################
+      #
+      # Methods
+      #
+      ##########################################################################
+
+      ####################################
+      # Encoding
+      ####################################
+
+      # Encodes the message body.
+      #
+      # @param [IO] buffer  the buffer to write the header to.
+      def encode_body(buffer)
+        buffer << prior_action_ids.to_msgpack
+      end
+    end
+  end
+end

data/lib/skydb/property.rb

@@ -0,0 +1,93 @@
+require 'skydb/property/type'
+
+class SkyDB
+  class Property
+    ##########################################################################
+    #
+    # Constructor
+    #
+    ##########################################################################
+
+    # Initializes the property.
+    def initialize(id=0, type=nil, data_type="String", name=nil)
+      self.id = id
+      self.type = type
+      self.data_type = data_type
+      self.name = name
+    end
+    
+
+    ##########################################################################
+    #
+    # Attributes
+    #
+    ##########################################################################
+
+    ##################################
+    # ID
+    ##################################
+
+    # The property identifier.
+    attr_reader :id
+    
+    def id=(value)
+      @id = value.to_i
+    end
+
+
+    ##################################
+    # Type
+    ##################################
+
+    # The property type. This can be either :object or :action.
+    attr_reader :type
+    
+    def type=(value)
+      value = :object unless [:object, :action].index(value)
+      @type = value
+    end
+
+
+    ##################################
+    # Data Type
+    ##################################
+
+    # The property data type. This can be either 'String', 'Int', 'Float',
+    # or 'Boolean'.
+    attr_reader :data_type
+    
+    def data_type=(value)
+      value = nil unless ['String', 'Int', 'Float', 'Boolean'].index(value)
+      @data_type = value
+    end
+
+
+    ##################################
+    # Name
+    ##################################
+
+    # The name of the property.
+    attr_reader :name
+    
+    def name=(value)
+      @name = value.to_s
+    end
+
+
+    ##########################################################################
+    #
+    # Methods
+    #
+    ##########################################################################
+    
+    # Encodes the property into MsgPack format.
+    def to_msgpack
+      return {
+        id:id,
+        type:SkyDB::Property::Type.encode(type),
+        dataType:data_type,
+        name:name
+      }.to_msgpack
+    end
+  end
+end

data/lib/skydb/property/type.rb

@@ -0,0 +1,40 @@
+class SkyDB
+  class Property
+    class Type
+      ########################################################################
+      #
+      # Constants
+      #
+      ########################################################################
+
+      OBJECT = 1
+
+      ACTION = 2
+
+
+      ########################################################################
+      #
+      # Static Methods
+      #
+      ########################################################################
+      
+      # Encodes the type into it's numeric enum value.
+      #
+      # @param type [Symbol]  the type.
+      # @return [Fixnum]  the enum value for the property type.
+      def self.encode(type)
+        return nil unless [:object, :action].index(type)
+        return type == :object ? OBJECT : ACTION
+      end
+
+      # Decodes the numeric enum value into its symbol.
+      #
+      # @param value [Fixnum]  the numeric enum value.
+      # @return [Symbol]  the type symbol.
+      def self.decode(value)
+        return nil unless [OBJECT, ACTION].index(value)
+        return value == OBJECT ? :object : :action
+      end
+    end
+  end
+end

data/lib/skydb/timestamp.rb

@@ -0,0 +1,22 @@
+class SkyDB
+  class Timestamp
+    ##########################################################################
+    #
+    # Static Methods
+    #
+    ##########################################################################
+
+    # Converts a Time object to a Sky timestamp value.
+    #
+    # @param [Time] time  the time object.
+    #
+    # @return [Fixnum]  the number of microseconds since the epoch (1970-01-01T00:00:00.000000Z).
+    def self.to_timestamp(time)
+      if time.nil?
+        return nil
+      else
+        return time.to_time.utc.to_i * 1_000_000
+      end
+    end
+  end
+end

data/lib/skydb/version.rb

@@ -0,0 +1,3 @@
+class SkyDB
+  VERSION = "0.2.0"
+end

data/test/client_test.rb

@@ -0,0 +1,71 @@
+# encoding: binary
+require 'test_helper'
+
+class TestClient < MiniTest::Unit::TestCase
+  def setup
+    @client = SkyDB::Client.new(host:'127.0.0.1', port:9000)
+    @message = mock('message')
+  end
+  
+  ######################################
+  # Action Messages
+  ######################################
+
+  def test_add_action
+    action = SkyDB::Action.new
+    @client.expects(:send_message).with(is_a(SkyDB::Message::AddAction))
+    @client.add_action(action)
+  end
+
+  def test_get_action
+    @client.expects(:send_message).with(is_a(SkyDB::Message::GetAction))
+    @client.get_action(12)
+  end
+
+  def test_get_actions
+    @client.expects(:send_message).with(is_a(SkyDB::Message::GetActions))
+    @client.get_actions
+  end
+
+
+  ######################################
+  # Property Messages
+  ######################################
+
+  def test_add_property
+    property = SkyDB::Property.new
+    @client.expects(:send_message).with(is_a(SkyDB::Message::AddProperty))
+    @client.add_property(property)
+  end
+
+  def test_get_property
+    @client.expects(:send_message).with(is_a(SkyDB::Message::GetProperty))
+    @client.get_property(12)
+  end
+
+  def test_get_properties
+    @client.expects(:send_message).with(is_a(SkyDB::Message::GetProperties))
+    @client.get_properties
+  end
+
+
+  ######################################
+  # Event Messages
+  ######################################
+
+  def test_add_event
+    event = SkyDB::Event.new
+    @client.expects(:send_message).with(is_a(SkyDB::Message::AddEvent))
+    @client.add_event(event)
+  end
+
+
+  ######################################
+  # Query Messages
+  ######################################
+
+  def test_next_actions
+    @client.expects(:send_message).with(is_a(SkyDB::Message::NextActions))
+    @client.next_actions([1, 2])
+  end
+end

data/test/event_test.rb

@@ -0,0 +1,37 @@
+# encoding: binary
+require 'test_helper'
+
+class TestEvent < MiniTest::Unit::TestCase
+  def setup
+    @event = SkyDB::Event.new()
+  end
+  
+  ######################################
+  # Object ID
+  ######################################
+  
+  def test_object_id
+    @event.object_id = 12
+    assert_equal 12, @event.object_id
+  end
+  
+  def test_invalid_object_id
+    @event.object_id = "foo"
+    assert_equal 0, @event.object_id
+  end
+  
+
+  ######################################
+  # Timestamp
+  ######################################
+  
+  def test_timestamp_with_time
+    @event.timestamp = Time.now
+    refute_nil @event.timestamp
+  end
+  
+  def test_invalid_timestamp
+    @event.object_id = "foo"
+    assert_nil @event.timestamp
+  end
+end