mirror of
https://github.com/discourse/discourse.git
synced 2024-12-14 01:38:24 +08:00
30990006a9
This reduces chances of errors where consumers of strings mutate inputs and reduces memory usage of the app. Test suite passes now, but there may be some stuff left, so we will run a few sites on a branch prior to merging
73 lines
2.2 KiB
Ruby
73 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'ipaddr'
|
|
|
|
# awkward TopicView is taken
|
|
class TopicViewItem < ActiveRecord::Base
|
|
self.table_name = 'topic_views'
|
|
belongs_to :user
|
|
validates_presence_of :topic_id, :ip_address, :viewed_at
|
|
|
|
def self.add(topic_id, ip, user_id = nil, at = nil, skip_redis = false)
|
|
# Only store a view once per day per thing per (user || ip)
|
|
at ||= Date.today
|
|
redis_key = +"view:#{topic_id}:#{at}"
|
|
if user_id
|
|
redis_key << ":user-#{user_id}"
|
|
else
|
|
redis_key << ":ip-#{ip}"
|
|
end
|
|
|
|
if skip_redis || $redis.setnx(redis_key, "1")
|
|
skip_redis || $redis.expire(redis_key, SiteSetting.topic_view_duration_hours.hours)
|
|
|
|
TopicViewItem.transaction do
|
|
# this is called real frequently, working hard to avoid exceptions
|
|
sql = "INSERT INTO topic_views (topic_id, ip_address, viewed_at, user_id)
|
|
SELECT :topic_id, :ip_address, :viewed_at, :user_id
|
|
WHERE NOT EXISTS (
|
|
SELECT 1 FROM topic_views
|
|
/*where*/
|
|
)"
|
|
|
|
builder = DB.build(sql)
|
|
|
|
if !user_id
|
|
builder.where("ip_address = :ip_address AND topic_id = :topic_id AND user_id IS NULL")
|
|
else
|
|
builder.where("user_id = :user_id AND topic_id = :topic_id")
|
|
ip = nil # do not store IP of logged in users
|
|
end
|
|
|
|
result = builder.exec(topic_id: topic_id, ip_address: ip, viewed_at: at, user_id: user_id)
|
|
|
|
Topic.where(id: topic_id).update_all 'views = views + 1'
|
|
|
|
if result > 0
|
|
UserStat.where(user_id: user_id).update_all 'topics_entered = topics_entered + 1' if user_id
|
|
end
|
|
|
|
# Update the views count in the parent, if it exists.
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: topic_views
|
|
#
|
|
# topic_id :integer not null
|
|
# viewed_at :date not null
|
|
# user_id :integer
|
|
# ip_address :inet
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_topic_views_on_topic_id_and_viewed_at (topic_id,viewed_at)
|
|
# index_topic_views_on_user_id_and_viewed_at (user_id,viewed_at)
|
|
# index_topic_views_on_viewed_at_and_topic_id (viewed_at,topic_id)
|
|
# uniq_ip_or_user_id_topic_views (user_id,ip_address,topic_id) UNIQUE
|
|
#
|