cover_rage 1.0.0 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5475081d89068cc543f39035786796e346ca90364f30c525680f4458118c211c
-  data.tar.gz: 954ed2e2a6d4173e99b533c8914fd4c517730b53262bdbce0193c937644ad524
+  metadata.gz: e35c3c64110ebafa0432d4ac37f2d05b19a552df8505263f751f4fb85a841525
+  data.tar.gz: 9cd16923a839fd2ddcd6f5b585d28bfc19ab40a466d810ac8a93378bf5c13b02
 SHA512:
-  metadata.gz: e73234c452933ed861dec4cde8735e455ed5ea36ccbc9fc682bd7f695f43573ebfc5f5fb723b57c490af2fdd56743480e5812efd67a521f3f8672169d2d7af7d
-  data.tar.gz: 36635a50188153c1f5d1df5b24eb024f12bcfc3b28496ec0deb2eaa918cd13cb571822945109b674f552b2f7fcad23519ef3347fc99d6e8f1ef91f85fdfdd179
+  metadata.gz: b1b158a2dc50611a60482aeae2f448c9f4f8ae59f3d1ebb6ecd29bc03fea38a82a934b4b84906a4f331a7b293ccfcb4780ab7d956fe24015c42c54a8d4a52cdc
+  data.tar.gz: b06bdca3118bf5d80f7225e4792d258a18ef17a27f9255e0993a0e9523aab81e1aa88051b6c6670241ea688bfd10df10118ce9934e84e1c4379d4fcae02b7282

data/lib/cover_rage/config.rb

@@ -27,16 +27,7 @@ module CoverRage
     end
 
     def self.interval
-      @interval ||= begin
-        args =
-          ENV.fetch('COVER_RAGE_INTERVAL', '60:90').split(':').map!(&:to_i).first(2)
-        args.push(args.first.succ) if args.length < 2
-        Range.new(*args, true)
-      end
-    end
-
-    def self.disable?
-      @disable ||= ENV.key?('COVER_RAGE_DISABLE')
+      @interval ||= ENV.fetch('COVER_RAGE_INTERVAL', '60').to_i
     end
   end
 end

data/lib/cover_rage/fork_hook.rb

@@ -7,5 +7,12 @@ module CoverRage
       CoverRage::Launcher.start if pid.zero?
       pid
     end
+
+    def daemon(...)
+      CoverRage::Launcher.recorder.save(Coverage.result(stop: false))
+      returned = super
+      CoverRage::Launcher.start
+      returned
+    end
   end
 end

data/lib/cover_rage/launcher.rb

@@ -7,6 +7,8 @@ require 'coverage'
 
 module CoverRage
   class Launcher
+    singleton_class.attr_reader :recorder
+
     def self.start(**kwargs)
       if @recorder.nil?
         @recorder = CoverRage::Recorder.new(
@@ -16,9 +18,6 @@ module CoverRage
         )
       end
       @recorder.start
-      return unless Process.respond_to?(:_fork)
-      return if Process.singleton_class < CoverRage::ForkHook
-
       Process.singleton_class.prepend(CoverRage::ForkHook)
     end
   end

data/lib/cover_rage/recorder.rb

@@ -6,7 +6,6 @@ require 'digest'
 
 module CoverRage
   class Recorder
-    INTERVAL = Config.interval
     attr_reader :store
 
     def initialize(path_prefix:, store:)
@@ -24,15 +23,15 @@ module CoverRage
         at_exit { save(Coverage.result) }
       end
       @thread = Thread.new do
+        interval = Config.interval
+        jitter = 0.15
         loop do
-          sleep(rand(INTERVAL))
+          sleep(interval + (rand * interval * jitter))
           save(Coverage.result(stop: false, clear: true))
         end
       end
     end
 
-    private
-
     def save(coverage_result)
       records = []
       coverage_result.map do |filepath, execution_count|
@@ -45,14 +44,21 @@ module CoverRage
 
         records << Record.new(
           path: relative_path,
-          revision: revision,
-          source: source,
-          execution_count: execution_count
+          revision:,
+          source:,
+          execution_count:
         )
       end
-      @store.import(records) if records.any?
+      return unless records.any?
+
+      @store.transaction do
+        records_to_save = Record.merge(@store.list, records)
+        @store.update(records_to_save)
+      end
     end
 
+    private
+
     def read_file_with_revision(path)
       return @file_cache[path] if @file_cache.key?(path)
 

data/lib/cover_rage/stores/pstore.rb

@@ -10,27 +10,42 @@ module CoverRage
         @store = PStore.new(path, true)
       end
 
-      def import(records)
+      def transaction
         @store.transaction do
-          persisted_records = @store.keys.map { @store[_1] }
-          records_to_save = Record.merge(persisted_records, records)
-          records_to_save.each { @store[_1.path] = _1 }
+          @transaction = true
+          yield
+        ensure
+          @transaction = false
         end
       end
 
-      def find(path)
-        @store.transaction { @store[path] }
+      def update(records)
+        if @transaction
+          records.each { @store[_1.path] = _1 }
+        else
+          @store.transaction do
+            records.each { @store[_1.path] = _1 }
+          end
+        end
       end
 
       def list
-        @store.transaction do
+        if @transaction
           @store.keys.map { @store[_1] }
+        else
+          @store.transaction do
+            @store.keys.map { @store[_1] }
+          end
         end
       end
 
       def clear
-        @store.transaction do
+        if @transaction
           @store.keys.each { @store.delete(_1) }
+        else
+          @store.transaction do
+            @store.keys.each { @store.delete(_1) }
+          end
         end
       end
     end

data/lib/cover_rage/stores/redis.rb

