rbhive 0.2.95 → 0.5.0

data/lib/thrift/queryplan_types.rb

@@ -1,253 +1,261 @@
 #
-# Autogenerated by Thrift
+# Autogenerated by Thrift Compiler (0.9.0)
 #
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 #
 
+require 'thrift'
 
-module AdjacencyType
-  CONJUNCTIVE = 0
-  DISJUNCTIVE = 1
-  VALUE_MAP = {0 => "CONJUNCTIVE", 1 => "DISJUNCTIVE"}
-  VALID_VALUES = Set.new([CONJUNCTIVE, DISJUNCTIVE]).freeze
-end
+module Hive
+  module Thrift
+    module AdjacencyType
+      CONJUNCTIVE = 0
+      DISJUNCTIVE = 1
+      VALUE_MAP = {0 => "CONJUNCTIVE", 1 => "DISJUNCTIVE"}
+      VALID_VALUES = Set.new([CONJUNCTIVE, DISJUNCTIVE]).freeze
+    end
 
-module NodeType
-  OPERATOR = 0
-  STAGE = 1
-  VALUE_MAP = {0 => "OPERATOR", 1 => "STAGE"}
-  VALID_VALUES = Set.new([OPERATOR, STAGE]).freeze
-end
+    module NodeType
+      OPERATOR = 0
+      STAGE = 1
+      VALUE_MAP = {0 => "OPERATOR", 1 => "STAGE"}
+      VALID_VALUES = Set.new([OPERATOR, STAGE]).freeze
+    end
 
-module OperatorType
-  JOIN = 0
-  MAPJOIN = 1
-  EXTRACT = 2
-  FILTER = 3
-  FORWARD = 4
-  GROUPBY = 5
-  LIMIT = 6
-  SCRIPT = 7
-  SELECT = 8
-  TABLESCAN = 9
-  FILESINK = 10
-  REDUCESINK = 11
-  UNION = 12
-  UDTF = 13
-  LATERALVIEWJOIN = 14
-  LATERALVIEWFORWARD = 15
-  VALUE_MAP = {0 => "JOIN", 1 => "MAPJOIN", 2 => "EXTRACT", 3 => "FILTER", 4 => "FORWARD", 5 => "GROUPBY", 6 => "LIMIT", 7 => "SCRIPT", 8 => "SELECT", 9 => "TABLESCAN", 10 => "FILESINK", 11 => "REDUCESINK", 12 => "UNION", 13 => "UDTF", 14 => "LATERALVIEWJOIN", 15 => "LATERALVIEWFORWARD"}
-  VALID_VALUES = Set.new([JOIN, MAPJOIN, EXTRACT, FILTER, FORWARD, GROUPBY, LIMIT, SCRIPT, SELECT, TABLESCAN, FILESINK, REDUCESINK, UNION, UDTF, LATERALVIEWJOIN, LATERALVIEWFORWARD]).freeze
-end
+    module OperatorType
+      JOIN = 0
+      MAPJOIN = 1
+      EXTRACT = 2
+      FILTER = 3
+      FORWARD = 4
+      GROUPBY = 5
+      LIMIT = 6
+      SCRIPT = 7
+      SELECT = 8
+      TABLESCAN = 9
+      FILESINK = 10
+      REDUCESINK = 11
+      UNION = 12
+      UDTF = 13
+      LATERALVIEWJOIN = 14
+      LATERALVIEWFORWARD = 15
+      HASHTABLESINK = 16
+      HASHTABLEDUMMY = 17
+      VALUE_MAP = {0 => "JOIN", 1 => "MAPJOIN", 2 => "EXTRACT", 3 => "FILTER", 4 => "FORWARD", 5 => "GROUPBY", 6 => "LIMIT", 7 => "SCRIPT", 8 => "SELECT", 9 => "TABLESCAN", 10 => "FILESINK", 11 => "REDUCESINK", 12 => "UNION", 13 => "UDTF", 14 => "LATERALVIEWJOIN", 15 => "LATERALVIEWFORWARD", 16 => "HASHTABLESINK", 17 => "HASHTABLEDUMMY"}
+      VALID_VALUES = Set.new([JOIN, MAPJOIN, EXTRACT, FILTER, FORWARD, GROUPBY, LIMIT, SCRIPT, SELECT, TABLESCAN, FILESINK, REDUCESINK, UNION, UDTF, LATERALVIEWJOIN, LATERALVIEWFORWARD, HASHTABLESINK, HASHTABLEDUMMY]).freeze
+    end
 
