Skip to content

Instantly share code, notes, and snippets.

@aladac

aladac/mongoid_sample.rb

Created April 10, 2014 21:05
Show Gist options
  • Save aladac/10423381 to your computer and use it in GitHub Desktop.
Save aladac/10423381 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
mongoid_sample.rb
class Bounty
include Mongoid::Document
embeds_many :kills
validates_uniqueness_of :ts, :scope => [ :char_id ]
validates_presence_of :ts, :char_id, :bounty
field :ts, type: Time
field :char_id, type: Integer
field :corp_id, type: Integer
field :tax, type: Integer
field :bounty, type: Integer
index({ ts: 1, char_id: 1 }, { unique: true, drop_dups: true })
index({ ts: -1 })
index({ ts: 1 })
index({ char_id: 1})
index({ tax: -1 })
index({ tax: 1 })
index({ bounty: 1 })
index({ bounty: -1 })
# index({ "kill.rat_id" => 1 }, { unique: true, drop_dups: true } )
def self.total_bounty(id)
Bounty.where(char_id: id).sum(:bounty) / 100
end
def self.total_tax(id)
Bounty.where(corp_id: id).sum(:tax) / 100
end
def self.tax_contrib(corp_id, limit=10)
limit -= 1
collection.aggregate(
{ "$match" => { "corp_id" => corp_id } },
{ "$group" => { "_id" =>
{
"corp_id" => "$corp_id",
"char_id" => "$char_id"
},
"sum" => { "$sum" => "$tax"} } },
{"$sort" => { "sum" => -1 } }
)[0..limit]
end
def self.total_kills(id)
Bounty.collection.aggregate({ "$match" => { "char_id" => id }}, { "$unwind" => "$kills" } ).count
end
def self.global_kills
collection.aggregate({ "$unwind" => "$kills" } ).count
end
def self.global_faction_kills(type)
rat_ids = Array.new
Rat.any_of(rat_type: /#{type}/ ).to_a.each do |rat|
rat_ids.push rat.rat_id
end
count = 0
Bounty.collection.aggregate( { "$unwind" => "$kills" } ).to_a.each do |row|
# "kills"=>{"_id"=>"51e06424ade21b90a700005f", "rat_id"=>23319, "rat_amount"=>2}
if rat_ids.include? row["kills"]["rat_id"]
count += row["kills"]["rat_amount"]
end
end
count
end
def self.tax_daily(corp_id, limit=10)
limit -= 1
collection.aggregate(
{ "$match" => { "corp_id" => corp_id } },
{ "$group" => { "_id" =>
{
"corp_id" => "$corp_id",
"year" => { "$year" => "$ts" },
"month" => { "$month" => "$ts" },
"day" => { "$dayOfMonth" => "$ts" }
},
"sum" => { "$sum" => "$tax"} } },
{"$sort" => { "_id" => -1 } }
)[0..limit]
end
def self.daily(char_id, limit=10)
limit -= 1
collection.aggregate(
{ "$match" => { "char_id" => char_id } },
{ "$group" => { "_id" =>
{
"char_id" => "$char_id",
"year" => { "$year" => "$ts" },
"month" => { "$month" => "$ts" },
"day" => { "$dayOfMonth" => "$ts" }
},
"sum" => { "$sum" => "$bounty"} } },
{"$sort" => { "_id" => -1 } }
)[0..limit]
end
def self.top_daily(char_id)
collection.aggregate(
{ "$match" => { "char_id" => char_id } },
{ "$group" => { "_id" =>
{
"year" => { "$year" => "$ts" },
"month" => { "$month" => "$ts" },
"day" => { "$dayOfMonth" => "$ts" }
},
"sum" => { "$sum" => "$bounty"} } },
{"$sort" => { "sum" => -1 } }
).first["sum"]/100
end
def self.top_kills_this_month(limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{"$unwind" => "$kills"},
{ "$group" =>
{
"_id" => "$char_id",
"kills" => { "$sum" => 1 }
}
},
{ "$sort" =>
{ "kills" => -1 }
}
)[0..limit]
end
def self.kills_this_month(id)
collection.aggregate(
{ "$match" => { "char_id" => id, "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{"$unwind" => "$kills"}
).count
end
def self.top_kills_days(days=10, limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc - days.days) } } },
{"$unwind" => "$kills"},
{ "$group" =>
{
"_id" => "$char_id",
"kills" => { "$sum" => 1 }
}
},
{ "$sort" =>
{ "kills" => -1 }
}
)[0..limit]
end
def self.top_kills(limit=5)
limit -= 1
collection.aggregate(
{"$unwind" => "$kills"},
{ "$group" =>
{
"_id" => "$char_id",
"kills" => { "$sum" => 1 }
}
},
{ "$sort" =>
{ "kills" => -1 }
}
)[0..limit]
end
def self.top_bounty_days(days=10, limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc - days.days) } } },
{ "$group" =>
{
"_id" => "$char_id",
"sum" =>
{ "$sum" => "$bounty"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.top_tax_days(days=10, limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc - days.days) } } },
{ "$group" =>
{
"_id" => "$corp_id",
"sum" =>
{ "$sum" => "$tax"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.top_corp_bounty_days(days=10, limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc - days.days) } } },
{ "$group" =>
{
"_id" => "$corp_id",
"sum" =>
{ "$sum" => "$bounty"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.top_corp_bounty_this_month(limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{ "$group" =>
{
"_id" => "$corp_id",
"sum" =>
{ "$sum" => "$bounty"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.top_tax_this_month(limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{ "$group" =>
{
"_id" => "$corp_id",
"sum" =>
{ "$sum" => "$tax"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.top_bounty_this_month(limit=5)
limit -= 1
collection.aggregate(
{ "$match" => { "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{ "$group" =>
{
"_id" => "$char_id",
"sum" =>
{ "$sum" => "$bounty"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
)[0..limit]
end
def self.bounty_this_month(id)
bounty = collection.aggregate(
{ "$match" => { "char_id" => id, "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{ "$group" =>
{
"_id" => "$char_id",
"sum" =>
{ "$sum" => "$bounty"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
).first
if bounty
bounty["sum"]/100
else
0
end
end
def self.tax_this_month(id)
tax = collection.aggregate(
{ "$match" => { "corp_id" => id, "ts" => { "$gt" => (Time.now.utc.at_beginning_of_month) } } },
{ "$group" =>
{
"_id" => "$corp_id",
"sum" =>
{ "$sum" => "$tax"}
}
},
{"$sort" =>
{ "sum" => -1 }
}
).first
if tax
tax["sum"]/100
else
0
end
end
def self.top_corp_bounty(limit=5)
limit -= 1
collection.aggregate(
{ "$group" =>
{ "_id" =>
"$corp_id", "sum" => { "$sum" => "$bounty"}
}
},
{"$sort" => { "sum" => -1 } }
)[0..limit]
end
def self.top_bounty(limit=5)
limit -= 1
collection.aggregate(
{ "$group" =>
{ "_id" =>
"$char_id", "sum" => { "$sum" => "$bounty"}
}
},
{"$sort" => { "sum" => -1 } }
)[0..limit]
end
def self.top_tax(limit=5)
limit -= 1
collection.aggregate(
{ "$group" =>
{ "_id" =>
"$corp_id", "sum" => { "$sum" => "$tax"}
}
},
{"$sort" => { "sum" => -1 } }
)[0..limit]
end
def self.bounty_rank(char_id)
rank=0
Bounty.top_bounty(0).each do |entry|
rank+=1
if entry["_id"] == char_id
result = rank
break
end
end
return rank
end
def self.kill_rank(char_id)
rank=0
Bounty.top_kills(0).each do |entry|
rank+=1
if entry["_id"] == char_id
break
end
end
return rank
end
def self.highest_tick_days(days=10, limit=5)
where(:ts.gt => (Time.now.utc - days.days)).sort(:bounty.desc).limit(limit)
end
def self.highest_tick(limit=5)
all.sort(:bounty.desc).limit(limit)
end
def self.top_tick(id)
top_tick = where(char_id: id).max(:bounty)
top_tick ? top_tick=top_tick/100 : 0
end
def self.avg_tick(id)
avg_tick = where(char_id: id).avg(:bounty)
avg_tick ? avg_tick=avg_tick/100 : 0
end
def self.tick_rank(char_id)
rank=0
Bounty.highest_tick(0).to_a.each do |tick|
rank+=1
if tick.char_id == char_id
break
end
end
return rank
end
def self.unique_rats
Bounty.collection.aggregate({ "$unwind" => "$kills" }, { "$group" => { "_id" => "$kills.rat_id" } } )
end
def self.rats(id, limit=5)
limit -= 1
Bounty.collection.aggregate(
{
"$match" => { "char_id" => id }
},
{
"$unwind" => "$kills"
},
{
"$group" => {
"_id" => {
"rat_id" => "$kills.rat_id",
"char_id" => "$char_id" },
"sum" => { "$sum" => 1 }
}
},
{
"$sort" => { "sum" => -1 }
}
)[0..limit]
end
def self.rat_kills
collection.aggregate( { "$unwind" => "$kills" })
end
def self.kills_by_rat_type(type)
rat_ids = Array.new
Rat.any_of(rat_type: /#{type}/ ).to_a.each do |rat|
rat_ids.push rat.rat_id
end
count = 0
map = %Q|
function() {
for (index in this.kills) {
emit(this.kills[index].rat_id, this.kills[index].rat_amount );
}
}
|
reduce = %Q|
function(key, values) {
var count = 0;
for (index in values) {
count += values[index];
}
return count;
}
|
result = 0
a = Bounty.map_reduce(map, reduce).out(inline: true)
a.to_a.each do |rat|
if rat_ids.include?(rat["_id"].to_i)
result += rat["value"]
end
end
result
end
def self.kills_by_rat_name(name)
rat_ids = Array.new
Rat.any_of(rat_name: /#{name}/ ).to_a.each do |rat|
rat_ids.push rat.rat_id
end
count = 0
Bounty.collection.aggregate(
{ "$unwind" => "$kills" },
{
"$sort" => { "ts" => -1 }
},
{ "$match" => {
"kills.rat_id" => {
"$in" => rat_ids
}
}
}
)
end
def self.rat_kills
a = collection.aggregate({ "$unwind" => "$kills" }, { "$project" => { "amount" => "$kills.rat_amount" } } )
result = 0
a.each do |row|
result = result + row["amount"]
end
result
end
def self.kills_by_rat_id
map = %Q|
function() {
for (index in this.kills) {
emit(this.kills[index].rat_id, this.kills[index].rat_amount );
}
}
|
reduce = %Q|
function(key, values) {
var count = 0;
for (index in values) {
count += values[index];
}
return count;
}
|
map_reduce(map, reduce).out(inline: true).to_a
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment