michaelmior / NoSE

module NoSE

  module Loader

    # Load some random data (mostly useful for testing)

    class RandomLoader < LoaderBase

      def initialize(workload = nil, backend = nil)

        @logger = Logging.logger['nose::loader::randomloader']

        @workload = workload

        @backend = backend

      end

      # Load a generated set of indexes with data from MySQL

      # @return [void]

      private

      # Load a single index into the backend

      # @return [void]

      def load_index(index, _config, show_progress, limit, skip_existing)

        # Skip this index if it's not empty

        if skip_existing && [email protected]_empty?(index)

          @logger.info "Skipping index #{index.inspect}" if show_progress

          return

        end

        @logger.info index.inspect if show_progress

        chunk = Array.new(limit) do

          Hash[index.all_fields.map do |field|

            [field.id, field.random_value]

          end]

        end

        @backend.index_insert_chunk index, chunk

      end

    end

  end

end