orientdb-binary 0.6.0

data/lib/orientdb_binary/protocols/base.rb

@@ -0,0 +1,42 @@
+module OrientdbBinary
+  module Protocols
+
+    module Base
+
+      def process(socket)
+        write(socket)
+
+        status = BinData::Int8.read(socket).to_i
+        errors = process_errors(socket, status)
+        unless errors
+          constantize("#{self.class.to_s}Answer").read(socket)
+        else
+          return {exceptions: errors[:exceptions][0..-2]}
+        end
+      end
+
+      private
+
+      def process_errors(socket, status)
+        if status == 1
+          errors = OrientdbBinary::Protocols::Errors.read(socket)
+          return errors
+        end
+        return nil
+      end
+
+      # this might go to String (as a refinment?)
+      def constantize(name)
+        names = name.split('::')
+        names.shift if names.empty? || names.first.empty?
+
+        constant = Object
+        names.each do |name|
+          constant = constant.const_defined?(name) ? constant.const_get(name) : constant.const_missing(name)
+        end
+        constant
+
+      end
+    end
+  end
+end

data/lib/orientdb_binary/protocols/bindata_primitives.rb

@@ -0,0 +1,46 @@
+module OrientdbBinary
+  module BinDataPrimitives
+    class ProtocolString < BinData::Primitive
+      endian :big
+
+      int32 :len, value: -> { data.length }
+      string :data, read_length: :len
+
+      def get
+        data
+      end
+
+      def set(v)
+        self.data = v
+      end
+    end
+
+    class RecordType < BinData::Primitive
+      endian :big
+
+      int8 :type
+
+      def get
+        type.chr
+      end
+
+      def set(v)
+        self.type = v.ord
+      end
+    end
+
+    class RecordObject < BinData::Primitive
+      endian :big
+
+      protocol_string :content
+
+      def get
+        self.content = OrientdbBinary::Parser::Deserializer.new.deserialize_document(self.content)
+      end
+
+      def set(v)
+        self.content = v
+      end
+    end
+  end
+end

data/lib/orientdb_binary/protocols/command.rb

