mirror of
https://github.com/discourse/discourse.git
synced 2024-12-21 14:24:27 +08:00
89 lines
2.3 KiB
Ruby
89 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
class ApplicationRequest < ActiveRecord::Base
|
|
enum :req_type,
|
|
{
|
|
http_total: 0,
|
|
http_2xx: 1,
|
|
http_background: 2,
|
|
http_3xx: 3,
|
|
http_4xx: 4,
|
|
http_5xx: 5,
|
|
page_view_crawler: 6,
|
|
page_view_logged_in: 7,
|
|
page_view_anon: 8,
|
|
page_view_logged_in_mobile: 9,
|
|
page_view_anon_mobile: 10,
|
|
api: 11,
|
|
user_api: 12,
|
|
page_view_anon_browser: 13,
|
|
page_view_anon_browser_mobile: 14,
|
|
page_view_logged_in_browser: 15,
|
|
page_view_logged_in_browser_mobile: 16,
|
|
}
|
|
|
|
include CachedCounting
|
|
|
|
def self.disable
|
|
@disabled = true
|
|
end
|
|
|
|
def self.enable
|
|
@disabled = false
|
|
end
|
|
|
|
def self.increment!(req_type)
|
|
return if @disabled
|
|
perform_increment!(req_type)
|
|
end
|
|
|
|
def self.write_cache!(req_type, count, date)
|
|
req_type_id = req_types[req_type]
|
|
|
|
DB.exec(<<~SQL, date: date, req_type_id: req_type_id, count: count)
|
|
INSERT INTO application_requests (date, req_type, count)
|
|
VALUES (:date, :req_type_id, :count)
|
|
ON CONFLICT (date, req_type)
|
|
DO UPDATE SET count = application_requests.count + excluded.count
|
|
SQL
|
|
end
|
|
|
|
def self.stats
|
|
s = HashWithIndifferentAccess.new({})
|
|
|
|
self.req_types.each do |key, i|
|
|
query = self.where(req_type: i)
|
|
s["#{key}_total"] = query.sum(:count)
|
|
s["#{key}_30_days"] = query.where("date > ?", 30.days.ago).sum(:count)
|
|
s["#{key}_28_days"] = query.where("date > ?", 28.days.ago).sum(:count)
|
|
s["#{key}_7_days"] = query.where("date > ?", 7.days.ago).sum(:count)
|
|
end
|
|
|
|
s
|
|
end
|
|
|
|
def self.request_type_count_for_period(type, since)
|
|
id = self.req_types[type]
|
|
if !id
|
|
raise ArgumentError.new(
|
|
"unknown request type #{type.inspect} in ApplicationRequest.req_types",
|
|
)
|
|
end
|
|
|
|
self.where(req_type: id).where("date >= ?", since).sum(:count)
|
|
end
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: application_requests
|
|
#
|
|
# id :integer not null, primary key
|
|
# date :date not null
|
|
# req_type :integer not null
|
|
# count :integer default(0), not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_application_requests_on_date_and_req_type (date,req_type) UNIQUE
|
|
#
|