Kohana ORM пользовательские методы в моделях - PullRequest
Новая
       23

Kohana ORM пользовательские методы в моделях

4 голосов
/ 31 января 2011

У меня есть две модели: 

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;

Метод подсчитывает кредиты пользователей (это может быть + кредиты и -кредиты) и возвращает общую сумму доступных кредитов. Пока я использую метод ORM :: factory для создания пользовательского объекта, я хотел бы реализовать метод total таким образом, чтобы он использовал загруженные ресурсы, а не выполнял новый запрос (я написал вышеупомянутый запрос, чтобы продемонстрировать бизнес логика).

Есть предложения? :)

1 Ответ

6 голосов
/ 31 января 2011 
<?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;
        }
    }
...