100 lines
2.7 KiB
PHP
100 lines
2.7 KiB
PHP
|
<?php
|
|||
|
|
|
|||
|
|
namespace bethrezen\arangodb;
|
|||
|
|
|
|||
|
|
use yii;
|
|||
|
|
|
|||
|
|
|
|||
|
|
use triagens\ArangoDb\Document;
|
|||
|
|
|
|||
|
|
class ArangoModel extends \yii\base\model {
|
|||
|
|
|
|||
|
|
private $_isNewRecord = true;
|
|||
|
|
|
|||
|
|
private $_doc = null;
|
|||
|
|
|
|||
|
|
|
|||
|
|
public static function findById($id)
|
|||
|
|
{
|
|||
|
|
$parts = explode("\\", $id);
|
|||
|
|
if (count($parts)==2) {
|
|||
|
|
$id = $parts[1]; // для формата "Collection\1237643123"
|
|||
|
|
} else {
|
|||
|
|
$parts = explode("/", $id); // для формата "Collection/123123321"
|
|||
|
|
if (count($parts)==2) {
|
|||
|
|
$id = $parts[1];
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
$model = new static;
|
|||
|
|
$model
|
|||
|
|
->setDocument(Yii::$app->arango->getDocument(static::class_to_collection(get_called_class()), $id))
|
|||
|
|
->setIsNewRecord(false);
|
|||
|
|
|
|||
|
|
return $model;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @todo функция должна возвращать true/false в зависимости от результата
|
|||
|
|
* Но аранга возвращает различный тип данных. Надо написать код
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
public function save()
|
|||
|
|
{
|
|||
|
|
if ($this->_isNewRecord) {
|
|||
|
|
// добавляем запись
|
|||
|
|
$this->_doc = Document::createFromArray($this->getAttributes());
|
|||
|
|
|
|||
|
|
return intval(Yii::$app->arango->documentHandler()->add(static::class_to_collection(get_called_class()), $this->_doc)) > 0;
|
|||
|
|
} else {
|
|||
|
|
// патчим!
|
|||
|
|
$doc_attributes = array_keys($this->_doc->getAll());
|
|||
|
|
|
|||
|
|
$attributes = $this->getAttributes();
|
|||
|
|
foreach ($attributes as $k=>$v) {
|
|||
|
|
$this->_doc->set($k, $v);
|
|||
|
|
unset($doc_attributes[$k]);
|
|||
|
|
}
|
|||
|
|
foreach ($doc_attributes as $key) {
|
|||
|
|
if ($key != '_key')
|
|||
|
|
unset($this->_doc->$key);
|
|||
|
|
}
|
|||
|
|
return Yii::$app->arango->documentHandler()->update($this->_doc);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static function class_to_collection($class)
|
|||
|
|
{
|
|||
|
|
$parts = explode("\\", $class);
|
|||
|
|
return end($parts);
|
|||
|
|
}
|
|||
|
|
private static function id_to_int($class)
|
|||
|
|
{
|
|||
|
|
$parts = explode("/", $class);
|
|||
|
|
return end($parts);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public function setIsNewRecord($state)
|
|||
|
|
{
|
|||
|
|
$this->_isNewRecord = $state;
|
|||
|
|
return $this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public function setDocument($doc)
|
|||
|
|
{
|
|||
|
|
$this->_doc = $doc;
|
|||
|
|
$all = $this->_doc->getAll();
|
|||
|
|
$this->_id = $this->_doc->getInternalId();
|
|||
|
|
$this->setAttributes($all, false);
|
|||
|
|
|
|||
|
|
return $this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public function delete()
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
Yii::$app->arango->documentHandler()->deleteById(
|
|||
|
|
static::class_to_collection(get_called_class()),
|
|||
|
|
static::id_to_int($this->_doc->getInternalId())
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|