From 1a13f685e91799e57d2871291f0752dd546a8d21 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mantas=20Marcinkevi=C4=8Dius?= <marc.mantas@gmail.com>
Date: Wed, 20 Jul 2016 12:00:07 +0300
Subject: [PATCH] changed abstract inner hit to nested
---
src/InnerHit/AbstractInnerHit.php | 30 -----
src/InnerHit/NestedInnerHit.php | 197 ++++++++++++++++++++++++++++++
2 files changed, 197 insertions(+), 30 deletions(-)
delete mode 100644 src/InnerHit/AbstractInnerHit.php
create mode 100644 src/InnerHit/NestedInnerHit.php
diff --git a/src/InnerHit/AbstractInnerHit.php b/src/InnerHit/AbstractInnerHit.php
deleted file mode 100644
index f2d92f9..0000000
--- a/src/InnerHit/AbstractInnerHit.php
+++ /dev/null
@@ -1,30 +0,0 @@
-<?php
-
-/*
- * This file is part of the ONGR package.
- *
- * (c) NFQ Technologies UAB <info@nfq.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
-
-namespace ONGR\ElasticsearchDSL\InnerHit;
-
-use ONGR\ElasticsearchDSL\BuilderInterface;
-
-/**
- * AbstractAggregation class.
- */
-abstract class AbstractInnerHit implements BuilderInterface
-{
- /**
- * {@inheritdoc}
- */
- abstract public function toArray();
-
- /**
- * {@inheritdoc}
- */
- abstract public function getType();
-}
diff --git a/src/InnerHit/NestedInnerHit.php b/src/InnerHit/NestedInnerHit.php
new file mode 100644
index 0000000..73dc853
--- /dev/null
+++ b/src/InnerHit/NestedInnerHit.php
@@ -0,0 +1,197 @@
+<?php
+
+/*
+ * This file is part of the ONGR package.
+ *
+ * (c) NFQ Technologies UAB <info@nfq.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ONGR\ElasticsearchDSL\InnerHit;
+
+use ONGR\ElasticsearchDSL\BuilderBag;
+use ONGR\ElasticsearchDSL\BuilderInterface;
+use ONGR\ElasticsearchDSL\NameAwareTrait;
+use ONGR\ElasticsearchDSL\ParametersTrait;
+
+/**
+ * Represents Elasticsearch top level nested inner hits.
+ *
+ * @link https://www.elastic.co/guide/en/elasticsearch/reference/current/search-request-inner-hits.html
+ */
+class NestedInnerHit implements BuilderInterface
+{
+ use ParametersTrait;
+ use NameAwareTrait;
+
+ /**
+ * @var string
+ */
+ private $path;
+
+ /**
+ * @var BuilderInterface
+ */
+ private $query;
+
+ /**
+ * @var BuilderBag
+ */
+ private $innerHits;
+
+ /**
+ * Inner hits container init.
+ *
+ * @param string $name
+ * @param string $path
+ * @param BuilderInterface $query
+ */
+ public function __construct($name, $path, BuilderInterface $query)
+ {
+ $this->setName($name);
+ $this->setPath($path);
+ $this->setQuery($query);
+ }
+
+ /**
+ * @return string
+ */
+ public function getPath()
+ {
+ return $this->path;
+ }
+
+ /**
+ * @param string $path
+ */
+ public function setPath($path)
+ {
+ $this->path = $path;
+ }
+
+ /**
+ * @return BuilderInterface
+ */
+ public function getQuery()
+ {
+ return $this->query;
+ }
+
+ /**
+ * @param BuilderInterface $query
+ */
+ public function setQuery(BuilderInterface $query)
+ {
+ $this->query = $query;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function getType()
+ {
+ return 'nested';
+ }
+
+ /**
+ * Adds a sub-innerHit.
+ *
+ * @param NestedInnerHit $innerHit
+ */
+ public function addInnerHit(NestedInnerHit $innerHit)
+ {
+ if (!$this->innerHits) {
+ $this->innerHits = new BuilderBag();
+ }
+
+ $this->innerHits->add($innerHit);
+ }
+
+ /**
+ * Returns all sub inner hits.
+ *
+ * @return BuilderInterface[]
+ */
+ public function getInnerHits()
+ {
+ if ($this->innerHits) {
+ return $this->innerHits->all();
+ } else {
+ return [];
+ }
+ }
+
+ /**
+ * Returns sub inner hit.
+ * @param string $name inner hit name to return.
+ *
+ * @return NestedInnerHit|null
+ */
+ public function getInnerHit($name)
+ {
+ if ($this->innerHits && $this->innerHits->has($name)) {
+ return $this->innerHits->get($name);
+ } else {
+ return null;
+ }
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function toArray()
+ {
+ $out = array_filter(
+ [
+ 'query' => $this->getQuery()->toArray(),
+ 'inner_hits' => $this->collectNestedInnerHits(),
+ ]
+ );
+
+ $out = [
+ $this->getPathType() => [
+ $this->getPath() => $this->processArray($out),
+ ],
+ ];
+
+ return $out;
+ }
+
+ /**
+ * Returns 'path' for neted and 'type' for parent inner hits
+ *
+ * @return null|string
+ */
+ private function getPathType()
+ {
+ switch ($this->getType()) {
+ case 'nested':
+ $type = 'path';
+ break;
+ case 'parent':
+ $type = 'type';
+ break;
+ default:
+ $type = null;
+ }
+ return $type;
+ }
+
+ /**
+ * Process all nested inner hits.
+ *
+ * @return array
+ */
+ private function collectNestedInnerHits()
+ {
+ $result = [];
+ /** @var NestedInnerHit $innerHit */
+ foreach ($this->getInnerHits() as $innerHit) {
+ $result[$innerHit->getName()] = $innerHit->toArray();
+ }
+
+ return $result;
+ }
+}
--
GitLab