impala 0.1.0

data/lib/impala/protocol/impala_plan_service.rb

@@ -0,0 +1,310 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'impala_plan_service_types'
+
+module Impala
+  module Protocol
+    module ImpalaPlanService
+      class Client
+        include ::Thrift::Client
+
+        def CreateExecRequest(clientRequest)
+          send_CreateExecRequest(clientRequest)
+          return recv_CreateExecRequest()
+        end
+
+        def send_CreateExecRequest(clientRequest)
+          send_message('CreateExecRequest', CreateExecRequest_args, :clientRequest => clientRequest)
+        end
+
+        def recv_CreateExecRequest()
+          result = receive_message(CreateExecRequest_result)
+          return result.success unless result.success.nil?
+          raise result.e unless result.e.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'CreateExecRequest failed: unknown result')
+        end
+
+        def RefreshMetadata()
+          send_RefreshMetadata()
+          recv_RefreshMetadata()
+        end
+
+        def send_RefreshMetadata()
+          send_message('RefreshMetadata', RefreshMetadata_args)
+        end
+
+        def recv_RefreshMetadata()
+          result = receive_message(RefreshMetadata_result)
+          return
+        end
+
+        def GetExplainString(queryRequest)
+          send_GetExplainString(queryRequest)
+          return recv_GetExplainString()
+        end
+
+        def send_GetExplainString(queryRequest)
+          send_message('GetExplainString', GetExplainString_args, :queryRequest => queryRequest)
+        end
+
+        def recv_GetExplainString()
+          result = receive_message(GetExplainString_result)
+          return result.success unless result.success.nil?
+          raise result.e unless result.e.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'GetExplainString failed: unknown result')
+        end
+
+        def UpdateMetastore(update)
+          send_UpdateMetastore(update)
+          recv_UpdateMetastore()
+        end
+
+        def send_UpdateMetastore(update)
+          send_message('UpdateMetastore', UpdateMetastore_args, :update => update)
+        end
+
+        def recv_UpdateMetastore()
+          result = receive_message(UpdateMetastore_result)
+          raise result.e unless result.e.nil?
+          return
+        end
+
+        def ShutdownServer()
+          send_ShutdownServer()
+          recv_ShutdownServer()
+        end
+
+        def send_ShutdownServer()
+          send_message('ShutdownServer', ShutdownServer_args)
+        end
+
+        def recv_ShutdownServer()
+          result = receive_message(ShutdownServer_result)
+          return
+        end
+
+      end
+
+      class Processor
+        include ::Thrift::Processor
+
+        def process_CreateExecRequest(seqid, iprot, oprot)
+          args = read_args(iprot, CreateExecRequest_args)
+          result = CreateExecRequest_result.new()
+          begin
+            result.success = @handler.CreateExecRequest(args.clientRequest)
+          rescue Impala::Protocol::TImpalaPlanServiceException => e
+            result.e = e
+          end
+          write_result(result, oprot, 'CreateExecRequest', seqid)
+        end
+
+        def process_RefreshMetadata(seqid, iprot, oprot)
+          args = read_args(iprot, RefreshMetadata_args)
+          result = RefreshMetadata_result.new()
+          @handler.RefreshMetadata()
+          write_result(result, oprot, 'RefreshMetadata', seqid)
+        end
+
+        def process_GetExplainString(seqid, iprot, oprot)
+          args = read_args(iprot, GetExplainString_args)
+          result = GetExplainString_result.new()
+          begin
+            result.success = @handler.GetExplainString(args.queryRequest)
+          rescue Impala::Protocol::TImpalaPlanServiceException => e
+            result.e = e
+          end
+          write_result(result, oprot, 'GetExplainString', seqid)
+        end
+
+        def process_UpdateMetastore(seqid, iprot, oprot)
+          args = read_args(iprot, UpdateMetastore_args)
+          result = UpdateMetastore_result.new()
+          begin
+            @handler.UpdateMetastore(args.update)
+          rescue Impala::Protocol::TImpalaPlanServiceException => e
+            result.e = e
+          end
+          write_result(result, oprot, 'UpdateMetastore', seqid)
+        end
+
+        def process_ShutdownServer(seqid, iprot, oprot)
+          args = read_args(iprot, ShutdownServer_args)
+          result = ShutdownServer_result.new()
+          @handler.ShutdownServer()
+          write_result(result, oprot, 'ShutdownServer', seqid)
+        end
+
+      end
+
+      # HELPER FUNCTIONS AND STRUCTURES
+
+      class CreateExecRequest_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        CLIENTREQUEST = 1
+
+        FIELDS = {
+        CLIENTREQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'clientRequest', :class => Impala::Protocol::TClientRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class CreateExecRequest_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+        E = 1
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TExecRequest },
+        E => { :type => ::Thrift::Types::STRUCT, :name => 'e', :class => Impala::Protocol::TImpalaPlanServiceException }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class RefreshMetadata_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class RefreshMetadata_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class GetExplainString_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        QUERYREQUEST = 1
+
+        FIELDS = {
+        QUERYREQUEST => { :type => ::Thrift::Types::STRUCT, :name => 'queryRequest', :class => Impala::Protocol::TClientRequest }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class GetExplainString_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+        E = 1
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRING, :name => 'success' },
+        E => { :type => ::Thrift::Types::STRUCT, :name => 'e', :class => Impala::Protocol::TImpalaPlanServiceException }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UpdateMetastore_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        UPDATE = 1
+
+        FIELDS = {
+        UPDATE => { :type => ::Thrift::Types::STRUCT, :name => 'update', :class => Impala::Protocol::TCatalogUpdate }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class UpdateMetastore_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        E = 1
+
+        FIELDS = {
+        E => { :type => ::Thrift::Types::STRUCT, :name => 'e', :class => Impala::Protocol::TImpalaPlanServiceException }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class ShutdownServer_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class ShutdownServer_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+    end
+
+  end
+end

data/lib/impala/protocol/impala_plan_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 'impala_plan_service_types'
+
+module Impala
+  module Protocol
+  end
+end

data/lib/impala/protocol/impala_plan_service_types.rb

@@ -0,0 +1,36 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'frontend_types'
+
+
+module Impala
+  module Protocol
+    class TImpalaPlanServiceException < ::Thrift::Exception
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      def initialize(message=nil)
+        super()
+        self.msg = message
+      end
+
+      def message; msg end
+
+      MSG = 1
+
+      FIELDS = {
+      MSG => { :type => ::Thrift::Types::STRING, :name => 'msg' }
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+end

data/lib/impala/protocol/impala_service.rb

@@ -0,0 +1,260 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'beeswax_service'
+require 'impala_service_types'
+
+module Impala
+  module Protocol
+    module ImpalaService
+      class Client < Impala::Protocol::Beeswax::BeeswaxService::Client
+        include ::Thrift::Client
+
+        def Cancel(query_id)
+          send_Cancel(query_id)
+          return recv_Cancel()
+        end
+
+        def send_Cancel(query_id)
+          send_message('Cancel', Cancel_args, :query_id => query_id)
+        end
+
+        def recv_Cancel()
+          result = receive_message(Cancel_result)
+          return result.success unless result.success.nil?
+          raise result.error unless result.error.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'Cancel failed: unknown result')
+        end
+
+        def ResetCatalog()
+          send_ResetCatalog()
+          return recv_ResetCatalog()
+        end
+
+        def send_ResetCatalog()
+          send_message('ResetCatalog', ResetCatalog_args)
+        end
+
+        def recv_ResetCatalog()
+          result = receive_message(ResetCatalog_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'ResetCatalog failed: unknown result')
+        end
+
+        def CloseInsert(handle)
+          send_CloseInsert(handle)
+          return recv_CloseInsert()
+        end
+
+        def send_CloseInsert(handle)
+          send_message('CloseInsert', CloseInsert_args, :handle => handle)
+        end
+
+        def recv_CloseInsert()
+          result = receive_message(CloseInsert_result)
+          return result.success unless result.success.nil?
+          raise result.error unless result.error.nil?
+          raise result.error2 unless result.error2.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'CloseInsert failed: unknown result')
+        end
+
+        def PingImpalaService()
+          send_PingImpalaService()
+          recv_PingImpalaService()
+        end
+
+        def send_PingImpalaService()
+          send_message('PingImpalaService', PingImpalaService_args)
+        end
+
+        def recv_PingImpalaService()
+          result = receive_message(PingImpalaService_result)
+          return
+        end
+
+      end
+
+      class Processor < Impala::Protocol::Beeswax::BeeswaxService::Processor
+        include ::Thrift::Processor
+
+        def process_Cancel(seqid, iprot, oprot)
+          args = read_args(iprot, Cancel_args)
+          result = Cancel_result.new()
+          begin
+            result.success = @handler.Cancel(args.query_id)
+          rescue Impala::Protocol::Beeswax::BeeswaxException => error
+            result.error = error
+          end
+          write_result(result, oprot, 'Cancel', seqid)
+        end
+
+        def process_ResetCatalog(seqid, iprot, oprot)
+          args = read_args(iprot, ResetCatalog_args)
+          result = ResetCatalog_result.new()
+          result.success = @handler.ResetCatalog()
+          write_result(result, oprot, 'ResetCatalog', seqid)
+        end
+
+        def process_CloseInsert(seqid, iprot, oprot)
+          args = read_args(iprot, CloseInsert_args)
+          result = CloseInsert_result.new()
+          begin
+            result.success = @handler.CloseInsert(args.handle)
+          rescue Impala::Protocol::Beeswax::QueryNotFoundException => error
+            result.error = error
+          rescue Impala::Protocol::Beeswax::BeeswaxException => error2
+            result.error2 = error2
+          end
+          write_result(result, oprot, 'CloseInsert', seqid)
+        end
+
+        def process_PingImpalaService(seqid, iprot, oprot)
+          args = read_args(iprot, PingImpalaService_args)
+          result = PingImpalaService_result.new()
+          @handler.PingImpalaService()
+          write_result(result, oprot, 'PingImpalaService', seqid)
+        end
+
+      end
+
+      # HELPER FUNCTIONS AND STRUCTURES
+
+      class Cancel_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        QUERY_ID = 1
+
+        FIELDS = {
+        QUERY_ID => { :type => ::Thrift::Types::STRUCT, :name => 'query_id', :class => Impala::Protocol::Beeswax::QueryHandle }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class Cancel_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+        ERROR = 1
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TStatus },
+        ERROR => { :type => ::Thrift::Types::STRUCT, :name => 'error', :class => Impala::Protocol::Beeswax::BeeswaxException }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class ResetCatalog_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class ResetCatalog_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TStatus }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class CloseInsert_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        HANDLE = 1
+
+        FIELDS = {
+        HANDLE => { :type => ::Thrift::Types::STRUCT, :name => 'handle', :class => Impala::Protocol::Beeswax::QueryHandle }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class CloseInsert_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+        ERROR = 1
+        ERROR2 = 2
+
+        FIELDS = {
+        SUCCESS => { :type => ::Thrift::Types::STRUCT, :name => 'success', :class => Impala::Protocol::TInsertResult },
+        ERROR => { :type => ::Thrift::Types::STRUCT, :name => 'error', :class => Impala::Protocol::Beeswax::QueryNotFoundException },
+        ERROR2 => { :type => ::Thrift::Types::STRUCT, :name => 'error2', :class => Impala::Protocol::Beeswax::BeeswaxException }
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class PingImpalaService_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class PingImpalaService_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+
+        FIELDS = {
+
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+    end
+
+  end
+end