Playing with iterators and results from elastic search

Author: lorenzo

Controller/AppController.php

@@ -38,8 +38,9 @@ class AppController extends Controller {
 		],
 		'Crud.Crud' => array(
 			'actions' => array('index', 'add', 'view', 'edit', 'delete'),
-			'listeners' => array('Api', 'ApiQueryLog')
-		)
+			'listeners' => array('Api', 'ApiQueryLog', 'ApiPagination')
+		),
+		'Paginator' => array('settings' => ['paramType' => 'querystring'])
 	);
 
 }

Controller/UsersController.php

@@ -1,6 +1,9 @@
 <?php
 App::uses('AppController', 'Controller');
 App::uses('CakeEvent', 'Event');
+App::uses('FavoriteIterator', 'Iterator');
+App::uses('SecurityIterator', 'Iterator');
+App::uses('ListIterator', 'Iterator');
 
 /**
  * Users Controller
@@ -11,6 +14,18 @@ class UsersController extends AppController {
 	public function index() {
 		$this->Crud->on('beforePaginate', function() {
 			$this->User->switchToElastic();
+			$this->Paginator->settings['limit'] = 20;
+			if ($this->request->query('name')) {
+				$this->Paginator->settings['query'] = [
+					'match' => ['_all' => $this->request->query('name')]
+				];
+			}
+		});
+		$this->Crud->on('afterPaginate', function($event) {
+			$event->subject()->items = new ListIterator(new SecurityIterator(new FavoriteIterator(
+				new ArrayIterator($event->subject()->items),
+				$this->Auth->user('id')
+			)));
 		});
 		return $this->Crud->executeAction();
 	}

Iterator/FavoriteIterator.php

@@ -0,0 +1,20 @@
+<?php
+
+class FavoriteIterator extends IteratorIterator {
+
+	protected $_user;
+
+	protected $_cache;
+
+	public function __construct($results, $userId) {
+		$this->_user = $userId;
+		$this->_cache = (array)Cache::read('favorites_' . $userId);
+		parent::__construct($results);
+	}
+
+	public function current() {
+		$current = parent::current();
+		$current['User']['favorite'] = !empty($this->_cache[$current['User']['id']]);
+		return $current;
+	}
+}

Iterator/ListIterator.php

@@ -0,0 +1,15 @@
+<?php
+
+class ListIterator extends IteratorIterator {
+
+	public function key() {
+		return parent::current()['User']['id'];
+	}
+
+	public function current() {
+		$current = parent::current();
+		return $current['User']['full_name'] . ' - ' . $current['User']['companies'][0]['country'];
+	}
+}
+
+

Iterator/SecurityIterator.php

@@ -0,0 +1,11 @@
+<?php
+
+class SecurityIterator extends IteratorIterator {
+
+	public function current() {
+		$current = parent::current();
+		unset($current['User']['last_name']);
+		return $current;
+	}
+}
+

Model/User.php

@@ -17,8 +17,8 @@ public function elasticMapping() {
 		return array(
 			'id' => array('type' => 'string', 'index' => 'not_analyzed'),
 			'full_name' => array('type' => 'string'),
-			'first_name' => array('type' => 'string'),
-			'last_name' => array('type' => 'string'),
+			'first_name' => array('type' => 'string', 'include_in_all' => false),
+			'last_name' => array('type' => 'string', 'include_in_all' => false),
 			'companies' => array(
 				'type'=> 'object',
 				'properties' => array(
@@ -29,7 +29,7 @@ public function elasticMapping() {
 					'state' => array('type' => 'string'),
 					'country' => array('type' => 'string'),
 					'zip' => array('type' => 'string'),
-					'website' => array('type' => 'string', 'length' => 2048),					
+					'website' => array('type' => 'string', 'length' => 2048),
 					'location' => array('type' => 'geo_point', 'lat_lon' => true),
 				)
 			),
@@ -103,7 +103,7 @@ function afterSave($created) {
 		$esData = $this->esDenormalize($this->data);
 		$this->switchToElastic();
 		$this->create();
-		$this->save($this->data, ['callbacks' => 'false']);
+		$this->save($esData, ['callbacks' => 'false']);
 		$this->switchToDatabase();
 	}