add identity and lock

[m6w6/pq-gateway] / lib / pq / Gateway / Table.php

diff --git a/lib/pq/Gateway/Table.php b/lib/pq/Gateway/Table.php
index 8fc85d7ad748ff68d816aa607afdf9fdfbc8cec1..fe5efbdd454309b2efa15fbdddb21d1f9a1a7a3e 100644 (file)
--- a/lib/pq/Gateway/Table.php
+++ b/lib/pq/Gateway/Table.php
@@ -47,6 +47,11 @@ class Table
         */
        protected $exec;
        
+       /**
+        * @var \pq\Gateway\Table\Identity
+        */
+       protected $identity;
+       
        /**
         * @var \pq\Gateway\Table\Relations
         */
@@ -57,6 +62,11 @@ class Table
         */
        protected $metadataCache;
        
+       /**
+        * @var \pq\Gateway\Table\LockInterface
+        */
+       protected $lock;
+       
        /**
         * @param string $table
         * @return \pq\Gateway\Table
@@ -179,6 +189,17 @@ class Table
                return $this;
        }
        
+       /**
+        * Get the primary key
+        * @return \pq\Gateway\Table\Identity
+        */
+       function getIdentity() {
+               if (!isset($this->identity)) {
+                       $this->identity = new Table\Identity($this);
+               }
+               return $this->identity;
+       }
+       
        /**
         * Get foreign key relations
         * @param string $to fkey
@@ -227,6 +248,24 @@ class Table
                return $this->name;
        }
 
+       /**
+        * Set a lock provider
+        * @param \pq\Gateway\Table\LockInterface $lock
+        * @return \pq\Gateway\Table
+        */
+       function setLock(Table\LockInterface $lock) {
+               $this->lock = $lock;
+               return $this;
+       }
+       
+       /**
+        * Get any set lock provider
+        * @return \pq\Gateway\Table\LockIntferace
+        */
+       function getLock() {
+               return $this->lock;
+       }
+       
        /**
         * Execute the query
         * @param \pq\Query\WriterInterface $query
@@ -262,9 +301,10 @@ class Table
         * @param array|string $order
         * @param int $limit
         * @param int $offset
+        * @param string $lock
         * @return mixed
         */
-       function find(array $where = null, $order = null, $limit = 0, $offset = 0) {
+       function find(array $where = null, $order = null, $limit = 0, $offset = 0, $lock = null) {
                $query = $this->getQueryWriter()->reset();
                $query->write("SELECT * FROM", $this->conn->quoteName($this->name));
                if ($where) {
@@ -276,7 +316,12 @@ class Table
                if ($limit) {
                        $query->write("LIMIT", $limit);
                }
-               $query->write("OFFSET", $offset);
+               if ($offset) {
+                       $query->write("OFFSET", $offset);
+               }
+               if ($lock) {
+                       $query->write("FOR", $lock);
+               }
                return $this->execute($query);
        }
        
@@ -293,7 +338,7 @@ class Table
                // select * from $this where $this->$foreignColumn = $foreign->$referencedColumn
                
                if (!isset($name)) {
-                       $name = $this->getName();
+                       $name = $foreign->getTable()->getName();
                }
                
                if (!$foreign->getTable()->hasRelation($name, $this->getName())) {