-module TaskType
-  MAP = 0
-  REDUCE = 1
-  OTHER = 2
-  VALUE_MAP = {0 => "MAP", 1 => "REDUCE", 2 => "OTHER"}
-  VALID_VALUES = Set.new([MAP, REDUCE, OTHER]).freeze
-end
+    module TaskType
+      MAP = 0
+      REDUCE = 1
+      OTHER = 2
+      VALUE_MAP = {0 => "MAP", 1 => "REDUCE", 2 => "OTHER"}
+      VALID_VALUES = Set.new([MAP, REDUCE, OTHER]).freeze
+    end
 
-module StageType
-  CONDITIONAL = 0
-  COPY = 1
-  DDL = 2
-  MAPRED = 3
-  EXPLAIN = 4
-  FETCH = 5
-  FUNC = 6
-  MAPREDLOCAL = 7
-  MOVE = 8
-  VALUE_MAP = {0 => "CONDITIONAL", 1 => "COPY", 2 => "DDL", 3 => "MAPRED", 4 => "EXPLAIN", 5 => "FETCH", 6 => "FUNC", 7 => "MAPREDLOCAL", 8 => "MOVE"}
-  VALID_VALUES = Set.new([CONDITIONAL, COPY, DDL, MAPRED, EXPLAIN, FETCH, FUNC, MAPREDLOCAL, MOVE]).freeze
-end
+    module StageType
+      CONDITIONAL = 0
+      COPY = 1
+      DDL = 2
+      MAPRED = 3
+      EXPLAIN = 4
+      FETCH = 5
+      FUNC = 6
+      MAPREDLOCAL = 7
+      MOVE = 8
+      STATS = 9
+      VALUE_MAP = {0 => "CONDITIONAL", 1 => "COPY", 2 => "DDL", 3 => "MAPRED", 4 => "EXPLAIN", 5 => "FETCH", 6 => "FUNC", 7 => "MAPREDLOCAL", 8 => "MOVE", 9 => "STATS"}
+      VALID_VALUES = Set.new([CONDITIONAL, COPY, DDL, MAPRED, EXPLAIN, FETCH, FUNC, MAPREDLOCAL, MOVE, STATS]).freeze
+    end
 
-class Adjacency
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  NODE = 1
-  CHILDREN = 2
-  ADJACENCYTYPE = 3
+    class Adjacency
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      NODE = 1
+      CHILDREN = 2
+      ADJACENCYTYPE = 3
 
-  FIELDS = {
-    NODE => {:type => ::Thrift::Types::STRING, :name => 'node'},
-    CHILDREN => {:type => ::Thrift::Types::LIST, :name => 'children', :element => {:type => ::Thrift::Types::STRING}},
-    ADJACENCYTYPE => {:type => ::Thrift::Types::I32, :name => 'adjacencyType', :enum_class => AdjacencyType}
-  }
+      FIELDS = {
+        NODE => {:type => ::Thrift::Types::STRING, :name => 'node'},
+        CHILDREN => {:type => ::Thrift::Types::LIST, :name => 'children', :element => {:type => ::Thrift::Types::STRING}},
+        ADJACENCYTYPE => {:type => ::Thrift::Types::I32, :name => 'adjacencyType', :enum_class => ::Hive::Thrift::AdjacencyType}
+      }
 
-  def struct_fields; FIELDS; end
+      def struct_fields; FIELDS; end
 
-  def validate
-    unless @adjacencyType.nil? || AdjacencyType::VALID_VALUES.include?(@adjacencyType)
-      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field adjacencyType!')
+      def validate
+        unless @adjacencyType.nil? || ::Hive::Thrift::AdjacencyType::VALID_VALUES.include?(@adjacencyType)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field adjacencyType!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
     end
