mysql_rewinder 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc9e1df489b72666d2df9e54a9d3467e008c33729cf9a2cc674a70e258570842
-  data.tar.gz: d470cfd7cfb5867a21c79c0b45f40480d2d34dbdbe048b78cfcbb751b7a97222
+  metadata.gz: 4a2e247f212ee2a165527969e69188036a554e5e812f156adb4f13136b302fbe
+  data.tar.gz: 9ca4cf5ad37d8ab1343882e0012ea2c350c4fd6305d043fc2dde901a3ae068dd
 SHA512:
-  metadata.gz: ba1ee1138a2d318fd6ebbc1376cb06cd43ea0962a3305c09b59e4e0cd82de72fa7001dc7146bb85cda1bade4fa1b82df0f11a10870e76b1ac8c867edc3ccabc2
-  data.tar.gz: 93c60a029034ad464c93c36ce7a3bc3fe292cca146e9fbaaad041a7c08b8ccd7093c0324edcff443b449cef197881f4f1bcbaf2375a2b925f67bfbfca6168d55
+  metadata.gz: dfe1906efd4de0b8ee5c3eafcb5e102d7fda767f24c9f16da594a19deb081f515281b1c9aec8f6646ae49023271eeb730876f48a6b725f4bbe724d91749e15f2
+  data.tar.gz: d7a6f5a4e43799a49270f1b65412a3aebb627e2d313be5ff7842ae0d19db70744a7768607aad8255214cad377d1229cf731d8897e088ad4094b3ec898fab3dbf

data/README.md

@@ -89,7 +89,7 @@ MysqlRewinder.setup(db_configs, adapter: :mysql2)
 If you want to use MysqlRewinder with ActiveRecord, do the following:
 
 * Generate db_configs from `ActiveRecord::Base.configurations`
-* Pass `ActiveRecord::SchemaMigration.new(nil).table_name` and `ActiveRecord::Base.internal_metadata_table_name` to `DatabaseRewinder.setup` as `except_tables`
+* Pass `ActiveRecord::SchemaMigration.new(nil).table_name` and `ActiveRecord::Base.internal_metadata_table_name` to `MysqlRewinder.setup` as `except_tables`
 
 ```ruby
 db_configs = ActiveRecord::Base.configurations.configs_for(env_name: 'test').map(&:configuration_hash)
@@ -105,6 +105,14 @@ except_tables = [
 MysqlRewinder.setup(db_configs, except_tables: except_tables)
 ```
 
+### Logging
+
+If you want to enable logging, specify `logger` for `MysqlRewinder.setup`
+
+```ruby
+MysqlRewinder.setup(db_configs, logger: Logger.new(STDOUT))
+```
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/DeNA/mysql_rewinder. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/DeNA/mysql_rewinder/blob/trunk/CODE_OF_CONDUCT.md).

data/lib/mysql_rewinder/cleaner.rb

@@ -6,10 +6,11 @@ class MysqlRewinder
   class Cleaner
     attr_reader :db_config
 
-    def initialize(db_config, except_tables:, adapter:)
+    def initialize(db_config, except_tables:, adapter:, logger: nil)
       @db_config = db_config
       @client = Adapter.generate(adapter, db_config.transform_keys(&:to_sym))
       @except_tables = except_tables
+      @logger = logger
     end
 
     def clean_all
@@ -20,8 +21,8 @@ class MysqlRewinder
       target_tables = (tables - @except_tables) & all_tables
       return if target_tables.empty?
 
-      @client.execute("SET FOREIGN_KEY_CHECKS = 0;")
-      @client.execute(target_tables.map { |table| "DELETE FROM #{table}" }.join(';'))
+      log_and_execute("SET FOREIGN_KEY_CHECKS = 0;")
+      log_and_execute(target_tables.map { |table| "DELETE FROM #{table}" }.join(';'))
     end
 
     def all_tables
@@ -31,5 +32,20 @@ class MysqlRewinder
         WHERE TABLE_SCHEMA = DATABASE()
       SQL
     end
+
+    private
+
+    def log_and_execute(sql)
+      return @client.execute(sql) unless @logger&.debug?
+
+      start_ts = Time.now
+      res = @client.execute(sql)
+      duration = (Time.now - start_ts) * 1000
+
+      name = "[MysqlRewinder] Cleaner SQL (#{duration.round(1)}ms)"
+      msg = "\e[1m\e[30m#{name}\e[0m  \e[34m#{sql}\e[0m"
+      @logger.debug msg
+      res
+    end
   end
 end

data/lib/mysql_rewinder/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class MysqlRewinder
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/mysql_rewinder.rb

@@ -6,27 +6,30 @@ require 'set'
 require 'tmpdir'
 require 'fileutils'
 require 'forwardable'
