forgeplus/app/services/cache/v2/user_statistic_service.rb

202 lines
7.6 KiB
Ruby

class Cache::V2::UserStatisticService < ApplicationService
attr_reader :user_id, :follow_count, :fork_count, :issue_count, :project_count, :project_language_count_key, :project_language_count, :project_praise_count, :project_watcher_count, :pullrequest_count
def initialize(user_id, params={})
@user_id = user_id
@follow_count = params[:follow_count]
@fork_count = params[:fork_count]
@issue_count = params[:issue_count]
@project_count = params[:project_count]
@project_language_count_key = params[:project_language_count_key]
@project_language_count = params[:project_language_count]
@project_praise_count = params[:project_praise_count]
@project_watcher_count = params[:project_watcher_count]
@pullrequest_count = params[:pullrequest_count]
Cache::V2::OwnerCommonService.new(user_id).read
end
def read
user_statistic
end
def call
set_user_statistic
end
def reset
reset_user_statistic
end
private
def user_statistic_key
"v2-user-statistic:#{@user_id}"
end
def follow_count_key
"follow-count"
end
def fork_count_key
"fork-count"
end
def issue_count_key
"issue-count"
end
def project_count_key
"project-count"
end
def project_language_key
"project-language"
end
def project_praise_count_key
"project-praise-count"
end
def project_watcher_count_key
"project-watcher-count"
end
def pullrequest_count_key
"pullrequest-count"
end
def user_statistic
result = $redis_cache.hgetall(user_statistic_key)
result.blank? ? reset_user_statistic : result
end
def set_user_statistic
if $redis_cache.hgetall(user_statistic_key).blank?
reset_user_statistic
return
end
if @follow_count.present?
if $redis_cache.hget(user_statistic_key, follow_count_key).nil?
reset_user_follow_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {follow_count: @follow_count})
else
$redis_cache.hincrby(user_statistic_key, follow_count_key, @follow_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {follow_count: @follow_count})
end
end
if @fork_count.present?
if $redis_cache.hget(user_statistic_key, fork_count_key).nil?
reset_user_fork_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {fork_count: @fork_count})
else
$redis_cache.hincrby(user_statistic_key, fork_count_key, @fork_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {fork_count: @fork_count})
end
end
if @issue_count.present?
if $redis_cache.hget(user_statistic_key, issue_count_key).nil?
reset_user_issue_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {issue_count: @issue_count})
else
$redis_cache.hincrby(user_statistic_key, issue_count_key, @issue_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {issue_count: @issue_count})
end
end
if @project_count.present?
if $redis_cache.hget(user_statistic_key, project_count_key).nil?
reset_user_project_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_count: @project_count})
else
$redis_cache.hincrby(user_statistic_key, project_count_key, @project_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_count: @project_count})
end
end
if @project_language_count_key.present? && project_language_count.present?
if $redis_cache.hget(user_statistic_key, project_language_key).nil?
reset_user_project_language
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_language_count_key: @project_language_count_key, project_language_count: @project_language_count})
else
result = JSON.parse($redis_cache.hget(user_statistic_key, project_language_key))
result[@project_language_count_key] ||= 0
result[@project_language_count_key] += project_language_count.to_i
result.delete(@project_language_count_key) if result[@project_language_count_key] == 0
$redis_cache.hset(user_statistic_key, project_language_key, result.to_json)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_language_count_key: @project_language_count_key, project_language_count: @project_language_count})
end
end
if @project_praise_count.present?
if $redis_cache.hget(user_statistic_key, project_praise_count_key).nil?
reset_user_project_praise_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_praise_count: @project_praise_count})
else
$redis_cache.hincrby(user_statistic_key, project_praise_count_key, @project_praise_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_praise_count: @project_praise_count})
end
end
if @project_watcher_count.present?
if $redis_cache.hget(user_statistic_key, project_watcher_count_key).nil?
reset_user_project_watcher_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_watcher_count: @project_watcher_count})
else
$redis_cache.hincrby(user_statistic_key, project_watcher_count_key, @project_watcher_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {project_watcher_count: @project_watcher_count})
end
end
if @pullrequest_count.present?
if $redis_cache.hget(user_statistic_key, pullrequest_count_key).nil?
reset_user_pullrequest_count
Cache::V2::UserDateRankService.call(@user_id, Date.today, {pullrequest_count: @pullrequest_count})
else
$redis_cache.hincrby(user_statistic_key, pullrequest_count_key, @pullrequest_count)
Cache::V2::UserDateRankService.call(@user_id, Date.today, {pullrequest_count: @pullrequest_count})
end
end
$redis_cache.hgetall(user_statistic_key)
end
def reset_user_follow_count
$redis_cache.hset(user_statistic_key, follow_count_key, Watcher.where(watchable_type: 'User', watchable_id: @user_id).count)
end
def reset_user_fork_count
$redis_cache.hset(user_statistic_key, fork_count_key, ForkUser.joins(:project).where(projects: {user_id: @user_id}).count)
end
def reset_user_issue_count
$redis_cache.hset(user_statistic_key, issue_count_key, Issue.where(author_id: @user_id).count)
end
def reset_user_project_count
$redis_cache.hset(user_statistic_key, project_count_key, Project.where(user_id: @user_id).count)
end
def reset_user_project_language
$redis_cache.hset(user_statistic_key, project_language_key, Project.where(user_id: @user_id).joins(:project_language).group("project_languages.name").count.to_json)
end
def reset_user_project_praise_count
$redis_cache.hset(user_statistic_key, project_praise_count_key, PraiseTread.where(praise_tread_object_type: 'Project', praise_tread_object_id: Project.where(user_id: @user_id)).count)
end
def reset_user_project_watcher_count
$redis_cache.hset(user_statistic_key, project_watcher_count_key, Watcher.where(watchable_type: 'Project', watchable_id: Project.where(user_id: @user_id)).count)
end
def reset_user_pullrequest_count
$redis_cache.hset(user_statistic_key, pullrequest_count_key, PullRequest.where(user_id: @user_id).count)
end
def reset_user_statistic
$redis_cache.del(user_statistic_key)
reset_user_follow_count
reset_user_fork_count
reset_user_issue_count
reset_user_project_count
reset_user_project_language
reset_user_project_praise_count
reset_user_project_watcher_count
reset_user_pullrequest_count
$redis_cache.hgetall(user_statistic_key)
end
end