|
- 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
|