+require 'logger'
 
 class MysqlRewinder
   class << self
     extend Forwardable
     delegate %i[clean clean_all record_inserted_table] => :@instance
 
-    def setup(db_configs, except_tables: [], adapter: :trilogy)
-      @instance = new(db_configs: db_configs, except_tables: except_tables, adapter: adapter)
+    def setup(db_configs, except_tables: [], adapter: :trilogy, logger: nil)
+      @instance = new(db_configs: db_configs, except_tables: except_tables, adapter: adapter, logger: logger)
     end
   end
 
-  def initialize(db_configs:, except_tables:, adapter:)
+  def initialize(db_configs:, except_tables:, adapter:, logger:)
     @initialized_pid = Process.pid
     @inserted_table_record_dir = Pathname(Dir.tmpdir)
     @cleaners = db_configs.map do |db_config|
       Cleaner.new(
         db_config.transform_keys(&:to_sym),
         except_tables: except_tables,
-        adapter: adapter
+        adapter: adapter,
+        logger: logger
       )
     end
+    @logger = logger
     reset_inserted_tables
   end
 

data/sig/mysql_rewinder/cleaner.rbs

@@ -4,12 +4,17 @@ class MysqlRewinder
     @client: Adapter
     @except_tables: Array[String]
     @all_tables: Array[String]
+    @logger: untyped
 
     attr_reader db_config: Hash[Symbol,String]
 
-    def initialize: (Hash[Symbol,String] db_config, except_tables: Array[String], adapter: Symbol) -> untyped
+    def initialize: (Hash[Symbol,String] db_config, except_tables: Array[String], adapter: Symbol, ?logger: untyped) -> untyped
     def clean_all: () -> void
     def clean: (tables: untyped) -> void
     def all_tables: () -> void
+
+    private
+
+    def log_and_execute: (String sql) -> void
   end
 end

data/sig/mysql_rewinder.rbs

@@ -5,13 +5,14 @@ class MysqlRewinder
   # @inserted_table_record_dir: Pathname
   @cleaners: Array[Cleaner]
   @inserted_tables: Set[String]
+  @logger: untyped
 
-  def self.setup: (Array[Hash[Symbol,String]] db_configs, ?except_tables: Array[String], ?adapter: ::Symbol) -> void
+  def self.setup: (Array[Hash[Symbol,String]] db_configs, ?except_tables: Array[String], ?adapter: ::Symbol, ?logger: untyped) -> void
   def self.clean_all: () -> void
   def self.clean: () -> void
   def self.record_inserted_table: (String sql) -> void
 
-  def initialize: (db_configs: Array[Hash[Symbol,String]], except_tables: Array[String], adapter: ::Symbol) -> untyped
+  def initialize: (db_configs: Array[Hash[Symbol,String]], except_tables: Array[String], adapter: ::Symbol, logger: untyped) -> untyped
   def record_inserted_table: (String sql) -> void
   def reset_inserted_tables: () -> void
   def calculate_inserted_tables: () -> void

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mysql_rewinder
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Yusuke Sangenya
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-29 00:00:00.000000000 Z
+date: 2024-07-04 00:00:00.000000000 Z
 dependencies: []
 description:
 email:
@@ -33,7 +33,6 @@ files:
 - lib/mysql_rewinder/ext/mysql2_client.rb
 - lib/mysql_rewinder/ext/trilogy.rb
 - lib/mysql_rewinder/version.rb
-- mysql_rewinder.gemspec
 - sig/mysql_rewinder.rbs
 - sig/mysql_rewinder/cleaner.rbs
 - sig/mysql_rewinder/cleaner/adapter.rbs
@@ -63,7 +62,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.3.26
 signing_key:
 specification_version: 4
 summary: Simple, stable, and fast database cleaner for mysql

data/mysql_rewinder.gemspec

@@ -1,22 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "lib/mysql_rewinder/version"
-
-Gem::Specification.new do |spec|
-  spec.name = "mysql_rewinder"
-  spec.version = MysqlRewinder::VERSION
-  spec.authors = ["Yusuke Sangenya"]
-  spec.email = ["longinus.eva@gmail.com"]
-
-  spec.summary = "Simple, stable, and fast database cleaner for mysql"
-  spec.homepage = "https://github.com/genya0407/mysql_rewinder"
-  spec.license = "MIT"
-  spec.required_ruby_version = ">= 3.0.0"
-
-  spec.files = Dir.chdir(__dir__) do
-    `git ls-files -z`.split("\x0").reject do |f|
-      (f == __FILE__) || f.match(%r{\A(?:(?:bin|test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
-    end
-  end
-  spec.require_paths = ["lib"]
-end