class RuboCop::Cop::Rails::PluckId

This cop enforces the use of `ids` over `pluck(:id)` and `pluck(primary_key)`.

@example

# bad
User.pluck(:id)
user.posts.pluck(:id)

def self.user_ids
  pluck(primary_key)
end

# good
User.ids
user.posts.ids

def self.user_ids
  ids
end

Constants

MSG
RESTRICT_ON_SEND

Public Instance Methods

on_send(node) click to toggle source
# File lib/rubocop/cop/rails/pluck_id.rb, line 37
def on_send(node)
  return if !pluck_id_call?(node) || in_where?(node)

  range = offense_range(node)
  message = format(MSG, bad_method: range.source)

  add_offense(range, message: message) do |corrector|
    corrector.replace(offense_range(node), 'ids')
  end
end

Private Instance Methods

offense_range(node) click to toggle source
# File lib/rubocop/cop/rails/pluck_id.rb, line 50
def offense_range(node)
  range_between(node.loc.selector.begin_pos, node.loc.expression.end_pos)
end