impala 0.1.0

data/lib/impala/protocol/runtime_profile_constants.rb

@@ -0,0 +1,12 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'runtime_profile_types'
+
+module Impala
+  module Protocol
+  end
+end

data/lib/impala/protocol/runtime_profile_types.rb

@@ -0,0 +1,97 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+
+module Impala
+  module Protocol
+    module TCounterType
+      UNIT = 0
+      UNIT_PER_SECOND = 1
+      TIME_MS = 2
+      CPU_TICKS = 3
+      BYTES = 4
+      BYTES_PER_SECOND = 5
+      VALUE_MAP = { 0 => "UNIT", 1 => "UNIT_PER_SECOND", 2 => "TIME_MS", 3 => "CPU_TICKS", 4 => "BYTES", 5 => "BYTES_PER_SECOND" }
+      VALID_VALUES = Set.new([UNIT, UNIT_PER_SECOND, TIME_MS, CPU_TICKS, BYTES, BYTES_PER_SECOND]).freeze
+    end
+
+    class TCounter
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      NAME = 1
+      TYPE = 2
+      VALUE = 3
+
+      FIELDS = {
+      NAME => { :type => ::Thrift::Types::STRING, :name => 'name' },
+      TYPE => { :type => ::Thrift::Types::I32, :name => 'type', :enum_class => Impala::Protocol::TCounterType },
+      VALUE => { :type => ::Thrift::Types::I64, :name => 'value' }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field name is unset!') unless @name
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field type is unset!') unless @type
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field value is unset!') unless @value
+        unless @type.nil? || Impala::Protocol::TCounterType::VALID_VALUES.include?(@type)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field type!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TRuntimeProfileNode
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      NAME = 1
+      NUM_CHILDREN = 2
+      COUNTERS = 3
+      METADATA = 4
+      INDENT = 5
+      INFO_STRINGS = 6
+
+      FIELDS = {
+      NAME => { :type => ::Thrift::Types::STRING, :name => 'name' },
+      NUM_CHILDREN => { :type => ::Thrift::Types::I32, :name => 'num_children' },
+      COUNTERS => { :type => ::Thrift::Types::LIST, :name => 'counters', :element => { :type => ::Thrift::Types::STRUCT, :class => Impala::Protocol::TCounter } },
+      METADATA => { :type => ::Thrift::Types::I64, :name => 'metadata' },
+      INDENT => { :type => ::Thrift::Types::BOOL, :name => 'indent' },
+      INFO_STRINGS => { :type => ::Thrift::Types::MAP, :name => 'info_strings', :key => { :type => ::Thrift::Types::STRING }, :value => { :type => ::Thrift::Types::STRING } }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field name is unset!') unless @name
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field num_children is unset!') unless @num_children
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field counters is unset!') unless @counters
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field metadata is unset!') unless @metadata
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field indent is unset!') if @indent.nil?
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field info_strings is unset!') unless @info_strings
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TRuntimeProfileTree
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      NODES = 1
+
+      FIELDS = {
+      NODES => { :type => ::Thrift::Types::LIST, :name => 'nodes', :element => { :type => ::Thrift::Types::STRUCT, :class => Impala::Protocol::TRuntimeProfileNode } }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field nodes is unset!') unless @nodes
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+end

data/lib/impala/protocol/state_store_service.rb

@@ -0,0 +1,244 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'state_store_service_types'
+
+module Impala
+  module Protocol
+    module StateStoreService
+      class Client
+        include ::Thrift::Client
+
+        def RegisterService(request)
+          send_RegisterService(request)
+          return recv_RegisterService()
+        end
+
+        def send_RegisterService(request)
+          send_message('RegisterService', RegisterService_args, :request => request)
+        end
+
+        def recv_RegisterService()
+          result = receive_message(RegisterService_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'RegisterService failed: unknown result')
+        end
+
+        def UnregisterService(request)
+          send_UnregisterService(request)
+          return recv_UnregisterService()
+        end
+
+        def send_UnregisterService(request)
+          send_message('UnregisterService', UnregisterService_args, :request => request)
+        end
+
+        def recv_UnregisterService()
+          result = receive_message(UnregisterService_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'UnregisterService failed: unknown result')
+        end
+
+        def RegisterSubscription(request)
+          send_RegisterSubscription(request)
+          return recv_RegisterSubscription()
+        end
+
+        def send_RegisterSubscription(request)
+          send_message('RegisterSubscription', RegisterSubscription_args, :request => request)
+        end
+
+        def recv_RegisterSubscription()
+          result = receive_message(RegisterSubscription_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'RegisterSubscription failed: unknown result')
+        end
+
+        def UnregisterSubscription(request)
+          send_UnregisterSubscription(request)
+          return recv_UnregisterSubscription()
+        end
+
+        def send_UnregisterSubscription(request)
+          send_message('UnregisterSubscription', UnregisterSubscription_args, :request => request)
+        end
+
+        def recv_UnregisterSubscription()
+          result = receive_message(UnregisterSubscription_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'UnregisterSubscription failed: unknown result')
+        end
+
+      end
+
+      class Processor
+        include ::Thrift::Processor
+
+        def process_RegisterService(seqid, iprot, oprot)
+          args = read_args(iprot, RegisterService_args)
+          result = RegisterService_result.new()
+          result.success = @handler.RegisterService(args.request)
+          write_result(result, oprot, 'RegisterService', seqid)
+        end
+
+        def process_UnregisterService(seqid, iprot, oprot)
+          args = read_args(iprot, UnregisterService_args)
+          result = UnregisterService_result.new()
+          result.success = @handler.UnregisterService(args.request)
+          write_result(result, oprot, 'UnregisterService', seqid)
+        end
+
+        def process_RegisterSubscription(seqid, iprot, oprot)
+          args = read_args(iprot, RegisterSubscription_args)
+          result = RegisterSubscription_result.new()
+          result.success = @handler.RegisterSubscription(args.request)
+          write_result(result, oprot, 'RegisterSubscription', seqid)
+        end
+
+        def process_UnregisterSubscription(seqid, iprot, oprot)
+          args = read_args(iprot, UnregisterSubscription_args)
+          result = UnregisterSubscription_result.new()
+          result.success = @handler.UnregisterSubscription(args.request)
+          write_result(result, oprot, 'UnregisterSubscription', seqid)
+        end
+
+      end
+
+      # HELPER FUNCTIONS AND STRUCTURES
+
+      class RegisterService_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+        REQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'request', :class => Impala::Protocol::TRegisterServiceRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class RegisterService_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TRegisterServiceResponse }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UnregisterService_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+        REQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'request', :class => Impala::Protocol::TUnregisterServiceRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UnregisterService_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TUnregisterServiceResponse }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class RegisterSubscription_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+        REQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'request', :class => Impala::Protocol::TRegisterSubscriptionRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class RegisterSubscription_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TRegisterSubscriptionResponse }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UnregisterSubscription_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+        REQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'request', :class => Impala::Protocol::TUnregisterSubscriptionRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UnregisterSubscription_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TUnregisterSubscriptionResponse }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+    end
+
+  end
+end

