Use commit timestamps for ElasticSearch and sort on last commit time.

[phorkie.git] / src / phorkie / Database / Adapter / Elasticsearch / Search.php

diff --git a/src/phorkie/Database/Adapter/Elasticsearch/Search.php b/src/phorkie/Database/Adapter/Elasticsearch/Search.php
index 88ad4ac8f0fb0dd6d73d38fa18ede6018d4f7f6e..ddbc13b7286e2793ac176aaf39f95fb7fc59dd15 100644 (file)
--- a/src/phorkie/Database/Adapter/Elasticsearch/Search.php
+++ b/src/phorkie/Database/Adapter/Elasticsearch/Search.php
@@ -1,13 +1,79 @@
 <?php
 namespace phorkie;
 
-class Database_Adapter_Elasticsearch_Search
+class Database_Adapter_Elasticsearch_Search implements Database_ISearch
 {
+    protected static $sortMap = array(
+        'id' => array('id', 'asc'),
+        'crdate' => array('crdate', 'desc'),
+        'modate' => array('modate', 'desc'),
+        'tstamp' => array('tstamp', 'desc'),
+    );
+
     public function __construct()
     {
         $this->searchInstance = $GLOBALS['phorkie']['cfg']['elasticsearch'];
     }
 
+    /**
+     * List all repositories
+     *
+     * @param integer $page    Page of search results, starting with 0
+     * @param integer $perPage Number of results per page
+     * @param string  $sort    Sort order. Allowed values:
+     *                         - id     - repository id
+     *                         - crdate - creation date
+     *                         - modate - modification date
+     *                         - tstamp - last index date
+     *
+     * @return Search_Result Search result object
+     */
+    public function listAll($page = 0, $perPage = 10, $sort = 'id', $sortOrder = null)
+    {
+        list($sortField, $orderField) = $this->getSortField($sort, $sortOrder);
+        $r = new Database_Adapter_Elasticsearch_HTTPRequest(
+            $this->searchInstance . 'repo/_search',
+            \HTTP_Request2::METHOD_GET
+        );
+        $r->setBody(
+            json_encode(
+                (object)array(
+                    'from'  => $page * $perPage,
+                    'size'  => $perPage,
+                    'sort'  => array(
+                        $sortField => $orderField
+                    ),
+                    'query' => (object)array(
+                        'match_all' => (object)array()
+                    ),
+                )
+            )
+        );
+        $httpRes = $r->send();
+        $jRes = json_decode($httpRes->getBody());
+        if (isset($jRes->error)) {
+            throw new Exception(
+                'Search exception: ' . $jRes->error, $jRes->status
+            );
+        }
+
+        $sres = new Search_Result();
+        $sres->results = $jRes->hits->total;
+        $sres->page    = $page;
+        $sres->perPage = $perPage;
+
+        foreach ($jRes->hits->hits as $hit) {
+            $r = new Repository();
+            $r->loadById($hit->_source->id);
+            $r->crdate = strtotime($hit->_source->crdate);
+            $r->modate = strtotime($hit->_source->modate);
+            $sres->repos[] = $r;
+        }
+
+        return $sres;
+    }
+
+
     /**
      * Search for a given term and return repositories that contain it
      * in their description, file names or file content
@@ -77,6 +143,22 @@ class Database_Adapter_Elasticsearch_Search
 
         return $sres;
     }
+
+    protected function getSortField($sort, $sortOrder)
+    {
+        if (!isset(self::$sortMap[$sort])) {
+            throw new Exception('Invalid sort parameter: ' . $sort);
+        }
+        if ($sortOrder !== 'asc' && $sortOrder !== 'desc') {
+            throw new Exception('Invalid sortOrder parameter: ' . $sortOrder);
+        }
+
+        $data = self::$sortMap[$sort];
+        if ($sortOrder !== null) {
+            $data[1] = $sortOrder;
+        }
+        return $data;
+    }
 }
 
 ?>