require 'mongoid'
require 'benchmark'
require 'yajl/json_gem'

ENV['MONGOID_ENV'] = 'development'

Mongoid.load!("config/mongoid.yml")

class Zine
  include Mongoid::Document
  field :name, type: String
end

class Band
  include Mongoid::Document
  field :name, type: String
  embeds_many :albums
end

class Album
  include Mongoid::Document
  field :name, type: String
  embedded_in :band
  embeds_many :tracks
end

class Track
  include Mongoid::Document
  field :name, type: String
  embedded_in :album
end


NUM_DOCS = 1000

Benchmark.bm do |x|

  Zine.delete_all
  NUM_DOCS.times do |i|
    Zine.create!(name:"Zine #{i}")
  end

  x.report("One") { Zine.all.to_json }

  Band.delete_all
  NUM_DOCS.times do |i|
    Band.create!(name:"Band #{i}")
  end
  x.report("One") { Band.all.to_json }


  Band.delete_all
  NUM_DOCS.times do |i|
    Band.create!(name:"Band #{i}", albums: [
      { name:"Album #{i}"},
      { name:"Album #{i}"}
    ])
  end
  x.report("Two") { Band.all.to_json }

  Band.delete_all
    NUM_DOCS.times do |i|
      Band.create!(name:"Band #{i}", albums: [
        { name:"Album #{i}", tracks: [
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}"
        ]},
        { name:"Album #{i}", tracks: [
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}",
          name: "Track #{i}"
        ]}

      ])
    end
  x.report("Three") { Band.all.to_json }

end

       user     system      total        real
One  0.210000   0.000000   0.210000 (  0.209466)
One  0.250000   0.000000   0.250000 (  0.254118)
Two  0.910000   0.010000   0.920000 (  0.917821)
Three  1.680000   0.000000   1.680000 (  1.689789)