如何在上传新文件时更新每个用户帐户的存储-在Rails 3中

Question

我希望发生的是每次新文件上传时，user.space_used都会更新。

但现在还没有发生。为了获得更新，我必须在控制台中手动运行user.update_space以更新特定用户，或者使用数组循环遍历所有用户并像这样更新它。

我如何让它在正确的时间做到这一点-此外，如果我能验证User模型上的space_used列具有Upload模型上为该用户上传的所有文件的文件大小的总和，那将是很好的。

我的用户模型是这样的:

# == Schema Information
# Schema version: 20110412170916
#
# Table name: users
#
#  id                   :integer         not null, primary key
#  email                :string(255)
#  encrypted_password   :string(128)
#  password_salt        :string(255)
#  reset_password_token :string(255)
#  remember_token       :string(255)
#  remember_created_at  :datetime
#  sign_in_count        :integer
#  current_sign_in_at   :datetime
#  last_sign_in_at      :datetime
#  current_sign_in_ip   :string(255)
#  last_sign_in_ip      :string(255)
#  username             :string(255)
#  first_name           :string(255)
#  last_name            :string(255)
#  created_at           :datetime
#  updated_at           :datetime
#  invitation_token     :string(60)
#  invitation_sent_at   :datetime
#  plan_id              :integer
#  current_state        :string(255)
#  confirmation_token   :string(255)
#  confirmed_at         :datetime
#  confirmation_sent_at :datetime
#  space_used           :integer         default(0), not null
#  failed_attempts      :integer         default(0)
#  unlock_token         :string(255)
#  locked_at            :datetime
#  trial_end_date       :date
#  active_subscription  :boolean
#
class User < ActiveRecord::Base
    acts_as_voter
  devise :database_authenticatable, :confirmable, :registerable, :timeoutable,
         :recoverable, :rememberable, :trackable, :validatable, :invitable, :lockable
  attr_accessible :email, :password, :password_confirmation, :remember_me, :username, :first_name, :last_name, :plan_id
    after_save  :update_space
  def role_symbols
    roles.map do |role|
      role.name.underscore.to_sym
    end
  end
  def update_space
       total_size = 0
         if self.uploads.count > 0
           self.uploads.each do |upload|
             total_size += upload[:image_file_size]
           end
         end
    self.space_used = total_size
  end
    def space_threshold_reached?
        self.plan.storage == self.space_used        
    end
    def space_left
        (self.plan.storage * 1024 * 1024 * 1024) - self.space_used.to_f     
    end

end

我的上传模型是这样的:

# == Schema Information
# Schema version: 20110330215959
#
# Table name: uploads
#
#  id                 :integer         not null, primary key
#  name               :string(255)
#  description        :string(255)
#  image_file_name    :string(255)
#  image_content_type :string(255)
#  image_file_size    :integer
#  image_updated_at   :datetime
#  stage_id           :integer
#  user_id            :integer
#  created_at         :datetime
#  updated_at         :datetime
class Upload < ActiveRecord::Base
    acts_as_voteable
    has_attached_file :image, :styles => { 
                                                                :thumb => "64x64" },
                                                            :storage => :s3,
                                                            :path => "/:style/:id/:filename"
    validates_attachment_presence :image                    
  validates_attachment_size :image, :less_than => 10.megabytes
  validates_attachment_content_type :image, :content_type => ['image/jpeg', 'image/png', 'image/gif', 'image/jpg', 'image/JPG']
    after_post_process :update_users_space_used
    def self.total_used
        total_size = 0
        all.each do |upload|
            total_size += upload.image_file_size
        end
        return total_size
    end
    def update_users_space_used
        Authorization.current_user.space_used += self.image_file_size       
    end

end

谢谢。

编辑1:顺便说一句，我用回形针来管理上传。

Edit2:在Upload.rb模型中，我将before_save更改为回曲针的after_post_process回调，但它仍然不起作用。

Answer 1

似乎Paperclip的after_post_process回调将是你想要的。