-  end
 
-  ::Thrift::Struct.generate_accessors self
-end
+    class Graph
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      NODETYPE = 1
+      ROOTS = 2
+      ADJACENCYLIST = 3
 
-class Graph
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  NODETYPE = 1
-  ROOTS = 2
-  ADJACENCYLIST = 3
+      FIELDS = {
+        NODETYPE => {:type => ::Thrift::Types::I32, :name => 'nodeType', :enum_class => ::Hive::Thrift::NodeType},
+        ROOTS => {:type => ::Thrift::Types::LIST, :name => 'roots', :element => {:type => ::Thrift::Types::STRING}},
+        ADJACENCYLIST => {:type => ::Thrift::Types::LIST, :name => 'adjacencyList', :element => {:type => ::Thrift::Types::STRUCT, :class => ::Hive::Thrift::Adjacency}}
+      }
 
-  FIELDS = {
-    NODETYPE => {:type => ::Thrift::Types::I32, :name => 'nodeType', :enum_class => NodeType},
-    ROOTS => {:type => ::Thrift::Types::LIST, :name => 'roots', :element => {:type => ::Thrift::Types::STRING}},
-    ADJACENCYLIST => {:type => ::Thrift::Types::LIST, :name => 'adjacencyList', :element => {:type => ::Thrift::Types::STRUCT, :class => Adjacency}}
-  }
+      def struct_fields; FIELDS; end
 
-  def struct_fields; FIELDS; end
+      def validate
+        unless @nodeType.nil? || ::Hive::Thrift::NodeType::VALID_VALUES.include?(@nodeType)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field nodeType!')
+        end
+      end
 
-  def validate
-    unless @nodeType.nil? || NodeType::VALID_VALUES.include?(@nodeType)
-      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field nodeType!')
+      ::Thrift::Struct.generate_accessors self
     end
-  end
-
-  ::Thrift::Struct.generate_accessors self
-end
 
