activerecord 4.2.0 → 5.0.0

Sign up to get free protection for your applications and to get access to all the features.

Potentially problematic release.


This version of activerecord might be problematic. Click here for more details.

Files changed (249) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +1537 -789
  3. data/MIT-LICENSE +2 -2
  4. data/README.rdoc +7 -8
  5. data/examples/performance.rb +2 -3
  6. data/examples/simple.rb +0 -1
  7. data/lib/active_record/aggregations.rb +37 -23
  8. data/lib/active_record/association_relation.rb +16 -3
  9. data/lib/active_record/associations/alias_tracker.rb +19 -16
  10. data/lib/active_record/associations/association.rb +23 -9
  11. data/lib/active_record/associations/association_scope.rb +74 -102
  12. data/lib/active_record/associations/belongs_to_association.rb +26 -29
  13. data/lib/active_record/associations/builder/association.rb +28 -34
  14. data/lib/active_record/associations/builder/belongs_to.rb +43 -18
  15. data/lib/active_record/associations/builder/collection_association.rb +12 -20
  16. data/lib/active_record/associations/builder/has_and_belongs_to_many.rb +22 -15
  17. data/lib/active_record/associations/builder/has_many.rb +4 -4
  18. data/lib/active_record/associations/builder/has_one.rb +11 -6
  19. data/lib/active_record/associations/builder/singular_association.rb +3 -10
  20. data/lib/active_record/associations/collection_association.rb +61 -33
  21. data/lib/active_record/associations/collection_proxy.rb +81 -35
  22. data/lib/active_record/associations/foreign_association.rb +11 -0
  23. data/lib/active_record/associations/has_many_association.rb +21 -57
  24. data/lib/active_record/associations/has_many_through_association.rb +15 -45
  25. data/lib/active_record/associations/has_one_association.rb +13 -5
  26. data/lib/active_record/associations/join_dependency/join_association.rb +20 -8
  27. data/lib/active_record/associations/join_dependency.rb +37 -21
  28. data/lib/active_record/associations/preloader/association.rb +51 -53
  29. data/lib/active_record/associations/preloader/collection_association.rb +0 -6
  30. data/lib/active_record/associations/preloader/has_many_through.rb +1 -1
  31. data/lib/active_record/associations/preloader/has_one.rb +0 -8
  32. data/lib/active_record/associations/preloader/through_association.rb +27 -14
  33. data/lib/active_record/associations/preloader.rb +18 -8
  34. data/lib/active_record/associations/singular_association.rb +8 -8
  35. data/lib/active_record/associations/through_association.rb +22 -9
  36. data/lib/active_record/associations.rb +321 -212
  37. data/lib/active_record/attribute/user_provided_default.rb +28 -0
  38. data/lib/active_record/attribute.rb +79 -15
  39. data/lib/active_record/attribute_assignment.rb +20 -141
  40. data/lib/active_record/attribute_decorators.rb +6 -5
  41. data/lib/active_record/attribute_methods/before_type_cast.rb +6 -1
  42. data/lib/active_record/attribute_methods/dirty.rb +51 -81
  43. data/lib/active_record/attribute_methods/primary_key.rb +2 -2
  44. data/lib/active_record/attribute_methods/query.rb +2 -2
  45. data/lib/active_record/attribute_methods/read.rb +31 -59
  46. data/lib/active_record/attribute_methods/serialization.rb +13 -16
  47. data/lib/active_record/attribute_methods/time_zone_conversion.rb +65 -14
  48. data/lib/active_record/attribute_methods/write.rb +14 -38
  49. data/lib/active_record/attribute_methods.rb +70 -45
  50. data/lib/active_record/attribute_mutation_tracker.rb +70 -0
  51. data/lib/active_record/attribute_set/builder.rb +37 -15
  52. data/lib/active_record/attribute_set.rb +34 -3
  53. data/lib/active_record/attributes.rb +199 -73
  54. data/lib/active_record/autosave_association.rb +73 -25
  55. data/lib/active_record/base.rb +35 -27
  56. data/lib/active_record/callbacks.rb +39 -43
  57. data/lib/active_record/coders/json.rb +1 -1
  58. data/lib/active_record/coders/yaml_column.rb +20 -8
  59. data/lib/active_record/collection_cache_key.rb +40 -0
  60. data/lib/active_record/connection_adapters/abstract/connection_pool.rb +457 -181
  61. data/lib/active_record/connection_adapters/abstract/database_limits.rb +3 -3
  62. data/lib/active_record/connection_adapters/abstract/database_statements.rb +83 -59
  63. data/lib/active_record/connection_adapters/abstract/query_cache.rb +3 -3
  64. data/lib/active_record/connection_adapters/abstract/quoting.rb +74 -9
  65. data/lib/active_record/connection_adapters/abstract/savepoints.rb +4 -4
  66. data/lib/active_record/connection_adapters/abstract/schema_creation.rb +61 -39
  67. data/lib/active_record/connection_adapters/abstract/schema_definitions.rb +246 -185
  68. data/lib/active_record/connection_adapters/abstract/schema_dumper.rb +72 -17
  69. data/lib/active_record/connection_adapters/abstract/schema_statements.rb +438 -136
  70. data/lib/active_record/connection_adapters/abstract/transaction.rb +53 -40
  71. data/lib/active_record/connection_adapters/abstract_adapter.rb +166 -66
  72. data/lib/active_record/connection_adapters/abstract_mysql_adapter.rb +429 -335
  73. data/lib/active_record/connection_adapters/column.rb +28 -43
  74. data/lib/active_record/connection_adapters/connection_specification.rb +15 -27
  75. data/lib/active_record/connection_adapters/determine_if_preparable_visitor.rb +22 -0
  76. data/lib/active_record/connection_adapters/mysql/column.rb +50 -0
  77. data/lib/active_record/connection_adapters/mysql/database_statements.rb +125 -0
  78. data/lib/active_record/connection_adapters/mysql/explain_pretty_printer.rb +70 -0
  79. data/lib/active_record/connection_adapters/mysql/quoting.rb +51 -0
  80. data/lib/active_record/connection_adapters/mysql/schema_creation.rb +67 -0
  81. data/lib/active_record/connection_adapters/mysql/schema_definitions.rb +93 -0
  82. data/lib/active_record/connection_adapters/mysql/schema_dumper.rb +54 -0
  83. data/lib/active_record/connection_adapters/mysql/type_metadata.rb +32 -0
  84. data/lib/active_record/connection_adapters/mysql2_adapter.rb +26 -177
  85. data/lib/active_record/connection_adapters/postgresql/column.rb +5 -10
  86. data/lib/active_record/connection_adapters/postgresql/database_statements.rb +11 -73
  87. data/lib/active_record/connection_adapters/postgresql/explain_pretty_printer.rb +42 -0
  88. data/lib/active_record/connection_adapters/postgresql/oid/array.rb +27 -56
  89. data/lib/active_record/connection_adapters/postgresql/oid/bit.rb +2 -2
  90. data/lib/active_record/connection_adapters/postgresql/oid/bytea.rb +2 -1
  91. data/lib/active_record/connection_adapters/postgresql/oid/cidr.rb +3 -1
  92. data/lib/active_record/connection_adapters/postgresql/oid/date_time.rb +7 -13
  93. data/lib/active_record/connection_adapters/postgresql/oid/enum.rb +3 -1
  94. data/lib/active_record/connection_adapters/postgresql/oid/hstore.rb +3 -3
  95. data/lib/active_record/connection_adapters/postgresql/oid/json.rb +1 -26
  96. data/lib/active_record/connection_adapters/postgresql/oid/jsonb.rb +2 -2
  97. data/lib/active_record/connection_adapters/postgresql/oid/money.rb +0 -4
  98. data/lib/active_record/connection_adapters/postgresql/oid/point.rb +4 -4
  99. data/lib/active_record/connection_adapters/postgresql/oid/rails_5_1_point.rb +50 -0
  100. data/lib/active_record/connection_adapters/postgresql/oid/range.rb +31 -17
  101. data/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb +17 -5
  102. data/lib/active_record/connection_adapters/postgresql/oid/uuid.rb +2 -2
  103. data/lib/active_record/connection_adapters/postgresql/oid/vector.rb +1 -1
  104. data/lib/active_record/connection_adapters/postgresql/oid/xml.rb +1 -1
  105. data/lib/active_record/connection_adapters/postgresql/oid.rb +1 -6
  106. data/lib/active_record/connection_adapters/postgresql/quoting.rb +26 -18
  107. data/lib/active_record/connection_adapters/postgresql/referential_integrity.rb +29 -10
  108. data/lib/active_record/connection_adapters/postgresql/schema_definitions.rb +107 -79
  109. data/lib/active_record/connection_adapters/postgresql/schema_dumper.rb +47 -0
  110. data/lib/active_record/connection_adapters/postgresql/schema_statements.rb +248 -154
  111. data/lib/active_record/connection_adapters/postgresql/type_metadata.rb +35 -0
  112. data/lib/active_record/connection_adapters/postgresql_adapter.rb +258 -170
  113. data/lib/active_record/connection_adapters/schema_cache.rb +36 -23
  114. data/lib/active_record/connection_adapters/sql_type_metadata.rb +32 -0
  115. data/lib/active_record/connection_adapters/sqlite3/explain_pretty_printer.rb +19 -0
  116. data/lib/active_record/connection_adapters/sqlite3/quoting.rb +48 -0
  117. data/lib/active_record/connection_adapters/sqlite3/schema_creation.rb +22 -0
  118. data/lib/active_record/connection_adapters/sqlite3_adapter.rb +150 -209
  119. data/lib/active_record/connection_adapters/statement_pool.rb +31 -12
  120. data/lib/active_record/connection_handling.rb +38 -15
  121. data/lib/active_record/core.rb +109 -114
  122. data/lib/active_record/counter_cache.rb +14 -25
  123. data/lib/active_record/dynamic_matchers.rb +1 -20
  124. data/lib/active_record/enum.rb +115 -79
  125. data/lib/active_record/errors.rb +88 -48
  126. data/lib/active_record/explain_registry.rb +1 -1
  127. data/lib/active_record/explain_subscriber.rb +2 -2
  128. data/lib/active_record/fixture_set/file.rb +26 -5
  129. data/lib/active_record/fixtures.rb +84 -46
  130. data/lib/active_record/gem_version.rb +2 -2
  131. data/lib/active_record/inheritance.rb +32 -40
  132. data/lib/active_record/integration.rb +4 -4
  133. data/lib/active_record/internal_metadata.rb +56 -0
  134. data/lib/active_record/legacy_yaml_adapter.rb +46 -0
  135. data/lib/active_record/locale/en.yml +3 -2
  136. data/lib/active_record/locking/optimistic.rb +27 -25
  137. data/lib/active_record/locking/pessimistic.rb +1 -1
  138. data/lib/active_record/log_subscriber.rb +43 -21
  139. data/lib/active_record/migration/command_recorder.rb +59 -18
  140. data/lib/active_record/migration/compatibility.rb +126 -0
  141. data/lib/active_record/migration.rb +372 -114
  142. data/lib/active_record/model_schema.rb +128 -38
  143. data/lib/active_record/nested_attributes.rb +71 -32
  144. data/lib/active_record/no_touching.rb +1 -1
  145. data/lib/active_record/null_relation.rb +16 -8
  146. data/lib/active_record/persistence.rb +124 -80
  147. data/lib/active_record/query_cache.rb +15 -18
  148. data/lib/active_record/querying.rb +10 -9
  149. data/lib/active_record/railtie.rb +28 -19
  150. data/lib/active_record/railties/controller_runtime.rb +1 -1
  151. data/lib/active_record/railties/databases.rake +67 -51
  152. data/lib/active_record/readonly_attributes.rb +1 -1
  153. data/lib/active_record/reflection.rb +318 -139
  154. data/lib/active_record/relation/batches/batch_enumerator.rb +67 -0
  155. data/lib/active_record/relation/batches.rb +139 -34
  156. data/lib/active_record/relation/calculations.rb +80 -102
  157. data/lib/active_record/relation/delegation.rb +7 -20
  158. data/lib/active_record/relation/finder_methods.rb +167 -97
  159. data/lib/active_record/relation/from_clause.rb +32 -0
  160. data/lib/active_record/relation/merger.rb +38 -41
  161. data/lib/active_record/relation/predicate_builder/array_handler.rb +12 -16
  162. data/lib/active_record/relation/predicate_builder/association_query_handler.rb +88 -0
  163. data/lib/active_record/relation/predicate_builder/base_handler.rb +17 -0
  164. data/lib/active_record/relation/predicate_builder/basic_object_handler.rb +17 -0
  165. data/lib/active_record/relation/predicate_builder/class_handler.rb +27 -0
  166. data/lib/active_record/relation/predicate_builder/polymorphic_array_handler.rb +57 -0
  167. data/lib/active_record/relation/predicate_builder/range_handler.rb +33 -0
  168. data/lib/active_record/relation/predicate_builder/relation_handler.rb +1 -1
  169. data/lib/active_record/relation/predicate_builder.rb +124 -82
  170. data/lib/active_record/relation/query_attribute.rb +19 -0
  171. data/lib/active_record/relation/query_methods.rb +323 -257
  172. data/lib/active_record/relation/record_fetch_warning.rb +49 -0
  173. data/lib/active_record/relation/spawn_methods.rb +11 -10
  174. data/lib/active_record/relation/where_clause.rb +174 -0
  175. data/lib/active_record/relation/where_clause_factory.rb +38 -0
  176. data/lib/active_record/relation.rb +176 -115
  177. data/lib/active_record/result.rb +4 -3
  178. data/lib/active_record/runtime_registry.rb +1 -1
  179. data/lib/active_record/sanitization.rb +95 -66
  180. data/lib/active_record/schema.rb +26 -22
  181. data/lib/active_record/schema_dumper.rb +62 -38
  182. data/lib/active_record/schema_migration.rb +11 -17
  183. data/lib/active_record/scoping/default.rb +24 -9
  184. data/lib/active_record/scoping/named.rb +49 -28
  185. data/lib/active_record/scoping.rb +32 -15
  186. data/lib/active_record/secure_token.rb +38 -0
  187. data/lib/active_record/serialization.rb +2 -4
  188. data/lib/active_record/statement_cache.rb +16 -14
  189. data/lib/active_record/store.rb +8 -3
  190. data/lib/active_record/suppressor.rb +58 -0
  191. data/lib/active_record/table_metadata.rb +68 -0
  192. data/lib/active_record/tasks/database_tasks.rb +59 -42
  193. data/lib/active_record/tasks/mysql_database_tasks.rb +32 -26
  194. data/lib/active_record/tasks/postgresql_database_tasks.rb +29 -9
  195. data/lib/active_record/tasks/sqlite_database_tasks.rb +5 -1
  196. data/lib/active_record/timestamp.rb +20 -9
  197. data/lib/active_record/touch_later.rb +58 -0
  198. data/lib/active_record/transactions.rb +159 -67
  199. data/lib/active_record/type/adapter_specific_registry.rb +130 -0
  200. data/lib/active_record/type/date.rb +2 -41
  201. data/lib/active_record/type/date_time.rb +2 -38
  202. data/lib/active_record/type/hash_lookup_type_map.rb +8 -2
  203. data/lib/active_record/type/internal/abstract_json.rb +29 -0
  204. data/lib/active_record/type/internal/timezone.rb +15 -0
  205. data/lib/active_record/type/serialized.rb +21 -14
  206. data/lib/active_record/type/time.rb +10 -16
  207. data/lib/active_record/type/type_map.rb +4 -4
  208. data/lib/active_record/type.rb +66 -17
  209. data/lib/active_record/type_caster/connection.rb +29 -0
  210. data/lib/active_record/type_caster/map.rb +19 -0
  211. data/lib/active_record/type_caster.rb +7 -0
  212. data/lib/active_record/validations/absence.rb +23 -0
  213. data/lib/active_record/validations/associated.rb +10 -3
  214. data/lib/active_record/validations/length.rb +24 -0
  215. data/lib/active_record/validations/presence.rb +11 -12
  216. data/lib/active_record/validations/uniqueness.rb +29 -18
  217. data/lib/active_record/validations.rb +33 -32
  218. data/lib/active_record.rb +9 -2
  219. data/lib/rails/generators/active_record/migration/migration_generator.rb +7 -4
  220. data/lib/rails/generators/active_record/migration/templates/create_table_migration.rb +8 -6
  221. data/lib/rails/generators/active_record/migration/templates/migration.rb +8 -7
  222. data/lib/rails/generators/active_record/migration.rb +7 -0
  223. data/lib/rails/generators/active_record/model/model_generator.rb +32 -15
  224. data/lib/rails/generators/active_record/model/templates/application_record.rb +5 -0
  225. data/lib/rails/generators/active_record/model/templates/model.rb +3 -0
  226. metadata +60 -34
  227. data/lib/active_record/connection_adapters/mysql_adapter.rb +0 -491
  228. data/lib/active_record/connection_adapters/postgresql/array_parser.rb +0 -93
  229. data/lib/active_record/connection_adapters/postgresql/oid/date.rb +0 -11
  230. data/lib/active_record/connection_adapters/postgresql/oid/float.rb +0 -21
  231. data/lib/active_record/connection_adapters/postgresql/oid/infinity.rb +0 -13
  232. data/lib/active_record/connection_adapters/postgresql/oid/integer.rb +0 -11
  233. data/lib/active_record/connection_adapters/postgresql/oid/time.rb +0 -11
  234. data/lib/active_record/serializers/xml_serializer.rb +0 -193
  235. data/lib/active_record/type/big_integer.rb +0 -13
  236. data/lib/active_record/type/binary.rb +0 -50
  237. data/lib/active_record/type/boolean.rb +0 -30
  238. data/lib/active_record/type/decimal.rb +0 -40
  239. data/lib/active_record/type/decimal_without_scale.rb +0 -11
  240. data/lib/active_record/type/decorator.rb +0 -14
  241. data/lib/active_record/type/float.rb +0 -19
  242. data/lib/active_record/type/integer.rb +0 -55
  243. data/lib/active_record/type/mutable.rb +0 -16
  244. data/lib/active_record/type/numeric.rb +0 -36
  245. data/lib/active_record/type/string.rb +0 -36
  246. data/lib/active_record/type/text.rb +0 -11
  247. data/lib/active_record/type/time_value.rb +0 -38
  248. data/lib/active_record/type/unsigned_integer.rb +0 -15
  249. data/lib/active_record/type/value.rb +0 -101
