/**
* Saves a foreign relationship where a property on the supplied object references an id for another object.
*
* Can be used to save a one-to-one relationship or the "one" side of a one-to-many relationship.
*
* @param DataObjectInterface[] $objects
* @param string $foreignIdColumn Property on this object that relates to the foreign tables id
*/
public function saveOne(array $objects, $foreignIdColumn)
{
$getter = 'get' . $this->inflector->methodNameFromColumn($foreignIdColumn);
/** @var DataObjectInterface[] $foreignObjectsByObjectId */
$foreignObjectsByObjectId = [];
foreach ($objects as $object) {
if (!method_exists($object, $getter)) {
throw new MethodNotImplementedException("{$getter} must be defined on {$object->getClassName()} to save relationship");
}
$objectIdSetter = 'set' . $this->inflector->idColumnFromClass(get_class($object));
$foreignObject = $object->{$getter}();
if ($foreignObject) {
$foreignObjectsByObjectId[$object->getId()] = $foreignObject;
if (method_exists($foreignObject, $objectIdSetter)) {
// for true one-to-one relationships
$foreignObject->{$objectIdSetter}($object->getId());
}
}
}
$this->objectMapper->saveAll($foreignObjectsByObjectId);
$idSetter = 'set' . ucfirst($foreignIdColumn);
$idGetter = 'get' . ucfirst($foreignIdColumn);
$objectsToUpdate = [];
foreach ($objects as $object) {
if (!method_exists($object, $idSetter)) {
throw new MethodNotImplementedException("{$idSetter} must be defined on {$object->getClassName()} to save relationship");
}
if (!method_exists($object, $idGetter)) {
throw new MethodNotImplementedException("{$idGetter} must be defined on {$object->getClassName()} to save relationship");
}
if (isset($foreignObjectsByObjectId[$object->getId()])) {
$foreignObject = $foreignObjectsByObjectId[$object->getId()];
if ($object->{$idGetter}() != $foreignObject->getId()) {
$object->{$idSetter}($foreignObject->getId());
$objectsToUpdate[] = $object;
}
}
}
$this->objectMapper->saveAll($objectsToUpdate);
}
/**
* Loads objects of the foreign class onto the supplied objects linked by a link table containing the id's of both objects
*
* @param DataObjectInterface[] $objects
* @param string $className Class name of foreign objects to load
* @param string $linkTable Table that links two objects together
* @param string $idColumn Column on link table = the id on this object
* @param string $foreignIdColumn Column on link table = the id on the foreign object table
* @return DataObjectInterface[] Loaded objects keyed by id
*/
public function loadManyToMany(array $objects, $className, $linkTable, $idColumn = null, $foreignIdColumn = null)
{
if (empty($objects)) {
return [];
}
$ids = DataObject::getIds($objects);
$queryHelper = $this->objectMapper->getQueryHelper();
$db = $queryHelper->getConnection();
$qb = $queryHelper->buildSelectQuery($linkTable, [$db->quoteIdentifier($idColumn) . ' AS id', $db->quoteIdentifier($foreignIdColumn) . ' AS ' . $db->quoteIdentifier('foreignId')], [$idColumn => $ids]);
$foreignIdsById = [];
$foreignIds = [];
$linkRows = $qb->execute();
$linkRows->setFetchMode(\PDO::FETCH_OBJ);
foreach ($linkRows as $linkRow) {
$foreignIdsById[$linkRow->id][] = $linkRow->foreignId;
$foreignIds[$linkRow->foreignId] = true;
}
$foreignObjects = $this->objectMapper->findByIds($className, array_keys($foreignIds));
unset($foreignIds);
$foreignObjectsById = [];
foreach ($foreignObjects as $foreignObject) {
$foreignObjectsById[$foreignObject->getId()] = $foreignObject;
}
unset($foreignObjects);
$setter = 'set' . $this->inflector->methodNameFromColumn($foreignIdColumn, true);
foreach ($objects as $object) {
if (method_exists($object, $setter)) {
$foreignObjects = [];
if (isset($foreignIdsById[$object->getId()])) {
$foreignIds = $foreignIdsById[$object->getId()];
foreach ($foreignIds as $foreignId) {
$foreignObjects[] = $foreignObjectsById[$foreignId];
}
}
$object->{$setter}($foreignObjects);
} else {
throw new MethodNotImplementedException("{$setter} must be defined on {$object->getClassName()} to load many-to-many relationship with {$className}");
}
}
return $foreignObjectsById;
}
