sequel-impala 1.0.0

data/lib/impala/protocol/impala_hive_server2_service.rb

@@ -0,0 +1,29 @@
+#
+# Autogenerated by Thrift Compiler (0.9.1)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 't_c_l_i_service'
+require 'impala_service_types'
+
+module Impala
+  module Protocol
+    module ImpalaHiveServer2Service
+      class Client < ::Impala::Protocol::Hive::TCLIService::Client 
+        include ::Thrift::Client
+
+      end
+
+      class Processor < ::Impala::Protocol::Hive::TCLIService::Processor 
+        include ::Thrift::Processor
+
+      end
+
+      # HELPER FUNCTIONS AND STRUCTURES
+
+    end
+
+  end
+end

data/lib/impala/protocol/impala_service.rb

@@ -0,0 +1,377 @@
+#
+# Autogenerated by Thrift Compiler (0.9.1)
+#
+# 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 ResetTable(request)
+          send_ResetTable(request)
+          return recv_ResetTable()
+        end
+
+        def send_ResetTable(request)
+          send_message('ResetTable', ResetTable_args, :request => request)
+        end
+
+        def recv_ResetTable()
+          result = receive_message(ResetTable_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'ResetTable failed: unknown result')
+        end
+
+        def GetRuntimeProfile(query_id)
+          send_GetRuntimeProfile(query_id)
+          return recv_GetRuntimeProfile()
+        end
+
+        def send_GetRuntimeProfile(query_id)
+          send_message('GetRuntimeProfile', GetRuntimeProfile_args, :query_id => query_id)
+        end
+
+        def recv_GetRuntimeProfile()
+          result = receive_message(GetRuntimeProfile_result)
+          return result.success unless result.success.nil?
+          raise result.error unless result.error.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'GetRuntimeProfile 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()
+          return recv_PingImpalaService()
+        end
+
+        def send_PingImpalaService()
+          send_message('PingImpalaService', PingImpalaService_args)
+        end
+
+        def recv_PingImpalaService()
+          result = receive_message(PingImpalaService_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'PingImpalaService failed: unknown result')
+        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_ResetTable(seqid, iprot, oprot)
+          args = read_args(iprot, ResetTable_args)
+          result = ResetTable_result.new()
+          result.success = @handler.ResetTable(args.request)
+          write_result(result, oprot, 'ResetTable', seqid)
+        end
+
+        def process_GetRuntimeProfile(seqid, iprot, oprot)
+          args = read_args(iprot, GetRuntimeProfile_args)
+          result = GetRuntimeProfile_result.new()
+          begin
+            result.success = @handler.GetRuntimeProfile(args.query_id)
+          rescue ::Impala::Protocol::Beeswax::BeeswaxException => error
+            result.error = error
+          end
+          write_result(result, oprot, 'GetRuntimeProfile', 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()
+          result.success = @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 ResetTable_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+          REQUEST => {:type => ::Thrift::Types::STRUCT, :name => 'request', :class => ::Impala::Protocol::TResetTableReq}
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class ResetTable_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 GetRuntimeProfile_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 GetRuntimeProfile_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+        ERROR = 1
+
+        FIELDS = {
+          SUCCESS => {:type => ::Thrift::Types::STRING, :name => 'success'},
+          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 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
+        SUCCESS = 0
+
+        FIELDS = {
+          SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ::Impala::Protocol::TPingImpalaServiceResp}
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+    end
+
+  end
+end

data/lib/impala/protocol/impala_service_constants.rb

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

data/lib/impala/protocol/impala_service_types.rb

@@ -0,0 +1,90 @@
+#
+# Autogenerated by Thrift Compiler (0.9.1)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'status_types'
+require 'beeswax_types'
+require 'cli_service_types'
+
+
+module Impala
+  module Protocol
+    module TImpalaQueryOptions
+      ABORT_ON_ERROR = 0
+      MAX_ERRORS = 1
+      DISABLE_CODEGEN = 2
+      BATCH_SIZE = 3
+      MEM_LIMIT = 4
+      NUM_NODES = 5
+      MAX_SCAN_RANGE_LENGTH = 6
+      MAX_IO_BUFFERS = 7
+      NUM_SCANNER_THREADS = 8
+      ALLOW_UNSUPPORTED_FORMATS = 9
+      DEFAULT_ORDER_BY_LIMIT = 10
+      DEBUG_ACTION = 11
+      ABORT_ON_DEFAULT_LIMIT_EXCEEDED = 12
+      PARQUET_COMPRESSION_CODEC = 13
+      HBASE_CACHING = 14
+      HBASE_CACHE_BLOCKS = 15
+      VALUE_MAP = {0 => "ABORT_ON_ERROR", 1 => "MAX_ERRORS", 2 => "DISABLE_CODEGEN", 3 => "BATCH_SIZE", 4 => "MEM_LIMIT", 5 => "NUM_NODES", 6 => "MAX_SCAN_RANGE_LENGTH", 7 => "MAX_IO_BUFFERS", 8 => "NUM_SCANNER_THREADS", 9 => "ALLOW_UNSUPPORTED_FORMATS", 10 => "DEFAULT_ORDER_BY_LIMIT", 11 => "DEBUG_ACTION", 12 => "ABORT_ON_DEFAULT_LIMIT_EXCEEDED", 13 => "PARQUET_COMPRESSION_CODEC", 14 => "HBASE_CACHING", 15 => "HBASE_CACHE_BLOCKS"}
+      VALID_VALUES = Set.new([ABORT_ON_ERROR, MAX_ERRORS, DISABLE_CODEGEN, BATCH_SIZE, MEM_LIMIT, NUM_NODES, MAX_SCAN_RANGE_LENGTH, MAX_IO_BUFFERS, NUM_SCANNER_THREADS, ALLOW_UNSUPPORTED_FORMATS, DEFAULT_ORDER_BY_LIMIT, DEBUG_ACTION, ABORT_ON_DEFAULT_LIMIT_EXCEEDED, PARQUET_COMPRESSION_CODEC, HBASE_CACHING, HBASE_CACHE_BLOCKS]).freeze
+    end
+
+    class TInsertResult
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      ROWS_APPENDED = 1
+
+      FIELDS = {
+        ROWS_APPENDED => {:type => ::Thrift::Types::MAP, :name => 'rows_appended', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field rows_appended is unset!') unless @rows_appended
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TPingImpalaServiceResp
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      VERSION = 1
+
+      FIELDS = {
+        VERSION => {:type => ::Thrift::Types::STRING, :name => 'version'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TResetTableReq
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      DB_NAME = 1
+      TABLE_NAME = 2
+
+      FIELDS = {
+        DB_NAME => {:type => ::Thrift::Types::STRING, :name => 'db_name'},
+        TABLE_NAME => {:type => ::Thrift::Types::STRING, :name => 'table_name'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field db_name is unset!') unless @db_name
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field table_name is unset!') unless @table_name
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+end