Error202
/
yii2-bootstrap
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

implemented redis AR ActiveQuery and find()

tags/2.0.0-beta
Carsten Brandt 12 years ago
parent
7efe3b1483
commit
34b6624410
2 changed files with 54 additions and 128 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 160
      framework/db/redis/ActiveQuery.php
  2. 22
      framework/db/redis/ActiveRecord.php

160
framework/db/redis/ActiveQuery.php
Unescape View File

@ -70,15 +70,20 @@ class ActiveQuery extends \yii\base\Component
*/
public $limit;
/**
* @var integer zero-based offset from where the records are to be returned. If not set or
* less than 0, it means starting from the beginning.
* @var integer zero-based offset from where the records are to be returned.
* If not set, it means starting from the beginning.
* If less than zero it means starting n elements from the end.
*/
public $offset;
/**
* @var string|array how to sort the query results. This refers to the ORDER BY clause in a SQL statement.
* It can be either a string (e.g. `'id ASC, name DESC'`) or an array (e.g. `array('id ASC', 'name DESC')`).
* @var array array of primary keys of the records to find.
*/
public $orderBy;
public $primaryKeys;
public function primaryKeys($primaryKeys) {
$this->primaryKeys = $primaryKeys;
return $this;
}
/**
* Executes query and returns all results as an array.
@ -86,9 +91,20 @@ class ActiveQuery extends \yii\base\Component
*/
public function all()
{
// TODO implement
$command = $this->createCommand();
$rows = $command->queryAll();
$modelClass = $this->modelClass;
/** @var Connection $db */
$db = $modelClass::getDb();
if (($primaryKeys = $this->primaryKeys) === null) {
$start = $this->offset === null ? 0 : $this->offset;
$end = $this->limit === null ? -1 : $start + $this->limit;
$primaryKeys = $db->executeCommand('LRANGE', array($modelClass::tableName(), $start, $end));
}
$rows = array();
foreach($primaryKeys as $pk) {
$key = $modelClass::tableName() . ':a:' . (is_array($pk) ? implode('-', $pk) : $pk); // TODO escape PK glue
// get attributes
$rows[] = $db->executeCommand('HGETALL', array($key));
}
if ($rows !== array()) {
$models = $this->createModels($rows);
if (!empty($this->with)) {
@ -108,9 +124,18 @@ class ActiveQuery extends \yii\base\Component
*/
public function one()
{
// TODO implement
$command = $this->createCommand();
$row = $command->queryRow();
$modelClass = $this->modelClass;
/** @var Connection $db */
$db = $modelClass::getDb();
if (($primaryKeys = $this->primaryKeys) === null) {
$start = $this->offset === null ? 0 : $this->offset;
$primaryKeys = $db->executeCommand('LRANGE', array($modelClass::tableName(), $start, $start + 1));
}
$pk = reset($primaryKeys);
$key = $modelClass::tableName() . ':a:' . (is_array($pk) ? implode('-', $pk) : $pk); // TODO escape PK glue
// get attributes
$row = $db->executeCommand('HGETALL', array($key));
// TODO check for empty list if key does not exist
if ($row !== false && !$this->asArray) {
/** @var $class ActiveRecord */
$class = $this->modelClass;
@ -132,63 +157,12 @@ class ActiveQuery extends \yii\base\Component
* Make sure you properly quote column names.
* @return integer number of records
*/
public function count($q = '*')
{
// TODO implement
$this->select = array("COUNT($q)");
return $this->createCommand()->queryScalar();
}
/**
* Returns the sum of the specified column values.
* @param string $q the column name or expression.
* Make sure you properly quote column names.
* @return integer the sum of the specified column values
*/
public function sum($q)
{
// TODO implement
$this->select = array("SUM($q)");
return $this->createCommand()->queryScalar();
}
/**
* Returns the average of the specified column values.
* @param string $q the column name or expression.
* Make sure you properly quote column names.
* @return integer the average of the specified column values.
*/
public function average($q)
{
// TODO implement
$this->select = array("AVG($q)");
return $this->createCommand()->queryScalar();
}
/**
* Returns the minimum of the specified column values.
* @param string $q the column name or expression.
* Make sure you properly quote column names.
* @return integer the minimum of the specified column values.
*/
public function min($q)
{
// TODO implement
$this->select = array("MIN($q)");
return $this->createCommand()->queryScalar();
}
/**
* Returns the maximum of the specified column values.
* @param string $q the column name or expression.
* Make sure you properly quote column names.
* @return integer the maximum of the specified column values.
*/
public function max($q)
public function count()
{
// TODO implement
$this->select = array("MAX($q)");
return $this->createCommand()->queryScalar();
$modelClass = $this->modelClass;
/** @var Connection $db */
$db = $modelClass::getDb();
return $db->executeCommand('LLEN', array($modelClass::tableName()));
}
/**
@ -197,10 +171,10 @@ class ActiveQuery extends \yii\base\Component
* @return string|boolean the value of the first column in the first row of the query result.
* False is returned if the query result is empty.
*/
public function scalar()
public function scalar($column)
{
// TODO implement
return $this->createCommand()->queryScalar();
$record = $this->one();
return $record->$column;
}
/**
@ -209,9 +183,7 @@ class ActiveQuery extends \yii\base\Component
*/
public function exists()
{
// TODO implement
$this->select = array(new Expression('1'));
return $this->scalar() !== false;
return $this->one() !== null;
}
@ -228,48 +200,6 @@ class ActiveQuery extends \yii\base\Component
}
/**
* Sets the ORDER BY part of the query.
* TODO: refactor, it is duplicated from yii/db/Query
* @param string|array $columns the columns (and the directions) to be ordered by.
* Columns can be specified in either a string (e.g. "id ASC, name DESC") or an array (e.g. array('id ASC', 'name DESC')).
* The method will automatically quote the column names unless a column contains some parenthesis
* (which means the column contains a DB expression).
* @return Query the query object itself
* @see addOrder()
*/
public function orderBy($columns)
{
$this->orderBy = $columns;
return $this;
}
/**
* Adds additional ORDER BY columns to the query.
* TODO: refactor, it is duplicated from yii/db/Query
* @param string|array $columns the columns (and the directions) to be ordered by.
* Columns can be specified in either a string (e.g. "id ASC, name DESC") or an array (e.g. array('id ASC', 'name DESC')).
* The method will automatically quote the column names unless a column contains some parenthesis
* (which means the column contains a DB expression).
* @return Query the query object itself
* @see order()
*/
public function addOrderBy($columns)
{
if (empty($this->orderBy)) {
$this->orderBy = $columns;
} else {
if (!is_array($this->orderBy)) {
$this->orderBy = preg_split('/\s*,\s*/', trim($this->orderBy), -1, PREG_SPLIT_NO_EMPTY);
}
if (!is_array($columns)) {
$columns = preg_split('/\s*,\s*/', trim($columns), -1, PREG_SPLIT_NO_EMPTY);
}
$this->orderBy = array_merge($this->orderBy, $columns);
}
return $this;
}
/**
* Sets the LIMIT part of the query.
* TODO: refactor, it is duplicated from yii/db/Query
* @param integer $limit the limit

22
framework/db/redis/ActiveRecord.php
Unescape View File

@ -60,16 +60,15 @@ abstract class ActiveRecord extends \yii\db\ActiveRecord
* returned (null will be returned if there is no matching).
* @see createQuery()
*/
public static function find($q = null)
public static function find($q = null) // TODO optimize API
{
// TODO
$query = static::createQuery();
if (is_array($q)) {
return $query->where($q)->one();
return $query->primaryKeys($q)->one();
} elseif ($q !== null) {
// query by primary key
$primaryKey = static::primaryKey();
return $query->where(array($primaryKey[0] => $q))->one();
return $query->primaryKeys(array($primaryKey[0] => $q))->one();
}
return $query;
}
@ -178,7 +177,7 @@ abstract class ActiveRecord extends \yii\db\ActiveRecord
}
}
// save pk in a findall pool
$db->executeCommand('SADD', array(static::tableName(), $pk));
$db->executeCommand('RPUSH', array(static::tableName(), $pk));
$key = static::tableName() . ':a:' . implode('-', $pk); // TODO escape PK glue
// save attributes
@ -214,7 +213,7 @@ abstract class ActiveRecord extends \yii\db\ActiveRecord
{
$db = static::getDb();
if ($condition==='') {
$condition = $db->executeCommand('SMEMBERS', array(static::tableName()));
$condition = $db->executeCommand('LRANGE', array(static::tableName(), 0, -1));
}
if (empty($attributes)) {
return 0;
@ -255,7 +254,7 @@ abstract class ActiveRecord extends \yii\db\ActiveRecord
{
$db = static::getDb();
if ($condition==='') {
$condition = $db->executeCommand('SMEMBERS', array(static::tableName()));
$condition = $db->executeCommand('LRANGE', array(static::tableName(), 0, -1));
}
$n=0;
foreach($condition as $pk) {
@ -287,23 +286,20 @@ abstract class ActiveRecord extends \yii\db\ActiveRecord
{
$db = static::getDb();
if ($condition==='') {
$condition = $db->executeCommand('SMEMBERS', array(static::tableName()));
$condition = $db->executeCommand('LRANGE', array(static::tableName(), 0, -1));
}
if (empty($condition)) {
return 0;
}
$smembers = array();
$attributeKeys = array();
foreach($condition as $pk) {
if (is_array($pk)) {
$pk = implode('-', $pk);
}
$smembers[] = $pk; // TODO escape PK glue
$db->executeCommand('LREM', array(static::tableName(), 0, $pk)); // TODO escape PK glue
$attributeKeys[] = static::tableName() . ':' . $pk . ':a'; // TODO escape PK glue
}
array_unshift($smembers, static::tableName());
$db->executeCommand('DEL', $attributeKeys);
return $db->executeCommand('SREM', $smembers);
return $db->executeCommand('DEL', $attributeKeys);
}
/**

Write Preview
Loading…
Cancel
Save