ec-pg 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: def4a2f65d7a836d927f53af5af0bf3cb0ef644803f2f5dd96e992d92019bff3
-  data.tar.gz: ff88e37f850e3d18bdc48c9a4ca522cdc776f793183852f50cf7c8e9a657ff33
+  metadata.gz: 927ed76ae446cafa9cfa09623eb2ed3b05adf7973023c21de25353a95e7f59b2
+  data.tar.gz: 59865a939e8b813ec9872d4c27e9e5b7313dae76edf001cc175d79f41f76cb66
 SHA512:
-  metadata.gz: a7b0d8add219551e2793c062f62ae4ed36bdaa2d53977b035c55993cce8f4a114a485698dd3a9377f4aa47a15aa10f7bfdac523813496fe194f949fe5b28fd26
-  data.tar.gz: fca5463a5e84112a75090f7b852b7abf88ed68c24c1e66b142f1a0a626fe6c95df24e1a240bcd01325b4b73f8eebf18ff0b05fdd156f9c0adade49c829ef63a3
+  metadata.gz: 2e54504bc522de9c78e56344bae59e0b4661cc4fdce52ebc393ea210ab5c09d1fe1d49e8d7fc49b610f0c3ac729747ec82d028703d41ec9b7fb0650fee0337e9
+  data.tar.gz: 067f8f312bc09007fef2544b45a94f91f7c7e5196dc74101c736bef801875b3795b57f04ec156db2b647bb181acf159e7e6e25a77915d7a01e49786eb84b7978

data/lib/ec/pg/schema_manager.rb

@@ -81,9 +81,12 @@ module Ec
 
       # Builds the search_path string and executes SET search_path on the connection.
       def apply_schema(schema_name)
-        registered_connections.each do |connection|
+        registered_models.each do |model|
+          connection = model.connection
           shared = Ec::Pg.configuration.shared_schemas
           path   = ([schema_name] + shared).uniq.join(", ")
+
+          Ec::Pg.dprint("%s: Switching to schema: %s" % [model, path], color: :green)
           connection.schema_search_path = path
         end
       end
@@ -94,8 +97,10 @@ module Ec
         shared = Ec::Pg.configuration.shared_schemas
         path   = ([default] + shared).uniq.join(", ")
 
-        registered_connections.each do |connection|
+        registered_models.each do |model|
+          connection = model.connection
           if connection && connection.pool.present?
+            Ec::Pg.dprint("%s: Restoring schema to: %s" % [model, path], color: :red)
             connection.schema_search_path = path
           end
         end
@@ -103,10 +108,8 @@ module Ec
         # Swallow errors during cleanup (connection may have been returned to pool)
       end
 
-      def registered_connections
-        Ec::Pg::SchemaMixin
-          .registered_models
-          .map(&:connection)
+      def registered_models
+        Ec::Pg::SchemaMixin.registered_models
       end
 
       # Guards against SQL injection via schema names.
@@ -120,7 +123,7 @@ module Ec
               "Only alphanumeric characters and underscores are allowed."
       end
 
-      module_function :apply_schema, :restore_schema, :validate_schema_name!, :registered_connections
+      module_function :apply_schema, :restore_schema, :validate_schema_name!, :registered_models
     end
   end
 end

data/lib/ec/pg/version.rb

@@ -2,6 +2,6 @@
 
 module Ec
   module Pg
-    VERSION = "0.1.7"
+    VERSION = "0.1.8"
   end
 end

data/lib/ec/pg.rb

@@ -36,9 +36,9 @@ module Ec
 
     module_function 
 
-    def dprint(*args)
+    def dprint(message, color: :white)
       if ENV['ECPG_DEBUG']
-        puts args
+        puts message.colorize(color)
       end
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ec-pg
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - gmhawash