discourse/lib/score_calculator.rb

class ScoreCalculator

  def self.default_score_weights
    {
      reply_count: 5,
      like_score: 15,
      incoming_link_count: 5,
      bookmark_count: 2,
      avg_time: 0.05,
      reads: 0.2
    }
  end

  def initialize(weightings=nil)
    @weightings = weightings || ScoreCalculator.default_score_weights
  end

  # Calculate the score for all posts based on the weightings
  def calculate(min_topic_age=nil)

    update_posts_score(min_topic_age)

    update_posts_rank(min_topic_age)

    update_topics_rank(min_topic_age)

    update_topics_percent_rank(min_topic_age)

  end


  private

  def update_posts_score(min_topic_age)
    limit = 20000

    components = []
    @weightings.each_key { |k| components << "COALESCE(#{k}, 0) * :#{k}" }
    components = components.join(" + ")

    builder = SqlBuilder.new <<SQL
       UPDATE posts p
        SET score = x.score
       FROM (
        SELECT id, #{components} as score FROM posts
        /*where*/
        limit #{limit}
       ) AS x
       WHERE x.id = p.id
SQL

    builder.where("score IS NULL OR score <> #{components}", @weightings)

    filter_topics(builder, min_topic_age)

    while builder.exec.cmd_tuples == limit
    end
  end

  def update_posts_rank(min_topic_age)
    limit = 20000

    builder = SqlBuilder.new <<SQL
UPDATE posts
SET percent_rank = X.percent_rank
FROM (
  SELECT posts.id, Y.percent_rank
  FROM posts
  JOIN (
    SELECT id, percent_rank()
                 OVER (PARTITION BY topic_id ORDER BY SCORE DESC) as percent_rank
    FROM posts
   ) Y ON Y.id = posts.id
  /*where*/
  LIMIT #{limit}
) AS X
WHERE posts.id = X.id
SQL

    builder.where("posts.percent_rank IS NULL OR Y.percent_rank <> posts.percent_rank")

    filter_topics(builder, min_topic_age)

    while builder.exec.cmd_tuples == limit
    end

  end

  def update_topics_rank(min_topic_age)
    builder = SqlBuilder.new("UPDATE topics AS t
              SET has_summary = (t.like_count >= :likes_required AND
                                 t.posts_count >= :posts_required AND
                                 x.max_score >= :score_required),
                  score = x.avg_score
              FROM (SELECT p.topic_id,
                           MAX(p.score) AS max_score,
                           AVG(p.score) AS avg_score
                    FROM posts AS p
                    GROUP BY p.topic_id) AS x
                    /*where*/")

    builder.where("x.topic_id = t.id AND
                        (
                          (t.score <> x.avg_score OR t.score IS NULL) OR
                          (t.has_summary IS NULL OR t.has_summary <> (
                            t.like_count >= :likes_required AND
                            t.posts_count >= :posts_required AND
                            x.max_score >= :score_required
                          ))
                        )
  ",
              likes_required: SiteSetting.summary_likes_required,
              posts_required: SiteSetting.summary_posts_required,
              score_required: SiteSetting.summary_score_threshold)

    if min_topic_age
      builder.where("t.bumped_at > :bumped_at ",
                   bumped_at: min_topic_age)
    end

    builder.exec
  end

  def update_topics_percent_rank(min_topic_age)

    builder = SqlBuilder.new("UPDATE topics SET percent_rank = x.percent_rank
          FROM (SELECT id, percent_rank()
                OVER (ORDER BY SCORE DESC) as percent_rank
                FROM topics) AS x
                /*where*/")

    builder.where("x.id = topics.id AND (topics.percent_rank <> x.percent_rank OR topics.percent_rank IS NULL)")


    if min_topic_age
      builder.where("topics.bumped_at > :bumped_at ",
                   bumped_at: min_topic_age)
    end


    builder.exec
  end


  def filter_topics(builder, min_topic_age)
    if min_topic_age
      builder.where('posts.topic_id IN
                    (SELECT id FROM topics WHERE bumped_at > :bumped_at)',
                   bumped_at: min_topic_age)
    end

    builder
  end

end