Move OnceOff to a post migration

We're not using OnceOffs any more, so to avoid having to maintain this, we're moving it to a migration. This post migration should ideally be a no-op, so it is fine to run anyway
This commit is contained in:
Nat 2024-07-17 16:44:21 +08:00
parent 05396b6fb4
commit 85eb26903e
No known key found for this signature in database
GPG Key ID: 4938B35D927EC773
3 changed files with 91 additions and 98 deletions

View File

@ -1,94 +0,0 @@
# frozen_string_literal: true
module Jobs
class VotingEnsureConsistency < ::Jobs::Onceoff
def execute_onceoff(args)
# archive votes to closed or archived or deleted topics
DB.exec(<<~SQL)
UPDATE discourse_voting_votes
SET archive=true
FROM topics
WHERE topics.id = discourse_voting_votes.topic_id
AND discourse_voting_votes.archive IS NOT TRUE
AND (topics.closed OR topics.archived OR topics.deleted_at IS NOT NULL)
SQL
# un-archive votes to open topics
DB.exec(<<~SQL)
UPDATE discourse_voting_votes
SET archive=false
FROM topics
WHERE topics.id = discourse_voting_votes.topic_id
AND discourse_voting_votes.archive IS TRUE
AND NOT topics.closed
AND NOT topics.archived
AND topics.deleted_at IS NULL
SQL
# delete duplicate votes
DB.exec(<<~SQL)
DELETE FROM discourse_voting_votes dvv1
USING discourse_voting_votes dvv2
WHERE dvv1.id < dvv2.id AND
dvv1.user_id = dvv2.user_id AND
dvv1.topic_id = dvv2.topic_id AND
dvv1.archive = dvv2.archive
SQL
# delete votes associated with no user
DB.exec(<<~SQL)
DELETE FROM discourse_voting_votes
WHERE user_id NOT IN (SELECT id FROM users)
SQL
# delete votes associated with no topics
DB.exec(<<~SQL)
DELETE FROM discourse_voting_votes
WHERE discourse_voting_votes.topic_id IS NULL
SQL
# delete duplicate vote counts for topics
DB.exec(<<~SQL)
DELETE FROM discourse_voting_topic_vote_count dvtvc
USING discourse_voting_topic_vote_count dvtvc2
WHERE dvtvc.id < dvtvc2.id AND
dvtvc.topic_id = dvtvc2.topic_id AND
dvtvc.votes_count = dvtvc2.votes_count
SQL
# insert missing vote counts for topics
# ensures we have "something" for every topic with votes
DB.exec(<<~SQL)
WITH missing_ids AS (
SELECT DISTINCT t.id FROM topics t
JOIN discourse_voting_votes dvv ON t.id = dvv.topic_id
LEFT JOIN discourse_voting_topic_vote_count dvtvc ON t.id = dvtvc.topic_id
WHERE dvtvc.topic_id IS NULL
)
INSERT INTO discourse_voting_topic_vote_count (votes_count, topic_id, created_at, updated_at)
SELECT '0', id, now(), now() FROM missing_ids
SQL
# remove all superflous vote count custom fields
DB.exec(<<~SQL)
DELETE FROM discourse_voting_topic_vote_count
WHERE topic_id IN (
SELECT t1.id FROM topics t1
LEFT JOIN discourse_voting_votes dvv
ON dvv.topic_id = t1.id
WHERE dvv.id IS NULL
)
SQL
# correct topics vote counts
DB.exec(<<~SQL)
UPDATE discourse_voting_topic_vote_count dvtvc
SET votes_count = (
SELECT COUNT(*) FROM discourse_voting_votes dvv
WHERE dvtvc.topic_id = dvv.topic_id
GROUP BY dvv.topic_id
)
SQL
end
end
end

View File

@ -0,0 +1,86 @@
# frozen_string_literal: true
class EnsureConsistency < ActiveRecord::Migration[7.0]
def up
# un-archive votes to open topics
DB.exec(<<~SQL)
UPDATE topic_voting_votes
SET archive=false
FROM topics
WHERE topics.id = topic_voting_votes.topic_id
AND topic_voting_votes.archive IS TRUE
AND NOT topics.closed
AND NOT topics.archived
AND topics.deleted_at IS NULL
SQL
# delete duplicate votes
DB.exec(<<~SQL)
DELETE FROM topic_voting_votes dvv1
USING topic_voting_votes dvv2
WHERE dvv1.id < dvv2.id AND
dvv1.user_id = dvv2.user_id AND
dvv1.topic_id = dvv2.topic_id AND
dvv1.archive = dvv2.archive
SQL
# delete votes associated with no user
DB.exec(<<~SQL)
DELETE FROM topic_voting_votes
WHERE user_id NOT IN (SELECT id FROM users)
SQL
# delete votes associated with no topics
DB.exec(<<~SQL)
DELETE FROM topic_voting_votes
WHERE topic_voting_votes.topic_id IS NULL
SQL
# delete duplicate vote counts for topics
DB.exec(<<~SQL)
DELETE FROM topic_voting_topic_vote_count dvtvc
USING topic_voting_topic_vote_count dvtvc2
WHERE dvtvc.id < dvtvc2.id AND
dvtvc.topic_id = dvtvc2.topic_id AND
dvtvc.votes_count = dvtvc2.votes_count
SQL
# insert missing vote counts for topics
# ensures we have "something" for every topic with votes
DB.exec(<<~SQL)
WITH missing_ids AS (
SELECT DISTINCT t.id FROM topics t
JOIN topic_voting_votes dvv ON t.id = dvv.topic_id
LEFT JOIN topic_voting_topic_vote_count dvtvc ON t.id = dvtvc.topic_id
WHERE dvtvc.topic_id IS NULL
)
INSERT INTO topic_voting_topic_vote_count (votes_count, topic_id, created_at, updated_at)
SELECT '0', id, now(), now() FROM missing_ids
SQL
# remove all superflous vote count custom fields
DB.exec(<<~SQL)
DELETE FROM topic_voting_topic_vote_count
WHERE topic_id IN (
SELECT t1.id FROM topics t1
LEFT JOIN topic_voting_votes dvv
ON dvv.topic_id = t1.id
WHERE dvv.id IS NULL
)
SQL
# correct topics vote counts
DB.exec(<<~SQL)
UPDATE topic_voting_topic_vote_count dvtvc
SET votes_count = (
SELECT COUNT(*) FROM topic_voting_votes dvv
WHERE dvtvc.topic_id = dvv.topic_id
GROUP BY dvv.topic_id
)
SQL
end
def down
raise ActiveRecord::IrreversibleMigration
end
end

View File

@ -1,10 +1,11 @@
# frozen_string_literal: true
require "rails_helper"
require Rails.root.join(
"plugins/discourse-topic-voting/db/post_migrate/20240711102255_ensure_consistency.rb",
)
describe Jobs::VotingEnsureConsistency do
subject(:job) { described_class.new }
describe EnsureConsistency do
it "ensures consistency" do
user = Fabricate(:user)
user2 = Fabricate(:user)
@ -24,7 +25,7 @@ describe Jobs::VotingEnsureConsistency do
DiscourseTopicVoting::Vote.create!(user: user, topic: two_vote_topic, archive: true)
DiscourseTopicVoting::Vote.create!(user: user2, topic: two_vote_topic)
job.execute_onceoff(nil)
EnsureConsistency.new.up
no_vote_topic.reload