pg_shrink 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 91758d0f6d6dda346377c31c7e26739d44d209ed
-  data.tar.gz: 3456597445c2de0dcd3a6d956673ddac607b331d
+  metadata.gz: 4d4bd5c0b682c29ae81072befc0f5c962319809a
+  data.tar.gz: d9d4aa0ceae5e3c816dfd8cea704a0f5b3f2a404
 SHA512:
-  metadata.gz: bf159b0db40e195a2a1caaabf7fd37e8a98eb415ade6ebaaa4258b7e7f1ccc21ed2f8ff3c45fd6388d56fcd2f0528cf68088b80dcfd3a65cac3283773e40fc3d
-  data.tar.gz: 10a7d63d18ab1a5cf52510b0cd6a4699def302188d53950256e7aa90808da92d80463e5795083ae2f3c5aae0f620f7af80ccb021584f0725091cac02d0e615eb
+  metadata.gz: 9183c4d2b6226d8e0c7b610066333de8f7fa1d731663e5f67241722601336f7749d285b42ef3c84b044c0039b0ba671e0a1085ea622af1d9e086999d1ec92db0
+  data.tar.gz: 0e55a008ff71f575854f2acc1add7dd7a7f7459385eac3fdf63346d94f469ea0c46e86c50b0b627a01249b8d2595c8c3ff375af1d9579d9e4b26acf8e7995b98

data/bin/pg_shrink

@@ -22,11 +22,16 @@ Please make sure you have a Shrinkfile or specify one using -c
     config_desc = '(Optional) Specify configuration file for how to shrink.  ' +
                   '  Will default to Shrinkfile in directory command is being ' +
                   'run from'
-              'time pg_shrink is run.  It will be modified in place.'
     opts.on('-c', '--config Shrinkfile', config_desc) do |config|
       options[:config] = config
     end
 
+    batch_desc = '(Optional) Specify batch size.  Increasing batch size will' +
+                  'increase memory requirements but speed execution.  Default 10000.'
+    opts.on('-b', '--batchsize [10000]', batch_desc)do |batch|
+      options[:batch_size] = batch
+    end
+
     force_desc = 'Force run without confirmation.'
     opts.on('--force', force_desc) do
       options[:force] = true

data/lib/pg_shrink/version.rb

@@ -1,3 +1,3 @@
 module PgShrink
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/lib/pg_shrink.rb

@@ -18,7 +18,8 @@ module PgShrink
     {
       url: nil,
       config: 'Shrinkfile',
-      force: false
+      force: false,
+      batch_size: 10000
     }
   end
 
@@ -53,8 +54,13 @@ module PgShrink
     end
 
     validate_pg_url!(options[:url])
+    batch_size = options[:batch_size].to_i
+    unless batch_size >= 1
+      abort("Batch size must be at least 1.  #{options[:batch_size]} is invalid!")
+    end
 
-    database = Database::Postgres.new(:postgres_url => options[:url])
+    database = Database::Postgres.new(:postgres_url => options[:url],
+                                      :batch_size => batch_size)
 
     database.instance_eval(File.read(options[:config]), options[:config], 1)
 

data/spec/pg_shrink_spec.rb

@@ -13,6 +13,13 @@ describe PgShrink do
     database.connection.disconnect
   end
 
+
+  it "should let you specify a different batch size" do
+    opts = PgSpecHelper.pg_config.merge(:batch_size => 20000)
+    db = PgShrink::Database::Postgres.new(opts)
+    expect(db.batch_size).to eq(20000)
+  end
+
   describe "simple foreign_key setup" do
     before(:all) do
       # Rspec doesn't want you using 'let' defined things in before(:all)
@@ -63,7 +70,7 @@ describe PgShrink do
             expect(PgShrink::Database::Postgres).to receive(:new) do |opts|
                 expect(opts[:postgres_url]).to eq(database.connection_string)
               end.and_return(database)
-            PgShrink.run(config: shrinkfile, url: url, force: true)
+            PgShrink.run(config: shrinkfile, url: url, force: true, batch_size: 5)
           end
         end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_shrink
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Kevin Ball
@@ -14,182 +14,182 @@ dependencies:
   name: pg
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: activesupport
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: sequel
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.6'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.6'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec-nc
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec-mocks
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: guard
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: guard-rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: pry-remote
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: pry-nav
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 description: pg_shrink makes it simple to shrink and sanitize a psql database
@@ -200,8 +200,8 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
-- .gitignore
-- .rspec
+- ".gitignore"
+- ".rspec"
 - Gemfile
 - Guardfile
 - README.md
@@ -237,17 +237,17 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.3
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: ''