class SqlBuilder def initialize(template,klass=nil) @args = {} @sql = template @sections = {} @klass = klass end [:set, :where2,:where,:order_by,:limit,:left_join,:join,:offset, :select].each do |k| define_method k do |data, args = {}| @args.merge!(args) @sections[k] ||= [] @sections[k] << data self end end def secure_category(secure_category_ids, category_alias = 'c') if secure_category_ids.present? where("NOT COALESCE(" << category_alias << ".read_restricted, false) OR " << category_alias << ".id IN (:secure_category_ids)", secure_category_ids: secure_category_ids) else where("NOT COALESCE(" << category_alias << ".read_restricted, false)") end self end def to_sql sql = @sql.dup @sections.each do |k,v| joined = nil case k when :select joined = "SELECT " << v.join(" , ") when :where, :where2 joined = "WHERE " << v.map{|c| "(" << c << ")" }.join(" AND ") when :join joined = v.map{|v| "JOIN " << v }.join("\n") when :left_join joined = v.map{|v| "LEFT JOIN " << v }.join("\n") when :limit joined = "LIMIT " << v.last.to_s when :offset joined = "OFFSET " << v.last.to_s when :order_by joined = "ORDER BY " << v.join(" , ") when :set joined = "SET " << v.join(" , ") end sql.sub!("/*#{k}*/", joined) end sql end def exec(args = {}) @args.merge!(args) sql = to_sql if @klass @klass.find_by_sql(ActiveRecord::Base.send(:sanitize_sql_array, [sql, @args])) else if @args == {} ActiveRecord::Base.exec_sql(sql) else ActiveRecord::Base.exec_sql(sql,@args) end end end #AS reloads this on tests remove_const :FTYPE_MAP if defined? FTYPE_MAP FTYPE_MAP = { 23 => :value_to_integer, 1114 => :string_to_time, 16 => :value_to_boolean } def self.map_exec(klass, sql, args = {}) self.new(sql).map_exec(klass, args) end def map_exec(klass = OpenStruct, args = {}) results = exec(args) setters = results.fields.each_with_index.map do |f, index| [(f.dup << "=").to_sym, FTYPE_MAP[results.ftype(index)]] end values = results.values values.map! do |row| mapped = klass.new setters.each_with_index do |mapper, index| translated = row[index] if mapper[1] && !translated.nil? translated = ActiveRecord::ConnectionAdapters::Column.send mapper[1], translated end mapped.send mapper[0], translated end mapped end end end class ActiveRecord::Base def self.sql_builder(template) SqlBuilder.new(template, self) end end