Traits: ElasticSearchAuditable

Author: rankarpan

src/Traits/ElasticSearchAuditable.php

@@ -0,0 +1,108 @@
+<?php
+
+/**
+ * This file is part of the Laravel Auditing package.
+ *
+ * @author     Arpan Rank <arpan@iconscout.com>
+ * @copyright  2018
+ *
+ * For the full copyright and license information,
+ * please view the LICENSE.md file that was distributed
+ * with this source code.
+ */
+
+namespace Iconscout\Auditing\Traits;
+
+use Elasticsearch\ClientBuilder;
+use Illuminate\Support\Facades\Config;
+use Illuminate\Database\Eloquent\Collection;
+
+trait ElasticSearchAuditable
+{
+    /**
+     * @var string
+     */
+    protected $client = null;
+
+    /**
+     * @var string
+     */
+    protected $index = null;
+
+    /**
+     * @var string
+     */
+    protected $type = null;
+
+    /**
+     * ElasticSearch constructor.
+     */
+    public function __construct()
+    {
+        parent::__construct();
+
+        $this->client = ClientBuilder::create()->setHosts(Config::get('audit.drivers.es.client.hosts', ['localhost:9200']))->build();
+        $this->index = Config::get('audit.drivers.es.index', 'laravel_auditing');
+        $this->type = Config::get('audit.drivers.es.type', 'audits');
+    }
+
+    public function esAudits($page = 1, $perPage = 10)
+    {
+        $from = ($page - 1) * $perPage;
+
+        $params = [
+            'index' => $this->index,
+            'type' => $this->type,
+            'size' => $perPage,
+            'from' => $from,
+            'body' => [
+                'query' => [
+                    'bool' => [
+                        'must' => [
+                            [
+                                'term' => [
+                                    'auditable_id' => $this->id
+                                ]
+                            ],
+                            [
+                                'term' => [
+                                    'auditable_type' => $this->getMorphClass()
+                                ]
+                            ]
+                        ]
+                    ]
+                ],
+                'sort' => [
+                    'created_at' => [
+                        'order' => 'desc'
+                    ]
+                ],
+                'track_scores' => true
+            ]
+        ];
+
+        $results = $this->client->search($params);
+        $hits = $results['hits'];
+
+        $collection = Collection::make();
+
+        foreach ($hits['hits'] as $key => $result) {
+            $audit['id'] = $result['_id'];
+            $audit = array_merge($audit, $result['_source']);
+            $audit['score'] = $result['_score'];
+
+            $collection->put($key, $audit);
+        }
+
+        return [
+            'total' => $hits['total'],
+            'per_page' => $perPage,
+            'data' => $collection
+        ];
+    }
+
+    public function getEsAuditsAttribute()
+    {
+        return $this->esAudits();
+    }
+}