这段代码是否有效?

时间:2015-01-13 08:50:59

标签: sql ruby-on-rails database-design

我只是在数据库中已存在变量tagcolor的情况下尝试分配它。如果颜色存在,此代码查询DB两次,确定它是否确实存在,另一次查询实际分配它。

如果它不存在,我只想将tagcolor分配给用户提供的颜色。

if Tagmap.where("name = ? AND user_id = ?", tag, current_user.id).first.present?
  tagcolor = Tagmap.where("name = ? AND user_id = ?", tag, current_user.id).first.color
else
  tagcolor = params[:color].downcase
end

我可以减少这个块,所以我只查询一次数据库吗?

1 个答案:

答案 0 :(得分:6)

如果您使用Rails 4,则可以执行以下操作:

tagcolor = Tagmap.find_by(name: tag, user: current_user).try(:color) || params[:color].downcase

在Rails 3中可以实现类似(但不是那么漂亮)的构造:

tagcolor = Tagmap.find_by_name_and_user(tag, current_user).try(:color) || params[:color].downcase