summaryrefslogtreecommitdiff
path: root/www/wiki/extensions/SemanticMediaWiki/src/Elastic/Indexer/Bulk.php
blob: 2392d1a31d58ea83606a9c2b055f6957eca82bd5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
<?php

namespace SMW\Elastic\Indexer;

use SMW\Elastic\Connection\Client as ElasticClient;

/**
 * @license GNU GPL v2+
 * @since 3.0
 *
 * @author mwjames
 */
class Bulk {

	/**
	 * @var ElasticClient
	 */
	private $connection;

	/**
	 * @var array
	 */
	private $bulk = [];

	/**
	 * @var array
	 */
	private $head = [];

	/**
	 * @since 3.0
	 */
	public function __construct( ElasticClient $connection ) {
		$this->connection = $connection;
	}

	/**
	 * @since 3.0
	 */
	public function clear() {
		$this->bulk = [];
		$this->head = [];
	}

	/**
	 * @since 3.0
	 *
	 * @param array $params
	 */
	public function head( array $params ) {
		$this->head = $params;
	}

	/**
	 * @since 3.0
	 *
	 * @param array $params
	 */
	public function delete( array $params ) {
		$this->bulk['body'][] = [ 'delete' => $params + $this->head ];
	}

	/**
	 * @since 3.0
	 *
	 * @param array $params
	 * @param array $source
	 */
	public function index( array $params, array $source ) {
		$this->bulk['body'][] = [ 'index' => $params + $this->head ];
		$this->bulk['body'][] = $source;
	}

	/**
	 * @since 3.0
	 *
	 * @param array $params
	 * @param array $source
	 */
	public function upsert( array $params, array $doc ) {
		$this->bulk['body'][] = [ 'update' => $params + $this->head ];
		$this->bulk['body'][] = [ 'doc' => $doc, "doc_as_upsert" => true ];
	}

	/**
	 * @since 3.0
	 */
	public function execute() {

		$response = $this->connection->bulk(
			$this->bulk
		);

		$this->bulk = [];

		return $response;
	}

	/**
	 * @since 3.0
	 *
	 * @return array
	 */
	public function toJson( $flags = 0 ) {
		return json_encode( $this->bulk, $flags );
	}

}