worker_tools 1.2 → 1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: df0b1a0aede242ffcdd30a20f86adc4b05d78cf1634e998ad25c5edb0b654b9e
-  data.tar.gz: be05c13b2ab05fc02d67692bc676087c97502b9ef8d7598910d54650c9122241
+  metadata.gz: 6a69e014d2d2959641972ad163b5567e957693c90b54ae9e7aec4a723f1f39c6
+  data.tar.gz: 3cc3536cf26ab79054761570671361d23bbcb2c708e4974ebfd07ea8671180f3
 SHA512:
-  metadata.gz: ea0934e1b034b0011707909f2ef17c98c52458885d1ead4ee537373c668cf50a72c44c5b3dfcb6dfc1e9beab7a97e12892372415fe9f73c1b4a2952b9c032584
-  data.tar.gz: 27550fcf2510fe836215ee316b24780f732afda39714d69a8d1d60047048c91c01c6595475cc0396cc2b64d838d4eb9e953efcf078e6660e52a3c36f13b565cc
+  metadata.gz: f4d622417f3afd6057ae548f2b0e688ce1d8087d1e75a176ff4fe85caef5b1a1469e3ad07db2e21da2be6a365e97ba0bae18fbdd818f2eaea47bc7bb4e2fd0b7
+  data.tar.gz: e479438605973cf104b4c9d60cf3a7f821d09c053c7e569370afa7a1bbcc5e52006f65863a8a2085caa31f878df987c85d8daa7461c41f229aaa71094f9a6c2b

data/.tool-versions

@@ -0,0 +1 @@
+ruby 2.7.5

data/README.md

@@ -349,6 +349,25 @@ class MyImporter
 end
 ```
 
+## MemoryUsage
+
+There is a memory usage wrapper that you can use to record the memory consumption of your worker tasks. The memory usage is recorded in megabytes and stored in `model.meta['memory_usage']`. The module uses the `get_process_mem` gem to measure memory usage before and after the task execution.
+
+```ruby
+class MyImporter
+  include WorkerTools::MemoryUsage
+  wrappers :memory_usage
+
+  def run
+    # do stuff that consumes memory
+  end
+
+  # ..
+end
+```
+
+The memory usage will be automatically recorded in the model's meta attribute as `memory_usage` with the value in megabytes (rounded to 2 decimal places). The measurement represents the difference in memory usage before and after the task execution.
+
 ## Module 'Notes'
 
 If you use ActiveRecord you may need to modify the serializer as well as deserializer from the note attribute. After that you can easily serialize hashes and array of hashes with indifferent access. For that purpose the gem provides two utility methods. (HashWithIndifferentAccessType, SerializedArrayType). There is an example of how you can use it.

data/lib/worker_tools/memory_usage.rb

@@ -0,0 +1,28 @@
+require 'get_process_mem'
+
+module WorkerTools
+  module MemoryUsage
+    extend ActiveSupport::Concern
+
+    included do
+      # rubocop:disable Metrics/MethodLength
+      def with_wrapper_memory_usage(&block)
+        memory_error = nil
+        start_memory = GetProcessMem.new.mb
+
+        begin
+          block.call
+        rescue StandardError => e
+          memory_error = e
+        ensure
+          end_memory = GetProcessMem.new.mb
+          memory_used = (end_memory - start_memory).round(2)
+          model.meta['memory_usage'] = memory_used if model.respond_to?(:meta)
+        end
+
+        raise memory_error if memory_error
+      end
+      # rubocop:enable Metrics/MethodLength
+    end
+  end
+end

data/lib/worker_tools/version.rb

@@ -1,3 +1,3 @@
 module WorkerTools
-  VERSION = '1.2'.freeze
+  VERSION = '1.3'.freeze
 end

data/worker_tools.gemspec

@@ -27,6 +27,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   spec.add_dependency 'activesupport'
+  spec.add_dependency 'get_process_mem'
   spec.add_dependency 'roo'
   spec.add_dependency 'rubyXL'
   spec.add_dependency 'slack-notifier'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: worker_tools
 version: !ruby/object:Gem::Version
-  version: '1.2'
+  version: '1.3'
 platform: ruby
 authors:
 - fsainz
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-03-18 00:00:00.000000000 Z
+date: 2025-09-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: get_process_mem
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: roo
   requirement: !ruby/object:Gem::Requirement
@@ -229,6 +243,7 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - ".rubocop.yml"
+- ".tool-versions"
 - ".travis.yml"
 - CHANGELOG.md
 - Gemfile
@@ -244,6 +259,7 @@ files:
 - lib/worker_tools/csv_input.rb
 - lib/worker_tools/csv_output.rb
 - lib/worker_tools/errors.rb
+- lib/worker_tools/memory_usage.rb
 - lib/worker_tools/recorder.rb
 - lib/worker_tools/slack_error_notifier.rb
 - lib/worker_tools/utils/hash_with_indifferent_access_type.rb