ankane · Aug 1, 2018 · Aug 1, 2018
Showing with 37 additions and 1 deletion.

+1 −0 .gitignore

+19 −1 lib/pgsync/client.rb

+6 −0 lib/pgsync/data_source.rb

+11 −0 lib/pgsync/table_list.rb
diff --git a/.gitignore b/.gitignore
@@ -8,3 +8,4 @@
 /spec/reports/
 /tmp/
 /.pgsync.yml
+.idea
diff --git a/lib/pgsync/client.rb b/lib/pgsync/client.rb
@@ -79,6 +79,15 @@ def sync(args, opts)
 
         pretty_list list_items
       else
+        if opts[:tenant_schema_first] || opts[:tenant_schema_only]
+          if opts[:preserve]
+            raise PgSync::Error, "Cannot use --preserve with --schema-first or --schema-only"
+          end
+
+          log "* Dumping Tenant schema schema"
+          sync_tenant_schema(source, destination, ['public', 'foodcellar_47th_rd'])
+        end
+
         if opts[:schema_first] || opts[:schema_only]
           if opts[:preserve]
             raise PgSync::Error, "Cannot use --preserve with --schema-first or --schema-only"
@@ -88,7 +97,7 @@ def sync(args, opts)
           sync_schema(source, destination, tables)
         end
 
-        unless opts[:schema_only]
+        unless opts[:schema_only] || opts[:tenant_schema_only]
           confirm_tables_exist(destination, tables, "destination")
 
           in_parallel(tables) do |table, table_opts|
@@ -151,6 +160,12 @@ def sync_schema(source, destination, tables)
       system("#{dump_command} | #{restore_command}")
     end
 
+    def sync_tenant_schema(source, destination, tenants)
+      dump_command = source.dump_tenant_command(tenants)
+      restore_command = destination.restore_command
+      system("#{dump_command} | #{restore_command}")
+    end
+
     def parse_args(args)
       opts = Slop.parse(args) do |o|
         o.banner = %{Usage:
@@ -159,6 +174,7 @@ def parse_args(args)
 Options:}
         o.string "-d", "--db", "database"
         o.string "-t", "--tables", "tables to sync"
+        o.string "-T", "--tenants", "tenants to sync"
         o.string "-g", "--groups", "groups to sync"
         o.string "--schemas", "schemas to sync"
         o.string "--from", "source"
@@ -176,6 +192,8 @@ def parse_args(args)
         o.boolean "--truncate", "truncate existing rows", default: false
         o.boolean "--schema-first", "schema first", default: false
         o.boolean "--schema-only", "schema only", default: false
+        o.boolean "--tenant-schema-only", "tenant schema only", default: false
+        o.boolean "--tenant-schema-first", "tenant schema first", default: false
         o.boolean "--all-schemas", "all schemas", default: false
         o.boolean "--no-rules", "do not apply data rules", default: false
         o.boolean "--setup", "setup", default: false

diff --git a/lib/pgsync/data_source.rb b/lib/pgsync/data_source.rb
@@ -112,9 +112,15 @@ def close
 
     def dump_command(tables)
       tables = tables.keys.map { |t| "-t #{Shellwords.escape(quote_ident_full(t))}" }.join(" ")
+      # tables = ""
       "pg_dump -Fc --verbose --schema-only --no-owner --no-acl #{tables} -d #{@url}"
     end
 
+    def dump_tenant_command(tenants)
+      tenants = tenants.map { |t| "-n #{Shellwords.escape(quote_ident_full(t))}" }.join(" ")
+      "pg_dump -Fc --verbose --schema-only --no-owner --no-acl #{tenants} -d #{@url}"
+    end
+
     def restore_command
       psql_version = `psql --version`.lines[0].chomp.split(" ")[-1].split(/[^\d.]/)[0]
       if_exists = Gem::Version.new(psql_version) >= Gem::Version.new("9.4.0")

diff --git a/lib/pgsync/table_list.rb b/lib/pgsync/table_list.rb
@@ -11,6 +11,17 @@ def initialize(args, options, source, config)
 
     def tables
       tables = nil
+      if opts[:tenants]
+        tables ||= Hash.new { |hash, key| hash[key] = {} }
+        specified_groups = to_arr(opts[:tenants])
+        specified_groups.map do |tag|
+          if (t = (config["tenants"] || {})[tag])
+            t.each { |tables_to_sync| tables["#{tag}.#{tables_to_sync}"] = {} }
+          else
+            raise PgSync::Error, "Tenant not found: #{tag}"
+          end
+        end
+      end
 
       if opts[:groups]
         tables ||= Hash.new { |hash, key| hash[key] = {} }
-Original file line number
+Diff line change
@@ @@ -8,3 +8,4 @@ @@
     /spec/reports/
     /tmp/
     /.pgsync.yml
+    .idea