data_keeper 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2fc356b01f1619f2ec815a556092bd13fc28b44b2b201707c6a820da00fcff70
-  data.tar.gz: 8f2d4dcc59035b7fee1131daa3cfed6dedb0bf5c635e781d48dc65c7c43a58e7
+  metadata.gz: 281f71c099696e1d51c88dd57e2cd0edf5f028bbbf4340c4457923e3a17b553a
+  data.tar.gz: 1592db229f9407735630803bff1a97bd85ed051ec30b878029952ffcfb272759
 SHA512:
-  metadata.gz: 5102f1e716045e8bc425d69c71b03c8cad46b805d3caf7db12a0eb9ff073f53c1c94cebfd70a9909b8d806f543c66e1ecd7f3c3d3d344a688bacf8bdf6d2f3f5
-  data.tar.gz: 0b4459b6441c79cb3768fa769271ccb7661f458cf4f69b75562a4fc92ebfa2e28502bf294102689112b399cf97938c25ee703775444132f39c2a38e6d402bfa4
+  metadata.gz: cf10459525a1cd3f210c6ef926b2b6d87a217505c8d8c8c155e99021d7a8837312fbf0038302987a683bf2a66f335d7507071451eff213ce004536ea42d26ca9
+  data.tar.gz: 30ff5218bb48851d938b96afba015e9076226b25342171e35ac47b5d78ecc8d8c3308afc2ffd6a94ded725495303da9a65f4394530a63cb193fcf005a90d4952

data/.gitignore

@@ -6,5 +6,6 @@
 /pkg/
 /spec/reports/
 /tmp/
+/lib/tmp
 
 Gemfile.lock

data/Gemfile

@@ -5,3 +5,4 @@ gemspec
 
 gem "rake", "~> 12.0"
 gem "minitest", "~> 5.0"
+gem 'pg'

data/lib/data_keeper/database_config.rb

@@ -1,7 +1,7 @@
 module DataKeeper
   module DatabaseConfig
     def database_connection_config
-      Rails.configuration.database_configuration[Rails.env]
+      @database_connection_config ||= DataKeeper.database_config.call
     end
 
     def psql_env

data/lib/data_keeper/definition.rb

@@ -33,6 +33,7 @@ module DataKeeper
       @tables = []
       @raw_sqls = {}
       @definition_block = definition_block
+      @on_after_load_block = nil
     end
 
     def evaluate!

data/lib/data_keeper/loader.rb

@@ -20,48 +20,66 @@ module DataKeeper
       end
 
       if @dump.on_after_load_block
+        ActiveRecord::Base.establish_connection
         @dump.on_after_load_block.call
       end
     end
 
     private
 
-    def load_full_database!
-      ensure_schema_compatibility!
+    def log_redirect
+      if Terrapin::CommandLine.logger
+        ""
+      else
+        "  2>/dev/null"
+      end
+    end
 
-      pg_restore = Terrapin::CommandLine.new(
-        'pg_restore',
-        "#{connection_args} -j 4 --no-owner --dbname #{database} #{@file.path} 2>/dev/null",
+    def set_ar_internal_metadata!
+      cmd = Terrapin::CommandLine.new(
+        'psql',
+        "#{connection_args} -d :database -c :sql",
         environment: psql_env
       )
 
-      pg_restore.run(
+      cmd.run(
         database: database,
         host: host,
-        port: port
+        port: port,
+        sql: "DELETE from ar_internal_metadata"
       )
 
-      cmd = Terrapin::CommandLine.new(
-        'psql',
-        "#{connection_args} -d :database -c :sql",
+      cmd.run(
+        database: database,
+        host: host,
+        port: port,
+        sql: "INSERT into ar_internal_metadata (key, value, created_at, updated_at) VALUES ('environment', 'development', '2020-04-03 12:25:54.094209', '2020-04-03 12:25:54.094209')"
+      )
+    end
+
+    def load_full_database!
+      ensure_schema_compatibility!
+
+      pg_restore = Terrapin::CommandLine.new(
+        'pg_restore',
+        "#{connection_args} -j 4 --no-owner --dbname #{database} #{@file.path}#{log_redirect}",
         environment: psql_env
       )
 
-      cmd.run(
+      pg_restore.run(
         database: database,
         host: host,
-        port: port,
-        sql: "UPDATE ar_internal_metadata SET value = 'development'"
+        port: port
       )
+
+      set_ar_internal_metadata!
     end
 
     def load_partial_database!
       inflate(@file.path) do |schema_path, tables_path, sql_files|
-        ensure_schema_compatibility!
-
         pg_restore = Terrapin::CommandLine.new(
           'pg_restore',
-          "#{connection_args} -j 4 --no-owner --dbname :database #{schema_path} 2>/dev/null",
+          "#{connection_args} -j 4 --no-owner -s --dbname :database #{schema_path}#{log_redirect}",
           environment: psql_env
         )
 
@@ -73,7 +91,7 @@ module DataKeeper
 
         pg_restore = Terrapin::CommandLine.new(
           'pg_restore',
-          "#{connection_args} --data-only -j 4 --no-owner --disable-triggers --dbname :database #{tables_path} 2>/dev/null",
+          "#{connection_args} --data-only -j 4 --no-owner --disable-triggers --dbname :database #{tables_path}#{log_redirect}",
           environment: psql_env
         )
 
@@ -99,7 +117,7 @@ module DataKeeper
           )
         end
 
-        Rake::Task['db:environment:set'].invoke
+        set_ar_internal_metadata!
       end
     end
 

data/lib/data_keeper/version.rb

@@ -1,3 +1,3 @@
 module DataKeeper
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

data/lib/data_keeper.rb

@@ -17,6 +17,7 @@ module DataKeeper
 
   @dump_definition_builders = {}
   @storage = nil
+  @database_config = -> { Rails.configuration.database_configuration[Rails.env] }
 
   def self.define_dump(name, type = :partial, &block)
     @dump_definition_builders[name.to_sym] = DefinitionBuilder.new(type, block)
@@ -59,6 +60,14 @@ module DataKeeper
     @storage = value
   end
 
+  def self.database_config=(value)
+    @database_config = value
+  end
+
+  def self.database_config
+    @database_config
+  end
+
   def self.clear_dumps!
     @dump_definition_builders = {}
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: data_keeper
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Roger Campos
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-09-02 00:00:00.000000000 Z
+date: 2021-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord