discourse/lib/tasks/bookmarks.rake

# frozen_string_literal: true

require_dependency "rake_helpers"

##
# This will create records in the new bookmarks table from PostAction
# records. The task is idempotent, it will not create additional bookmark
# records for PostActions that have already been created in the new table.
# You can provide a sync_limit for a smaller batch run.
#
desc "migrates old PostAction bookmarks to the new Bookmark model & table"
task "bookmarks:sync_to_table", [:sync_limit] => :environment do |_t, args|
  sync_limit = args[:sync_limit] || 0
  post_action_bookmarks = PostAction
    .select('post_actions.id', 'post_actions.post_id', 'posts.topic_id', 'post_actions.user_id')
    .where(post_action_type_id: PostActionType.types[:bookmark])
    .joins(:post)
    .where(deleted_at: nil)
    .joins('LEFT JOIN bookmarks ON bookmarks.post_id = post_actions.post_id AND bookmarks.user_id = post_actions.user_id')
    .where('bookmarks.id IS NULL')

  # if sync_limit is provided as an argument this will cap
  # the number of bookmarks that will be created in a run of
  # this task (for huge bookmark count communities)
  if sync_limit > 0
    post_action_bookmarks = post_action_bookmarks.limit(sync_limit)
  end

  post_action_bookmark_count = post_action_bookmarks.count('post_actions.id')
  bookmarks_to_create = []
  i = 0

  Bookmark.transaction do
    post_action_bookmarks.find_each(batch_size: 1000) do |pab|
      RakeHelpers.print_status_with_label("Creating post new bookmarks.......", i, post_action_bookmark_count)
      now = Time.zone.now
      bookmarks_to_create << {
        topic_id: pab.topic_id,
        post_id: pab.post_id,
        user_id: pab.user_id,
        created_at: now,
        updated_at: now
      }

      i += 1
    end

    # this will ignore conflicts in the bookmarks table so
    # if the user already has a post bookmarked in the new way,
    # then we don't error and keep on truckin'
    #
    # we shouldn't have duplicates here at any rate because of
    # the above LEFT JOIN but best to be safe knowing this
    # won't blow up
    Bookmark.insert_all(bookmarks_to_create)
  end

  RakeHelpers.print_status_with_label("Bookmark creation complete!            ", i, post_action_bookmark_count)
  puts ""
end