@@ -1,491 +0,0 @@
1
- require 'active_record/connection_adapters/abstract_mysql_adapter'
2
- require 'active_record/connection_adapters/statement_pool'
3
- require 'active_support/core_ext/hash/keys'
4
-
5
- gem 'mysql', '~> 2.9'
6
- require 'mysql'
7
-
8
- class Mysql
9
- class Time
10
- def to_date
11
- Date.new(year, month, day)
12
- end
13
- end
14
- class Stmt; include Enumerable end
15
- class Result; include Enumerable end
16
- end
17
-
18
- module ActiveRecord
19
- module ConnectionHandling # :nodoc:
20
- # Establishes a connection to the database that's used by all Active Record objects.
21
- def mysql_connection(config)
22
- config = config.symbolize_keys
23
- host = config[:host]
24
- port = config[:port]
25
- socket = config[:socket]
26
- username = config[:username] ? config[:username].to_s : 'root'
27
- password = config[:password].to_s
28
- database = config[:database]
29
-
30
- mysql = Mysql.init
31
- mysql.ssl_set(config[:sslkey], config[:sslcert], config[:sslca], config[:sslcapath], config[:sslcipher]) if config[:sslca] || config[:sslkey]
32
-
33
- default_flags = Mysql.const_defined?(:CLIENT_MULTI_RESULTS) ? Mysql::CLIENT_MULTI_RESULTS : 0
34
- default_flags |= Mysql::CLIENT_FOUND_ROWS if Mysql.const_defined?(:CLIENT_FOUND_ROWS)
35
- options = [host, username, password, database, port, socket, default_flags]
36
- ConnectionAdapters::MysqlAdapter.new(mysql, logger, options, config)
37
- rescue Mysql::Error => error
38
- if error.message.include?("Unknown database")
39
- raise ActiveRecord::NoDatabaseError.new(error.message, error)
40
- else
41
- raise
42
- end
43
- end
44
- end
45
-
46
- module ConnectionAdapters
47
- # The MySQL adapter will work with both Ruby/MySQL, which is a Ruby-based MySQL adapter that comes bundled with Active Record, and with
48
- # the faster C-based MySQL/Ruby adapter (available both as a gem and from http://www.tmtm.org/en/mysql/ruby/).
49
- #
50
- # Options:
51
- #
52
- # * <tt>:host</tt> - Defaults to "localhost".
53
- # * <tt>:port</tt> - Defaults to 3306.
54
- # * <tt>:socket</tt> - Defaults to "/tmp/mysql.sock".
55
- # * <tt>:username</tt> - Defaults to "root"
56
- # * <tt>:password</tt> - Defaults to nothing.
57
- # * <tt>:database</tt> - The name of the database. No default, must be provided.
58
- # * <tt>:encoding</tt> - (Optional) Sets the client encoding by executing "SET NAMES <encoding>" after connection.
59
- # * <tt>:reconnect</tt> - Defaults to false (See MySQL documentation: http://dev.mysql.com/doc/refman/5.0/en/auto-reconnect.html).
60
- # * <tt>:strict</tt> - Defaults to true. Enable STRICT_ALL_TABLES. (See MySQL documentation: http://dev.mysql.com/doc/refman/5.0/en/server-sql-mode.html)
61
- # * <tt>:variables</tt> - (Optional) A hash session variables to send as <tt>SET @@SESSION.key = value</tt> on each database connection. Use the value +:default+ to set a variable to its DEFAULT value. (See MySQL documentation: http://dev.mysql.com/doc/refman/5.0/en/set-statement.html).
62
- # * <tt>:sslca</tt> - Necessary to use MySQL with an SSL connection.
63
- # * <tt>:sslkey</tt> - Necessary to use MySQL with an SSL connection.
64
- # * <tt>:sslcert</tt> - Necessary to use MySQL with an SSL connection.
65
- # * <tt>:sslcapath</tt> - Necessary to use MySQL with an SSL connection.
66
- # * <tt>:sslcipher</tt> - Necessary to use MySQL with an SSL connection.
67
- #
68
- class MysqlAdapter < AbstractMysqlAdapter
69
- ADAPTER_NAME = 'MySQL'.freeze
70
-
71
- class StatementPool < ConnectionAdapters::StatementPool
72
- def initialize(connection, max = 1000)
73
- super
74
- @cache = Hash.new { |h,pid| h[pid] = {} }
75
- end
76
-
77
- def each(&block); cache.each(&block); end
78
- def key?(key); cache.key?(key); end
79
- def [](key); cache[key]; end
80
- def length; cache.length; end
81
- def delete(key); cache.delete(key); end
82
-
83
- def []=(sql, key)
84
- while @max <= cache.size
85
- cache.shift.last[:stmt].close
86
- end
87
- cache[sql] = key
88
- end
89
-
90
- def clear
91
- cache.each_value do |hash|
92
- hash[:stmt].close
93
- end
94
- cache.clear
95
- end
96
-
97
- private
98
- def cache
99
- @cache[Process.pid]
100
- end
101
- end
102
-
103
- def initialize(connection, logger, connection_options, config)
104
- super
105
- @statements = StatementPool.new(@connection,
106
- self.class.type_cast_config_to_integer(config.fetch(:statement_limit) { 1000 }))
107
- @client_encoding = nil
108
- connect
109
- end
110
-
111
- # Returns true, since this connection adapter supports prepared statement
112
- # caching.
113
- def supports_statement_cache?
114
- true
115
- end
116
-
117
- # HELPER METHODS ===========================================
118
-
119
- def each_hash(result) # :nodoc:
120
- if block_given?
121
- result.each_hash do |row|
122
- row.symbolize_keys!
123
- yield row
124
- end
125
- else
126
- to_enum(:each_hash, result)
127
- end
128
- end
129
-
130
- def error_number(exception) # :nodoc:
131
- exception.errno if exception.respond_to?(:errno)
132
- end
133
-
134
- # QUOTING ==================================================
135
-
136
- def quote_string(string) #:nodoc:
137
- @connection.quote(string)
138
- end
139
-
140
- #--
141
- # CONNECTION MANAGEMENT ====================================
142
- #++
143
-
144
- def active?
145
- if @connection.respond_to?(:stat)
146
- @connection.stat
147
- else
148
- @connection.query 'select 1'
149
- end
150
-
151
- # mysql-ruby doesn't raise an exception when stat fails.
152
- if @connection.respond_to?(:errno)
153
- @connection.errno.zero?
154
- else
155
- true
156
- end
157
- rescue Mysql::Error
158
- false
159
- end
160
-
161
- def reconnect!
162
- super
163
- disconnect!
164
- connect
165
- end
166
-
167
- # Disconnects from the database if already connected. Otherwise, this
168
- # method does nothing.
169
- def disconnect!
170
- super
171
- @connection.close rescue nil
172
- end
173
-
174
- def reset!
175
- if @connection.respond_to?(:change_user)
176
- # See http://bugs.mysql.com/bug.php?id=33540 -- the workaround way to
177
- # reset the connection is to change the user to the same user.
178
- @connection.change_user(@config[:username], @config[:password], @config[:database])
179
- configure_connection
180
- end
181
- end
182
-
183
- #--
184
- # DATABASE STATEMENTS ======================================
185
- #++
186
-
187
- def select_rows(sql, name = nil, binds = [])
188
- @connection.query_with_result = true
189
- rows = exec_query(sql, name, binds).rows
190
- @connection.more_results && @connection.next_result # invoking stored procedures with CLIENT_MULTI_RESULTS requires this to tidy up else connection will be dropped
191
- rows
192
- end
193
-
194
- # Clears the prepared statements cache.
195
- def clear_cache!
196
- super
197
- @statements.clear
198
- end
199
-
200
- # Taken from here:
201
- # https://github.com/tmtm/ruby-mysql/blob/master/lib/mysql/charset.rb
202
- # Author: TOMITA Masahiro <tommy@tmtm.org>
203
- ENCODINGS = {
204
- "armscii8" => nil,
205
- "ascii" => Encoding::US_ASCII,
206
- "big5" => Encoding::Big5,
207
- "binary" => Encoding::ASCII_8BIT,
208
- "cp1250" => Encoding::Windows_1250,
209
- "cp1251" => Encoding::Windows_1251,
210
- "cp1256" => Encoding::Windows_1256,
211
- "cp1257" => Encoding::Windows_1257,
212
- "cp850" => Encoding::CP850,
213
- "cp852" => Encoding::CP852,
214
- "cp866" => Encoding::IBM866,
215
- "cp932" => Encoding::Windows_31J,
216
- "dec8" => nil,
217
- "eucjpms" => Encoding::EucJP_ms,
218
- "euckr" => Encoding::EUC_KR,
219
- "gb2312" => Encoding::EUC_CN,
220
- "gbk" => Encoding::GBK,
221
- "geostd8" => nil,
222
- "greek" => Encoding::ISO_8859_7,
223
- "hebrew" => Encoding::ISO_8859_8,
224
- "hp8" => nil,
225
- "keybcs2" => nil,
226
- "koi8r" => Encoding::KOI8_R,
227
- "koi8u" => Encoding::KOI8_U,
228
- "latin1" => Encoding::ISO_8859_1,
229
- "latin2" => Encoding::ISO_8859_2,
230
- "latin5" => Encoding::ISO_8859_9,
231
- "latin7" => Encoding::ISO_8859_13,
232
- "macce" => Encoding::MacCentEuro,
233
- "macroman" => Encoding::MacRoman,
234
- "sjis" => Encoding::SHIFT_JIS,
235
- "swe7" => nil,
236
- "tis620" => Encoding::TIS_620,
237
- "ucs2" => Encoding::UTF_16BE,
238
- "ujis" => Encoding::EucJP_ms,
239
- "utf8" => Encoding::UTF_8,
240
- "utf8mb4" => Encoding::UTF_8,
241
- }
242
-
243
- # Get the client encoding for this database
244
- def client_encoding
245
- return @client_encoding if @client_encoding
246
-
247
- result = exec_query(
248
- "SHOW VARIABLES WHERE Variable_name = 'character_set_client'",
249
- 'SCHEMA')
250
- @client_encoding = ENCODINGS[result.rows.last.last]
251
- end
252
-
253
- def exec_query(sql, name = 'SQL', binds = [])
254
- if without_prepared_statement?(binds)
255
- result_set, affected_rows = exec_without_stmt(sql, name)
256
- else
257
- result_set, affected_rows = exec_stmt(sql, name, binds)
258
- end
259
-
260
- yield affected_rows if block_given?
261
-
262
- result_set
263
- end
264
-
265
- def last_inserted_id(result)
266
- @connection.insert_id
267
- end
268
-
269
- module Fields # :nodoc:
270
- class DateTime < Type::DateTime # :nodoc:
271
- def cast_value(value)
272
- if Mysql::Time === value
273
- new_time(
274
- value.year,
275
- value.month,
276
- value.day,
277
- value.hour,
278
- value.minute,
279
- value.second,
280
- value.second_part)
281
- else
282
- super
283
- end
284
- end
285
- end
286
-
287
- class Time < Type::Time # :nodoc:
288
- def cast_value(value)
289
- if Mysql::Time === value
290
- new_time(
291
- 2000,
292
- 01,
293
- 01,
294
- value.hour,
295
- value.minute,
296
- value.second,
297
- value.second_part)
298
- else
299
- super
300
- end
301
- end
302
- end
303
-
304
- class << self
305
- TYPES = Type::HashLookupTypeMap.new # :nodoc:
306
-
307
- delegate :register_type, :alias_type, to: :TYPES
308
-
309
- def find_type(field)
310
- if field.type == Mysql::Field::TYPE_TINY && field.length > 1
311
- TYPES.lookup(Mysql::Field::TYPE_LONG)
312
- else
313
- TYPES.lookup(field.type)
314
- end
315
- end
316
- end
317
-
318
- register_type Mysql::Field::TYPE_TINY, Type::Boolean.new
319
- register_type Mysql::Field::TYPE_LONG, Type::Integer.new
320
- alias_type Mysql::Field::TYPE_LONGLONG, Mysql::Field::TYPE_LONG
321
- alias_type Mysql::Field::TYPE_NEWDECIMAL, Mysql::Field::TYPE_LONG
322
-
323
- register_type Mysql::Field::TYPE_DATE, Type::Date.new
324
- register_type Mysql::Field::TYPE_DATETIME, Fields::DateTime.new
325
- register_type Mysql::Field::TYPE_TIME, Fields::Time.new
326
- register_type Mysql::Field::TYPE_FLOAT, Type::Float.new
327
- end
328
-
329
- def initialize_type_map(m) # :nodoc:
330
- super
331
- m.register_type %r(datetime)i, Fields::DateTime.new
332
- m.register_type %r(time)i, Fields::Time.new
333
- end
334
-
335
- def exec_without_stmt(sql, name = 'SQL') # :nodoc:
336
- # Some queries, like SHOW CREATE TABLE don't work through the prepared
337
- # statement API. For those queries, we need to use this method. :'(
338
- log(sql, name) do
339
- result = @connection.query(sql)
340
- affected_rows = @connection.affected_rows
341
-
342
- if result
343
- types = {}
344
- fields = []
345
- result.fetch_fields.each { |field|
346
- field_name = field.name
347
- fields << field_name
348
-
349
- if field.decimals > 0
350
- types[field_name] = Type::Decimal.new
351
- else
352
- types[field_name] = Fields.find_type field
353
- end
354
- }
355
-
356
- result_set = ActiveRecord::Result.new(fields, result.to_a, types)
357
- result.free
358
- else
359
- result_set = ActiveRecord::Result.new([], [])
360
- end
361
-
362
- [result_set, affected_rows]
363
- end
364
- end
365
-
366
- def execute_and_free(sql, name = nil) # :nodoc:
367
- result = execute(sql, name)
368
- ret = yield result
369
- result.free
370
- ret
371
- end
372
-
373
- def insert_sql(sql, name = nil, pk = nil, id_value = nil, sequence_name = nil) #:nodoc:
374
- super sql, name
375
- id_value || @connection.insert_id
376
- end
377
- alias :create :insert_sql
378
-
379
- def exec_delete(sql, name, binds) # :nodoc:
380
- affected_rows = 0
381
-
382
- exec_query(sql, name, binds) do |n|
383
- affected_rows = n
384
- end
385
-
386
- affected_rows
387
- end
388
- alias :exec_update :exec_delete
389
-
390
- def begin_db_transaction #:nodoc:
391
- exec_query "BEGIN"
392
- end
393
-
394
- private
395
-
396
- def exec_stmt(sql, name, binds)
397
- cache = {}
398
- type_casted_binds = binds.map { |col, val|
399
- [col, type_cast(val, col)]
400
- }
401
-
402
- log(sql, name, type_casted_binds) do
403
- if binds.empty?
404
- stmt = @connection.prepare(sql)
405
- else
406
- cache = @statements[sql] ||= {
407
- :stmt => @connection.prepare(sql)
408
- }
409
- stmt = cache[:stmt]
410
- end
411
-
412
- begin
413
- stmt.execute(*type_casted_binds.map { |_, val| val })
414
- rescue Mysql::Error => e
415
- # Older versions of MySQL leave the prepared statement in a bad
416
- # place when an error occurs. To support older MySQL versions, we
417
- # need to close the statement and delete the statement from the
418
- # cache.
419
- stmt.close
420
- @statements.delete sql
421
- raise e
422
- end
423
-
424
- cols = nil
425
- if metadata = stmt.result_metadata
426
- cols = cache[:cols] ||= metadata.fetch_fields.map { |field|
427
- field.name
428
- }
429
- metadata.free
430
- end
431
-
432
- result_set = ActiveRecord::Result.new(cols, stmt.to_a) if cols
433
- affected_rows = stmt.affected_rows
434
-
435
- stmt.free_result
436
- stmt.close if binds.empty?
437
-
438
- [result_set, affected_rows]
439
- end
440
- end
441
-
442
- def connect
443
- encoding = @config[:encoding]
444
- if encoding
445
- @connection.options(Mysql::SET_CHARSET_NAME, encoding) rescue nil
446
- end
447
-
448
- if @config[:sslca] || @config[:sslkey]
449
- @connection.ssl_set(@config[:sslkey], @config[:sslcert], @config[:sslca], @config[:sslcapath], @config[:sslcipher])
450
- end
451
-
452
- @connection.options(Mysql::OPT_CONNECT_TIMEOUT, @config[:connect_timeout]) if @config[:connect_timeout]
453
- @connection.options(Mysql::OPT_READ_TIMEOUT, @config[:read_timeout]) if @config[:read_timeout]
454
- @connection.options(Mysql::OPT_WRITE_TIMEOUT, @config[:write_timeout]) if @config[:write_timeout]
455
-
456
- @connection.real_connect(*@connection_options)
457
-
458
- # reconnect must be set after real_connect is called, because real_connect sets it to false internally
459
- @connection.reconnect = !!@config[:reconnect] if @connection.respond_to?(:reconnect=)
460
-
461
- configure_connection
462
- end
463
-
464
- # Many Rails applications monkey-patch a replacement of the configure_connection method
465
- # and don't call 'super', so leave this here even though it looks superfluous.
466
- def configure_connection
467
- super
468
- end
469
-
470
- def select(sql, name = nil, binds = [])
471
- @connection.query_with_result = true
472
- rows = super
473
- @connection.more_results && @connection.next_result # invoking stored procedures with CLIENT_MULTI_RESULTS requires this to tidy up else connection will be dropped
474
- rows
475
- end
476
-
477
- # Returns the full version of the connected MySQL server.
478
- def full_version
479
- @full_version ||= @connection.server_info
480
- end
481
-
482
- def set_field_encoding field_name
483
- field_name.force_encoding(client_encoding)
484
- if internal_enc = Encoding.default_internal
485
- field_name = field_name.encode!(internal_enc)
486
- end
487
- field_name
488
- end
489
- end
490
- end
491
- end
@@ -1,93 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module ArrayParser # :nodoc:
5
-
6
- DOUBLE_QUOTE = '"'
7
- BACKSLASH = "\\"
8
- COMMA = ','
9
- BRACKET_OPEN = '{'
10
- BRACKET_CLOSE = '}'
11
-
12
- def parse_pg_array(string) # :nodoc:
13
- local_index = 0
14
- array = []
15
- while(local_index < string.length)
16
- case string[local_index]
17
- when BRACKET_OPEN
18
- local_index,array = parse_array_contents(array, string, local_index + 1)
19
- when BRACKET_CLOSE
20
- return array
21
- end
22
- local_index += 1
23
- end
24
-
25
- array
26
- end
27
-
28
- private
29
-
30
- def parse_array_contents(array, string, index)
31
- is_escaping = false
32
- is_quoted = false
33
- was_quoted = false
34
- current_item = ''
35
-
36
- local_index = index
37
- while local_index
38
- token = string[local_index]
39
- if is_escaping
40
- current_item << token
41
- is_escaping = false
42
- else
43
- if is_quoted
44
- case token
45
- when DOUBLE_QUOTE
46
- is_quoted = false
47
- was_quoted = true
48
- when BACKSLASH
49
- is_escaping = true
50
- else
51
- current_item << token
52
- end
53
- else
54
- case token
55
- when BACKSLASH
56
- is_escaping = true
57
- when COMMA
58
- add_item_to_array(array, current_item, was_quoted)
59
- current_item = ''
60
- was_quoted = false
61
- when DOUBLE_QUOTE
62
- is_quoted = true
63
- when BRACKET_OPEN
64
- internal_items = []
65
- local_index,internal_items = parse_array_contents(internal_items, string, local_index + 1)
66
- array.push(internal_items)
67
- when BRACKET_CLOSE
68
- add_item_to_array(array, current_item, was_quoted)
69
- return local_index,array
70
- else
71
- current_item << token
72
- end
73
- end
74
- end
75
-
76
- local_index += 1
77
- end
78
- return local_index,array
79
- end
80
-
81
- def add_item_to_array(array, current_item, quoted)
82
- return if !quoted && current_item.length == 0
83
-
84
- if !quoted && current_item == 'NULL'
85
- array.push nil
86
- else
87
- array.push current_item
88
- end
89
- end
90
- end
91
- end
92
- end
93
- end
@@ -1,11 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module OID # :nodoc:
5
- class Date < Type::Date # :nodoc:
6
- include Infinity
7
- end
8
- end
9
- end
10
- end
11
- end
@@ -1,21 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module OID # :nodoc:
5
- class Float < Type::Float # :nodoc:
6
- include Infinity
7
-
8
- def cast_value(value)
9
- case value
10
- when ::Float then value
11
- when 'Infinity' then ::Float::INFINITY
12
- when '-Infinity' then -::Float::INFINITY
13
- when 'NaN' then ::Float::NAN
14
- else value.to_f
15
- end
16
- end
17
- end
18
- end
19
- end
20
- end
21
- end
@@ -1,13 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module OID # :nodoc:
5
- module Infinity # :nodoc:
6
- def infinity(options = {})
7
- options[:negative] ? -::Float::INFINITY : ::Float::INFINITY
8
- end
9
- end
10
- end
11
- end
12
- end
13
- end
@@ -1,11 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module OID # :nodoc:
5
- class Integer < Type::Integer # :nodoc:
6
- include Infinity
7
- end
8
- end
9
- end
10
- end
11
- end
@@ -1,11 +0,0 @@
1
- module ActiveRecord
2
- module ConnectionAdapters
3
- module PostgreSQL
4
- module OID # :nodoc:
5
- class Time < Type::Time # :nodoc:
6
- include Infinity
7
- end
8
- end
9
- end
10
- end
11
- end