ec-pg 0.1.8 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 927ed76ae446cafa9cfa09623eb2ed3b05adf7973023c21de25353a95e7f59b2
-  data.tar.gz: 59865a939e8b813ec9872d4c27e9e5b7313dae76edf001cc175d79f41f76cb66
+  metadata.gz: ffe4813266a9c664a415549779456ef3da7f2ef8ba7be5fa27c7a9f355ec76af
+  data.tar.gz: 2d20b237a2e705ff121e29cfbfc6fc4d19afef4b89843c240d354c2fc1471f06
 SHA512:
-  metadata.gz: 2e54504bc522de9c78e56344bae59e0b4661cc4fdce52ebc393ea210ab5c09d1fe1d49e8d7fc49b610f0c3ac729747ec82d028703d41ec9b7fb0650fee0337e9
-  data.tar.gz: 067f8f312bc09007fef2544b45a94f91f7c7e5196dc74101c736bef801875b3795b57f04ec156db2b647bb181acf159e7e6e25a77915d7a01e49786eb84b7978
+  metadata.gz: 62504aab93841d6051442cb3836afbf60edda564753e35770f72179239e85a44f8333f09a04c1a3990eb45d76e9c1d1c0b8a703d139f515c89ca35adbf195b37
+  data.tar.gz: 27deef7ab9f3213accfd76fbdedbe84baa9befb96a56b18b9f3700a509d2fb439bdca5c54e569e957614b3c4ad7c697c37dde1f9a79ccd6daedd259edb5f9220

data/lib/ec/pg/migration_mixin.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Ec
+  module Pg
+    module MigrationMixin
+      def create_type(name, allow_blank: true)
+        values = allow_blank ? ['', 'null'] : []
+        values += I18n.t('database_types.%s' % name).absolute_deepest_keys.map(&:to_s)
+        quoted_values = values.map{|x| quote(x)}.join(',')
+        execute <<-SQL
+        CREATE TYPE #{Ec::Pg::SchemaManager.current_schema}.#{name} as ENUM (#{quoted_values});
+        SQL
+      end
+
+      def drop_type(name)
+        execute <<-SQL
+        DROP TYPE IF EXISTS #{Ec::Pg::SchemaManager.current_schema}.#{name};
+        SQL
+
+      end
+
+      def add_values_to_type(name, values, instruction = '')
+        execute('commit;')   # The Alter Type query cannont happen inside a SQL transaction, so quit it.
+        values.each do |value|
+          execute "ALTER TYPE #{Ec::Pg::SchemaManager.current_schema}.#{name} ADD VALUE IF NOT EXISTS #{quote(value)} #{instruction}"
+        end
+      end
+    end
+  end
+end
+
+if Object.const_defined?("ActiveRecord::Migration")
+  ActiveRecord::Migration.send(:include, Ec::Pg::MigrationMixin)
+end

data/lib/ec/pg/migrator.rb

@@ -16,7 +16,9 @@ module Ec
               Ec::Pg.switch(shard: shard) do 
                 schemas.each do |schema_name|
                   Ec::Pg.dprint("migrating model: #{model}, shard: #{shard}, schema: #{schema_name}".red)
-                  migrate_schema(model.connection, schema_name)
+                  Context.with(schema: schema_name) do
+                    migrate_schema(model.connection, schema_name)
+                  end
                 end
               end
             end
@@ -31,7 +33,9 @@ module Ec
               Ec::Pg.switch(shard: shard) do 
                 schemas.each do |schema_name|
                   Ec::Pg.dprint("rolling back model: #{model}, shard: #{shard}, schema: #{schema_name}".red)
-                  rollback_schema(model.connection, schema_name, steps)
+                  Context.with(schema: schema_name) do
+                    rollback_schema(model.connection, schema_name, steps)
+                  end
                 end
               end
             end

data/lib/ec/pg/version.rb

@@ -2,6 +2,7 @@
 
 module Ec
   module Pg
-    VERSION = "0.1.8"
+    MINOR = 10
+    VERSION = "0.1.#{MINOR}"
   end
 end

data/lib/ec/pg.rb

@@ -7,6 +7,7 @@ require_relative "pg/version"
 require_relative "pg/context"
 require_relative "pg/tenant_context"
 require_relative "pg/configuration"
+require_relative "pg/migration_mixin"
 
 if ENV['RAILS_ENV'] == 'test'
   require 'debug'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ec-pg
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.10
 platform: ruby
 authors:
 - gmhawash
@@ -153,6 +153,7 @@ files:
 - lib/ec/pg/configuration.rb
 - lib/ec/pg/context.rb
 - lib/ec/pg/middleware/context_switcher.rb
+- lib/ec/pg/migration_mixin.rb
 - lib/ec/pg/migrator.rb
 - lib/ec/pg/mixin.rb
 - lib/ec/pg/railtie.rb
@@ -192,7 +193,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.3
+rubygems_version: 4.0.10
 specification_version: 4
 summary: Multitenant implemntations using shards, schemas and RLS
 test_files: []