1: 2: 3: 4: 5: 6: 7: 8: 9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 44: 45: 46: 47: 48: 49: 50: 51: 52: 53: 54: 55: 56: 57: 58: 59: 60: 61: 62: 63: 64: 65: 66: 67: 68: 69: 70: 71: 72: 73: 74: 75: 76: 77: 78: 79: 80: 81: 82: 83: 84: 85: 86: 87: 88: 89: 90: 91: 92: 93: 94: 95: 96: 97: 98: 99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 184: 185: 186: 187: 188: 189: 190: 191: 192: 193: 194: 195: 196: 197: 198: 199: 200: 201: 202: 203: 204: 205: 206: 207: 208: 209: 210: 211: 212: 213: 214: 215: 216: 217: 218: 219: 220: 221: 222: 223: 224: 225: 226: 227: 228: 229: 230: 231: 232: 233: 234: 235: 236: 237: 238: 239: 240: 241: 242: 243: 244: 245: 246: 247: 248: 249: 250: 251: 252: 253: 254: 255: 256: 257: 258: 259: 260: 261: 262: 263: 264: 265: 266: 267: 268: 269: 270: 271: 272: 273: 274: 275: 276: 277: 278: 279: 280: 281: 282: 283: 284: 285: 286: 287: 288: 289: 290: 291: 292: 293: 294: 295: 296: 297: 298: 299: 300: 301: 302: 303: 304: 305: 306: 307: 308: 309: 310: 311: 312: 313: 314: 315: 316: 317: 318: 319: 320: 321: 322: 323: 324: 325: 326: 327: 328: 329: 330: 331:
<?php
namespace LeanMapper;
use DibiRow;
use LeanMapper\Exception\InvalidArgumentException;
use LeanMapper\Exception\InvalidStateException;
use LeanMapper\Reflection\AnnotationsParser;
use ReflectionClass;
abstract class Repository
{
protected $connection;
protected $mapper;
protected $table;
protected $entityClass;
protected $events;
private $docComment;
private $tableAnnotationChecked = false;
public function __construct(Connection $connection, IMapper $mapper)
{
$this->connection = $connection;
$this->mapper = $mapper;
$this->events = new Events;
$this->initEvents();
}
public function &__get($name)
{
if (preg_match('#^on[A-Z]#', $name)) {
return $this->events->getCallbacksReference(lcfirst(substr($name, 2)));
}
}
protected function initEvents()
{
}
public function persist(Entity $entity)
{
$this->checkEntityType($entity);
$this->events->invokeCallbacks(Events::EVENT_BEFORE_PERSIST, $entity);
if ($entity->isModified()) {
if ($entity->isDetached()) {
$entity->useMapper($this->mapper);
$this->events->invokeCallbacks(Events::EVENT_BEFORE_CREATE, $entity);
$result = $id = $this->insertIntoDatabase($entity);
$entity->markAsAttached($id, $this->getTable(), $this->connection);
$this->events->invokeCallbacks(Events::EVENT_AFTER_CREATE, $entity);
} else {
$this->events->invokeCallbacks(Events::EVENT_BEFORE_UPDATE, $entity);
$result = $this->updateInDatabase($entity);
$entity->markAsUpdated();
$this->events->invokeCallbacks(Events::EVENT_AFTER_UPDATE, $entity);
}
}
$this->persistHasManyChanges($entity);
$this->events->invokeCallbacks(Events::EVENT_AFTER_PERSIST, $entity);
return isset($result) ? $result : null;
}
public function delete($arg)
{
$this->events->invokeCallbacks(Events::EVENT_BEFORE_DELETE, $arg);
if ($arg instanceof Entity) {
$this->checkEntityType($arg);
if ($arg->isDetached()) {
throw new InvalidStateException('Cannot delete detached entity.');
}
}
$this->deleteFromDatabase($arg);
if ($arg instanceof Entity) {
$arg->detach();
}
$this->events->invokeCallbacks(Events::EVENT_AFTER_DELETE, $arg);
}
protected function insertIntoDatabase(Entity $entity)
{
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
$values = $entity->getModifiedRowData();
$this->connection->query(
'INSERT INTO %n %v', $this->getTable(), $values
);
return isset($values[$primaryKey]) ? $values[$primaryKey] : $this->connection->getInsertId();
}
protected function updateInDatabase(Entity $entity)
{
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
$idField = $this->mapper->getEntityField($this->getTable(), $primaryKey);
$values = $entity->getModifiedRowData();
return $this->connection->query(
'UPDATE %n SET %a WHERE %n = ?', $this->getTable(), $values, $primaryKey, $entity->$idField
);
}
protected function deleteFromDatabase($arg)
{
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
$idField = $this->mapper->getEntityField($this->getTable(), $primaryKey);
$id = ($arg instanceof Entity) ? $arg->$idField : $arg;
$this->connection->query(
'DELETE FROM %n WHERE %n = ?', $this->getTable(), $primaryKey, $id
);
}
protected function persistHasManyChanges(Entity $entity)
{
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
$idField = $this->mapper->getEntityField($this->getTable(), $primaryKey);
foreach ($entity->getHasManyRowDifferences() as $key => $difference) {
list($columnReferencingSourceTable, $relationshipTable, $columnReferencingTargetTable) = explode(':', $key);
$multiInsert = array();
foreach ($difference as $value => $count) {
if ($count > 0) {
for ($i = 0; $i < $count; $i++) {
$multiInsert[] = array(
$columnReferencingSourceTable => $entity->$idField,
$columnReferencingTargetTable => $value,
);
}
} else {
$this->connection->query(
'DELETE FROM %n WHERE %n = ? AND %n = ? %lmt', $relationshipTable, $columnReferencingSourceTable, $entity->$idField, $columnReferencingTargetTable, $value, -$count
);
}
}
if (!empty($multiInsert)) {
$this->connection->query(
'INSERT INTO %n %ex', $relationshipTable, $multiInsert
);
}
}
}
protected function createEntity(DibiRow $dibiRow, $entityClass = null, $table = null)
{
if ($table === null) {
$table = $this->getTable();
}
$result = Result::getInstance($dibiRow, $table, $this->connection, $this->mapper);
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
$row = $result->getRow($dibiRow->$primaryKey);
if ($entityClass === null) {
$entityClass = $this->mapper->getEntityClass($this->getTable(), $row);
}
return new $entityClass($row);
}
protected function createEntities(array $rows, $entityClass = null, $table = null)
{
if ($table === null) {
$table = $this->getTable();
}
$entities = array();
$collection = Result::getInstance($rows, $table, $this->connection, $this->mapper);
$primaryKey = $this->mapper->getPrimaryKey($this->getTable());
if ($entityClass !== null) {
foreach ($rows as $dibiRow) {
$entities[$dibiRow->$primaryKey] = new $entityClass($collection->getRow($dibiRow->$primaryKey));
}
} else {
foreach ($rows as $dibiRow) {
$row = $collection->getRow($dibiRow->$primaryKey);
$entityClass = $this->mapper->getEntityClass($this->getTable(), $row);
$entities[$dibiRow->$primaryKey] = new $entityClass($row);
}
}
return $this->createCollection($entities);
}
protected function getTable()
{
if ($this->table === null) {
if (!$this->tableAnnotationChecked) {
$this->tableAnnotationChecked = true;
$table = AnnotationsParser::parseSimpleAnnotationValue('table', $this->getDocComment());
if ($table !== null) {
return $this->table = $table;
}
}
return $this->mapper->getTableByRepositoryClass(get_called_class());
}
return $this->table;
}
protected function createCollection(array $entities)
{
return $entities;
}
protected function checkEntityType(Entity $entity)
{
$entityClass = $this->mapper->getEntityClass($this->getTable());
if (!($entity instanceof $entityClass)) {
throw new InvalidArgumentException('Repository ' . get_called_class() . ' cannot handle ' . get_class($entity) . ' entity.');
}
}
private function getDocComment()
{
if ($this->docComment === null) {
$reflection = new ReflectionClass(get_called_class());
$this->docComment = $reflection->getDocComment();
}
return $this->docComment;
}
}