Class: Jvertica::ResultSet

Inherits:

Object

• Object
• Jvertica::ResultSet

Includes:

Enumerable

Defined in:

lib/jvertica.rb

Instance Method Summary

• #close ⇒ Object
• #closed? ⇒ Boolean
• #each ⇒ Object
• #initialize(rset, &close_callback) ⇒ ResultSet constructor

  A new instance of ResultSet.

Constructor Details

#initialize(rset, &close_callback) ⇒ ResultSet

# File 'lib/jvertica.rb', line 214

def initialize rset, &close_callback
  unless rset.respond_to? :get_meta_data
    rset.close if rset
    @closed = true
    return
  end

  @close_callback = close_callback
  @rset           = rset
  @rsmd           = @rset.get_meta_data
  @num_col        = @rsmd.get_column_count
  @getters        = []
  @col_labels     = []
  @col_labels_d   = []
  (1..@num_col).each do |i|
    type = @rsmd.get_column_type i

    @getters <<
      case type
      when java.sql.Types::NUMERIC, java.sql.Types::DECIMAL
        precision = @rsmd.get_precision i
        scale = @rsmd.get_scale i

        if precision > 0 and scale >= 0
          if scale > 0 then :getBigDecimal
          else

            if precision <= 9     then :getInt
            elsif precision <= 18 then :getLong
            else                       :getBigNum
            end

          end
        else :getBigDecimal
        end

      else Jvertica::Constant::GETTER_MAP.fetch type, :get_string
      end

    label = @rsmd.get_column_label i
    @col_labels << label
    @col_labels_d << label.downcase
  end

  @rownum = -1
  @nrow   = @rset.next
  @closed = false
end

Instance Method Details

#close ⇒ Object

# File 'lib/jvertica.rb', line 202

def close
  return if closed?

  @rset.close
  @close_callback.call if @close_callback
  @closed = true
end

#closed? ⇒ Boolean

# File 'lib/jvertica.rb', line 210

def closed?
  @closed
end

#each ⇒ Object

# File 'lib/jvertica.rb', line 172

def each
  return enum_for(:each) unless block_given?
  return if closed?

  while @nrow
    idx = 0
    row = Jvertica::Row.new(
      @col_labels,
      @col_labels_d,
      @getters.map{|gt|
        case gt
        when :getBigNum
          v = @rset.getBigDecimal idx+=1
          @rset.was_null ? nil : v.toPlainString.to_i
        when :getBigDecimal
          v = @rset.getBigDecimal idx+=1
          @rset.was_null ? nil : BigDecimal.new(v.toPlainString)
        else
          v = @rset.send gt, idx+=1
          @rset.was_null ? nil : v
        end
      },
      @rownum += 1
    )
    close unless @nrow = @rset.next
    yield row
  end
  close
end