activerecord 5.2.3 → 6.0.1

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 (269) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +691 -542
  3. data/MIT-LICENSE +3 -1
  4. data/README.rdoc +4 -2
  5. data/examples/performance.rb +1 -1
  6. data/lib/active_record.rb +9 -2
  7. data/lib/active_record/aggregations.rb +4 -2
  8. data/lib/active_record/association_relation.rb +15 -6
  9. data/lib/active_record/associations.rb +20 -15
  10. data/lib/active_record/associations/association.rb +61 -20
  11. data/lib/active_record/associations/association_scope.rb +4 -6
  12. data/lib/active_record/associations/belongs_to_association.rb +36 -42
  13. data/lib/active_record/associations/belongs_to_polymorphic_association.rb +0 -4
  14. data/lib/active_record/associations/builder/association.rb +14 -18
  15. data/lib/active_record/associations/builder/belongs_to.rb +19 -52
  16. data/lib/active_record/associations/builder/collection_association.rb +5 -15
  17. data/lib/active_record/associations/builder/has_and_belongs_to_many.rb +17 -38
  18. data/lib/active_record/associations/builder/has_many.rb +2 -0
  19. data/lib/active_record/associations/builder/has_one.rb +35 -1
  20. data/lib/active_record/associations/builder/singular_association.rb +2 -0
  21. data/lib/active_record/associations/collection_association.rb +6 -21
  22. data/lib/active_record/associations/collection_proxy.rb +12 -15
  23. data/lib/active_record/associations/foreign_association.rb +7 -0
  24. data/lib/active_record/associations/has_many_association.rb +2 -10
  25. data/lib/active_record/associations/has_many_through_association.rb +18 -25
  26. data/lib/active_record/associations/has_one_association.rb +28 -30
  27. data/lib/active_record/associations/has_one_through_association.rb +5 -5
  28. data/lib/active_record/associations/join_dependency.rb +28 -28
  29. data/lib/active_record/associations/join_dependency/join_association.rb +27 -7
  30. data/lib/active_record/associations/join_dependency/join_part.rb +2 -2
  31. data/lib/active_record/associations/preloader.rb +39 -31
  32. data/lib/active_record/associations/preloader/association.rb +38 -36
  33. data/lib/active_record/associations/preloader/through_association.rb +48 -39
  34. data/lib/active_record/associations/singular_association.rb +2 -16
  35. data/lib/active_record/attribute_assignment.rb +7 -10
  36. data/lib/active_record/attribute_methods.rb +28 -100
  37. data/lib/active_record/attribute_methods/before_type_cast.rb +4 -1
  38. data/lib/active_record/attribute_methods/dirty.rb +111 -40
  39. data/lib/active_record/attribute_methods/primary_key.rb +15 -22
  40. data/lib/active_record/attribute_methods/query.rb +2 -3
  41. data/lib/active_record/attribute_methods/read.rb +15 -53
  42. data/lib/active_record/attribute_methods/serialization.rb +1 -1
  43. data/lib/active_record/attribute_methods/time_zone_conversion.rb +1 -1
  44. data/lib/active_record/attribute_methods/write.rb +17 -24
  45. data/lib/active_record/attributes.rb +13 -0
  46. data/lib/active_record/autosave_association.rb +22 -8
  47. data/lib/active_record/base.rb +2 -3
  48. data/lib/active_record/callbacks.rb +5 -19
  49. data/lib/active_record/connection_adapters/abstract/connection_pool.rb +126 -19
  50. data/lib/active_record/connection_adapters/abstract/database_limits.rb +17 -4
  51. data/lib/active_record/connection_adapters/abstract/database_statements.rb +99 -123
  52. data/lib/active_record/connection_adapters/abstract/query_cache.rb +21 -11
  53. data/lib/active_record/connection_adapters/abstract/quoting.rb +68 -17
  54. data/lib/active_record/connection_adapters/abstract/schema_creation.rb +19 -12
  55. data/lib/active_record/connection_adapters/abstract/schema_definitions.rb +76 -48
  56. data/lib/active_record/connection_adapters/abstract/schema_dumper.rb +1 -3
  57. data/lib/active_record/connection_adapters/abstract/schema_statements.rb +133 -54
  58. data/lib/active_record/connection_adapters/abstract/transaction.rb +96 -56
  59. data/lib/active_record/connection_adapters/abstract_adapter.rb +187 -43
  60. data/lib/active_record/connection_adapters/abstract_mysql_adapter.rb +138 -195
  61. data/lib/active_record/connection_adapters/column.rb +17 -13
  62. data/lib/active_record/connection_adapters/connection_specification.rb +53 -43
  63. data/lib/active_record/connection_adapters/determine_if_preparable_visitor.rb +6 -10
  64. data/lib/active_record/connection_adapters/mysql/database_statements.rb +75 -13
  65. data/lib/active_record/connection_adapters/mysql/quoting.rb +44 -7
  66. data/lib/active_record/connection_adapters/mysql/schema_creation.rb +3 -4
  67. data/lib/active_record/connection_adapters/mysql/schema_definitions.rb +40 -32
  68. data/lib/active_record/connection_adapters/mysql/schema_dumper.rb +14 -6
  69. data/lib/active_record/connection_adapters/mysql/schema_statements.rb +129 -13
  70. data/lib/active_record/connection_adapters/mysql/type_metadata.rb +6 -10
  71. data/lib/active_record/connection_adapters/mysql2_adapter.rb +26 -9
  72. data/lib/active_record/connection_adapters/postgresql/column.rb +17 -31
  73. data/lib/active_record/connection_adapters/postgresql/database_statements.rb +22 -1
  74. data/lib/active_record/connection_adapters/postgresql/oid/array.rb +1 -1
  75. data/lib/active_record/connection_adapters/postgresql/oid/bit.rb +1 -4
  76. data/lib/active_record/connection_adapters/postgresql/oid/hstore.rb +1 -1
  77. data/lib/active_record/connection_adapters/postgresql/oid/legacy_point.rb +1 -1
  78. data/lib/active_record/connection_adapters/postgresql/oid/money.rb +2 -2
  79. data/lib/active_record/connection_adapters/postgresql/oid/point.rb +1 -1
  80. data/lib/active_record/connection_adapters/postgresql/oid/range.rb +1 -1
  81. data/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb +9 -7
  82. data/lib/active_record/connection_adapters/postgresql/oid/uuid.rb +6 -3
  83. data/lib/active_record/connection_adapters/postgresql/quoting.rb +44 -7
  84. data/lib/active_record/connection_adapters/postgresql/schema_creation.rb +12 -1
  85. data/lib/active_record/connection_adapters/postgresql/schema_definitions.rb +107 -91
  86. data/lib/active_record/connection_adapters/postgresql/schema_statements.rb +55 -53
  87. data/lib/active_record/connection_adapters/postgresql/type_metadata.rb +24 -27
  88. data/lib/active_record/connection_adapters/postgresql_adapter.rb +164 -74
  89. data/lib/active_record/connection_adapters/schema_cache.rb +37 -14
  90. data/lib/active_record/connection_adapters/sql_type_metadata.rb +11 -8
  91. data/lib/active_record/connection_adapters/sqlite3/database_statements.rb +120 -0
  92. data/lib/active_record/connection_adapters/sqlite3/quoting.rb +42 -6
  93. data/lib/active_record/connection_adapters/sqlite3/schema_statements.rb +42 -11
  94. data/lib/active_record/connection_adapters/sqlite3_adapter.rb +131 -143
  95. data/lib/active_record/connection_handling.rb +155 -26
  96. data/lib/active_record/core.rb +103 -59
  97. data/lib/active_record/counter_cache.rb +4 -29
  98. data/lib/active_record/database_configurations.rb +233 -0
  99. data/lib/active_record/database_configurations/database_config.rb +37 -0
  100. data/lib/active_record/database_configurations/hash_config.rb +50 -0
  101. data/lib/active_record/database_configurations/url_config.rb +79 -0
  102. data/lib/active_record/dynamic_matchers.rb +1 -1
  103. data/lib/active_record/enum.rb +37 -7
  104. data/lib/active_record/errors.rb +15 -7
  105. data/lib/active_record/explain.rb +1 -1
  106. data/lib/active_record/fixture_set/model_metadata.rb +33 -0
  107. data/lib/active_record/fixture_set/render_context.rb +17 -0
  108. data/lib/active_record/fixture_set/table_row.rb +153 -0
  109. data/lib/active_record/fixture_set/table_rows.rb +47 -0
  110. data/lib/active_record/fixtures.rb +145 -472
  111. data/lib/active_record/gem_version.rb +3 -3
  112. data/lib/active_record/inheritance.rb +13 -3
  113. data/lib/active_record/insert_all.rb +179 -0
  114. data/lib/active_record/integration.rb +68 -16
  115. data/lib/active_record/internal_metadata.rb +10 -2
  116. data/lib/active_record/locking/optimistic.rb +5 -6
  117. data/lib/active_record/locking/pessimistic.rb +3 -3
  118. data/lib/active_record/log_subscriber.rb +7 -26
  119. data/lib/active_record/middleware/database_selector.rb +75 -0
  120. data/lib/active_record/middleware/database_selector/resolver.rb +88 -0
  121. data/lib/active_record/middleware/database_selector/resolver/session.rb +45 -0
  122. data/lib/active_record/migration.rb +100 -81
  123. data/lib/active_record/migration/command_recorder.rb +50 -6
  124. data/lib/active_record/migration/compatibility.rb +76 -49
  125. data/lib/active_record/model_schema.rb +33 -9
  126. data/lib/active_record/nested_attributes.rb +2 -2
  127. data/lib/active_record/no_touching.rb +7 -0
  128. data/lib/active_record/persistence.rb +228 -24
  129. data/lib/active_record/query_cache.rb +11 -4
  130. data/lib/active_record/querying.rb +32 -20
  131. data/lib/active_record/railtie.rb +80 -43
  132. data/lib/active_record/railties/collection_cache_association_loading.rb +34 -0
  133. data/lib/active_record/railties/controller_runtime.rb +30 -35
  134. data/lib/active_record/railties/databases.rake +196 -46
  135. data/lib/active_record/reflection.rb +42 -44
  136. data/lib/active_record/relation.rb +311 -80
  137. data/lib/active_record/relation/batches.rb +13 -10
  138. data/lib/active_record/relation/calculations.rb +58 -51
  139. data/lib/active_record/relation/delegation.rb +26 -43
  140. data/lib/active_record/relation/finder_methods.rb +24 -28
  141. data/lib/active_record/relation/merger.rb +11 -20
  142. data/lib/active_record/relation/predicate_builder.rb +4 -6
  143. data/lib/active_record/relation/predicate_builder/array_handler.rb +5 -4
  144. data/lib/active_record/relation/predicate_builder/association_query_value.rb +1 -4
  145. data/lib/active_record/relation/predicate_builder/base_handler.rb +1 -2
  146. data/lib/active_record/relation/predicate_builder/basic_object_handler.rb +1 -2
  147. data/lib/active_record/relation/predicate_builder/polymorphic_array_value.rb +1 -4
  148. data/lib/active_record/relation/predicate_builder/range_handler.rb +3 -23
  149. data/lib/active_record/relation/query_attribute.rb +13 -8
  150. data/lib/active_record/relation/query_methods.rb +219 -79
  151. data/lib/active_record/relation/spawn_methods.rb +1 -1
  152. data/lib/active_record/relation/where_clause.rb +14 -10
  153. data/lib/active_record/relation/where_clause_factory.rb +1 -2
  154. data/lib/active_record/result.rb +30 -11
  155. data/lib/active_record/sanitization.rb +32 -40
  156. data/lib/active_record/schema.rb +2 -11
  157. data/lib/active_record/schema_dumper.rb +22 -7
  158. data/lib/active_record/schema_migration.rb +5 -1
  159. data/lib/active_record/scoping.rb +8 -8
  160. data/lib/active_record/scoping/default.rb +6 -7
  161. data/lib/active_record/scoping/named.rb +19 -15
  162. data/lib/active_record/statement_cache.rb +32 -5
  163. data/lib/active_record/store.rb +87 -8
  164. data/lib/active_record/table_metadata.rb +10 -17
  165. data/lib/active_record/tasks/database_tasks.rb +194 -25
  166. data/lib/active_record/tasks/mysql_database_tasks.rb +5 -5
  167. data/lib/active_record/tasks/postgresql_database_tasks.rb +5 -7
  168. data/lib/active_record/tasks/sqlite_database_tasks.rb +2 -8
  169. data/lib/active_record/test_databases.rb +23 -0
  170. data/lib/active_record/test_fixtures.rb +224 -0
  171. data/lib/active_record/timestamp.rb +39 -25
  172. data/lib/active_record/touch_later.rb +4 -2
  173. data/lib/active_record/transactions.rb +57 -66
  174. data/lib/active_record/translation.rb +1 -1
  175. data/lib/active_record/type.rb +3 -4
  176. data/lib/active_record/type/adapter_specific_registry.rb +1 -8
  177. data/lib/active_record/type_caster/connection.rb +15 -14
  178. data/lib/active_record/type_caster/map.rb +1 -4
  179. data/lib/active_record/validations.rb +1 -0
  180. data/lib/active_record/validations/uniqueness.rb +15 -27
  181. data/lib/arel.rb +58 -0
  182. data/lib/arel/alias_predication.rb +9 -0
  183. data/lib/arel/attributes.rb +22 -0
  184. data/lib/arel/attributes/attribute.rb +37 -0
  185. data/lib/arel/collectors/bind.rb +24 -0
  186. data/lib/arel/collectors/composite.rb +31 -0
  187. data/lib/arel/collectors/plain_string.rb +20 -0
  188. data/lib/arel/collectors/sql_string.rb +20 -0
  189. data/lib/arel/collectors/substitute_binds.rb +28 -0
  190. data/lib/arel/crud.rb +42 -0
  191. data/lib/arel/delete_manager.rb +18 -0
  192. data/lib/arel/errors.rb +9 -0
  193. data/lib/arel/expressions.rb +29 -0
  194. data/lib/arel/factory_methods.rb +49 -0
  195. data/lib/arel/insert_manager.rb +49 -0
  196. data/lib/arel/math.rb +45 -0
  197. data/lib/arel/nodes.rb +68 -0
  198. data/lib/arel/nodes/and.rb +32 -0
  199. data/lib/arel/nodes/ascending.rb +23 -0
  200. data/lib/arel/nodes/binary.rb +52 -0
  201. data/lib/arel/nodes/bind_param.rb +36 -0
  202. data/lib/arel/nodes/case.rb +55 -0
  203. data/lib/arel/nodes/casted.rb +50 -0
  204. data/lib/arel/nodes/comment.rb +29 -0
  205. data/lib/arel/nodes/count.rb +12 -0
  206. data/lib/arel/nodes/delete_statement.rb +45 -0
  207. data/lib/arel/nodes/descending.rb +23 -0
  208. data/lib/arel/nodes/equality.rb +18 -0
  209. data/lib/arel/nodes/extract.rb +24 -0
  210. data/lib/arel/nodes/false.rb +16 -0
  211. data/lib/arel/nodes/full_outer_join.rb +8 -0
  212. data/lib/arel/nodes/function.rb +44 -0
  213. data/lib/arel/nodes/grouping.rb +8 -0
  214. data/lib/arel/nodes/in.rb +8 -0
  215. data/lib/arel/nodes/infix_operation.rb +80 -0
  216. data/lib/arel/nodes/inner_join.rb +8 -0
  217. data/lib/arel/nodes/insert_statement.rb +37 -0
  218. data/lib/arel/nodes/join_source.rb +20 -0
  219. data/lib/arel/nodes/matches.rb +18 -0
  220. data/lib/arel/nodes/named_function.rb +23 -0
  221. data/lib/arel/nodes/node.rb +50 -0
  222. data/lib/arel/nodes/node_expression.rb +13 -0
  223. data/lib/arel/nodes/outer_join.rb +8 -0
  224. data/lib/arel/nodes/over.rb +15 -0
  225. data/lib/arel/nodes/regexp.rb +16 -0
  226. data/lib/arel/nodes/right_outer_join.rb +8 -0
  227. data/lib/arel/nodes/select_core.rb +67 -0
  228. data/lib/arel/nodes/select_statement.rb +41 -0
  229. data/lib/arel/nodes/sql_literal.rb +16 -0
  230. data/lib/arel/nodes/string_join.rb +11 -0
  231. data/lib/arel/nodes/table_alias.rb +27 -0
  232. data/lib/arel/nodes/terminal.rb +16 -0
  233. data/lib/arel/nodes/true.rb +16 -0
  234. data/lib/arel/nodes/unary.rb +45 -0
  235. data/lib/arel/nodes/unary_operation.rb +20 -0
  236. data/lib/arel/nodes/unqualified_column.rb +22 -0
  237. data/lib/arel/nodes/update_statement.rb +41 -0
  238. data/lib/arel/nodes/values_list.rb +9 -0
  239. data/lib/arel/nodes/window.rb +126 -0
  240. data/lib/arel/nodes/with.rb +11 -0
  241. data/lib/arel/order_predications.rb +13 -0
  242. data/lib/arel/predications.rb +257 -0
  243. data/lib/arel/select_manager.rb +271 -0
  244. data/lib/arel/table.rb +110 -0
  245. data/lib/arel/tree_manager.rb +72 -0
  246. data/lib/arel/update_manager.rb +34 -0
  247. data/lib/arel/visitors.rb +20 -0
  248. data/lib/arel/visitors/depth_first.rb +204 -0
  249. data/lib/arel/visitors/dot.rb +297 -0
  250. data/lib/arel/visitors/ibm_db.rb +34 -0
  251. data/lib/arel/visitors/informix.rb +62 -0
  252. data/lib/arel/visitors/mssql.rb +157 -0
  253. data/lib/arel/visitors/mysql.rb +83 -0
  254. data/lib/arel/visitors/oracle.rb +159 -0
  255. data/lib/arel/visitors/oracle12.rb +66 -0
  256. data/lib/arel/visitors/postgresql.rb +110 -0
  257. data/lib/arel/visitors/sqlite.rb +39 -0
  258. data/lib/arel/visitors/to_sql.rb +889 -0
  259. data/lib/arel/visitors/visitor.rb +46 -0
  260. data/lib/arel/visitors/where_sql.rb +23 -0
  261. data/lib/arel/window_predications.rb +9 -0
  262. data/lib/rails/generators/active_record/migration.rb +14 -1
  263. data/lib/rails/generators/active_record/migration/migration_generator.rb +2 -5
  264. data/lib/rails/generators/active_record/migration/templates/create_table_migration.rb.tt +1 -1
  265. data/lib/rails/generators/active_record/migration/templates/migration.rb.tt +4 -2
  266. data/lib/rails/generators/active_record/model/model_generator.rb +1 -0
  267. data/lib/rails/generators/active_record/model/templates/model.rb.tt +10 -1
  268. metadata +111 -26
  269. data/lib/active_record/collection_cache_key.rb +0 -53
