store_controller.rb

来自「ruby on rails web敏捷开发之路第二版 源代码」· RB 代码 · 共 44 行

#---
# Excerpted from "Agile Web Development with Rails, 2nd Ed."
# We make no guarantees that this code is fit for any purpose. 
# Visit http://www.pragmaticprogrammer.com/titles/rails2 for more book information.
#---
class StoreController < ApplicationController
  
  def index
    @products = Product.find_products_for_sale
    @cart = find_cart
  end


  
  def add_to_cart
    begin                     
      product = Product.find(params[:id])  
    rescue ActiveRecord::RecordNotFound
      logger.error("Attempt to access invalid product #{params[:id]}")
      redirect_to_index("Invalid product")
    else
      @cart = find_cart
      @cart.add_product(product)
    end
  end
  

  def empty_cart
    session[:cart] = nil
    redirect_to_index("Your cart is currently empty")
  end

  private
  
  def redirect_to_index(msg)
    flash[:notice] = msg
    redirect_to :action => :index
  end
    
  def find_cart
    session[:cart] ||= Cart.new
  end
end