1
|
|
|
<?php |
2
|
|
|
/** |
3
|
|
|
* Class ElasticConfigureSynonymsTask|Firesphere\ElasticSearch\Tasks\ElasticConfigureSynonymsTask Index Elastic cores |
4
|
|
|
* |
5
|
|
|
* @package Firesphere\Elastic\Search |
6
|
|
|
* @author Simon `Firesphere` Erkelens; Marco `Sheepy` Hermo |
7
|
|
|
* @copyright Copyright (c) 2018 - now() Firesphere & Sheepy |
8
|
|
|
*/ |
9
|
|
|
|
10
|
|
|
namespace Firesphere\ElasticSearch\Tasks; |
11
|
|
|
|
12
|
|
|
use Elastic\Elasticsearch\Exception\ClientResponseException; |
13
|
|
|
use Elastic\Elasticsearch\Exception\MissingParameterException; |
14
|
|
|
use Elastic\Elasticsearch\Exception\ServerResponseException; |
15
|
|
|
use Firesphere\ElasticSearch\Models\SynonymSet; |
16
|
|
|
use Firesphere\ElasticSearch\Services\ElasticCoreService; |
17
|
|
|
use Firesphere\SearchBackend\Helpers\Synonyms as BaseSynonyms; |
18
|
|
|
use Firesphere\SearchBackend\Models\SearchSynonym; |
19
|
|
|
use SilverStripe\Core\Injector\Injector; |
20
|
|
|
use SilverStripe\Dev\BuildTask; |
21
|
|
|
|
22
|
|
|
/** |
23
|
|
|
* Class ElasticConfigureSynonymsTask |
24
|
|
|
* |
25
|
|
|
* @description Index synonyms to Elastic through a tasks |
26
|
|
|
* @package Firesphere\Elastic\Search |
27
|
|
|
*/ |
28
|
|
|
class ElasticConfigureSynonymsTask extends BuildTask |
29
|
|
|
{ |
30
|
|
|
/** |
31
|
|
|
* URLSegment of this task |
32
|
|
|
* |
33
|
|
|
* @var string |
34
|
|
|
*/ |
35
|
|
|
private static $segment = 'ElasticSynonymTask'; |
|
|
|
|
36
|
|
|
/** |
37
|
|
|
* My name |
38
|
|
|
* |
39
|
|
|
* @var string |
40
|
|
|
*/ |
41
|
|
|
protected $title = 'Add/update synonyms in Elasticsearch'; |
42
|
|
|
/** |
43
|
|
|
* What do I do? |
44
|
|
|
* |
45
|
|
|
* @var string |
46
|
|
|
*/ |
47
|
|
|
protected $description = 'Add or update synonyms to Elastic.'; |
48
|
|
|
|
49
|
|
|
/** |
50
|
|
|
* @throws ClientResponseException |
51
|
|
|
* @throws ServerResponseException |
52
|
|
|
* @throws MissingParameterException |
53
|
|
|
*/ |
54
|
|
|
public function run($request) |
55
|
|
|
{ |
56
|
|
|
$service = Injector::inst()->get(ElasticCoreService::class); |
57
|
|
|
$client = $service->getClient(); |
58
|
|
|
$baseSynonyms = BaseSynonyms::getSynonymsFlattened(); |
59
|
|
|
$baseSynonyms = $this->transformBaseSynonyms($baseSynonyms); |
60
|
|
|
$configuredSynonyms = SearchSynonym::get()->map('getModifiedID', 'getCombinedSynonym')->toArray(); |
61
|
|
|
$configuredSynonyms = $this->transformBaseSynonyms($configuredSynonyms, ''); |
62
|
|
|
// Note, the Elastic synonym class is not suitable for a bulk import |
63
|
|
|
$client->synonyms()->putSynonym([ |
64
|
|
|
'id' => SynonymSet::get()->first()->Key, |
65
|
|
|
'body' => ["synonyms_set" => array_merge($baseSynonyms, $configuredSynonyms)] |
66
|
|
|
]); |
67
|
|
|
} |
68
|
|
|
|
69
|
|
|
private function transformBaseSynonyms(array $synonyms, string $prefix = 'base') |
70
|
|
|
{ |
71
|
|
|
$return = []; |
72
|
|
|
foreach ($synonyms as $key => $synonym) { |
73
|
|
|
if (strlen($prefix) > 0) { |
74
|
|
|
$key = sprintf("%s-%s", $prefix, $key); |
75
|
|
|
} |
76
|
|
|
$return[] = ["id" => $key, "synonyms" => $synonym]; |
77
|
|
|
} |
78
|
|
|
|
79
|
|
|
return $return; |
80
|
|
|
} |
81
|
|
|
} |
82
|
|
|
|