Allow a list of schemas when switching using schemas

Author: Ryan Brunner

README.md

@@ -84,6 +84,14 @@ would not allow a full new database to be created.
 
 One can optionally use the full database creation instead if they want, though this is not recommended
 
+When using schemas, you can also pass in a list of schemas if desired. Any tables defined in a schema earlier in the chain will be referenced first, so this is only useful if you have a schema with only some of the tables defined:
+
+```ruby
+Apartment::Tenant.switch(['tenant_1', 'tenant_2']) do
+  # ...
+end
+```
+
 ### Switching Tenants
 
 To switch tenants using Apartment, use the following command:

lib/apartment/adapters/jdbc_postgresql_adapter.rb

@@ -38,9 +38,10 @@ class JDBCPostgresqlSchemaAdapter < PostgresqlSchemaAdapter
       #
       def connect_to_new(tenant = nil)
         return reset if tenant.nil?
+        raise ActiveRecord::StatementInvalid, "Could not find schema #{tenant}" unless schema_exists?(tenant)
 
-        tenant = tenant.to_s
-        raise ActiveRecord::StatementInvalid, "Could not find schema #{tenant}" unless tenant_exists?(tenant)
+        @current = tenant.is_a?(Array) ? tenant.map(&:to_s) : tenant.to_s
+        Apartment.connection.schema_search_path = full_search_path
 
         @current = tenant
         Apartment.connection.schema_search_path = full_search_path

lib/apartment/adapters/postgresql_adapter.rb

@@ -72,11 +72,9 @@ def drop_command(conn, tenant)
       #
       def connect_to_new(tenant = nil)
         return reset if tenant.nil?
+        raise ActiveRecord::StatementInvalid, "Could not find schema #{tenant}" unless schema_exists?(tenant)
 
-        tenant = tenant.to_s
-        raise ActiveRecord::StatementInvalid, "Could not find schema #{tenant}" unless tenant_exists?(tenant)
-
-        @current = tenant
+        @current = tenant.is_a?(Array) ? tenant.map(&:to_s) : tenant.to_s
         Apartment.connection.schema_search_path = full_search_path
 
         # When the PostgreSQL version is < 9.3,
@@ -148,6 +146,9 @@ def reset_sequence_names
           # unreproduceable error we're checking before trying to remove it
           c.remove_instance_variable :@sequence_name if c.instance_variable_defined?(:@sequence_name)
         end
+
+      def schema_exists?(schemas)
+        [*schemas].all? { |schema| Apartment.connection.schema_exists?(schema.to_s) }
       end
     end
 

spec/examples/schema_adapter_examples.rb

@@ -127,6 +127,13 @@
       expect(connection.schema_search_path).to start_with %("#{public_schema}")
       expect(User.sequence_name).to eq "#{public_schema}.#{User.table_name}_id_seq"
     end
+
+    it "allows a list of schemas" do
+      subject.switch([schema1, schema2]) do
+        expect(connection.schema_search_path).to include %{"#{schema1}"}
+        expect(connection.schema_search_path).to include %{"#{schema2}"}
+      end
+    end
   end
 
   describe '#reset' do