-
Notifications
You must be signed in to change notification settings - Fork 21.6k
/
mysql2_adapter.rb
196 lines (156 loc) · 5.49 KB
/
mysql2_adapter.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
# frozen_string_literal: true
require "active_record/connection_adapters/abstract_mysql_adapter"
require "active_record/connection_adapters/mysql2/database_statements"
gem "mysql2", "~> 0.5"
require "mysql2"
module ActiveRecord
module ConnectionAdapters
# = Active Record MySQL2 Adapter
class Mysql2Adapter < AbstractMysqlAdapter
ER_BAD_DB_ERROR = 1049
ER_DBACCESS_DENIED_ERROR = 1044
ER_ACCESS_DENIED_ERROR = 1045
ER_CONN_HOST_ERROR = 2003
ER_UNKNOWN_HOST_ERROR = 2005
ADAPTER_NAME = "Mysql2"
include Mysql2::DatabaseStatements
class << self
def new_client(config)
::Mysql2::Client.new(config)
rescue ::Mysql2::Error => error
case error.error_number
when ER_BAD_DB_ERROR
raise ActiveRecord::NoDatabaseError.db_error(config[:database])
when ER_DBACCESS_DENIED_ERROR, ER_ACCESS_DENIED_ERROR
raise ActiveRecord::DatabaseConnectionError.username_error(config[:username])
when ER_CONN_HOST_ERROR, ER_UNKNOWN_HOST_ERROR
raise ActiveRecord::DatabaseConnectionError.hostname_error(config[:host])
else
raise ActiveRecord::ConnectionNotEstablished, error.message
end
end
private
def initialize_type_map(m)
super
m.register_type(%r(char)i) do |sql_type|
limit = extract_limit(sql_type)
Type.lookup(:string, adapter: :mysql2, limit: limit)
end
m.register_type %r(^enum)i, Type.lookup(:string, adapter: :mysql2)
m.register_type %r(^set)i, Type.lookup(:string, adapter: :mysql2)
end
end
TYPE_MAP = Type::TypeMap.new.tap { |m| initialize_type_map(m) }
def initialize(...)
super
@affected_rows_before_warnings = nil
@config[:flags] ||= 0
if @config[:flags].kind_of? Array
@config[:flags].push "FOUND_ROWS"
else
@config[:flags] |= ::Mysql2::Client::FOUND_ROWS
end
@connection_parameters ||= @config
end
def supports_json?
!mariadb? && database_version >= "5.7.8"
end
def supports_comments?
true
end
def supports_comments_in_create?
true
end
def supports_savepoints?
true
end
def savepoint_errors_invalidate_transactions?
true
end
def supports_lazy_transactions?
true
end
# HELPER METHODS ===========================================
def error_number(exception)
exception.error_number if exception.respond_to?(:error_number)
end
#--
# CONNECTION MANAGEMENT ====================================
#++
def connected?
!(@raw_connection.nil? || @raw_connection.closed?)
end
def active?
connected? && @lock.synchronize { @raw_connection&.ping } || false
end
alias :reset! :reconnect!
# Disconnects from the database if already connected.
# Otherwise, this method does nothing.
def disconnect!
@lock.synchronize do
super
@raw_connection&.close
@raw_connection = nil
end
end
def discard! # :nodoc:
@lock.synchronize do
super
@raw_connection&.automatic_close = false
@raw_connection = nil
end
end
private
def text_type?(type)
TYPE_MAP.lookup(type).is_a?(Type::String) || TYPE_MAP.lookup(type).is_a?(Type::Text)
end
def connect
@raw_connection = self.class.new_client(@connection_parameters)
rescue ConnectionNotEstablished => ex
raise ex.set_pool(@pool)
end
def reconnect
@lock.synchronize do
@raw_connection&.close
@raw_connection = nil
connect
end
end
def configure_connection
@raw_connection.query_options[:as] = :array
@raw_connection.query_options[:database_timezone] = default_timezone
super
end
def full_version
database_version.full_version_string
end
def get_full_version
any_raw_connection.server_info[:version]
end
def translate_exception(exception, message:, sql:, binds:)
if exception.is_a?(::Mysql2::Error::TimeoutError) && !exception.error_number
ActiveRecord::AdapterTimeout.new(message, sql: sql, binds: binds, connection_pool: @pool)
elsif exception.is_a?(::Mysql2::Error::ConnectionError)
if exception.message.match?(/MySQL client is not connected/i)
ActiveRecord::ConnectionNotEstablished.new(exception, connection_pool: @pool)
else
ActiveRecord::ConnectionFailed.new(message, sql: sql, binds: binds, connection_pool: @pool)
end
else
super
end
end
def default_prepared_statements
false
end
ActiveRecord::Type.register(:immutable_string, adapter: :mysql2) do |_, **args|
Type::ImmutableString.new(true: "1", false: "0", **args)
end
ActiveRecord::Type.register(:string, adapter: :mysql2) do |_, **args|
Type::String.new(true: "1", false: "0", **args)
end
ActiveRecord::Type.register(:unsigned_integer, Type::UnsignedInteger, adapter: :mysql2)
end
ActiveSupport.run_load_hooks(:active_record_mysql2adapter, Mysql2Adapter)
end
end