require 'spec_helper'
require 'benchmark'

describe Mongoid::Report, long: true do
  let(:klass) { Model }

  it 'aggregates fields by app in threads' do
    report_klass = Class.new do
      include Mongoid::Report

      attach_to Model, as: 'field1-aggregation' do
        column :field1
      end

      attach_to Model, as: 'field2-aggregation' do
        column :field2
      end
    end

    times = 30000

    times.times { klass.create!(field1: 1, field2: 1) }

    report = report_klass.new
    scoped = report.aggregate

    Mongoid::Report::Config.use_threads_on_aggregate = true
    time1 = Benchmark.measure do
      rows = scoped.all
    end

    Mongoid::Report::Config.use_threads_on_aggregate = false
    time2 = Benchmark.measure do
      rows = scoped.all
    end

    puts time2
    puts time1

    time2.real.should > time1.real
  end

  it 'should work faster using batches in threads on aggregate' do
    report_klass1 = Class.new do
      include Mongoid::Report

      report 'example' do
        attach_to Model do
          group_by :day
          column :field1
        end
      end
    end

    report_klass2 = Class.new do
      include Mongoid::Report

      report 'example' do
        attach_to Model do
          group_by :day
          batches pool_size: 4
          column :field1
        end
      end
    end

    times = 10

    times.times.map do |i|
      Thread.new do
        10000.times { klass.create!(day: i.days.ago, field1: 1) }
      end
    end.map(&:join)

    report1 = report_klass1.new
    scoped = report1.aggregate

    time1 = Benchmark.measure do
      rows = scoped.all
      expect(rows['example']['models'].rows[0]['field1']).to eq(10000)
    end

    report2 = report_klass2.new
    scoped = report2.aggregate_for('example', 'models')

    time2 = Benchmark.measure do
      scoped = scoped
        .in_batches(day: (5.days.ago.to_date..0.days.from_now.to_date))
        .all
      expect(scoped.rows[0]['field1']).to eq(10000)
    end

    puts time2
    puts time1

    time1.real.should > time2.real
  end

  it 'should merge properly results on splitted requests' do
    report_klass = Class.new do
      include Mongoid::Report

      report 'example' do
        attach_to Model do
          group_by :field1
          batches pool_size: 2
          column :field1, :field2
        end
      end
    end

    klass.create!(day: 0.days.ago, field1: 1, field2: 1)
    klass.create!(day: 1.days.ago, field1: 1, field2: 1)
    klass.create!(day: 1.days.ago, field1: 2, field2: 2)
    klass.create!(day: 2.days.ago, field1: 3, field2: 3)
    klass.create!(day: 3.days.ago, field1: 1, field2: 1)
    klass.create!(day: 4.days.ago, field1: 1, field2: 1)

    report = report_klass.new

    scoped = report.aggregate_for('example', 'models')
    scoped = scoped
      .in_batches(day: (5.days.ago.to_date..0.days.from_now.to_date))
      .all

    values = scoped.rows.map {|row| row['field2']}
    expect(values).to include(4)
    expect(values).to include(2)
    expect(values).to include(3)
  end

  it 'should merge properly results with multiple groups' do
    report_klass = Class.new do
      include Mongoid::Report

      report 'example' do
        attach_to Model do
          group_by :field1, :field2
          batches pool_size: 2
          column :field1, :field2, :field3
        end
      end
    end

    klass.create!(day: 0.days.ago, field1: 1, field2: 4, field3: 1)
    klass.create!(day: 1.days.ago, field1: 1, field2: 5, field3: 1)
    klass.create!(day: 1.days.ago, field1: 2, field2: 6, field3: 2)
    klass.create!(day: 2.days.ago, field1: 3, field2: 7, field3: 3)
    klass.create!(day: 3.days.ago, field1: 1, field2: 8, field3: 1)
    klass.create!(day: 4.days.ago, field1: 1, field2: 9, field3: 1)

    report = report_klass.new

    scoped = report.aggregate_for('example', 'models')
    scoped = scoped
      .in_batches(day: (5.days.ago.to_date..0.days.from_now.to_date))
      .all

    expect(scoped.summary['field3']).to eq(9)
    expect(scoped.rows.size).to eq(6)
    expect(scoped.rows).to include({"field3"=>2, "field1"=>2, "field2"=>6})
    expect(scoped.rows).to include({"field3"=>3, "field1"=>3, "field2"=>7})
    expect(scoped.rows).to include({"field3"=>1, "field1"=>1, "field2"=>5})
    expect(scoped.rows).to include({"field3"=>1, "field1"=>1, "field2"=>4})
    expect(scoped.rows).to include({"field3"=>1, "field1"=>1, "field2"=>9})
    expect(scoped.rows).to include({"field3"=>1, "field1"=>1, "field2"=>8})
  end
end