Kohana ORM模型中的自定义方法

Question

I have these two models:

class Model_user extends ORM {
    protected $_has_many = array('credits', array('model'=>'credit', 'foreign_key'=>'user'));
}

class Model_credit extends ORM {
    protected $_belongs_to = array('user', array('model'=>'user', 'foreign_key'=>'user'));
    protected $_tb = 'credits';
    protected $_pk = 'id'; 
    // method to be implemented
    public function total() {
        return $total_credits_available;
    }
}

// accessing the 'total' method
$user = ORM::factory('user', 1);
$total_credits = $user->credits->total();

The thing is how to implement the 'total' method, which does something like:

return DB::select(array(DB::expr('SUM(qty * sign)'), 'total'))
    ->from($this->_tb)
    ->where('user', '=', $user_id)
    ->execute()
    ->total;

The method counts users's credits (it can be +credits and -credits) and return the total amount of credits available. As long as I use ORM::factory method to create the user object, I'd like to implement the 'total' method in such manner it uses loaded resources rather than running a new query (I wrote the above query to demonstrate the business logic).

Any suggestions? :)

Answer 1

<?php
class Model_user extends ORM {

        protected static $_totals = array();

        protected $_has_many = array('credits', array('model'=>'credit', 'foreign_key'=>'user'));

        public function total()
        {
            $total = Arr::get(Model_User::$_totals, $this->id, FALSE);

            if ($total !== FALSE)
                return $total;

            return Model_User::$_totals[$this->id] = $this->credits
                ->select(array(DB::expr('SUM(qty * sign)'), 'total'))
                ->find()
                ->total;
        }
    }