@@ -12,32 +12,33 @@ module CoverRage
       def initialize(url)
         @redis =
           if url.start_with?('rediss')
-            ::Redis.new(url: url, ssl_params: { verify_mode: OpenSSL::SSL::VERIFY_NONE })
+            ::Redis.new(url:, ssl_params: { verify_mode: OpenSSL::SSL::VERIFY_NONE })
           else
-            ::Redis.new(url: url)
+            ::Redis.new(url:)
           end
       end
 
-      def import(records)
+      def transaction(&)
         loop do
           break if @redis.watch(KEY) do
-            records_to_save = Record.merge(list, records)
-            if records_to_save.any?
-              arguments = []
-              records_to_save.each do |record|
-                arguments.push(record.path, JSON.dump(record.to_h))
-              end
-              @redis.multi { _1.hset(KEY, *arguments) }
+            @redis.multi do |multi|
+              Thread.current[:redis_multi] = multi
+              yield
+            ensure
+              Thread.current[:redis_multi] = nil
             end
           end
         end
       end
 
-      def find(path)
-        result = @redis.hget(KEY, path)
-        return nil if result.nil?
+      def update(records)
+        arguments = []
+        records.each do |record|
+          arguments.push(record.path, JSON.dump(record.to_h))
+        end
 
-        Record.new(**JSON.parse(result))
+        client = Thread.current[:redis_multi] || @redis
+        client.hset(KEY, *arguments)
       end
 
       def list

data/lib/cover_rage/stores/sqlite.rb

@@ -8,6 +8,8 @@ module CoverRage
   module Stores
     class Sqlite
       def initialize(path)
+        @mutex = Mutex.new
+        @path = path
         @db = SQLite3::Database.new(path)
         @db.execute <<-SQL
           create table if not exists records (
@@ -17,53 +19,50 @@ module CoverRage
             execution_count text not null
           )
         SQL
-      end
-
-      def import(records)
-        @db.transaction do
-          records_to_save = Record.merge(list, records)
-          if records_to_save.any?
-            @db.execute(
-              "insert or replace into records (path, revision, source, execution_count) values #{
-                (['(?,?,?,?)'] * records_to_save.length).join(',')
-              }",
-              records_to_save.each_with_object([]) do |record, memo|
-                memo.push(
-                  record.path,
-                  record.revision,
-                  record.source,
-                  JSON.dump(record.execution_count)
-                )
-              end
-            )
+        process_ext = Module.new
+        process_ext.module_exec(self) do |store|
+          define_method :_fork do
+            store.instance_variable_get(:@db).close
+            pid = super()
+            store.instance_variable_set(:@db, SQLite3::Database.new(path))
+            pid
           end
         end
+        Process.singleton_class.prepend(process_ext)
       end
 
-      def find(path)
-        rows = @db.execute(
-          'select revision, source, execution_count from records where path = ? limit 1',
-          [path]
-        )
-        return nil if rows.empty?
+      def transaction(&)
+        @mutex.synchronize do
+          @db.transaction(:exclusive, &)
+        rescue SQLite3::BusyException
+          retry
+        end
+      end
 
-        revision, source, execution_count = rows.first
-        Record.new(
-          path: path,
-          revision: revision,
-          source: source,
-          execution_count: JSON.parse(execution_count)
+      def update(records)
+        @db.execute(
+          "insert or replace into records (path, revision, source, execution_count) values #{
+            (['(?,?,?,?)'] * records.length).join(',')
+          }",
+          records.each_with_object([]) do |record, memo|
+            memo.push(
+              record.path,
+              record.revision,
+              record.source,
+              JSON.dump(record.execution_count)
+            )
+          end
         )
       end
 
       def list
         @db
           .execute('select path, revision, source, execution_count from records')
-          .map! do |(path, revision, source, execution_count)|
+          .map do |(path, revision, source, execution_count)|
             Record.new(
-              path: path,
-              revision: revision,
-              source: source,
+              path:,
+              revision:,
+              source:,
               execution_count: JSON.parse(execution_count)
             )
           end

metadata

@@ -1,15 +1,28 @@
 --- !ruby/object:Gem::Specification
 name: cover_rage
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.1
 platform: ruby
 authors:
 - Weihang Jian
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-25 00:00:00.000000000 Z
+date: 2024-12-30 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.18'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.18'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -25,26 +38,41 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '13.0'
 - !ruby/object:Gem::Dependency
-  name: minitest
+  name: redis
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.18'
+        version: '5.3'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.18'
+        version: '5.3'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
 description: |
-  A Ruby production code coverage tool designed to assist you in identifying unused code, offering the following features:
+  cover_rage is a Ruby code coverage tool designed to be simple and easy to use. It can be used not only for test coverage but also in production services to identify unused code.
+
+  Key features:
 
-  1. zero dependencies
-  2. super easy setup
-  3. support process forking without additional configuration
-  4. minimal performance overhead
+  1. Runs in continuous processes (e.g., Rails servers)
+  2. Zero dependencies
+  3. Supports forking and daemonization without additional setup
 email: tonytonyjan@gmail.com
 executables:
 - cover_rage
@@ -67,8 +95,8 @@ files:
 homepage: https://github.com/tonytonyjan/cover_rage
 licenses:
 - MIT
-metadata: {}
-post_install_message:
+metadata:
+  rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
 - lib
@@ -76,15 +104,16 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.3.0
+      version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.23
-signing_key:
+rubygems_version: 3.6.2
 specification_version: 4
-summary: A Ruby production code coverage tool
+summary: cover_rage is a Ruby code coverage tool designed to be simple and easy to
+  use. It can be used not only for test coverage but also in production services to
+  identify unused code.
 test_files: []