Merge pull request #3590 from consul/add_upgrade_task

Add task to upgrade to a new release

config/deploy.rb

@@ -51,6 +51,13 @@ def deploysecret(key)
   after :published, "refresh_sitemap"
 
   after :finishing, "deploy:cleanup"
+
+
+  desc "Deploys and runs the tasks needed to upgrade to a new release"
+  task :upgrade do
+    after "add_new_settings", "execute_release_tasks"
+    invoke "deploy"
+  end
 end
 
 task :install_bundler_gem do
@@ -70,11 +77,21 @@ def deploysecret(key)
 end
 
 task :add_new_settings do
-  on roles(:app) do
+  on roles(:db) do
     within release_path do
       with rails_env: fetch(:rails_env) do
         execute :rake, "settings:add_new_settings"
       end
     end
   end
 end
+
+task :execute_release_tasks do
+  on roles(:app) do
+    within release_path do
+      with rails_env: fetch(:rails_env) do
+        execute :rake, "consul:execute_release_tasks"
+      end
+    end
+  end
+end

lib/application_logger.rb

@@ -0,0 +1,11 @@
+class ApplicationLogger
+  def info(message)
+    logger.info(message)
+  end
+
+  def logger
+    @logger ||= Logger.new(STDOUT).tap do |logger|
+      logger.formatter = proc { |severity, _datetime, _progname, msg| "#{severity} #{msg}\n" }
+    end
+  end
+end

lib/tasks/budgets.rake

@@ -1,4 +1,14 @@
 namespace :budgets do
+  desc "Regenerate ballot_lines_count cache"
+  task calculate_ballot_lines: :environment do
+    ApplicationLogger.new.info "Calculating ballot lines"
+
+    Budget::Ballot.find_each.with_index do |ballot, index|
+      Budget::Ballot.reset_counters ballot.id, :lines
+      print "." if (index % 10_000).zero?
+    end
+  end
+
   namespace :email do
 
     desc "Sends emails to authors of selected investments"

lib/tasks/consul.rake

@@ -0,0 +1,13 @@
+namespace :consul do
+  desc "Runs tasks needed to upgrade to the latest version"
+  task execute_release_tasks: "execute_release_1.0.0_tasks"
+
+  desc "Runs tasks needed to upgrade from 1.0.0-beta to 1.0.0"
+  task "execute_release_1.0.0_tasks": [
+    "poll:generate_slugs",
+    "stats_and_results:migrate_to_reports",
+    "budgets:calculate_ballot_lines",
+    "settings:remove_deprecated_settings",
+    "stats:generate"
+  ]
+end

lib/tasks/polls.rake

@@ -1,6 +1,8 @@
 namespace :poll do
   desc "Generate slugs polls"
   task generate_slugs: :environment do
+    ApplicationLogger.new.info "Generating poll slugs"
+
     Poll.find_each do |poll|
       poll.update_columns(slug: poll.generate_slug, updated_at: Time.current) if poll.generate_slug?
     end

lib/tasks/settings.rake

@@ -2,6 +2,8 @@ namespace :settings do
 
   desc "Remove deprecated settings"
   task remove_deprecated_settings: :environment do
+    ApplicationLogger.new.info "Removing deprecated settings"
+
     deprecated_keys = [
       "place_name",
       "banner-style.banner-style-one",

lib/tasks/stats.rake

@@ -1,6 +1,8 @@
 namespace :stats do
   desc "Generates stats which are not cached yet"
   task generate: :environment do
+    ApplicationLogger.new.info "Updating budget and poll stats"
+
     Budget.find_each do |budget|
       Budget::Stats.new(budget).generate
       print "."

lib/tasks/stats_and_results.rake

@@ -1,6 +1,7 @@
 namespace :stats_and_results do
   desc "Migrates stats_enabled and results_enabled data to enabled reports"
   task migrate_to_reports: :environment do
+    ApplicationLogger.new.info "Migrating stats and results"
     Migrations::Reports.new.migrate
   end
 end