aboutsummaryrefslogtreecommitdiffstats
path: root/lib/arel/relations
diff options
context:
space:
mode:
Diffstat (limited to 'lib/arel/relations')
-rw-r--r--lib/arel/relations/alias.rb19
-rw-r--r--lib/arel/relations/compound.rb16
-rw-r--r--lib/arel/relations/deletion.rb25
-rw-r--r--lib/arel/relations/grouping.rb19
-rw-r--r--lib/arel/relations/insertion.rb28
-rw-r--r--lib/arel/relations/join.rb92
-rw-r--r--lib/arel/relations/nil.rb12
-rw-r--r--lib/arel/relations/order.rb21
-rw-r--r--lib/arel/relations/projection.rb23
-rw-r--r--lib/arel/relations/relation.rb158
-rw-r--r--lib/arel/relations/selection.rb21
-rw-r--r--lib/arel/relations/skip.rb15
-rw-r--r--lib/arel/relations/table.rb48
-rw-r--r--lib/arel/relations/take.rb15
-rw-r--r--lib/arel/relations/update.rb30
-rw-r--r--lib/arel/relations/writing.rb4
16 files changed, 546 insertions, 0 deletions
diff --git a/lib/arel/relations/alias.rb b/lib/arel/relations/alias.rb
new file mode 100644
index 0000000000..2dab52515f
--- /dev/null
+++ b/lib/arel/relations/alias.rb
@@ -0,0 +1,19 @@
+module Arel
+ class Alias < Compound
+ attr_reader :alias
+
+ def initialize(relation, aliaz)
+ @relation, @alias = relation, aliaz
+ end
+
+ def alias?
+ true
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ @alias == other.alias
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/compound.rb b/lib/arel/relations/compound.rb
new file mode 100644
index 0000000000..4ffac6d1c3
--- /dev/null
+++ b/lib/arel/relations/compound.rb
@@ -0,0 +1,16 @@
+module Arel
+ class Compound < Relation
+ attr_reader :relation
+
+ hash_on :relation
+
+ delegate :joins, :selects, :orders, :groupings, :table_sql, :inserts, :taken,
+ :skipped, :name, :alias, :aggregation?, :alias?, :prefix_for, :column_for,
+ :engine,
+ :to => :relation
+
+ def attributes
+ relation.attributes.collect { |a| a.bind(self) }
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/deletion.rb b/lib/arel/relations/deletion.rb
new file mode 100644
index 0000000000..6c802ba905
--- /dev/null
+++ b/lib/arel/relations/deletion.rb
@@ -0,0 +1,25 @@
+module Arel
+ class Deletion < Writing
+ def initialize(relation)
+ @relation = relation
+ end
+
+ def to_sql(formatter = nil)
+ [
+ "DELETE",
+ "FROM #{table_sql}",
+ ("WHERE #{selects.collect(&:to_sql).join('\n\tAND ')}" unless selects.blank? ),
+ ("LIMIT #{taken}" unless taken.blank? ),
+ ].compact.join("\n")
+ end
+
+ def call(connection = engine.connection)
+ connection.delete(to_sql)
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/grouping.rb b/lib/arel/relations/grouping.rb
new file mode 100644
index 0000000000..ccca600360
--- /dev/null
+++ b/lib/arel/relations/grouping.rb
@@ -0,0 +1,19 @@
+module Arel
+ class Grouping < Compound
+ attr_reader :expressions, :groupings
+
+ def initialize(relation, *groupings)
+ @relation, @groupings = relation, groupings.collect { |g| g.bind(relation) }
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ groupings == other.groupings
+ end
+
+ def aggregation?
+ true
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/insertion.rb b/lib/arel/relations/insertion.rb
new file mode 100644
index 0000000000..37e7be8757
--- /dev/null
+++ b/lib/arel/relations/insertion.rb
@@ -0,0 +1,28 @@
+module Arel
+ class Insertion < Writing
+ attr_reader :record
+
+ def initialize(relation, record)
+ @relation, @record = relation, record.bind(relation)
+ end
+
+ def to_sql(formatter = nil)
+ [
+ "INSERT",
+ "INTO #{table_sql}",
+ "(#{record.keys.collect(&:to_sql).join(', ')})",
+ "VALUES (#{record.collect { |key, value| key.format(value) }.join(', ')})"
+ ].join("\n")
+ end
+
+ def call(connection = engine.connection)
+ connection.insert(to_sql)
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ record == other.record
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/join.rb b/lib/arel/relations/join.rb
new file mode 100644
index 0000000000..fb51ea0260
--- /dev/null
+++ b/lib/arel/relations/join.rb
@@ -0,0 +1,92 @@
+module Arel
+ class Join < Relation
+ attr_reader :join_sql, :relation1, :relation2, :predicates
+
+ delegate :engine, :to => :relation1
+
+ hash_on :relation1
+
+ def initialize(join_sql, relation1, relation2 = Nil.new, *predicates)
+ @join_sql, @relation1, @relation2, @predicates = join_sql, relation1, relation2, predicates
+ end
+
+ def ==(other)
+ self.class == other.class and
+ predicates == other.predicates and (
+ (relation1 == other.relation1 and relation2 == other.relation2) or
+ (relation2 == other.relation1 and relation1 == other.relation2)
+ )
+ end
+
+ def attributes
+ (externalize(relation1).attributes +
+ externalize(relation2).attributes).collect { |a| a.bind(self) }
+ end
+
+ def prefix_for(attribute)
+ if relation1[attribute] && !relation2[attribute]
+ externalize(relation1).prefix_for(attribute)
+ elsif relation2[attribute] && !relation1[attribute]
+ externalize(relation2).prefix_for(attribute)
+ else
+ if (attribute % relation1[attribute]).size < (attribute % relation2[attribute]).size
+ externalize(relation1).prefix_for(attribute)
+ else
+ externalize(relation2).prefix_for(attribute)
+ end
+ end
+ end
+
+ def joins
+ this_join = [
+ join_sql,
+ externalize(relation2).table_sql,
+ ("ON" unless predicates.blank?),
+ predicates.collect { |p| p.bind(self).to_sql }.join(' AND ')
+ ].compact.join(" ")
+ [relation1.joins, relation2.joins, this_join].compact.join(" ")
+ end
+
+ def selects
+ externalize(relation1).selects + externalize(relation2).selects
+ end
+
+ def table_sql
+ externalize(relation1).table_sql
+ end
+
+ private
+ def externalize(relation)
+ Externalizer.new(relation)
+ end
+
+ Externalizer = Struct.new(:relation) do
+ delegate :engine, :to => :relation
+
+ def table_sql
+ case
+ when relation.aggregation?
+ relation.to_sql(Sql::TableReference.new(engine))
+ when relation.alias?
+ relation.table_sql + ' AS ' + engine.quote_table_name(relation.alias.to_s)
+ else
+ relation.table_sql
+ end
+ end
+
+ def selects
+ relation.aggregation?? [] : relation.selects
+ end
+
+ def attributes
+ relation.aggregation?? relation.attributes.collect(&:to_attribute) : relation.attributes
+ end
+
+ def prefix_for(attribute)
+ if relation[attribute]
+ relation.alias?? relation.alias : relation.prefix_for(attribute)
+ end
+ end
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/nil.rb b/lib/arel/relations/nil.rb
new file mode 100644
index 0000000000..3c1d413953
--- /dev/null
+++ b/lib/arel/relations/nil.rb
@@ -0,0 +1,12 @@
+module Arel
+ class Nil < Relation
+ def table_sql; '' end
+
+ def to_s; '' end
+
+ def ==(other)
+ self.class == other.class
+ end
+ end
+
+end \ No newline at end of file
diff --git a/lib/arel/relations/order.rb b/lib/arel/relations/order.rb
new file mode 100644
index 0000000000..91526da02c
--- /dev/null
+++ b/lib/arel/relations/order.rb
@@ -0,0 +1,21 @@
+module Arel
+ class Order < Compound
+ attr_reader :ordering
+
+ def initialize(relation, *orders)
+ ordering = orders.pop
+ @relation = orders.empty?? relation : Order.new(relation, *orders)
+ @ordering = ordering.bind(@relation)
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ ordering == other.ordering
+ end
+
+ def orders
+ relation.orders + [ordering]
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/projection.rb b/lib/arel/relations/projection.rb
new file mode 100644
index 0000000000..f09d4f894b
--- /dev/null
+++ b/lib/arel/relations/projection.rb
@@ -0,0 +1,23 @@
+module Arel
+ class Projection < Compound
+ attr_reader :projections
+
+ def initialize(relation, *projections)
+ @relation, @projections = relation, projections
+ end
+
+ def attributes
+ projections.collect { |p| p.bind(self) }
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ projections == other.projections
+ end
+
+ def aggregation?
+ attributes.any?(&:aggregation?)
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/relation.rb b/lib/arel/relations/relation.rb
new file mode 100644
index 0000000000..60fb7bd00a
--- /dev/null
+++ b/lib/arel/relations/relation.rb
@@ -0,0 +1,158 @@
+module Arel
+ class Relation
+ def session
+ Session.new
+ end
+
+ module Enumerable
+ include ::Enumerable
+
+ def each(&block)
+ session.read(self).each(&block)
+ end
+
+ def first
+ session.read(self).first
+ end
+ end
+ include Enumerable
+
+ module Operations
+ def join(other = nil)
+ case other
+ when String
+ Join.new(other, self)
+ when Relation
+ JoinOperation.new("INNER JOIN", self, other)
+ else
+ self
+ end
+ end
+
+ def outer_join(other)
+ JoinOperation.new("LEFT OUTER JOIN", self, other)
+ end
+
+ def [](index)
+ case index
+ when Symbol, String
+ attribute_for_name(index)
+ when Attribute, Expression
+ attribute_for_attribute(index)
+ end
+ end
+
+ def select(*predicates)
+ predicates.all?(&:blank?) ? self : Selection.new(self, *predicates)
+ end
+
+ def project(*attributes)
+ attributes.all?(&:blank?) ? self : Projection.new(self, *attributes)
+ end
+
+ def as(aliaz = nil)
+ aliaz.blank?? self : Alias.new(self, aliaz)
+ end
+
+ def order(*attributes)
+ attributes.all?(&:blank?) ? self : Order.new(self, *attributes)
+ end
+
+ def take(taken = nil)
+ taken.blank?? self : Take.new(self, taken)
+ end
+
+ def skip(skipped = nil)
+ skipped.blank?? self : Skip.new(self, skipped)
+ end
+
+ def group(*groupings)
+ groupings.all?(&:blank?) ? self : Grouping.new(self, *groupings)
+ end
+
+ module Writes
+ def insert(record)
+ session.create Insertion.new(self, record); self
+ end
+
+ def update(assignments)
+ session.update Update.new(self, assignments); self
+ end
+
+ def delete
+ session.delete Deletion.new(self); self
+ end
+ end
+ include Writes
+
+ JoinOperation = Struct.new(:join_sql, :relation1, :relation2) do
+ def on(*predicates)
+ Join.new(join_sql, relation1, relation2, *predicates)
+ end
+ end
+ end
+ include Operations
+
+ module Externalizable
+ def aggregation?
+ false
+ end
+
+ def alias?
+ false
+ end
+ end
+ include Externalizable
+
+ def to_sql(formatter = Sql::SelectStatement.new(engine))
+ formatter.select [
+ "SELECT #{attributes.collect { |a| a.to_sql(Sql::SelectClause.new(engine)) }.join(', ')}",
+ "FROM #{table_sql}",
+ (joins unless joins.blank? ),
+ ("WHERE #{selects.collect { |s| s.to_sql(Sql::WhereClause.new(engine)) }.join("\n\tAND ")}" unless selects.blank? ),
+ ("ORDER BY #{orders.collect { |o| o.to_sql(Sql::OrderClause.new(engine)) }.join(', ')}" unless orders.blank? ),
+ ("GROUP BY #{groupings.collect(&:to_sql)}" unless groupings.blank? ),
+ ("LIMIT #{taken}" unless taken.blank? ),
+ ("OFFSET #{skipped}" unless skipped.blank? )
+ ].compact.join("\n"), self.alias
+ end
+ alias_method :to_s, :to_sql
+
+ def inclusion_predicate_sql
+ "IN"
+ end
+
+ def call(connection = engine.connection)
+ connection.select_all(to_sql)
+ end
+
+ module AttributeAccessors
+ def attribute_for_name(name)
+ attributes.detect { |a| a.alias_or_name.to_s == name.to_s }
+ end
+
+ def attribute_for_attribute(attribute)
+ attributes.detect { |a| a =~ attribute }
+ end
+ end
+ include AttributeAccessors
+
+ def bind(relation)
+ self
+ end
+
+ def format(object)
+ object.to_sql(Sql::WhereCondition.new(engine))
+ end
+
+ def attributes; [] end
+ def selects; [] end
+ def orders; [] end
+ def inserts; [] end
+ def groupings; [] end
+ def joins; nil end
+ def taken; nil end
+ def skipped; nil end
+ def alias; nil end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/selection.rb b/lib/arel/relations/selection.rb
new file mode 100644
index 0000000000..38a40e1b76
--- /dev/null
+++ b/lib/arel/relations/selection.rb
@@ -0,0 +1,21 @@
+module Arel
+ class Selection < Compound
+ attr_reader :predicate
+
+ def initialize(relation, *predicates)
+ predicate = predicates.shift
+ @relation = predicates.empty?? relation : Selection.new(relation, *predicates)
+ @predicate = predicate.bind(@relation)
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ predicate == other.predicate
+ end
+
+ def selects
+ relation.selects + [predicate]
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/skip.rb b/lib/arel/relations/skip.rb
new file mode 100644
index 0000000000..f17e439ebf
--- /dev/null
+++ b/lib/arel/relations/skip.rb
@@ -0,0 +1,15 @@
+module Arel
+ class Skip < Compound
+ attr_reader :skipped
+
+ def initialize(relation, skipped)
+ @relation, @skipped = relation, skipped
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ skipped == other.skipped
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/table.rb b/lib/arel/relations/table.rb
new file mode 100644
index 0000000000..cdc03df623
--- /dev/null
+++ b/lib/arel/relations/table.rb
@@ -0,0 +1,48 @@
+module Arel
+ class Table < Relation
+ cattr_accessor :engine
+ attr_reader :name, :engine
+
+ hash_on :name
+
+ def initialize(name, engine = Table.engine)
+ @name, @engine = name.to_s, engine
+ end
+
+ def attributes
+ @attributes ||= columns.collect do |column|
+ Attribute.new(self, column.name.to_sym)
+ end
+ end
+
+ def prefix_for(attribute)
+ self[attribute] and name
+ end
+
+ def column_for(attribute)
+ self[attribute] and columns.detect { |c| c.name == attribute.name.to_s }
+ end
+
+ def ==(other)
+ self.class == other.class and
+ name == other.name
+ end
+
+ def columns
+ @columns ||= engine.columns(name, "#{name} Columns")
+ end
+
+ def reset
+ @attributes = @columns = nil
+ end
+
+ def table_sql
+ engine.quote_table_name(name)
+ end
+
+ private
+ def qualifications
+ attributes.zip(attributes.collect(&:qualified_name)).to_hash
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/take.rb b/lib/arel/relations/take.rb
new file mode 100644
index 0000000000..d2743d7a6e
--- /dev/null
+++ b/lib/arel/relations/take.rb
@@ -0,0 +1,15 @@
+module Arel
+ class Take < Compound
+ attr_reader :taken
+
+ def initialize(relation, taken)
+ @relation, @taken = relation, taken
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ taken == other.taken
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/update.rb b/lib/arel/relations/update.rb
new file mode 100644
index 0000000000..f1f6776f15
--- /dev/null
+++ b/lib/arel/relations/update.rb
@@ -0,0 +1,30 @@
+module Arel
+ class Update < Writing
+ attr_reader :assignments
+
+ def initialize(relation, assignments)
+ @relation, @assignments = relation, assignments.bind(relation)
+ end
+
+ def to_sql(formatter = nil)
+ [
+ "UPDATE #{table_sql} SET",
+ assignments.collect do |attribute, value|
+ "#{value.format(attribute)} = #{attribute.format(value)}"
+ end.join(",\n"),
+ ("WHERE #{selects.collect(&:to_sql).join('\n\tAND ')}" unless selects.blank? ),
+ ("LIMIT #{taken}" unless taken.blank? )
+ ].join("\n")
+ end
+
+ def call(connection = engine.connection)
+ connection.update(to_sql)
+ end
+
+ def ==(other)
+ self.class == other.class and
+ relation == other.relation and
+ assignments == other.assignments
+ end
+ end
+end \ No newline at end of file
diff --git a/lib/arel/relations/writing.rb b/lib/arel/relations/writing.rb
new file mode 100644
index 0000000000..b871e5a520
--- /dev/null
+++ b/lib/arel/relations/writing.rb
@@ -0,0 +1,4 @@
+module Arel
+ class Writing < Compound
+ end
+end \ No newline at end of file
generated by cgit v1.2.3 (git 2.25.1) at 2025-04-04 14:05:37 +0000