Contents

class Arel::Visitors::MySQL
  INDEX_HINT_BY_TABLE_THREAD_KEY = "INDEX_HINT_BY_TABLE_THREAD_KEY"

  def accept_with_index_hint(object)
    retval = accept_without_index_hint(object)
    clear_index_hint_by_table
    retval
  end

  alias_method_chain :accept, :index_hint

  def visit_Arel_Table_with_index_hint(o, a)
    sql = visit_Arel_Table_without_index_hint(o, a)
    index_hint = get_index_hint(o.name)

    if index_hint
      append_index_hint(sql, index_hint)
    else
      sql
    end
  end

  alias_method_chain :visit_Arel_Table, :index_hint

  private

  def get_index_hint(table)
    index_hint_by_table = Thread.current[INDEX_HINT_BY_TABLE_THREAD_KEY]

    if index_hint_by_table.nil? or index_hint_by_table.empty?
      return nil
    end

    if index_hint_by_table.values.any? {|i| i.is_a?(Hash) }
      index_hint_by_table[table]
    else
      index_hint_by_table
    end
  end

  def clear_index_hint_by_table
    Thread.current[INDEX_HINT_BY_TABLE_THREAD_KEY] = nil
  end

  def append_index_hint(sql, index_hint)
    sql + " " + index_hint.map {|index, hint_type|
      index = Array(index)
      "#{hint_type} INDEX (#{index.join(', ')})"
    }.join(", ")
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
arel-mysql-index-hint-0.1.1	lib/arel-mysql-index-hint/arel-visitors-mysql.rb