X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fpq%2FMapper%2FStorage.php;h=042b6947874b0addc923f1f39684d14d1139f8a6;hb=861bb294d28aba35e011d9d35aad750d93c7dbd7;hp=4882eda881fb512232d6eb85b4455a92ad6e4afc;hpb=5c080bb03b5fee867da94cb3c1108d59529f741a;p=m6w6%2Fpq-gateway

diff --git a/lib/pq/Mapper/Storage.php b/lib/pq/Mapper/Storage.php
index 4882eda..042b694 100644
--- a/lib/pq/Mapper/Storage.php
+++ b/lib/pq/Mapper/Storage.php
@@ -13,27 +13,29 @@ class Storage implements StorageInterface
 	 * The mapping of this storage
 	 * @var MapInterface
 	 */
-	var $map;
+	private $map;
 
 	/**
-	 * The underlying table gateway
-	 * @var Table
+	 * The mapper
+	 * @var Mapper
 	 */
-	private $gateway;
+	private $mapper;
 
 	/**
-	 * Buffered transaction
-	 * @var Transaction
+	 * The underlying table gateway
+	 * @var Table
 	 */
-	private $xaction;
+	private $gateway;
 
 	/**
 	 * Create a storage for $map
-	 * @param MapInterface $map
+	 * @param Mapper $mapper
+	 * @param string $class
 	 */
-	function __construct(MapInterface $map) {
-		$this->map = $map;
-		$this->gateway = $map->getGateway();
+	function __construct(Mapper $mapper, $class) {
+		$this->mapper = $mapper;
+		$this->map = $mapper->mapOf($class);
+		$this->gateway = $this->map->getGateway();
 	}
 
 	/**
@@ -44,18 +46,20 @@ class Storage implements StorageInterface
 	function get($pk) {
 		$id = $this->gateway->getIdentity();
 		if (count($id) == 1 && is_scalar($pk)) {
-			$pk = [current($id->getColumns()) => $pk];
-		} elseif (!is_array($pk) || count($pk) !== count($id)) {
+			$vals = [$pk];
+		} elseif (is_array($pk) && count($pk) === count($id)) {
+			$vals = $pk;
+		} else {
 			throw InvalidArgumentException(
 				"Insufficient identity provided; not all fields of %s are provided in %s",
 				json_encode($id->getColumns()), json_encode($pk));
 		}
 
-		$where = [];
-		foreach ($pk as $k => $v) {
-			$where["$k="] = $v;
-		}
-		$rowset = $this->gateway->find($where);
+		$keys = array_map(function($v) {
+			return "$v=";
+		}, $id->getColumns());
+
+		$rowset = $this->gateway->find(array_combine($keys, $vals));
 		
 		return $this->map->map($rowset->current());
 	}
@@ -74,6 +78,30 @@ class Storage implements StorageInterface
 		return $this->map->mapAll($rowset);
 	}
 
+	/**
+	 * Find parent
+	 * @param object $object
+	 * @param string $refName
+	 * @return object
+	 */
+	function by($object, $refName) {
+		$row = $this->mapper->mapOf($object)->getObjects()->getRow($object);
+		$this->map->refOf($row, $refName, $objects);
+		return current($objects);
+	}
+
+	/**
+	 * Find childs
+	 * @param object $object
+	 * @param string $refName
+	 * @return array
+	 */
+	function of($object, $refName) {
+		$row = $this->mapper->mapOf($object)->getObjects()->getRow($object);
+		$this->map->allOf($row, $refName, $objects);
+		return $objects;
+	}
+
 	/**
 	 * Delete
 	 * @param object $object