lib/rom/sql/gateway.rb in rom-sql-0.6.1 vs lib/rom/sql/gateway.rb in rom-sql-0.7.0.beta1

@@ -17,10 +17,14 @@
     # @api public
     class Gateway < ROM::Gateway
       include Options
       include Migration
 
+      class << self
+        attr_accessor :instance
+      end
+
       # Return optionally configured logger
       #
       # @return [Object] logger
       #
       # @api public
@@ -61,16 +65,18 @@
       #   gateway = ROM::SQL::Gateway.new(DB)
       #
       # @api public
       def initialize(uri, options = {})
         repo_options = self.class.option_definitions.names
-        conn_options = options.reject { |k,_| repo_options.include?(k) }
+        conn_options = options.reject { |k, _| repo_options.include?(k) }
 
         @connection = connect(uri, conn_options)
         @schema = connection.tables
         add_extensions(Array(options[:extensions])) if options[:extensions]
 
-        super(uri, options.reject { |k,_| conn_options.keys.include?(k) })
+        super(uri, options.reject { |k, _| conn_options.keys.include?(k) })
+
+        self.class.instance = self
       end
 
       # Disconnect from database
       #
       # @api public