abstract_adapter.rb 7.1 KB
Newer Older
D
Initial  
David Heinemeier Hansson 已提交
1
require 'date'
2 3
require 'bigdecimal'
require 'bigdecimal/util'
4
require 'active_support/core_ext/benchmark'
D
Initial  
David Heinemeier Hansson 已提交
5

J
Joshua Peek 已提交
6
# TODO: Autoload these files
7 8 9 10
require 'active_record/connection_adapters/abstract/schema_definitions'
require 'active_record/connection_adapters/abstract/schema_statements'
require 'active_record/connection_adapters/abstract/database_statements'
require 'active_record/connection_adapters/abstract/quoting'
N
Nick 已提交
11
require 'active_record/connection_adapters/abstract/connection_pool'
12
require 'active_record/connection_adapters/abstract/connection_specification'
13
require 'active_record/connection_adapters/abstract/query_cache'
14

D
Initial  
David Heinemeier Hansson 已提交
15 16
module ActiveRecord
  module ConnectionAdapters # :nodoc:
P
Pratik Naik 已提交
17 18 19 20 21 22 23
    # ActiveRecord supports multiple database systems. AbstractAdapter and
    # related classes form the abstraction layer which makes this possible.
    # An AbstractAdapter represents a connection to a database, and provides an
    # abstract interface for database-specific functionality such as establishing
    # a connection, escaping values, building the right SQL fragments for ':offset'
    # and ':limit' options, etc.
    #
D
Initial  
David Heinemeier Hansson 已提交
24
    # All the concrete database adapters follow the interface laid down in this class.
P
Pratik Naik 已提交
25 26
    # ActiveRecord::Base.connection returns an AbstractAdapter object, which
    # you can use.
27
    #
P
Pratik Naik 已提交
28 29
    # Most of the methods in the adapter are useful during migrations. Most
    # notably, the instance methods provided by SchemaStatement are very useful.
D
Initial  
David Heinemeier Hansson 已提交
30
    class AbstractAdapter
31
      include Quoting, DatabaseStatements, SchemaStatements
32
      include QueryCache
33
      include ActiveSupport::Callbacks
34

35
      define_callbacks :checkout, :checkin
36

D
Initial  
David Heinemeier Hansson 已提交
37
      @@row_even = true
38

39
      def initialize(connection, logger = nil) #:nodoc:
40
        @active = nil
D
Initial  
David Heinemeier Hansson 已提交
41 42
        @connection, @logger = connection, logger
        @runtime = 0
43
        @query_cache_enabled = false
44
        @query_cache = {}
D
Initial  
David Heinemeier Hansson 已提交
45 46
      end

47 48
      # Returns the human-readable name of the adapter.  Use mixed case - one
      # can always use downcase if needed.
49
      def adapter_name
50 51
        'Abstract'
      end
52

53
      # Does this adapter support migrations?  Backend specific, as the
54
      # abstract adapter always returns +false+.
55 56
      def supports_migrations?
        false
57
      end
58

59 60 61 62 63 64 65
      # Can this adapter determine the primary key for tables not attached
      # to an ActiveRecord class, such as join tables?  Backend specific, as
      # the abstract adapter always returns +false+.
      def supports_primary_key?
        false
      end

66 67 68 69 70
      # Does this adapter support using DISTINCT within COUNT?  This is +true+
      # for all adapters except sqlite.
      def supports_count_distinct?
        true
      end
71

72 73 74 75 76 77
      # Does this adapter support DDL rollbacks in transactions?  That is, would
      # CREATE TABLE or ALTER TABLE get rolled back by a transaction?  PostgreSQL,
      # SQL Server, and others support this.  MySQL and others do not.
      def supports_ddl_transactions?
        false
      end
78

79 80 81 82 83
      # Does this adapter support savepoints? PostgreSQL and MySQL do, SQLite
      # does not.
      def supports_savepoints?
        false
      end
84

85 86 87 88
      # Should primary key values be selected from their corresponding
      # sequence before the insert statement?  If true, next_sequence_value
      # is called before each insert to set the record's primary key.
      # This is false for all adapters but Firebird.
89
      def prefetch_primary_key?(table_name = nil)