-class Operator
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  OPERATORID = 1
-  OPERATORTYPE = 2
-  OPERATORATTRIBUTES = 3
-  OPERATORCOUNTERS = 4
-  DONE = 5
-  STARTED = 6
-
-  FIELDS = {
-    OPERATORID => {:type => ::Thrift::Types::STRING, :name => 'operatorId'},
-    OPERATORTYPE => {:type => ::Thrift::Types::I32, :name => 'operatorType', :enum_class => OperatorType},
-    OPERATORATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'operatorAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
-    OPERATORCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'operatorCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
-    DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
-    STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
-  }
-
-  def struct_fields; FIELDS; end
-
-  def validate
-    unless @operatorType.nil? || OperatorType::VALID_VALUES.include?(@operatorType)
-      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field operatorType!')
+    class Operator
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      OPERATORID = 1
+      OPERATORTYPE = 2
+      OPERATORATTRIBUTES = 3
+      OPERATORCOUNTERS = 4
+      DONE = 5
+      STARTED = 6
+
+      FIELDS = {
+        OPERATORID => {:type => ::Thrift::Types::STRING, :name => 'operatorId'},
+        OPERATORTYPE => {:type => ::Thrift::Types::I32, :name => 'operatorType', :enum_class => ::Hive::Thrift::OperatorType},
+        OPERATORATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'operatorAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
+        OPERATORCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'operatorCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
+        DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
+        STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        unless @operatorType.nil? || ::Hive::Thrift::OperatorType::VALID_VALUES.include?(@operatorType)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field operatorType!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
     end
-  end
-
-  ::Thrift::Struct.generate_accessors self
-end
 
-class Task
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  TASKID = 1
-  TASKTYPE = 2
-  TASKATTRIBUTES = 3
-  TASKCOUNTERS = 4
-  OPERATORGRAPH = 5
-  OPERATORLIST = 6
-  DONE = 7
-  STARTED = 8
-
-  FIELDS = {
-    TASKID => {:type => ::Thrift::Types::STRING, :name => 'taskId'},
-    TASKTYPE => {:type => ::Thrift::Types::I32, :name => 'taskType', :enum_class => TaskType},
-    TASKATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'taskAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
-    TASKCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'taskCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
-    OPERATORGRAPH => {:type => ::Thrift::Types::STRUCT, :name => 'operatorGraph', :class => Graph, :optional => true},
-    OPERATORLIST => {:type => ::Thrift::Types::LIST, :name => 'operatorList', :element => {:type => ::Thrift::Types::STRUCT, :class => Operator}, :optional => true},
-    DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
-    STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
-  }
-
-  def struct_fields; FIELDS; end
-
-  def validate
-    unless @taskType.nil? || TaskType::VALID_VALUES.include?(@taskType)
-      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field taskType!')
+    class Task
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      TASKID = 1
+      TASKTYPE = 2
+      TASKATTRIBUTES = 3
+      TASKCOUNTERS = 4
+      OPERATORGRAPH = 5
+      OPERATORLIST = 6
+      DONE = 7
+      STARTED = 8
+
+      FIELDS = {
+        TASKID => {:type => ::Thrift::Types::STRING, :name => 'taskId'},
+        TASKTYPE => {:type => ::Thrift::Types::I32, :name => 'taskType', :enum_class => ::Hive::Thrift::TaskType},
+        TASKATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'taskAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
+        TASKCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'taskCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
+        OPERATORGRAPH => {:type => ::Thrift::Types::STRUCT, :name => 'operatorGraph', :class => ::Hive::Thrift::Graph, :optional => true},
+        OPERATORLIST => {:type => ::Thrift::Types::LIST, :name => 'operatorList', :element => {:type => ::Thrift::Types::STRUCT, :class => ::Hive::Thrift::Operator}, :optional => true},
+        DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
+        STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        unless @taskType.nil? || ::Hive::Thrift::TaskType::VALID_VALUES.include?(@taskType)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field taskType!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
     end
-  end
-
-  ::Thrift::Struct.generate_accessors self
-end
 
-class Stage
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  STAGEID = 1
-  STAGETYPE = 2
-  STAGEATTRIBUTES = 3
-  STAGECOUNTERS = 4
-  TASKLIST = 5
-  DONE = 6
-  STARTED = 7
-
-  FIELDS = {
-    STAGEID => {:type => ::Thrift::Types::STRING, :name => 'stageId'},
-    STAGETYPE => {:type => ::Thrift::Types::I32, :name => 'stageType', :enum_class => StageType},
-    STAGEATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'stageAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
-    STAGECOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'stageCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
-    TASKLIST => {:type => ::Thrift::Types::LIST, :name => 'taskList', :element => {:type => ::Thrift::Types::STRUCT, :class => Task}},
-    DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
-    STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
-  }
-
-  def struct_fields; FIELDS; end
-
-  def validate
-    unless @stageType.nil? || StageType::VALID_VALUES.include?(@stageType)
-      raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field stageType!')
+    class Stage
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STAGEID = 1
+      STAGETYPE = 2
+      STAGEATTRIBUTES = 3
+      STAGECOUNTERS = 4
+      TASKLIST = 5
+      DONE = 6
+      STARTED = 7
+
+      FIELDS = {
+        STAGEID => {:type => ::Thrift::Types::STRING, :name => 'stageId'},
+        STAGETYPE => {:type => ::Thrift::Types::I32, :name => 'stageType', :enum_class => ::Hive::Thrift::StageType},
+        STAGEATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'stageAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
+        STAGECOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'stageCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
+        TASKLIST => {:type => ::Thrift::Types::LIST, :name => 'taskList', :element => {:type => ::Thrift::Types::STRUCT, :class => ::Hive::Thrift::Task}},
+        DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
+        STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        unless @stageType.nil? || ::Hive::Thrift::StageType::VALID_VALUES.include?(@stageType)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field stageType!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
     end
-  end
 
-  ::Thrift::Struct.generate_accessors self
-end
+    class Query
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      QUERYID = 1
+      QUERYTYPE = 2
+      QUERYATTRIBUTES = 3
+      QUERYCOUNTERS = 4
+      STAGEGRAPH = 5
+      STAGELIST = 6
+      DONE = 7
+      STARTED = 8
+
+      FIELDS = {
+        QUERYID => {:type => ::Thrift::Types::STRING, :name => 'queryId'},
+        QUERYTYPE => {:type => ::Thrift::Types::STRING, :name => 'queryType'},
+        QUERYATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'queryAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
+        QUERYCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'queryCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
+        STAGEGRAPH => {:type => ::Thrift::Types::STRUCT, :name => 'stageGraph', :class => ::Hive::Thrift::Graph},
+        STAGELIST => {:type => ::Thrift::Types::LIST, :name => 'stageList', :element => {:type => ::Thrift::Types::STRUCT, :class => ::Hive::Thrift::Stage}},
+        DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
+        STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
 
-class Query
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  QUERYID = 1
-  QUERYTYPE = 2
-  QUERYATTRIBUTES = 3
-  QUERYCOUNTERS = 4
-  STAGEGRAPH = 5
-  STAGELIST = 6
-  DONE = 7
-  STARTED = 8
-
-  FIELDS = {
-    QUERYID => {:type => ::Thrift::Types::STRING, :name => 'queryId'},
-    QUERYTYPE => {:type => ::Thrift::Types::STRING, :name => 'queryType'},
-    QUERYATTRIBUTES => {:type => ::Thrift::Types::MAP, :name => 'queryAttributes', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}},
-    QUERYCOUNTERS => {:type => ::Thrift::Types::MAP, :name => 'queryCounters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::I64}},
-    STAGEGRAPH => {:type => ::Thrift::Types::STRUCT, :name => 'stageGraph', :class => Graph},
-    STAGELIST => {:type => ::Thrift::Types::LIST, :name => 'stageList', :element => {:type => ::Thrift::Types::STRUCT, :class => Stage}},
-    DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
-    STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
-  }
-
-  def struct_fields; FIELDS; end
-
-  def validate
-  end
+    class QueryPlan
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      QUERIES = 1
+      DONE = 2
+      STARTED = 3
 
-  ::Thrift::Struct.generate_accessors self
-end
+      FIELDS = {
+        QUERIES => {:type => ::Thrift::Types::LIST, :name => 'queries', :element => {:type => ::Thrift::Types::STRUCT, :class => ::Hive::Thrift::Query}},
+        DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
+        STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
+      }
 
-class QueryPlan
-  include ::Thrift::Struct, ::Thrift::Struct_Union
-  QUERIES = 1
-  DONE = 2
-  STARTED = 3
+      def struct_fields; FIELDS; end
 
-  FIELDS = {
-    QUERIES => {:type => ::Thrift::Types::LIST, :name => 'queries', :element => {:type => ::Thrift::Types::STRUCT, :class => Query}},
-    DONE => {:type => ::Thrift::Types::BOOL, :name => 'done'},
-    STARTED => {:type => ::Thrift::Types::BOOL, :name => 'started'}
-  }
+      def validate
+      end
 
-  def struct_fields; FIELDS; end
+      ::Thrift::Struct.generate_accessors self
+    end
 
-  def validate
   end
-
-  ::Thrift::Struct.generate_accessors self
 end
-

data/lib/thrift/sasl_client_transport.rb

@@ -0,0 +1,97 @@
+module Thrift
+  class SaslClientTransport < BufferedTransport
+    attr_reader :challenge
+
+    STATUS_BYTES = 1
+    PAYLOAD_LENGTH_BYTES = 4
+    AUTH_MECHANISM = 'PLAIN'
+    NEGOTIATION_STATUS = {
+      START:    0x01,
+      OK:       0x02,
+      BAD:      0x03,
+      ERROR:    0x04,
+      COMPLETE: 0x05
+    }
+
+    def initialize(transport, sasl_params={})
+      super(transport)
+      @challenge = nil
+      @sasl_username = sasl_params.fetch(:username, 'anonymous')
+      @sasl_password = sasl_params.fetch(:password, 'anonymous')
+    end
+
+    def read(sz)
+      len, = @transport.read(PAYLOAD_LENGTH_BYTES).unpack('l>') if @rbuf.nil?
+      sz = len if len && sz > len
+      @index += sz
+      ret = @rbuf.slice(@index - sz, sz) || Bytes.empty_byte_buffer
+      if ret.length == 0
+        @rbuf = @transport.read(len) rescue Bytes.empty_byte_buffer
+        @index = sz
+        ret = @rbuf.slice(0, sz) || Bytes.empty_byte_buffer
+      end
+      ret
+    end
+
+    def read_byte
+      reset_buffer! if @index >= @rbuf.size
+      @index += 1
+      Bytes.get_string_byte(@rbuf, @index - 1)
+    end
+
+    def read_into_buffer(buffer, size)
+      i = 0
+      while i < size
+        reset_buffer! if @index >= @rbuf.size
+        byte = Bytes.get_string_byte(@rbuf, @index)
+        Bytes.set_string_byte(buffer, i, byte)
+        @index += 1
+        i += 1
+      end
+      i
+    end
+
+    def write(buf)
+      initiate_hand_shake if @challenge.nil?
+      header = [buf.length].pack('l>')
+      @wbuf << (header + Bytes.force_binary_encoding(buf))
+    end
+
+    protected
+
+    def initiate_hand_shake
+      header = [NEGOTIATION_STATUS[:START], AUTH_MECHANISM.length].pack('cl>')
+      @transport.write header + AUTH_MECHANISM
+      message = "[#{AUTH_MECHANISM}]\u0000#{@sasl_username}\u0000#{@sasl_password}"
+      header = [NEGOTIATION_STATUS[:OK], message.length].pack('cl>')
+      @transport.write header + message
+      status, len = @transport.read(STATUS_BYTES + PAYLOAD_LENGTH_BYTES).unpack('cl>')
+      case status
+      when NEGOTIATION_STATUS[:BAD], NEGOTIATION_STATUS[:ERROR]
+        raise @transport.to_io.read(len)
+      when NEGOTIATION_STATUS[:COMPLETE]
+        @challenge = @transport.to_io.read len
+      when NEGOTIATION_STATUS[:OK]
+        raise "Failed to complete challenge exchange: only NONE supported currently"
+      end
+    end
+
+    private
+
+    def reset_buffer!
+      len, = @transport.read(PAYLOAD_LENGTH_BYTES).unpack('l>')
+      @rbuf = @transport.read(len)
+      while @rbuf.size < len
+        @rbuf << @transport.read(len - @rbuf.size)
+      end
+      @index = 0
+    end
+  end
+
+  class SaslClientTransportFactory < BaseTransportFactory
+    def get_transport(transport)
+      return SaslClientTransport.new(transport)
+    end
+  end
+
+end

data/lib/thrift/serde_constants.rb

@@ -1,10 +1,11 @@
 #
-# Autogenerated by Thrift
+# Autogenerated by Thrift Compiler (0.9.0)
 #
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 #
 
-require File.join(File.dirname(__FILE__), *%w[serde_types])
+require 'thrift'
+require 'serde_types'
 
 SERIALIZATION_LIB = %q"serialization.lib"
 
@@ -64,6 +65,8 @@ MAP_TYPE_NAME = %q"map"
 
 STRUCT_TYPE_NAME = %q"struct"
 
+UNION_TYPE_NAME = %q"uniontype"
+
 LIST_COLUMNS = %q"columns"
 
 LIST_COLUMN_TYPES = %q"columns.types"
@@ -87,4 +90,3 @@ CollectionTypes = Set.new([
   %q"array",
   %q"map",
 ])
-

data/lib/thrift/serde_types.rb

@@ -1,7 +1,7 @@
 #
-# Autogenerated by Thrift
+# Autogenerated by Thrift Compiler (0.9.0)
 #
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 #
 
-
+require 'thrift'