Contents

require 'apartment/adapters/postgresql_adapter'

module Apartment
  module Tenant

    def self.jdbc_postgresql_adapter(config)
      Apartment.use_schemas ?
        Adapters::JDBCPostgresqlSchemaAdapter.new(config) :
        Adapters::JDBCPostgresqlAdapter.new(config)
    end
  end

  module Adapters

    # Default adapter when not using Postgresql Schemas
    class JDBCPostgresqlAdapter < PostgresqlAdapter

    private

      def multi_tenantify_with_tenant_db_name(config, tenant)
        config[:url] = "#{config[:url].gsub(/(\S+)\/.+$/, '\1')}/#{environmentify(tenant)}"
      end

      def create_tenant_command(conn, tenant)
        conn.create_database(environmentify(tenant), { :thisisahack => '' })
      end

      def rescue_from
        ActiveRecord::JDBCError
      end
    end

    # Separate Adapter for Postgresql when using schemas
    class JDBCPostgresqlSchemaAdapter < PostgresqlSchemaAdapter

      #   Set schema search path to new schema
      #
      def connect_to_new(tenant = nil)
        return reset if tenant.nil?
        raise ActiveRecord::StatementInvalid.new("Could not find schema #{tenant}") unless Apartment.connection.all_schemas.include? tenant.to_s

        @current = tenant.to_s
        Apartment.connection.schema_search_path = full_search_path

      rescue ActiveRecord::StatementInvalid, ActiveRecord::JDBCError
        raise TenantNotFound, "One of the following schema(s) is invalid: #{full_search_path}"
      end

    private

      def rescue_from
        ActiveRecord::JDBCError
      end
    end
  end
end

Version data entries

9 entries across 9 versions & 2 rubygems

Version	Path
ros-apartment-2.3.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb
ros-apartment-2.3.0.alpha2	lib/apartment/adapters/jdbc_postgresql_adapter.rb
ros-apartment-2.3.0.alpha1	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-2.2.1	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-2.2.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-2.1.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-2.0.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-1.2.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb
apartment-1.1.0	lib/apartment/adapters/jdbc_postgresql_adapter.rb