data/lib/impala/protocol/state_store_service_constants.rb

@@ -0,0 +1,12 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'state_store_service_types'
+
+module Impala
+  module Protocol
+  end
+end

data/lib/impala/protocol/state_store_service_types.rb

@@ -0,0 +1,185 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'statestore_types_types'
+require 'status_types'
+require 'types_types'
+
+
+module Impala
+  module Protocol
+    module StateStoreServiceVersion
+      V1 = 0
+      VALUE_MAP = { 0 => "V1" }
+      VALID_VALUES = Set.new([V1]).freeze
+    end
+
+    class TRegisterServiceRequest
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      PROTOCOL_VERSION = 1
+      SUBSCRIBER_ADDRESS = 2
+      SERVICE_ID = 3
+      SERVICE_ADDRESS = 4
+
+      FIELDS = {
+      PROTOCOL_VERSION => { :type => ::Thrift::Types::I32, :name => 'protocol_version', :default =>             0, :enum_class => Impala::Protocol::StateStoreServiceVersion },
+      SUBSCRIBER_ADDRESS => { :type => ::Thrift::Types::STRUCT, :name => 'subscriber_address', :class => Impala::Protocol::THostPort, :optional => true },
+      SERVICE_ID => { :type => ::Thrift::Types::STRING, :name => 'service_id', :optional => true },
+      SERVICE_ADDRESS => { :type => ::Thrift::Types::STRUCT, :name => 'service_address', :class => Impala::Protocol::THostPort, :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field protocol_version is unset!') unless @protocol_version
+        unless @protocol_version.nil? || Impala::Protocol::StateStoreServiceVersion::VALID_VALUES.include?(@protocol_version)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field protocol_version!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TRegisterServiceResponse
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STATUS = 1
+
+      FIELDS = {
+      STATUS => { :type => ::Thrift::Types::STRUCT, :name => 'status', :class => Impala::Protocol::TStatus, :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TUnregisterServiceRequest
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      PROTOCOL_VERSION = 1
+      SUBSCRIBER_ADDRESS = 2
+      SERVICE_ID = 3
+
+      FIELDS = {
+      PROTOCOL_VERSION => { :type => ::Thrift::Types::I32, :name => 'protocol_version', :default =>             0, :enum_class => Impala::Protocol::StateStoreServiceVersion },
+      SUBSCRIBER_ADDRESS => { :type => ::Thrift::Types::STRUCT, :name => 'subscriber_address', :class => Impala::Protocol::THostPort, :optional => true },
+      SERVICE_ID => { :type => ::Thrift::Types::STRING, :name => 'service_id', :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field protocol_version is unset!') unless @protocol_version
+        unless @protocol_version.nil? || Impala::Protocol::StateStoreServiceVersion::VALID_VALUES.include?(@protocol_version)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field protocol_version!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TUnregisterServiceResponse
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STATUS = 1
+
+      FIELDS = {
+      STATUS => { :type => ::Thrift::Types::STRUCT, :name => 'status', :class => Impala::Protocol::TStatus, :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TRegisterSubscriptionRequest
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      PROTOCOL_VERSION = 1
+      SUBSCRIBER_ADDRESS = 2
+      SERVICES = 3
+      SUBSCRIPTION_ID = 4
+
+      FIELDS = {
+      PROTOCOL_VERSION => { :type => ::Thrift::Types::I32, :name => 'protocol_version', :default =>             0, :enum_class => Impala::Protocol::StateStoreServiceVersion },
+      SUBSCRIBER_ADDRESS => { :type => ::Thrift::Types::STRUCT, :name => 'subscriber_address', :class => Impala::Protocol::THostPort, :optional => true },
+      SERVICES => { :type => ::Thrift::Types::SET, :name => 'services', :element => { :type => ::Thrift::Types::STRING }, :optional => true },
+      SUBSCRIPTION_ID => { :type => ::Thrift::Types::STRING, :name => 'subscription_id', :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field protocol_version is unset!') unless @protocol_version
+        unless @protocol_version.nil? || Impala::Protocol::StateStoreServiceVersion::VALID_VALUES.include?(@protocol_version)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field protocol_version!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TRegisterSubscriptionResponse
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STATUS = 1
+
+      FIELDS = {
+      STATUS => { :type => ::Thrift::Types::STRUCT, :name => 'status', :class => Impala::Protocol::TStatus, :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TUnregisterSubscriptionRequest
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      PROTOCOL_VERSION = 1
+      SUBSCRIBER_ADDRESS = 2
+      SUBSCRIPTION_ID = 3
+
+      FIELDS = {
+      PROTOCOL_VERSION => { :type => ::Thrift::Types::I32, :name => 'protocol_version', :default =>             0, :enum_class => Impala::Protocol::StateStoreServiceVersion },
+      SUBSCRIBER_ADDRESS => { :type => ::Thrift::Types::STRUCT, :name => 'subscriber_address', :class => Impala::Protocol::THostPort, :optional => true },
+      SUBSCRIPTION_ID => { :type => ::Thrift::Types::STRING, :name => 'subscription_id', :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field protocol_version is unset!') unless @protocol_version
+        unless @protocol_version.nil? || Impala::Protocol::StateStoreServiceVersion::VALID_VALUES.include?(@protocol_version)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field protocol_version!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TUnregisterSubscriptionResponse
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STATUS = 1
+
+      FIELDS = {
+      STATUS => { :type => ::Thrift::Types::STRUCT, :name => 'status', :class => Impala::Protocol::TStatus, :optional => true }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+end