blue-shift 0.1.6 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2ba2efeff34013ecda9d1470847300d4c4672fef
-  data.tar.gz: 1084eda477b23e024194b0786ee5a52014d72ee6
+  metadata.gz: 4f0cdc1858c2e2f39ce26325ccca7fe95e656038
+  data.tar.gz: 1c7fe3e6aac9b6e32959e381ae03c06de172c4b4
 SHA512:
-  metadata.gz: 6fa65136a66fb3283351223c052fcc84cb58cccdd5687af8d5b9e0f2bc6b2e264342f78251e64d7f2ef5893e342ee89ce827e2ab8671c8829bd641cd7a8a8d35
-  data.tar.gz: ed64339075b3032198be0e039c576f5bee471add340a7f122d46159cc72f7c824ad89f4087c686633de81096750afa880fcf271bd9e776210c78e3bf2e19cc89
+  metadata.gz: ff1293408c000baa3b5a34f0e717fa6842d614970615e49c15019e89f31387c00884542a7c87dd8e4422b56aea96105e64625970fb2b72c52c19f4dae80919d2
+  data.tar.gz: 2d4665a820d82718069b3aad88d57fb8db136555663965ba628ebbb16d7a49af33ff9dea86d72cc4fff35e36e70dc8ddee67eaa237f0cc441a93ef992323f9de

data/lib/blueshift.rb

@@ -4,6 +4,7 @@ require 'blueshift/migration'
 require 'sequel'
 require 'sequel/adapters/redshift'
 require 'sequel/extensions/schema_dumper_ext'
+require 'sequel/extensions/redshift_schema_dumper'
 
 module Blueshift
   def self.migration(&block)

data/lib/blueshift/version.rb

@@ -1,3 +1,3 @@
 module Blueshift
-  VERSION = '0.1.6'
+  VERSION = '0.2.0'
 end

data/lib/sequel/adapters/redshift.rb

@@ -9,6 +9,23 @@ module Sequel
       SORTSTYLES = [:compound, :interleaved].freeze
       DISTSTYLES = [:even, :key, :all].freeze
 
+      def optimize_table(table, create_options)
+        extension :redshift_schema_dumper
+        transaction do
+          gen = dump_table_generator(table)
+
+          rename_table table, :"old_#{table}"
+          create_table(:"new_#{table}", create_options) do
+            instance_eval(gen.dump_columns,     __FILE__, __LINE__)
+            instance_eval(gen.dump_constraints, __FILE__, __LINE__)
+            instance_eval(gen.dump_indexes,     __FILE__, __LINE__)
+          end
+          run %Q{INSERT INTO "new_#{table}" (SELECT * FROM "old_#{table}")}
+          rename_table :"new_#{table}", table
+          drop_table :"old_#{table}"
+        end
+      end
+
       def serial_primary_key_options
         # redshift doesn't support serial type
         super.merge(serial: false)

data/lib/sequel/extensions/redshift_schema_dumper.rb

@@ -4,6 +4,7 @@ module Sequel
   module Redshift
     module SchemaDumper
       include Sequel::SchemaDumper
+      DISTSTYLE = { 0 => :even, 1 => :key, 8 => :all }.freeze
 
       def dump_table_schema(table, options=OPTS)
         gen = dump_table_generator(table, options)
@@ -14,6 +15,7 @@ module Sequel
 
       def table_options(table, gen, options)
         s = {distkey: table_distkey(table),
+         diststyle: table_diststyle(table),
          sortkeys: table_sortkeys(table),
          sortstyle: table_sortstyle(table),
          ignore_index_errors: (!options[:same_db] && options[:indexes] != false && !gen.indexes.empty?)
@@ -24,6 +26,10 @@ module Sequel
 
       private
 
+      def table_diststyle(table)
+        self[:pg_class].where(relname: table).map(:reldiststyle).first
+      end
+
       def table_distkey(table)
         key = pg_table_def(table).filter(distkey: true).map(:column).first
         key.to_sym if key

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: blue-shift
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.2.0
 platform: ruby
 authors:
 - Gabriel Mansour
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-04-11 00:00:00.000000000 Z
+date: 2016-05-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel