Contents

module Daru
  module IO
    class SqlDataSource
      # @private
      class Adapter
        def initialize(conn, query)
          @conn = conn
          @query = query
        end

        def result_hash
          column_names
            .map(&:to_sym)
            .zip(rows.transpose)
            .to_h
        end
      end

      # Private adapter class for DBI::DatabaseHandle
      # @private
      class DbiAdapter < Adapter
        private

        def column_names
          result.column_names
        end

        def rows
          result.to_a.map(&:to_a)
        end

        def result
          @result ||= @conn.execute(@query)
        end
      end

      # Private adapter class for connections of ActiveRecord
      # @private
      class ActiveRecordConnectionAdapter < Adapter
        private

        def column_names
          result.columns
        end

        def rows
          result.cast_values
        end

        def result
          @result ||= @conn.exec_query(@query)
        end
      end

      private_constant :DbiAdapter
      private_constant :ActiveRecordConnectionAdapter

      def self.make_dataframe(db, query)
        new(db, query).make_dataframe
      end

      def initialize(db, query)
        @adapter = init_adapter(db, query)
      end

      def make_dataframe
        Daru::DataFrame.new(@adapter.result_hash).tap(&:update)
      end

      private

      def init_adapter(db, query)
        query = String.try_convert(query) or
          raise ArgumentError, "Query must be a string, #{query.class} received"

        case db
        when DBI::DatabaseHandle
          DbiAdapter.new(db, query)
        when ActiveRecord::ConnectionAdapters::AbstractAdapter
          ActiveRecordConnectionAdapter.new(db, query)
        else
          raise ArgumentError, "Unknown database adapter type #{db.class}"
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
daru-0.1.5	lib/daru/io/sql_data_source.rb
daru-0.1.4.1	lib/daru/io/sql_data_source.rb
daru-0.1.4	lib/daru/io/sql_data_source.rb