Kohana ORM模型中的自定义方法

时间:2011-01-31 12:38:42

标签: php orm model kohana kohana-3

我有这两个模型:

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();

关键是如何实现'total'方法,它的作用如下:

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

该方法计算用户的学分(可以是+学分和-credits)并返回可用学分总数。只要我使用ORM :: factory方法创建用户对象,我就想以这样的方式实现'total'方法,它使用加载的资源而不是运行新的查询(我写了上面的查询来演示业务逻辑)。

有什么建议吗? :)

1 个答案:

答案 0 :(得分:6)

<?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;
        }
    }