我正在运行一个在线手提包商店,手提包可以是四种颜色 - 黑色,棕色,橙色和红色。我注意到,黑色手袋早于棕色手袋等等。这意味着人们最喜欢黑色手袋。如何改进这段代码
在网上商店的主页上,我想选择和显示网格布局中的10袋。所以我首先选择黑色的包包。如果我的库存中有10个或更多的黑色袋子,那么我会停下来,不要去寻找其他颜色的袋子。但是,如果我有5个黑色包包,我会继续寻找棕色包包。如果我还没有10袋,那么加入那些棕色袋子后,我会寻找橙色袋子等等。
下面是我在实施这一方案为Rails的模型方法的尝试:
class Handbag < ActiveRecord::Base
belongs_to :store
attr_accessor :color
end
class Store < ActiveRecord::Base
has_many :handbags
def handags_for_display
selected_handbags = []
["black", "brown", "orange", "red"].each do |color|
bags = get_handbags_by_color(color)
selected_bags += bags
if selected_bags.size >= 10
selected_handbags = selected_handbags[0..9]
break
end
end
selected_handbags
end
private
def get_handbags_by_color(color)
handbags.where("color = ?", color).limit(10)
end
end
虽然这个作品,我很好奇,如果有更好的方式来写它。特别是,我认为这个代码可以转换为使用Ruby的枚举器。
这对[Code Review.SE](http://s.tk/review)是一个更好的问题。 –