@@ -0,0 +1,166 @@
+module OrientdbBinary
+  module Protocols
+
+    # class SqlCommandPayload < BinData::Record
+    #   endian :big
+
+    #   protocol_string :text
+    #   int32 :non_text_limit
+    #   protocol_string :fetchplan
+    #   protocol_string :serialized_params, value: "params:[name:\"admin\"]"
+    # end
+
+    # class SqlCommandPayload < BinData::Record
+    #   endian :big
+
+    #   protocol_string :class_name
+    #   protocol_string :text
+    #   int32 :non_text_limit, value: -1
+    #   protocol_string :fetchplan, value: '*:-1'
+    #   protocol_string :serialized_params
+    #   # int8 :with_params, value: 1
+    #   # int8 :composite_key_params_present, value: 0
+    # end
+    class Recordd < BinData::Primitive
+      endian :big
+
+      array :records do
+            int16 :marker
+            int8 :record_type
+            int16 :cluster_id
+            int64 :position
+            int32 :version
+            protocol_string :record_content
+      end
+    end
+
+    class SqlCommandPayload < BinData::Record
+      endian :big
+
+      protocol_string :class_name
+      protocol_string :text
+      int32 :non_text_limit, value: -1
+      protocol_string :fetch_plan, initial_value: '*:0'
+      protocol_string :serialized_params
+    end
+
+    class CommandAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      int8 :synch_result_type
+
+      int32 :collection_size, onlyif: -> {synch_result_type == 108}
+      array :collection, initial_length: :collection_size, onlyif: -> {synch_result_type == 108} do
+        int16 :marker
+        int8 :record_type
+        int16 :cluster_id
+        int64 :position
+        int32 :version
+        protocol_string :record_content
+      end
+
+      int16 :marker, onlyif: -> {synch_result_type == 114}
+      int8 :record_type, onlyif: -> {synch_result_type == 114}
+      int16 :cluster_id, onlyif: -> {synch_result_type == 114}
+      int64 :position, onlyif: -> {synch_result_type == 114}
+      int32 :version, onlyif: -> {synch_result_type == 114}
+      protocol_string :record_content, onlyif: -> {synch_result_type == 114 or synch_result_type == 97}
+
+      array :prefetched_records, read_until: -> {element.payload_status == 0} do
+        int8  :payload_status
+        int16 :marker, onlyif: -> {payload_status > 0}
+        int8 :record_type, onlyif: -> {payload_status > 0}
+        int16 :cluster_id, onlyif: -> {payload_status > 0}
+        int64 :position, onlyif: -> {payload_status > 0}
+        int32 :version, onlyif: -> {payload_status > 0}
+        protocol_string :record_content, onlyif: -> {payload_status > 0}
+      end
+
+    end
+
+
+    # class SqlCommandPayload < BinData::Record
+    #   endian :big
+
+    #   protocol_string :class_name, value: 'com.orientechnologies.orient.core.sql.OCommandSQL'
+    #   protocol_string :text, value: 'SELECT FROM OUser WHERE name = :name'
+    #   # protocol_string :fetchplan, value: '*:1'
+    #   int8 :with_params, value: 1
+    #   protocol_string :serialized_params, value: 'params:{"name":"admin"}'
+    #   int8 :composite_key_params_present, value: 0
+    # end
+
+    class ScriptCommandPayload < BinData::Record
+      endian :big
+
+      protocol_string :language
+      protocol_string :text
+      int32 :non_text_limit
+      protocol_string :fetchplan
+      int32 :serialized_params, value: 0
+    end
+
+    class Command < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_COMMAND
+      int32 :session
+
+      bit8 :mode
+      protocol_string :command_payload
+    end
+
+    # class CommandAnswer < BinData::Record
+    #   endian :big
+
+    #   int32 :session
+    #   int8 :synch_result_type
+    #   int32 :collection_size
+    #   array :collection, initial_length: :collection_size do
+    #     int16 :marker
+    #     int8 :record_type
+    #     int16 :cluster_id
+    #     int64 :position
+    #     int32 :version
+    #     record_content :record_content
+    #   end
+    # end
+  end
+end
+
+
+# if ($this->mode == OrientDB::COMMAND_QUERY || $this->mode == OrientDB::COMMAND_SELECT_SYNC || $this->mode == OrientDB::COMMAND_SELECT_GREMLIN) {
+#             $this->addByte(self::MODE_SYNC);
+#         } else {
+#             $this->addByte(self::MODE_ASYNC);
+#         }
+#         if ($this->mode == OrientDB::COMMAND_SELECT_ASYNC) {
+#             $objName = 'com.orientechnologies.orient.core.sql.query.OSQLAsynchQuery';
+#         } elseif ($this->mode == OrientDB::COMMAND_SELECT_SYNC) {
+#             $objName = 'com.orientechnologies.orient.core.sql.query.OSQLSynchQuery';
+#         } elseif ($this->mode == OrientDB::COMMAND_SELECT_GREMLIN) {
+#             $objName = 'com.orientechnologies.orient.graph.gremlin.OCommandGremlin';
+#         } else {
+#             $objName = 'com.orientechnologies.orient.core.sql.OCommandSQL';
+#         }
+# $buff = '';
+#         // Java query object name serialization
+#         $buff .= pack('N', strlen($objName));
+#         $buff .= $objName;
+#         // Query text serialization in TEXT mode
+#         $buff .= pack('N', strlen($this->query));
+#         $buff .= $this->query;
+#         if ($this->mode == OrientDB::COMMAND_SELECT_ASYNC || $this->mode == OrientDB::COMMAND_SELECT_SYNC || $this->mode == OrientDB::COMMAND_SELECT_GREMLIN) {
+#             // Limit set to -1 to ignore and use TEXT MODE
+#             $buff .= pack('N', -1);
+#             // Add a fetchplan
+#             $buff .= pack('N', strlen($this->fetchPlan));
+#             $buff .= $this->fetchPlan;
+#         }
+#         // Params serialization, we have 0 params
+#         $buff .= pack('N', 0);
+#         // Now query object serialization complete, add it to command bytes
+#         $this->addString($buff);

data/lib/orientdb_binary/protocols/config_get.rb

