Refactoring rails nested methods

Question

Im trying to refactor my code in the Team model. Teams has:many Projections, and Projections has stats for a player (goals, assists, hits etc). I want to implement one method that can tally a specific total if you pass it the parameter you are looking for.

Something like this, where I could call: example_team.total(goals) or example_team.total(assists)

class Team < ApplicationRecord

# def total(value)
  #   total = 0    
  #   self.projections.each do |projection|
  #     total += projection.value
  #   end 
  #   return total
  # end

The issue im having is that im essentially trying to pass a method name within a different method as a parameter. Im getting errors when im trying to do this. How can I get this refactoring to work???

This is my current unfactored code:

class Team < ApplicationRecord

def goals
    total = 0    
    self.projections.each do |projection|
      total += projection.goals
    end 
    return total
  end 

  def assists
    total = 0    
    self.projections.each do |projection|
      total += projection.assists
    end 
    return total
  end 

  def pp_points
    total = 0    
    self.projections.each do |projection|
      total += projection.pp_points
    end 
    return total
  end 
  
  def hits
    total = 0    
    self.projections.each do |projection|
      total += projection.hits
    end 
    return total
  end 

  def blocks
    total = 0    
    self.projections.each do |projection|
      total += projection.blocks
    end 
    return total
  end

yzalavin · Accepted Answer · 2021-04-21 15:45:12Z

3

To send a method dynamically you can use public_send.

class Team < ApplicationRecord
  def total(field)
    total = 0    
    self.projections.each do |projection|
      total += projection.public_send(field)
    end
    return total
  end
end

To simplify the example we can use sum that takes a symbol as an argument.

class Team < ApplicationRecord
  def total(field)
    projections.sum(field)
  end
end

To make it work you should use something like this example_team.total(:goals).

answered Apr 21, 2021 at 15:45

yzalavin

1,8461 gold badge15 silver badges19 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Refactoring rails nested methods

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related