@@ -0,0 +1,34 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Arel # :nodoc: all
4
+ class UpdateManager < Arel::TreeManager
5
+ include TreeManager::StatementMethods
6
+
7
+ def initialize
8
+ super
9
+ @ast = Nodes::UpdateStatement.new
10
+ @ctx = @ast
11
+ end
12
+
13
+ ###
14
+ # UPDATE +table+
15
+ def table(table)
16
+ @ast.relation = table
17
+ self
18
+ end
19
+
20
+ def set(values)
21
+ if String === values
22
+ @ast.values = [values]
23
+ else
24
+ @ast.values = values.map { |column, value|
25
+ Nodes::Assignment.new(
26
+ Nodes::UnqualifiedColumn.new(column),
27
+ value
28
+ )
29
+ }
30
+ end
31
+ self
32
+ end
33
+ end
34
+ end
@@ -0,0 +1,20 @@
1
+ # frozen_string_literal: true
2
+
3
+ require "arel/visitors/visitor"
4
+ require "arel/visitors/depth_first"
5
+ require "arel/visitors/to_sql"
6
+ require "arel/visitors/sqlite"
7
+ require "arel/visitors/postgresql"
8
+ require "arel/visitors/mysql"
9
+ require "arel/visitors/mssql"
10
+ require "arel/visitors/oracle"
11
+ require "arel/visitors/oracle12"
12
+ require "arel/visitors/where_sql"
13
+ require "arel/visitors/dot"
14
+ require "arel/visitors/ibm_db"
15
+ require "arel/visitors/informix"
16
+
17
+ module Arel # :nodoc: all
18
+ module Visitors
19
+ end
20
+ end
@@ -0,0 +1,204 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Arel # :nodoc: all
4
+ module Visitors
5
+ class DepthFirst < Arel::Visitors::Visitor
6
+ def initialize(block = nil)
7
+ @block = block || Proc.new
8
+ super()
9
+ end
10
+
11
+ private
12
+
13
+ def visit(o, _ = nil)
14
+ super
15
+ @block.call o
16
+ end
17
+
18
+ def unary(o)
19
+ visit o.expr
20
+ end
21
+ alias :visit_Arel_Nodes_Else :unary
22
+ alias :visit_Arel_Nodes_Group :unary
23
+ alias :visit_Arel_Nodes_Cube :unary
24
+ alias :visit_Arel_Nodes_RollUp :unary
25
+ alias :visit_Arel_Nodes_GroupingSet :unary
26
+ alias :visit_Arel_Nodes_GroupingElement :unary
27
+ alias :visit_Arel_Nodes_Grouping :unary
28
+ alias :visit_Arel_Nodes_Having :unary
29
+ alias :visit_Arel_Nodes_Lateral :unary
30
+ alias :visit_Arel_Nodes_Limit :unary
31
+ alias :visit_Arel_Nodes_Not :unary
32
+ alias :visit_Arel_Nodes_Offset :unary
33
+ alias :visit_Arel_Nodes_On :unary
34
+ alias :visit_Arel_Nodes_Ordering :unary
35
+ alias :visit_Arel_Nodes_Ascending :unary
36
+ alias :visit_Arel_Nodes_Descending :unary
37
+ alias :visit_Arel_Nodes_UnqualifiedColumn :unary
38
+ alias :visit_Arel_Nodes_OptimizerHints :unary
39
+ alias :visit_Arel_Nodes_ValuesList :unary
40
+
41
+ def function(o)
42
+ visit o.expressions
43
+ visit o.alias
44
+ visit o.distinct
45
+ end
46
+ alias :visit_Arel_Nodes_Avg :function
47
+ alias :visit_Arel_Nodes_Exists :function
48
+ alias :visit_Arel_Nodes_Max :function
49
+ alias :visit_Arel_Nodes_Min :function
50
+ alias :visit_Arel_Nodes_Sum :function
51
+
52
+ def visit_Arel_Nodes_NamedFunction(o)
53
+ visit o.name
54
+ visit o.expressions
55
+ visit o.distinct
56
+ visit o.alias
57
+ end
58
+
59
+ def visit_Arel_Nodes_Count(o)
60
+ visit o.expressions
61
+ visit o.alias
62
+ visit o.distinct
63
+ end
64
+
65
+ def visit_Arel_Nodes_Case(o)
66
+ visit o.case
67
+ visit o.conditions
68
+ visit o.default
69
+ end
70
+
71
+ def nary(o)
72
+ o.children.each { |child| visit child }
73
+ end
74
+ alias :visit_Arel_Nodes_And :nary
75
+
76
+ def binary(o)
77
+ visit o.left
78
+ visit o.right
79
+ end
80
+ alias :visit_Arel_Nodes_As :binary
81
+ alias :visit_Arel_Nodes_Assignment :binary
82
+ alias :visit_Arel_Nodes_Between :binary
83
+ alias :visit_Arel_Nodes_Concat :binary
84
+ alias :visit_Arel_Nodes_DeleteStatement :binary
85
+ alias :visit_Arel_Nodes_DoesNotMatch :binary
86
+ alias :visit_Arel_Nodes_Equality :binary
87
+ alias :visit_Arel_Nodes_FullOuterJoin :binary
88
+ alias :visit_Arel_Nodes_GreaterThan :binary
89
+ alias :visit_Arel_Nodes_GreaterThanOrEqual :binary
90
+ alias :visit_Arel_Nodes_In :binary
91
+ alias :visit_Arel_Nodes_InfixOperation :binary
92
+ alias :visit_Arel_Nodes_JoinSource :binary
93
+ alias :visit_Arel_Nodes_InnerJoin :binary
94
+ alias :visit_Arel_Nodes_LessThan :binary
95
+ alias :visit_Arel_Nodes_LessThanOrEqual :binary
96
+ alias :visit_Arel_Nodes_Matches :binary
97
+ alias :visit_Arel_Nodes_NotEqual :binary
98
+ alias :visit_Arel_Nodes_NotIn :binary
99
+ alias :visit_Arel_Nodes_NotRegexp :binary
100
+ alias :visit_Arel_Nodes_IsNotDistinctFrom :binary
101
+ alias :visit_Arel_Nodes_IsDistinctFrom :binary
102
+ alias :visit_Arel_Nodes_Or :binary
103
+ alias :visit_Arel_Nodes_OuterJoin :binary
104
+ alias :visit_Arel_Nodes_Regexp :binary
105
+ alias :visit_Arel_Nodes_RightOuterJoin :binary
106
+ alias :visit_Arel_Nodes_TableAlias :binary
107
+ alias :visit_Arel_Nodes_When :binary
108
+
109
+ def visit_Arel_Nodes_StringJoin(o)
110
+ visit o.left
111
+ end
112
+
113
+ def visit_Arel_Attribute(o)
114
+ visit o.relation
115
+ visit o.name
116
+ end
117
+ alias :visit_Arel_Attributes_Integer :visit_Arel_Attribute
118
+ alias :visit_Arel_Attributes_Float :visit_Arel_Attribute
119
+ alias :visit_Arel_Attributes_String :visit_Arel_Attribute
120
+ alias :visit_Arel_Attributes_Time :visit_Arel_Attribute
121
+ alias :visit_Arel_Attributes_Boolean :visit_Arel_Attribute
122
+ alias :visit_Arel_Attributes_Attribute :visit_Arel_Attribute
123
+ alias :visit_Arel_Attributes_Decimal :visit_Arel_Attribute
124
+
125
+ def visit_Arel_Table(o)
126
+ visit o.name
127
+ end
128
+
129
+ def terminal(o)
130
+ end
131
+ alias :visit_ActiveSupport_Multibyte_Chars :terminal
132
+ alias :visit_ActiveSupport_StringInquirer :terminal
133
+ alias :visit_Arel_Nodes_Lock :terminal
134
+ alias :visit_Arel_Nodes_Node :terminal
135
+ alias :visit_Arel_Nodes_SqlLiteral :terminal
136
+ alias :visit_Arel_Nodes_BindParam :terminal
137
+ alias :visit_Arel_Nodes_Window :terminal
138
+ alias :visit_Arel_Nodes_True :terminal
139
+ alias :visit_Arel_Nodes_False :terminal
140
+ alias :visit_BigDecimal :terminal
141
+ alias :visit_Class :terminal
142
+ alias :visit_Date :terminal
143
+ alias :visit_DateTime :terminal
144
+ alias :visit_FalseClass :terminal
145
+ alias :visit_Float :terminal
146
+ alias :visit_Integer :terminal
147
+ alias :visit_NilClass :terminal
148
+ alias :visit_String :terminal
149
+ alias :visit_Symbol :terminal
150
+ alias :visit_Time :terminal
151
+ alias :visit_TrueClass :terminal
152
+
153
+ def visit_Arel_Nodes_InsertStatement(o)
154
+ visit o.relation
155
+ visit o.columns
156
+ visit o.values
157
+ end
158
+
159
+ def visit_Arel_Nodes_SelectCore(o)
160
+ visit o.projections
161
+ visit o.source
162
+ visit o.wheres
163
+ visit o.groups
164
+ visit o.windows
165
+ visit o.havings
166
+ end
167
+
168
+ def visit_Arel_Nodes_SelectStatement(o)
169
+ visit o.cores
170
+ visit o.orders
171
+ visit o.limit
172
+ visit o.lock
173
+ visit o.offset
174
+ end
175
+
176
+ def visit_Arel_Nodes_UpdateStatement(o)
177
+ visit o.relation
178
+ visit o.values
179
+ visit o.wheres
180
+ visit o.orders
181
+ visit o.limit
182
+ end
183
+
184
+ def visit_Arel_Nodes_Comment(o)
185
+ visit o.values
186
+ end
187
+
188
+ def visit_Array(o)
189
+ o.each { |i| visit i }
190
+ end
191
+ alias :visit_Set :visit_Array
192
+
193
+ def visit_Hash(o)
194
+ o.each { |k, v| visit(k); visit(v) }
195
+ end
196
+
197
+ DISPATCH = dispatch_cache
198
+
199
+ def get_dispatch_cache
200
+ DISPATCH
201
+ end
202
+ end
203
+ end
204
+ end
@@ -0,0 +1,297 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Arel # :nodoc: all
4
+ module Visitors
5
+ class Dot < Arel::Visitors::Visitor
6
+ class Node # :nodoc:
7
+ attr_accessor :name, :id, :fields
8
+
9
+ def initialize(name, id, fields = [])
10
+ @name = name
11
+ @id = id
12
+ @fields = fields
13
+ end
14
+ end
15
+
16
+ class Edge < Struct.new :name, :from, :to # :nodoc:
17
+ end
18
+
19
+ def initialize
20
+ super()
21
+ @nodes = []
22
+ @edges = []
23
+ @node_stack = []
24
+ @edge_stack = []
25
+ @seen = {}
26
+ end
27
+
28
+ def accept(object, collector)
29
+ visit object
30
+ collector << to_dot
31
+ end
32
+
33
+ private
34
+
35
+ def visit_Arel_Nodes_Ordering(o)
36
+ visit_edge o, "expr"
37
+ end
38
+
39
+ def visit_Arel_Nodes_TableAlias(o)
40
+ visit_edge o, "name"
41
+ visit_edge o, "relation"
42
+ end
43
+
44
+ def visit_Arel_Nodes_Count(o)
45
+ visit_edge o, "expressions"
46
+ visit_edge o, "distinct"
47
+ end
48
+
49
+ def visit_Arel_Nodes_ValuesList(o)
50
+ visit_edge o, "rows"
51
+ end
52
+
53
+ def visit_Arel_Nodes_StringJoin(o)
54
+ visit_edge o, "left"
55
+ end
56
+
57
+ def visit_Arel_Nodes_InnerJoin(o)
58
+ visit_edge o, "left"
59
+ visit_edge o, "right"
60
+ end
61
+ alias :visit_Arel_Nodes_FullOuterJoin :visit_Arel_Nodes_InnerJoin
62
+ alias :visit_Arel_Nodes_OuterJoin :visit_Arel_Nodes_InnerJoin
63
+ alias :visit_Arel_Nodes_RightOuterJoin :visit_Arel_Nodes_InnerJoin
64
+
65
+ def visit_Arel_Nodes_DeleteStatement(o)
66
+ visit_edge o, "relation"
67
+ visit_edge o, "wheres"
68
+ end
69
+
70
+ def unary(o)
71
+ visit_edge o, "expr"
72
+ end
73
+ alias :visit_Arel_Nodes_Group :unary
74
+ alias :visit_Arel_Nodes_Cube :unary
75
+ alias :visit_Arel_Nodes_RollUp :unary
76
+ alias :visit_Arel_Nodes_GroupingSet :unary
77
+ alias :visit_Arel_Nodes_GroupingElement :unary
78
+ alias :visit_Arel_Nodes_Grouping :unary
79
+ alias :visit_Arel_Nodes_Having :unary
80
+ alias :visit_Arel_Nodes_Limit :unary
81
+ alias :visit_Arel_Nodes_Not :unary
82
+ alias :visit_Arel_Nodes_Offset :unary
83
+ alias :visit_Arel_Nodes_On :unary
84
+ alias :visit_Arel_Nodes_UnqualifiedColumn :unary
85
+ alias :visit_Arel_Nodes_OptimizerHints :unary
86
+ alias :visit_Arel_Nodes_Preceding :unary
87
+ alias :visit_Arel_Nodes_Following :unary
88
+ alias :visit_Arel_Nodes_Rows :unary
89
+ alias :visit_Arel_Nodes_Range :unary
90
+
91
+ def window(o)
92
+ visit_edge o, "partitions"
93
+ visit_edge o, "orders"
94
+ visit_edge o, "framing"
95
+ end
96
+ alias :visit_Arel_Nodes_Window :window
97
+
98
+ def named_window(o)
99
+ visit_edge o, "partitions"
100
+ visit_edge o, "orders"
101
+ visit_edge o, "framing"
102
+ visit_edge o, "name"
103
+ end
104
+ alias :visit_Arel_Nodes_NamedWindow :named_window
105
+
106
+ def function(o)
107
+ visit_edge o, "expressions"
108
+ visit_edge o, "distinct"
109
+ visit_edge o, "alias"
110
+ end
111
+ alias :visit_Arel_Nodes_Exists :function
112
+ alias :visit_Arel_Nodes_Min :function
113
+ alias :visit_Arel_Nodes_Max :function
114
+ alias :visit_Arel_Nodes_Avg :function
115
+ alias :visit_Arel_Nodes_Sum :function
116
+
117
+ def extract(o)
118
+ visit_edge o, "expressions"
119
+ visit_edge o, "alias"
120
+ end
121
+ alias :visit_Arel_Nodes_Extract :extract
122
+
123
+ def visit_Arel_Nodes_NamedFunction(o)
124
+ visit_edge o, "name"
125
+ visit_edge o, "expressions"
126
+ visit_edge o, "distinct"
127
+ visit_edge o, "alias"
128
+ end
129
+
130
+ def visit_Arel_Nodes_InsertStatement(o)
131
+ visit_edge o, "relation"
132
+ visit_edge o, "columns"
133
+ visit_edge o, "values"
134
+ end
135
+
136
+ def visit_Arel_Nodes_SelectCore(o)
137
+ visit_edge o, "source"
138
+ visit_edge o, "projections"
139
+ visit_edge o, "wheres"
140
+ visit_edge o, "windows"
141
+ end
142
+
143
+ def visit_Arel_Nodes_SelectStatement(o)
144
+ visit_edge o, "cores"
145
+ visit_edge o, "limit"
146
+ visit_edge o, "orders"
147
+ visit_edge o, "offset"
148
+ end
149
+
150
+ def visit_Arel_Nodes_UpdateStatement(o)
151
+ visit_edge o, "relation"
152
+ visit_edge o, "wheres"
153
+ visit_edge o, "values"
154
+ end
155
+
156
+ def visit_Arel_Table(o)
157
+ visit_edge o, "name"
158
+ end
159
+
160
+ def visit_Arel_Nodes_Casted(o)
161
+ visit_edge o, "val"
162
+ visit_edge o, "attribute"
163
+ end
164
+
165
+ def visit_Arel_Attribute(o)
166
+ visit_edge o, "relation"
167
+ visit_edge o, "name"
168
+ end
169
+ alias :visit_Arel_Attributes_Integer :visit_Arel_Attribute
170
+ alias :visit_Arel_Attributes_Float :visit_Arel_Attribute
171
+ alias :visit_Arel_Attributes_String :visit_Arel_Attribute
172
+ alias :visit_Arel_Attributes_Time :visit_Arel_Attribute
173
+ alias :visit_Arel_Attributes_Boolean :visit_Arel_Attribute
174
+ alias :visit_Arel_Attributes_Attribute :visit_Arel_Attribute
175
+
176
+ def nary(o)
177
+ o.children.each_with_index do |x, i|
178
+ edge(i) { visit x }
179
+ end
180
+ end
181
+ alias :visit_Arel_Nodes_And :nary
182
+
183
+ def binary(o)
184
+ visit_edge o, "left"
185
+ visit_edge o, "right"
186
+ end
187
+ alias :visit_Arel_Nodes_As :binary
188
+ alias :visit_Arel_Nodes_Assignment :binary
189
+ alias :visit_Arel_Nodes_Between :binary
190
+ alias :visit_Arel_Nodes_Concat :binary
191
+ alias :visit_Arel_Nodes_DoesNotMatch :binary
192
+ alias :visit_Arel_Nodes_Equality :binary
193
+ alias :visit_Arel_Nodes_GreaterThan :binary
194
+ alias :visit_Arel_Nodes_GreaterThanOrEqual :binary
195
+ alias :visit_Arel_Nodes_In :binary
196
+ alias :visit_Arel_Nodes_JoinSource :binary
197
+ alias :visit_Arel_Nodes_LessThan :binary
198
+ alias :visit_Arel_Nodes_LessThanOrEqual :binary
199
+ alias :visit_Arel_Nodes_IsNotDistinctFrom :binary
200
+ alias :visit_Arel_Nodes_IsDistinctFrom :binary
201
+ alias :visit_Arel_Nodes_Matches :binary
202
+ alias :visit_Arel_Nodes_NotEqual :binary
203
+ alias :visit_Arel_Nodes_NotIn :binary
204
+ alias :visit_Arel_Nodes_Or :binary
205
+ alias :visit_Arel_Nodes_Over :binary
206
+
207
+ def visit_String(o)
208
+ @node_stack.last.fields << o
209
+ end
210
+ alias :visit_Time :visit_String
211
+ alias :visit_Date :visit_String
212
+ alias :visit_DateTime :visit_String
213
+ alias :visit_NilClass :visit_String
214
+ alias :visit_TrueClass :visit_String
215
+ alias :visit_FalseClass :visit_String
216
+ alias :visit_Integer :visit_String
217
+ alias :visit_BigDecimal :visit_String
218
+ alias :visit_Float :visit_String
219
+ alias :visit_Symbol :visit_String
220
+ alias :visit_Arel_Nodes_SqlLiteral :visit_String
221
+
222
+ def visit_Arel_Nodes_BindParam(o); end
223
+
224
+ def visit_Hash(o)
225
+ o.each_with_index do |pair, i|
226
+ edge("pair_#{i}") { visit pair }
227
+ end
228
+ end
229
+
230
+ def visit_Array(o)
231
+ o.each_with_index do |x, i|
232
+ edge(i) { visit x }
233
+ end
234
+ end
235
+ alias :visit_Set :visit_Array
236
+
237
+ def visit_Arel_Nodes_Comment(o)
238
+ visit_edge(o, "values")
239
+ end
240
+
241
+ def visit_edge(o, method)
242
+ edge(method) { visit o.send(method) }
243
+ end
244
+
245
+ def visit(o)
246
+ if node = @seen[o.object_id]
247
+ @edge_stack.last.to = node
248
+ return
249
+ end
250
+
251
+ node = Node.new(o.class.name, o.object_id)
252
+ @seen[node.id] = node
253
+ @nodes << node
254
+ with_node node do
255
+ super
256
+ end
257
+ end
258
+
259
+ def edge(name)
260
+ edge = Edge.new(name, @node_stack.last)
261
+ @edge_stack.push edge
262
+ @edges << edge
263
+ yield
264
+ @edge_stack.pop
265
+ end
266
+
267
+ def with_node(node)
268
+ if edge = @edge_stack.last
269
+ edge.to = node
270
+ end
271
+
272
+ @node_stack.push node
273
+ yield
274
+ @node_stack.pop
275
+ end
276
+
277
+ def quote(string)
278
+ string.to_s.gsub('"', '\"')
279
+ end
280
+
281
+ def to_dot
282
+ "digraph \"Arel\" {\nnode [width=0.375,height=0.25,shape=record];\n" +
283
+ @nodes.map { |node|
284
+ label = "<f0>#{node.name}"
285
+
286
+ node.fields.each_with_index do |field, i|
287
+ label += "|<f#{i + 1}>#{quote field}"
288
+ end
289
+
290
+ "#{node.id} [label=\"#{label}\"];"
291
+ }.join("\n") + "\n" + @edges.map { |edge|
292
+ "#{edge.from.id} -> #{edge.to.id} [label=\"#{edge.name}\"];"
293
+ }.join("\n") + "\n}"
294
+ end
295
+ end
296
+ end
297
+ end