@@ -0,0 +1,22 @@
+module OrientdbBinary
+  module Protocols
+
+    class ConfigGet < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_CONFIG_GET
+      int32 :session
+
+      protocol_string :option_key
+    end
+
+    class ConfigGetAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      protocol_string :option_value
+    end
+  end
+end

data/lib/orientdb_binary/protocols/config_list.rb

@@ -0,0 +1,24 @@
+module OrientdbBinary
+  module Protocols
+
+    class ConfigList < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_CONFIG_LIST
+      int32 :session
+    end
+
+    class ConfigListAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      int16 :config_count
+      array :config_list, initial_length: :config_count do
+        protocol_string :option_key
+        protocol_string :option_value
+      end
+    end
+  end
+end

data/lib/orientdb_binary/protocols/config_set.rb

@@ -0,0 +1,22 @@
+module OrientdbBinary
+  module Protocols
+
+    class ConfigSet < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_CONFIG_SET
+      int32 :session
+
+      protocol_string :option_key
+      protocol_string :option_value
+    end
+
+    class ConfigSetAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+    end
+  end
+end

data/lib/orientdb_binary/protocols/connect.rb

@@ -0,0 +1,26 @@
+module OrientdbBinary
+  module Protocols
+    class Connect < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_CONNECT
+      int32 :session, value: OrientdbBinary::OperationTypes::NEW_SESSION
+
+      protocol_string :driver, value: OrientdbBinary::NAME
+      protocol_string :driver_version, value: OrientdbBinary::VERSION
+      int16 :protocol, value: OrientdbBinary::PROTOCOL_VERSION
+      protocol_string :client_id, value: "#{OrientdbBinary::NAME} #{OrientdbBinary::VERSION}, protocol v#{OrientdbBinary::PROTOCOL_VERSION}"
+      protocol_string :user
+      protocol_string :password
+    end
+
+    class ConnectAnswer < BinData::Record
+      endian :big
+
+      skip length: 4
+      int32 :session
+    end
+  end
+end

data/lib/orientdb_binary/protocols/datacluster_add.rb

@@ -0,0 +1,26 @@
+module OrientdbBinary
+  module Protocols
+
+    class DataclusterAdd < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_DATACLUSTER_ADD
+      int32 :session
+
+      protocol_string :cluster_type
+      protocol_string :name
+      protocol_string :location
+      protocol_string :datasegment_name
+      int16 :cluster_id, initial_value: -1
+    end
+
+    class DataclusterAddAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      int16 :cluster_id
+    end
+  end
+end

data/lib/orientdb_binary/protocols/datacluster_count.rb

@@ -0,0 +1,26 @@
+module OrientdbBinary
+  module Protocols
+
+    class DataclusterCount < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_DATACLUSTER_COUNT
+      int32 :session
+
+      int16 :cluster_count
+      array :cluster_ids, initial_length: :cluster_count do
+        int16 :cluster_number
+      end
+      int8 :count_tombstones
+    end
+
+    class DataclusterCountAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      int64 :records_in_clusters
+    end
+  end
+end

data/lib/orientdb_binary/protocols/datacluster_datarange.rb

@@ -0,0 +1,23 @@
+module OrientdbBinary
+  module Protocols
+
+    class DataclusterDatarange < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_DATACLUSTER_DATARANGE
+      int32 :session
+
+      int16 :cluster_id
+    end
+
+    class DataclusterDatarangeAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      int64 :record_id_begin
+      int64 :record_id_end
+    end
+  end
+end

data/lib/orientdb_binary/protocols/datacluster_drop.rb

@@ -0,0 +1,22 @@
+module OrientdbBinary
+  module Protocols
+
+    class DataclusterDrop < BinData::Record
+      include OrientdbBinary::Protocols::Base
+
+      endian :big
+
+      int8 :operation, value: OrientdbBinary::OperationTypes::REQUEST_DATACLUSTER_DROP
+      int32 :session
+
+      int16 :cluster_id
+    end
+
+    class DataclusterDropAnswer < BinData::Record
+      endian :big
+
+      int32 :session
+      bit4 :succeed
+    end
+  end
+end