90 91 92
        false
      end

93
      def reset_runtime #:nodoc:
94 95
        rt, @runtime = @runtime, 0
        rt
96
      end
97

98 99
      # QUOTING ==================================================

100 101 102 103 104
      # Override to return the quoted table name. Defaults to column quoting.
      def quote_table_name(name)
        quote_column_name(name)
      end

105 106
      # REFERENTIAL INTEGRITY ====================================

P
Pratik Naik 已提交
107
      # Override to turn off referential integrity while executing <tt>&block</tt>.
108 109 110 111
      def disable_referential_integrity(&block)
        yield
      end

112 113
      # CONNECTION MANAGEMENT ====================================

P
Pratik Naik 已提交
114 115 116
      # Checks whether the connection to the database is still active. This includes
      # checking whether the database is actually capable of responding, i.e. whether
      # the connection isn't stale.
117
      def active?
118
        @active != false
119 120
      end

P
Pratik Naik 已提交
121 122
      # Disconnects from the database if already connected, and establishes a
      # new connection with the database.
123
      def reconnect!
124 125 126
        @active = true
      end

P
Pratik Naik 已提交
127 128
      # Disconnects from the database if already connected. Otherwise, this
      # method does nothing.
129 130
      def disconnect!
        @active = false
131 132
      end

133 134
      # Reset the state of this connection, directing the DBMS to clear
      # transactions and other connection-related server-side state. Usually a
P
Pratik Naik 已提交
135 136 137 138
      # database-dependent operation.
      #
      # The default implementation does nothing; the implementation should be
      # overridden by concrete adapters.
139
      def reset!
140
        # this should be overridden by concrete adapters
141 142
      end

143
      # Returns true if its safe to reload the connection between requests for development mode.
144
      def requires_reloading?
145
        true
146 147
      end

P
Pratik Naik 已提交
148 149 150
      # Checks whether the connection to the database is still active (i.e. not stale).
      # This is done under the hood by calling <tt>active?</tt>. If the connection
      # is no longer active, then this method will reconnect to the database.
151 152
      def verify!(*ignored)
        reconnect! unless active?
153
      end
154

P
Pratik Naik 已提交
155 156 157 158 159 160
      # Provides access to the underlying database driver for this adapter. For
      # example, this method returns a Mysql object in case of MysqlAdapter,
      # and a PGconn object in case of PostgreSQLAdapter.
      #
      # This is useful for when you need to call a proprietary method such as
      # PostgreSQL's lo_* methods.
161 162 163
      def raw_connection
        @connection
      end
164

165 166 167 168 169 170 171 172 173 174 175 176
      def open_transactions
        @open_transactions ||= 0
      end

      def increment_open_transactions
        @open_transactions ||= 0
        @open_transactions += 1
      end

      def decrement_open_transactions
        @open_transactions -= 1
      end
177 178 179 180 181

      def transaction_joinable=(joinable)
        @transaction_joinable = joinable
      end

J
Jonathan Viney 已提交
182 183
      def create_savepoint
      end
184

J
Jonathan Viney 已提交
185 186
      def rollback_to_savepoint
      end
187

J
Jonathan Viney 已提交
188 189
      def release_savepoint
      end
190

J
Jonathan Viney 已提交
191
      def current_savepoint_name
192
        "active_record_#{open_transactions}"
J
Jonathan Viney 已提交
193
      end
194

195
      protected
196

197
        def log(sql, name)
198
          name ||= "SQL"
199
          result = nil
200
          ActiveSupport::Notifications.instrument("sql.active_record",
201
            :sql => sql, :name => name, :connection_id => self.object_id) do
202 203 204
            @runtime += Benchmark.ms { result = yield }
          end
          result
205
        rescue Exception => e
206
          message = "#{e.class.name}: #{e.message}: #{sql}"
207
          @logger.debug message if @logger
208
          raise translate_exception(e, message)
D
Initial  
David Heinemeier Hansson 已提交
209 210
        end

211 212 213 214
        def translate_exception(e, message)
          # override in derived class
          ActiveRecord::StatementInvalid.new(message)
        end
215

216
    end
D
Initial  
David Heinemeier Hansson 已提交
217
  end
218
end