[~] save progress method

README.md

@@ -178,7 +178,7 @@ To use a different formatting value than the default, you need to specify it exp
 
 ## Resumable Tasks
 
-When working with long-running code that processes a large dataset, it makes sense to store progress in the database to allow resuming the task later. You can do this by using the `DeployPin::Task` instance methods: `#progress` and `#increment_progress!(num)`.
+When working with long-running code that processes a large dataset, it makes sense to store progress in the database to allow resuming the task later. You can do this by using the `DeployPin::Task` instance methods: `#progress`, `#save_progress!(num)` and `#increment_progress!(num)`.
 
 Here is an example of how to use these methods:
 
@@ -191,6 +191,7 @@ Here is an example of how to use these methods:
 Users.where(id: progress..).find_each do |user|
   # Do some work
   increment_progress!(1) # Increment progress by 1 and store it in the database so you can resume the task from this point
+  # or save_progress!(user.id) # Save the progress as the user id
 end
 ```
 

lib/deploy_pin/task.rb

@@ -54,6 +54,12 @@ def increment_progress!(incrementor)
       record.increment!(:progress, incrementor)
     end
 
+    def save_progress!(value)
+      raise NotImplementedError, 'Recurring tasks do not support progress yet.' if recurring
+
+      record.update(progress: value)
+    end
+
     def done?
       return if recurring
       return unless record

lib/deploy_pin/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DeployPin
-  VERSION = '1.7.1'
+  VERSION = '1.7.2'
 end

test/lib/deploy_pin/task_test.rb

@@ -35,6 +35,17 @@ class DeployPin::Task::Test < ActiveSupport::TestCase
     assert_equal @task.progress, 0
     assert_nothing_raised { @task.increment_progress!(77) }
     assert_equal @task.progress, 77
+    assert_nothing_raised { @task.increment_progress!(13) }
+    assert_equal @task.progress, 90
+  end
+
+  test 'save_progress!' do
+    @task.prepare
+    assert_equal @task.progress, 0
+    assert_nothing_raised { @task.save_progress!(13) }
+    assert_equal @task.progress, 13
+    assert_nothing_raised { @task.save_progress!(1) }
+    assert_equal @task.progress, 1
   end
 
   test 'parse' do