Contents

# frozen_string_literal: true

require_relative 'type_faker'

module RSpec
  module Hive
    class QueryBuilder
      class RowTransformer
        def initialize(schema, missing_column_strategy)
          @schema = schema
          @strategy = missing_column_strategy
        end

        def transform(row)
          if row.respond_to?(:each_pair)
            mock_hive_row(row)
          elsif row.respond_to?(:each)
            array_row(row)
          else
            raise ArgumentError, 'Array or Hash required!'
          end
        end

        private

        attr_reader :schema, :strategy

        HIVE_NIL = '\N'

        def array_row(row)
          size = schema.columns.size
          missing = size - row.size
          if missing.positive?
            row_with_missing_columns(row)
          else
            row
          end
        end

        def row_with_missing_columns(row)
          schema.columns.map.with_index do |column, index|
            if index > row.size
              strategy.missing(column)
            else
              row[index]
            end
          end
        end

        def mock_hive_row(partial_row)
          symbolized_row = Hash[partial_row.map { |k, v| [k.to_sym, v] }]

          schema.columns.map do |column|
            value = symbolized_row.fetch(column.name.to_sym) { strategy.missing(column) }
            nil_to_null(value)
          end
        end

        def nil_to_null(value)
          value.nil? ? HIVE_NIL : value
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
rspec-hive-0.6.3	lib/rspec/hive/query_builder/row_transformer.rb
rspec-hive-0.6.2	lib/rspec/hive/query_builder/row_transformer.rb
rspec-hive-0.6.1	lib/rspec/hive/query_builder/row_transformer.rb
rspec-hive-0.6.0	lib/rspec/hive/query_builder/row_transformer.rb