Return prepare query cache in sdk

Author: ilyakharev

src/Session.php

@@ -416,11 +416,18 @@ public function prepare($yql)
     {
         $statement = new Statement($this, $yql);
 
+        if ($statement->isCached())
+        {
+            return $statement;
+        }
+
         $result = $this->request('PrepareDataQuery', [
             'session_id' => $this->session_id,
             'yql_text' => $yql,
         ]);
 
+        $statement->saveInCache();
+
         return $statement;
     }
 

src/Statement.php

@@ -21,11 +21,26 @@ class Statement
      */
     protected $yql;
 
+    /**
+     * @var string
+     */
+    protected $qhash;
+
+    /**
+     * @var bool
+     */
+    protected $cached = false;
+
     /**
      * @var array
      */
     protected $params = [];
 
+    /**
+     * @var array
+     */
+    protected static $qcache = [];
+
     /**
      * @param Session $session
      * @param string $yql
@@ -35,6 +50,7 @@ public function __construct(Session $session, $yql)
         $this->session = $session;
         $this->yql = $yql;
 
+        $this->checkQueryCache();
         $this->detectParams();
     }
 
@@ -52,6 +68,14 @@ public function execute(array $parameters = [])
         return $this->session->query($q, $this->prepareParameters($parameters));
     }
 
+    /**
+     * @return bool
+     */
+    public function isCached()
+    {
+        return $this->cached;
+    }
+
     /**
      * @param array $parameters
      * @return array
@@ -81,6 +105,18 @@ protected function prepareParameters($parameters)
         return $data;
     }
 
+    /**
+     * @return void
+     */
+    protected function checkQueryCache()
+    {
+        $this->qhash = sha1(trim($this->yql));
+        if (in_array($this->qhash, static::$qcache))
+        {
+            $this->cached = true;
+        }
+    }
+
     /**
      * @return void
      */
@@ -95,4 +131,8 @@ protected function detectParams()
             }
         }
     }
+
+    public function saveInCache(){
+        static::$qcache[$this->qhash] = $this->qhash;
+    }
 }

tests/PreparedQueryTest.php

@@ -31,7 +31,7 @@ public function test(){
 
         $session = $table->createSession();
 
-        $prepared_query = $ydb->table()->session()->prepare('
+        $prepared_query = $session->prepare('
 declare $pk as Int64;
 select $pk;');
         $x = 2;