Added Gitlab::SQL::Union class
This class can be used to join multiple AcitveRecord::Relation objects together using a SQL UNION statement. ActiveRecord < 5.0 sadly doesn't support UNION and existing Gems out there don't handle prepared statements (e.g. they never incremented the variable bindings).
This commit is contained in:
parent
054f2f98ed
commit
d769596aec
2 changed files with 50 additions and 0 deletions
34
lib/gitlab/sql/union.rb
Normal file
34
lib/gitlab/sql/union.rb
Normal file
|
@ -0,0 +1,34 @@
|
|||
module Gitlab
|
||||
module SQL
|
||||
# Class for building SQL UNION statements.
|
||||
#
|
||||
# ORDER BYs are dropped from the relations as the final sort order is not
|
||||
# guaranteed any way.
|
||||
#
|
||||
# Example usage:
|
||||
#
|
||||
# union = Gitlab::SQL::Union.new(user.personal_projects, user.projects)
|
||||
# sql = union.to_sql
|
||||
#
|
||||
# Project.where("id IN (#{sql})")
|
||||
class Union
|
||||
def initialize(relations)
|
||||
@relations = relations
|
||||
end
|
||||
|
||||
def to_sql
|
||||
# Some relations may include placeholders for prepared statements, these
|
||||
# aren't incremented properly when joining relations together this way.
|
||||
# By using "unprepared_statements" we remove the usage of placeholders
|
||||
# (thus fixing this problem), at a slight performance cost.
|
||||
fragments = ActiveRecord::Base.connection.unprepared_statement do
|
||||
@relations.map do |rel|
|
||||
"(#{rel.reorder(nil).to_sql})"
|
||||
end
|
||||
end
|
||||
|
||||
fragments.join(' UNION ')
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
16
spec/lib/gitlab/sql/union_spec.rb
Normal file
16
spec/lib/gitlab/sql/union_spec.rb
Normal file
|
@ -0,0 +1,16 @@
|
|||
require 'spec_helper'
|
||||
|
||||
describe Gitlab::SQL::Union do
|
||||
describe '#to_sql' do
|
||||
it 'returns a String joining relations together using a UNION' do
|
||||
rel1 = User.where(email: 'alice@example.com')
|
||||
rel2 = User.where(email: 'bob@example.com')
|
||||
union = described_class.new([rel1, rel2])
|
||||
|
||||
sql1 = rel1.reorder(nil).to_sql
|
||||
sql2 = rel2.reorder(nil).to_sql
|
||||
|
||||
expect(union.to_sql).to eq("(#{sql1}) UNION (#{sql2})")
|
||||
end
|
||||
end
|
||||
end
|
Loading…
Reference in a new issue