@retter/elasticsearch 8.4.0
Sign up to get free protection for your applications and to get access to all the features.
- package/.dockerignore +5 -0
- package/CHANGELOG.md +3 -0
- package/LICENSE +202 -0
- package/NOTICE.txt +2 -0
- package/README.md +186 -0
- package/codecov.yml +14 -0
- package/index.d.ts +27 -0
- package/index.js +55 -0
- package/lib/api/api/async_search.d.ts +23 -0
- package/lib/api/api/async_search.js +131 -0
- package/lib/api/api/async_search.js.map +1 -0
- package/lib/api/api/autoscaling.d.ts +23 -0
- package/lib/api/api/autoscaling.js +109 -0
- package/lib/api/api/autoscaling.js.map +1 -0
- package/lib/api/api/bulk.d.ts +10 -0
- package/lib/api/api/bulk.js +54 -0
- package/lib/api/api/bulk.js.map +1 -0
- package/lib/api/api/cat.d.ts +89 -0
- package/lib/api/api/cat.js +637 -0
- package/lib/api/api/cat.js.map +1 -0
- package/lib/api/api/ccr.d.ts +50 -0
- package/lib/api/api/ccr.js +320 -0
- package/lib/api/api/ccr.js.map +1 -0
- package/lib/api/api/clear_scroll.d.ts +10 -0
- package/lib/api/api/clear_scroll.js +54 -0
- package/lib/api/api/clear_scroll.js.map +1 -0
- package/lib/api/api/close_point_in_time.d.ts +10 -0
- package/lib/api/api/close_point_in_time.js +53 -0
- package/lib/api/api/close_point_in_time.js.map +1 -0
- package/lib/api/api/cluster.d.ts +56 -0
- package/lib/api/api/cluster.js +392 -0
- package/lib/api/api/cluster.js.map +1 -0
- package/lib/api/api/count.d.ts +10 -0
- package/lib/api/api/count.js +62 -0
- package/lib/api/api/count.js.map +1 -0
- package/lib/api/api/create.d.ts +10 -0
- package/lib/api/api/create.js +46 -0
- package/lib/api/api/create.js.map +1 -0
- package/lib/api/api/dangling_indices.d.ts +20 -0
- package/lib/api/api/dangling_indices.js +85 -0
- package/lib/api/api/dangling_indices.js.map +1 -0
- package/lib/api/api/delete.d.ts +10 -0
- package/lib/api/api/delete.js +39 -0
- package/lib/api/api/delete.js.map +1 -0
- package/lib/api/api/delete_by_query.d.ts +10 -0
- package/lib/api/api/delete_by_query.js +53 -0
- package/lib/api/api/delete_by_query.js.map +1 -0
- package/lib/api/api/delete_by_query_rethrottle.d.ts +10 -0
- package/lib/api/api/delete_by_query_rethrottle.js +39 -0
- package/lib/api/api/delete_by_query_rethrottle.js.map +1 -0
- package/lib/api/api/delete_script.d.ts +10 -0
- package/lib/api/api/delete_script.js +39 -0
- package/lib/api/api/delete_script.js.map +1 -0
- package/lib/api/api/enrich.d.ts +26 -0
- package/lib/api/api/enrich.js +142 -0
- package/lib/api/api/enrich.js.map +1 -0
- package/lib/api/api/eql.d.ts +23 -0
- package/lib/api/api/eql.js +115 -0
- package/lib/api/api/eql.js.map +1 -0
- package/lib/api/api/exists.d.ts +10 -0
- package/lib/api/api/exists.js +39 -0
- package/lib/api/api/exists.js.map +1 -0
- package/lib/api/api/exists_source.d.ts +10 -0
- package/lib/api/api/exists_source.js +39 -0
- package/lib/api/api/exists_source.js.map +1 -0
- package/lib/api/api/explain.d.ts +10 -0
- package/lib/api/api/explain.js +53 -0
- package/lib/api/api/explain.js.map +1 -0
- package/lib/api/api/features.d.ts +17 -0
- package/lib/api/api/features.js +69 -0
- package/lib/api/api/features.js.map +1 -0
- package/lib/api/api/field_caps.d.ts +10 -0
- package/lib/api/api/field_caps.js +61 -0
- package/lib/api/api/field_caps.js.map +1 -0
- package/lib/api/api/fleet.d.ts +20 -0
- package/lib/api/api/fleet.js +113 -0
- package/lib/api/api/fleet.js.map +1 -0
- package/lib/api/api/get.d.ts +10 -0
- package/lib/api/api/get.js +39 -0
- package/lib/api/api/get.js.map +1 -0
- package/lib/api/api/get_script.d.ts +10 -0
- package/lib/api/api/get_script.js +39 -0
- package/lib/api/api/get_script.js.map +1 -0
- package/lib/api/api/get_script_context.d.ts +10 -0
- package/lib/api/api/get_script_context.js +40 -0
- package/lib/api/api/get_script_context.js.map +1 -0
- package/lib/api/api/get_script_languages.d.ts +10 -0
- package/lib/api/api/get_script_languages.js +40 -0
- package/lib/api/api/get_script_languages.js.map +1 -0
- package/lib/api/api/get_source.d.ts +10 -0
- package/lib/api/api/get_source.js +39 -0
- package/lib/api/api/get_source.js.map +1 -0
- package/lib/api/api/graph.d.ts +14 -0
- package/lib/api/api/graph.js +64 -0
- package/lib/api/api/graph.js.map +1 -0
- package/lib/api/api/ilm.d.ts +44 -0
- package/lib/api/api/ilm.js +275 -0
- package/lib/api/api/ilm.js.map +1 -0
- package/lib/api/api/index.d.ts +10 -0
- package/lib/api/api/index.js +54 -0
- package/lib/api/api/index.js.map +1 -0
- package/lib/api/api/indices.d.ts +170 -0
- package/lib/api/api/indices.js +1378 -0
- package/lib/api/api/indices.js.map +1 -0
- package/lib/api/api/info.d.ts +10 -0
- package/lib/api/api/info.js +40 -0
- package/lib/api/api/info.js.map +1 -0
- package/lib/api/api/ingest.d.ts +29 -0
- package/lib/api/api/ingest.js +183 -0
- package/lib/api/api/ingest.js.map +1 -0
- package/lib/api/api/knn_search.d.ts +10 -0
- package/lib/api/api/knn_search.js +53 -0
- package/lib/api/api/knn_search.js.map +1 -0
- package/lib/api/api/license.d.ts +32 -0
- package/lib/api/api/license.js +172 -0
- package/lib/api/api/license.js.map +1 -0
- package/lib/api/api/logstash.d.ts +20 -0
- package/lib/api/api/logstash.js +91 -0
- package/lib/api/api/logstash.js.map +1 -0
- package/lib/api/api/mget.d.ts +10 -0
- package/lib/api/api/mget.js +62 -0
- package/lib/api/api/mget.js.map +1 -0
- package/lib/api/api/migration.d.ts +20 -0
- package/lib/api/api/migration.js +95 -0
- package/lib/api/api/migration.js.map +1 -0
- package/lib/api/api/ml.d.ts +224 -0
- package/lib/api/api/ml.js +1929 -0
- package/lib/api/api/ml.js.map +1 -0
- package/lib/api/api/monitoring.d.ts +14 -0
- package/lib/api/api/monitoring.js +57 -0
- package/lib/api/api/monitoring.js.map +1 -0
- package/lib/api/api/msearch.d.ts +10 -0
- package/lib/api/api/msearch.js +54 -0
- package/lib/api/api/msearch.js.map +1 -0
- package/lib/api/api/msearch_template.d.ts +10 -0
- package/lib/api/api/msearch_template.js +54 -0
- package/lib/api/api/msearch_template.js.map +1 -0
- package/lib/api/api/mtermvectors.d.ts +10 -0
- package/lib/api/api/mtermvectors.js +62 -0
- package/lib/api/api/mtermvectors.js.map +1 -0
- package/lib/api/api/nodes.d.ts +32 -0
- package/lib/api/api/nodes.js +243 -0
- package/lib/api/api/nodes.js.map +1 -0
- package/lib/api/api/open_point_in_time.d.ts +10 -0
- package/lib/api/api/open_point_in_time.js +39 -0
- package/lib/api/api/open_point_in_time.js.map +1 -0
- package/lib/api/api/ping.d.ts +10 -0
- package/lib/api/api/ping.js +40 -0
- package/lib/api/api/ping.js.map +1 -0
- package/lib/api/api/put_script.d.ts +10 -0
- package/lib/api/api/put_script.js +61 -0
- package/lib/api/api/put_script.js.map +1 -0
- package/lib/api/api/rank_eval.d.ts +10 -0
- package/lib/api/api/rank_eval.js +61 -0
- package/lib/api/api/rank_eval.js.map +1 -0
- package/lib/api/api/reindex.d.ts +10 -0
- package/lib/api/api/reindex.js +53 -0
- package/lib/api/api/reindex.js.map +1 -0
- package/lib/api/api/reindex_rethrottle.d.ts +10 -0
- package/lib/api/api/reindex_rethrottle.js +39 -0
- package/lib/api/api/reindex_rethrottle.js.map +1 -0
- package/lib/api/api/render_search_template.d.ts +10 -0
- package/lib/api/api/render_search_template.js +62 -0
- package/lib/api/api/render_search_template.js.map +1 -0
- package/lib/api/api/rollup.d.ts +38 -0
- package/lib/api/api/rollup.js +239 -0
- package/lib/api/api/rollup.js.map +1 -0
- package/lib/api/api/scripts_painless_execute.d.ts +10 -0
- package/lib/api/api/scripts_painless_execute.js +54 -0
- package/lib/api/api/scripts_painless_execute.js.map +1 -0
- package/lib/api/api/scroll.d.ts +10 -0
- package/lib/api/api/scroll.js +53 -0
- package/lib/api/api/scroll.js.map +1 -0
- package/lib/api/api/search.d.ts +10 -0
- package/lib/api/api/search.js +69 -0
- package/lib/api/api/search.js.map +1 -0
- package/lib/api/api/search_mvt.d.ts +10 -0
- package/lib/api/api/search_mvt.js +53 -0
- package/lib/api/api/search_mvt.js.map +1 -0
- package/lib/api/api/search_shards.d.ts +10 -0
- package/lib/api/api/search_shards.js +48 -0
- package/lib/api/api/search_shards.js.map +1 -0
- package/lib/api/api/search_template.d.ts +10 -0
- package/lib/api/api/search_template.js +62 -0
- package/lib/api/api/search_template.js.map +1 -0
- package/lib/api/api/searchable_snapshots.d.ts +23 -0
- package/lib/api/api/searchable_snapshots.js +142 -0
- package/lib/api/api/searchable_snapshots.js.map +1 -0
- package/lib/api/api/security.d.ts +173 -0
- package/lib/api/api/security.js +1344 -0
- package/lib/api/api/security.js.map +1 -0
- package/lib/api/api/shutdown.d.ts +20 -0
- package/lib/api/api/shutdown.js +107 -0
- package/lib/api/api/shutdown.js.map +1 -0
- package/lib/api/api/slm.d.ts +38 -0
- package/lib/api/api/slm.js +214 -0
- package/lib/api/api/slm.js.map +1 -0
- package/lib/api/api/snapshot.d.ts +47 -0
- package/lib/api/api/snapshot.js +315 -0
- package/lib/api/api/snapshot.js.map +1 -0
- package/lib/api/api/sql.d.ts +29 -0
- package/lib/api/api/sql.js +178 -0
- package/lib/api/api/sql.js.map +1 -0
- package/lib/api/api/ssl.d.ts +14 -0
- package/lib/api/api/ssl.js +51 -0
- package/lib/api/api/ssl.js.map +1 -0
- package/lib/api/api/tasks.d.ts +20 -0
- package/lib/api/api/tasks.js +94 -0
- package/lib/api/api/tasks.js.map +1 -0
- package/lib/api/api/terms_enum.d.ts +10 -0
- package/lib/api/api/terms_enum.js +53 -0
- package/lib/api/api/terms_enum.js.map +1 -0
- package/lib/api/api/termvectors.d.ts +10 -0
- package/lib/api/api/termvectors.js +61 -0
- package/lib/api/api/termvectors.js.map +1 -0
- package/lib/api/api/text_structure.d.ts +14 -0
- package/lib/api/api/text_structure.js +57 -0
- package/lib/api/api/text_structure.js.map +1 -0
- package/lib/api/api/transform.d.ts +41 -0
- package/lib/api/api/transform.js +264 -0
- package/lib/api/api/transform.js.map +1 -0
- package/lib/api/api/update.d.ts +10 -0
- package/lib/api/api/update.js +53 -0
- package/lib/api/api/update.js.map +1 -0
- package/lib/api/api/update_by_query.d.ts +10 -0
- package/lib/api/api/update_by_query.js +53 -0
- package/lib/api/api/update_by_query.js.map +1 -0
- package/lib/api/api/update_by_query_rethrottle.d.ts +10 -0
- package/lib/api/api/update_by_query_rethrottle.js +39 -0
- package/lib/api/api/update_by_query_rethrottle.js.map +1 -0
- package/lib/api/api/watcher.d.ts +44 -0
- package/lib/api/api/watcher.js +291 -0
- package/lib/api/api/watcher.js.map +1 -0
- package/lib/api/api/xpack.d.ts +17 -0
- package/lib/api/api/xpack.js +69 -0
- package/lib/api/api/xpack.js.map +1 -0
- package/lib/api/index.d.ts +225 -0
- package/lib/api/index.js +520 -0
- package/lib/api/index.js.map +1 -0
- package/lib/api/types.d.ts +15306 -0
- package/lib/api/types.js +21 -0
- package/lib/api/types.js.map +1 -0
- package/lib/api/typesWithBodyKey.d.ts +15718 -0
- package/lib/api/typesWithBodyKey.js +21 -0
- package/lib/api/typesWithBodyKey.js.map +1 -0
- package/lib/client.d.ts +70 -0
- package/lib/client.js +287 -0
- package/lib/client.js.map +1 -0
- package/lib/helpers.d.ts +149 -0
- package/lib/helpers.js +776 -0
- package/lib/helpers.js.map +1 -0
- package/lib/sniffingTransport.d.ts +4 -0
- package/lib/sniffingTransport.js +56 -0
- package/lib/sniffingTransport.js.map +1 -0
- package/package.json +93 -0
- package/tsconfig.json +37 -0
package/lib/helpers.js
ADDED
@@ -0,0 +1,776 @@
|
|
1
|
+
"use strict";
|
2
|
+
/*
|
3
|
+
* Licensed to Elasticsearch B.V. under one or more contributor
|
4
|
+
* license agreements. See the NOTICE file distributed with
|
5
|
+
* this work for additional information regarding copyright
|
6
|
+
* ownership. Elasticsearch B.V. licenses this file to you under
|
7
|
+
* the Apache License, Version 2.0 (the "License") you may
|
8
|
+
* not use this file except in compliance with the License.
|
9
|
+
* You may obtain a copy of the License at
|
10
|
+
*
|
11
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
12
|
+
*
|
13
|
+
* Unless required by applicable law or agreed to in writing,
|
14
|
+
* software distributed under the License is distributed on an
|
15
|
+
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
16
|
+
* KIND, either express or implied. See the License for the
|
17
|
+
* specific language governing permissions and limitations
|
18
|
+
* under the License.
|
19
|
+
*/
|
20
|
+
var _a, _b, _c;
|
21
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
22
|
+
const tslib_1 = require("tslib");
|
23
|
+
/* eslint-disable @typescript-eslint/naming-convention */
|
24
|
+
/* eslint-disable @typescript-eslint/promise-function-async */
|
25
|
+
/* eslint-disable @typescript-eslint/no-unnecessary-type-assertion */
|
26
|
+
const assert_1 = tslib_1.__importDefault(require("assert"));
|
27
|
+
const util_1 = require("util");
|
28
|
+
const stream_1 = require("stream");
|
29
|
+
const transport_1 = require("@elastic/transport");
|
30
|
+
const { ResponseError, ConfigurationError } = transport_1.errors;
|
31
|
+
const sleep = (0, util_1.promisify)(setTimeout);
|
32
|
+
const pImmediate = (0, util_1.promisify)(setImmediate);
|
33
|
+
/* istanbul ignore next */
|
34
|
+
const noop = () => { };
|
35
|
+
const kClient = Symbol('elasticsearch-client');
|
36
|
+
const kMetaHeader = Symbol('meta header');
|
37
|
+
const kMaxRetries = Symbol('max retries');
|
38
|
+
class Helpers {
|
39
|
+
constructor(opts) {
|
40
|
+
Object.defineProperty(this, _a, {
|
41
|
+
enumerable: true,
|
42
|
+
configurable: true,
|
43
|
+
writable: true,
|
44
|
+
value: void 0
|
45
|
+
});
|
46
|
+
Object.defineProperty(this, _b, {
|
47
|
+
enumerable: true,
|
48
|
+
configurable: true,
|
49
|
+
writable: true,
|
50
|
+
value: void 0
|
51
|
+
});
|
52
|
+
Object.defineProperty(this, _c, {
|
53
|
+
enumerable: true,
|
54
|
+
configurable: true,
|
55
|
+
writable: true,
|
56
|
+
value: void 0
|
57
|
+
});
|
58
|
+
this[kClient] = opts.client;
|
59
|
+
this[kMetaHeader] = opts.metaHeader;
|
60
|
+
this[kMaxRetries] = opts.maxRetries;
|
61
|
+
}
|
62
|
+
/**
|
63
|
+
* Runs a search operation. The only difference between client.search and this utility,
|
64
|
+
* is that we are only returning the hits to the user and not the full ES response.
|
65
|
+
* This helper automatically adds `filter_path=hits.hits._source` to the querystring,
|
66
|
+
* as it will only need the documents source.
|
67
|
+
* @param {object} params - The Elasticsearch's search parameters.
|
68
|
+
* @param {object} options - The client optional configuration for this request.
|
69
|
+
* @return {array} The documents that matched the request.
|
70
|
+
*/
|
71
|
+
async search(params, options = {}) {
|
72
|
+
var _d;
|
73
|
+
appendFilterPath('hits.hits._source', params, true);
|
74
|
+
options.meta = true;
|
75
|
+
const { body: result } = await this[kClient].search(params, options);
|
76
|
+
if (((_d = result.hits) === null || _d === void 0 ? void 0 : _d.hits) != null) {
|
77
|
+
return result.hits.hits.map(d => d._source);
|
78
|
+
}
|
79
|
+
return [];
|
80
|
+
}
|
81
|
+
/**
|
82
|
+
* Runs a scroll search operation. This function returns an async iterator, allowing
|
83
|
+
* the user to use a for await loop to get all the results of a given search.
|
84
|
+
* ```js
|
85
|
+
* for await (const result of client.helpers.scrollSearch({ params })) {
|
86
|
+
* console.log(result)
|
87
|
+
* }
|
88
|
+
* ```
|
89
|
+
* Each result represents the entire body of a single scroll search request,
|
90
|
+
* if you just need to scroll the results, use scrollDocuments.
|
91
|
+
* This function handles automatically retries on 429 status code.
|
92
|
+
* @param {object} params - The Elasticsearch's search parameters.
|
93
|
+
* @param {object} options - The client optional configuration for this request.
|
94
|
+
* @return {iterator} the async iterator
|
95
|
+
*/
|
96
|
+
async *scrollSearch(params, options = {}) {
|
97
|
+
var _d, _e, _f, _g;
|
98
|
+
options.meta = true;
|
99
|
+
if (this[kMetaHeader] !== null) {
|
100
|
+
options.headers = (_d = options.headers) !== null && _d !== void 0 ? _d : {};
|
101
|
+
options.headers['x-elastic-client-meta'] = `${this[kMetaHeader]},h=s`;
|
102
|
+
}
|
103
|
+
const wait = (_e = options.wait) !== null && _e !== void 0 ? _e : 5000;
|
104
|
+
const maxRetries = (_f = options.maxRetries) !== null && _f !== void 0 ? _f : this[kMaxRetries];
|
105
|
+
if (Array.isArray(options.ignore)) {
|
106
|
+
options.ignore.push(429);
|
107
|
+
}
|
108
|
+
else {
|
109
|
+
options.ignore = [429];
|
110
|
+
}
|
111
|
+
params.scroll = (_g = params.scroll) !== null && _g !== void 0 ? _g : '1m';
|
112
|
+
appendFilterPath('_scroll_id', params, false);
|
113
|
+
let response;
|
114
|
+
for (let i = 0; i <= maxRetries; i++) {
|
115
|
+
response = await this[kClient].search(params, options);
|
116
|
+
if (response.statusCode !== 429)
|
117
|
+
break;
|
118
|
+
await sleep(wait);
|
119
|
+
}
|
120
|
+
(0, assert_1.default)(response !== undefined, 'The response is undefined, please file a bug report');
|
121
|
+
if (response.statusCode === 429) {
|
122
|
+
throw new ResponseError(response);
|
123
|
+
}
|
124
|
+
let scroll_id = response.body._scroll_id;
|
125
|
+
let stop = false;
|
126
|
+
const clear = async () => {
|
127
|
+
stop = true;
|
128
|
+
await this[kClient].clearScroll({ scroll_id }, { ignore: [400], ...options });
|
129
|
+
};
|
130
|
+
while (response.body.hits != null && response.body.hits.hits.length > 0) {
|
131
|
+
// scroll id is always present in the response, but it might
|
132
|
+
// change over time based on the number of shards
|
133
|
+
scroll_id = response.body._scroll_id;
|
134
|
+
// @ts-expect-error
|
135
|
+
response.clear = clear;
|
136
|
+
addDocumentsGetter(response);
|
137
|
+
// @ts-expect-error
|
138
|
+
yield response;
|
139
|
+
if (stop) {
|
140
|
+
break;
|
141
|
+
}
|
142
|
+
for (let i = 0; i <= maxRetries; i++) {
|
143
|
+
const r = await this[kClient].scroll({
|
144
|
+
scroll: params.scroll,
|
145
|
+
rest_total_hits_as_int: params.rest_total_hits_as_int,
|
146
|
+
scroll_id
|
147
|
+
}, options);
|
148
|
+
response = r;
|
149
|
+
(0, assert_1.default)(response !== undefined, 'The response is undefined, please file a bug report');
|
150
|
+
if (response.statusCode !== 429)
|
151
|
+
break;
|
152
|
+
await sleep(wait);
|
153
|
+
}
|
154
|
+
if (response.statusCode === 429) {
|
155
|
+
throw new ResponseError(response);
|
156
|
+
}
|
157
|
+
}
|
158
|
+
if (!stop) {
|
159
|
+
await clear();
|
160
|
+
}
|
161
|
+
}
|
162
|
+
/**
|
163
|
+
* Runs a scroll search operation. This function returns an async iterator, allowing
|
164
|
+
* the user to use a for await loop to get all the documents of a given search.
|
165
|
+
* ```js
|
166
|
+
* for await (const document of client.helpers.scrollSearch({ params })) {
|
167
|
+
* console.log(document)
|
168
|
+
* }
|
169
|
+
* ```
|
170
|
+
* Each document is what you will find by running a scrollSearch and iterating on the hits array.
|
171
|
+
* This helper automatically adds `filter_path=hits.hits._source` to the querystring,
|
172
|
+
* as it will only need the documents source.
|
173
|
+
* @param {object} params - The Elasticsearch's search parameters.
|
174
|
+
* @param {object} options - The client optional configuration for this request.
|
175
|
+
* @return {iterator} the async iterator
|
176
|
+
*/
|
177
|
+
async *scrollDocuments(params, options = {}) {
|
178
|
+
appendFilterPath('hits.hits._source', params, true);
|
179
|
+
for await (const { documents } of this.scrollSearch(params, options)) {
|
180
|
+
for (const document of documents) {
|
181
|
+
yield document;
|
182
|
+
}
|
183
|
+
}
|
184
|
+
}
|
185
|
+
/**
|
186
|
+
* Creates a msearch helper instance. Once you configure it, you can use the provided
|
187
|
+
* `search` method to add new searches in the queue.
|
188
|
+
* @param {object} options - The configuration of the msearch operations.
|
189
|
+
* @param {object} reqOptions - The client optional configuration for this request.
|
190
|
+
* @return {object} The possible operations to run.
|
191
|
+
*/
|
192
|
+
msearch(options = {}, reqOptions = {}) {
|
193
|
+
const client = this[kClient];
|
194
|
+
const { operations = 5, concurrency = 5, flushInterval = 500, retries = this[kMaxRetries], wait = 5000, ...msearchOptions } = options;
|
195
|
+
reqOptions.meta = true;
|
196
|
+
let stopReading = false;
|
197
|
+
let stopError = null;
|
198
|
+
let timeoutRef = null;
|
199
|
+
const operationsStream = new stream_1.Readable({
|
200
|
+
objectMode: true,
|
201
|
+
read(size) { }
|
202
|
+
});
|
203
|
+
const p = iterate();
|
204
|
+
const helper = {
|
205
|
+
[Symbol.toStringTag]: 'Promise',
|
206
|
+
then(onFulfilled, onRejected) {
|
207
|
+
return p.then(onFulfilled, onRejected);
|
208
|
+
},
|
209
|
+
catch(onRejected) {
|
210
|
+
return p.catch(onRejected);
|
211
|
+
},
|
212
|
+
finally(onFinally) {
|
213
|
+
return p.finally(onFinally);
|
214
|
+
},
|
215
|
+
stop(error = null) {
|
216
|
+
if (stopReading)
|
217
|
+
return;
|
218
|
+
stopReading = true;
|
219
|
+
stopError = error;
|
220
|
+
operationsStream.push(null);
|
221
|
+
},
|
222
|
+
// TODO: support abort a single search?
|
223
|
+
// NOTE: the validation checks are synchronous and the callback/promise will
|
224
|
+
// be resolved in the same tick. We might want to fix this in the future.
|
225
|
+
search(header, body) {
|
226
|
+
if (stopReading) {
|
227
|
+
const error = stopError === null
|
228
|
+
? new ConfigurationError('The msearch processor has been stopped')
|
229
|
+
: stopError;
|
230
|
+
return Promise.reject(error);
|
231
|
+
}
|
232
|
+
if (!(typeof header === 'object' && header !== null && !Array.isArray(header))) {
|
233
|
+
return Promise.reject(new ConfigurationError('The header should be an object'));
|
234
|
+
}
|
235
|
+
if (!(typeof body === 'object' && body !== null && !Array.isArray(body))) {
|
236
|
+
return Promise.reject(new ConfigurationError('The body should be an object'));
|
237
|
+
}
|
238
|
+
let onFulfilled = null;
|
239
|
+
let onRejected = null;
|
240
|
+
const promise = new Promise((resolve, reject) => {
|
241
|
+
onFulfilled = resolve;
|
242
|
+
onRejected = reject;
|
243
|
+
});
|
244
|
+
const callback = function callback(err, result) {
|
245
|
+
err !== null ? onRejected(err) : onFulfilled(result);
|
246
|
+
};
|
247
|
+
operationsStream.push([header, body, callback]);
|
248
|
+
return promise;
|
249
|
+
}
|
250
|
+
};
|
251
|
+
return helper;
|
252
|
+
async function iterate() {
|
253
|
+
const { semaphore, finish } = buildSemaphore();
|
254
|
+
const msearchBody = [];
|
255
|
+
const callbacks = [];
|
256
|
+
let loadedOperations = 0;
|
257
|
+
timeoutRef = setTimeout(onFlushTimeout, flushInterval); // eslint-disable-line
|
258
|
+
try {
|
259
|
+
for await (const operation of operationsStream) {
|
260
|
+
timeoutRef.refresh();
|
261
|
+
loadedOperations += 1;
|
262
|
+
msearchBody.push(operation[0], operation[1]);
|
263
|
+
callbacks.push(operation[2]);
|
264
|
+
if (loadedOperations >= operations) {
|
265
|
+
const send = await semaphore();
|
266
|
+
send(msearchBody.slice(), callbacks.slice());
|
267
|
+
msearchBody.length = 0;
|
268
|
+
callbacks.length = 0;
|
269
|
+
loadedOperations = 0;
|
270
|
+
}
|
271
|
+
}
|
272
|
+
}
|
273
|
+
finally {
|
274
|
+
clearTimeout(timeoutRef);
|
275
|
+
}
|
276
|
+
// In some cases the previos http call does not have finished,
|
277
|
+
// or we didn't reach the flush bytes threshold, so we force one last operation.
|
278
|
+
if (loadedOperations > 0) {
|
279
|
+
const send = await semaphore();
|
280
|
+
send(msearchBody, callbacks);
|
281
|
+
}
|
282
|
+
await finish();
|
283
|
+
if (stopError !== null) {
|
284
|
+
throw stopError;
|
285
|
+
}
|
286
|
+
async function onFlushTimeout() {
|
287
|
+
if (loadedOperations === 0)
|
288
|
+
return;
|
289
|
+
const msearchBodyCopy = msearchBody.slice();
|
290
|
+
const callbacksCopy = callbacks.slice();
|
291
|
+
msearchBody.length = 0;
|
292
|
+
callbacks.length = 0;
|
293
|
+
loadedOperations = 0;
|
294
|
+
try {
|
295
|
+
const send = await semaphore();
|
296
|
+
send(msearchBodyCopy, callbacksCopy);
|
297
|
+
}
|
298
|
+
catch (err) {
|
299
|
+
/* istanbul ignore next */
|
300
|
+
// @ts-expect-error
|
301
|
+
helper.stop(err);
|
302
|
+
}
|
303
|
+
}
|
304
|
+
}
|
305
|
+
// This function builds a semaphore using the concurrency
|
306
|
+
// options of the msearch helper. It is used inside the iterator
|
307
|
+
// to guarantee that no more than the number of operations
|
308
|
+
// allowed to run at the same time are executed.
|
309
|
+
// It returns a semaphore function which resolves in the next tick
|
310
|
+
// if we didn't reach the maximim concurrency yet, otherwise it returns
|
311
|
+
// a promise that resolves as soon as one of the running request has finshed.
|
312
|
+
// The semaphore function resolves a send function, which will be used
|
313
|
+
// to send the actual msearch request.
|
314
|
+
// It also returns a finish function, which returns a promise that is resolved
|
315
|
+
// when there are no longer request running.
|
316
|
+
function buildSemaphore() {
|
317
|
+
let resolveSemaphore = null;
|
318
|
+
let resolveFinish = null;
|
319
|
+
let running = 0;
|
320
|
+
return { semaphore, finish };
|
321
|
+
function finish() {
|
322
|
+
return new Promise((resolve, reject) => {
|
323
|
+
if (running === 0) {
|
324
|
+
resolve();
|
325
|
+
}
|
326
|
+
else {
|
327
|
+
resolveFinish = resolve;
|
328
|
+
}
|
329
|
+
});
|
330
|
+
}
|
331
|
+
function semaphore() {
|
332
|
+
if (running < concurrency) {
|
333
|
+
running += 1;
|
334
|
+
return pImmediate(send);
|
335
|
+
}
|
336
|
+
else {
|
337
|
+
return new Promise((resolve, reject) => {
|
338
|
+
resolveSemaphore = resolve;
|
339
|
+
});
|
340
|
+
}
|
341
|
+
}
|
342
|
+
function send(msearchBody, callbacks) {
|
343
|
+
/* istanbul ignore if */
|
344
|
+
if (running > concurrency) {
|
345
|
+
throw new Error('Max concurrency reached');
|
346
|
+
}
|
347
|
+
msearchOperation(msearchBody, callbacks, () => {
|
348
|
+
running -= 1;
|
349
|
+
if (resolveSemaphore !== null) {
|
350
|
+
running += 1;
|
351
|
+
resolveSemaphore(send);
|
352
|
+
resolveSemaphore = null;
|
353
|
+
}
|
354
|
+
else if (resolveFinish != null && running === 0) {
|
355
|
+
resolveFinish();
|
356
|
+
}
|
357
|
+
});
|
358
|
+
}
|
359
|
+
}
|
360
|
+
function msearchOperation(msearchBody, callbacks, done) {
|
361
|
+
let retryCount = retries;
|
362
|
+
// Instead of going full on async-await, which would make the code easier to read,
|
363
|
+
// we have decided to use callback style instead.
|
364
|
+
// This because every time we use async await, V8 will create multiple promises
|
365
|
+
// behind the scenes, making the code slightly slower.
|
366
|
+
tryMsearch(msearchBody, callbacks, retrySearch);
|
367
|
+
function retrySearch(msearchBody, callbacks) {
|
368
|
+
if (msearchBody.length > 0 && retryCount > 0) {
|
369
|
+
retryCount -= 1;
|
370
|
+
setTimeout(tryMsearch, wait, msearchBody, callbacks, retrySearch);
|
371
|
+
return;
|
372
|
+
}
|
373
|
+
done();
|
374
|
+
}
|
375
|
+
// This function never returns an error, if the msearch operation fails,
|
376
|
+
// the error is dispatched to all search executors.
|
377
|
+
function tryMsearch(msearchBody, callbacks, done) {
|
378
|
+
client.msearch(Object.assign({}, msearchOptions, { body: msearchBody }), reqOptions)
|
379
|
+
.then(results => {
|
380
|
+
const retryBody = [];
|
381
|
+
const retryCallbacks = [];
|
382
|
+
const { responses } = results.body;
|
383
|
+
for (let i = 0, len = responses.length; i < len; i++) {
|
384
|
+
const response = responses[i];
|
385
|
+
if (response.status === 429 && retryCount > 0) {
|
386
|
+
retryBody.push(msearchBody[i * 2]);
|
387
|
+
retryBody.push(msearchBody[(i * 2) + 1]);
|
388
|
+
retryCallbacks.push(callbacks[i]);
|
389
|
+
continue;
|
390
|
+
}
|
391
|
+
const result = { ...results, body: response };
|
392
|
+
// @ts-expect-error
|
393
|
+
addDocumentsGetter(result);
|
394
|
+
if (response.status != null && response.status >= 400) {
|
395
|
+
callbacks[i](new ResponseError(result), result);
|
396
|
+
}
|
397
|
+
else {
|
398
|
+
callbacks[i](null, result);
|
399
|
+
}
|
400
|
+
}
|
401
|
+
done(retryBody, retryCallbacks);
|
402
|
+
})
|
403
|
+
.catch(err => {
|
404
|
+
for (const callback of callbacks) {
|
405
|
+
callback(err, null);
|
406
|
+
}
|
407
|
+
return done([], []);
|
408
|
+
});
|
409
|
+
}
|
410
|
+
}
|
411
|
+
}
|
412
|
+
/**
|
413
|
+
* Creates a bulk helper instance. Once you configure it, you can pick which operation
|
414
|
+
* to execute with the given dataset, index, create, update, and delete.
|
415
|
+
* @param {object} options - The configuration of the bulk operation.
|
416
|
+
* @param {object} reqOptions - The client optional configuration for this request.
|
417
|
+
* @return {object} The possible operations to run with the datasource.
|
418
|
+
*/
|
419
|
+
bulk(options, reqOptions = {}) {
|
420
|
+
var _d;
|
421
|
+
const client = this[kClient];
|
422
|
+
const { serializer } = client;
|
423
|
+
if (this[kMetaHeader] !== null) {
|
424
|
+
reqOptions.headers = (_d = reqOptions.headers) !== null && _d !== void 0 ? _d : {};
|
425
|
+
reqOptions.headers['x-elastic-client-meta'] = `${this[kMetaHeader]},h=bp`;
|
426
|
+
}
|
427
|
+
reqOptions.meta = true;
|
428
|
+
const { datasource, onDocument, flushBytes = 5000000, flushInterval = 30000, concurrency = 5, retries = this[kMaxRetries], wait = 5000, onDrop = noop, refreshOnCompletion = false, ...bulkOptions } = options;
|
429
|
+
if (datasource === undefined) {
|
430
|
+
// @ts-expect-error
|
431
|
+
return Promise.reject(new ConfigurationError('bulk helper: the datasource is required'));
|
432
|
+
}
|
433
|
+
if (!(Array.isArray(datasource) || Buffer.isBuffer(datasource) || isReadableStream(datasource) || isAsyncIterator(datasource))) {
|
434
|
+
// @ts-expect-error
|
435
|
+
return Promise.reject(new ConfigurationError('bulk helper: the datasource must be an array or a buffer or a readable stream or an async generator'));
|
436
|
+
}
|
437
|
+
if (onDocument === undefined) {
|
438
|
+
// @ts-expect-error
|
439
|
+
return Promise.reject(new ConfigurationError('bulk helper: the onDocument callback is required'));
|
440
|
+
}
|
441
|
+
let shouldAbort = false;
|
442
|
+
let timeoutRef = null;
|
443
|
+
const stats = {
|
444
|
+
total: 0,
|
445
|
+
failed: 0,
|
446
|
+
retry: 0,
|
447
|
+
successful: 0,
|
448
|
+
noop: 0,
|
449
|
+
time: 0,
|
450
|
+
bytes: 0,
|
451
|
+
aborted: false
|
452
|
+
};
|
453
|
+
const p = iterate();
|
454
|
+
const helper = {
|
455
|
+
[Symbol.toStringTag]: 'Promise',
|
456
|
+
then(onFulfilled, onRejected) {
|
457
|
+
return p.then(onFulfilled, onRejected);
|
458
|
+
},
|
459
|
+
catch(onRejected) {
|
460
|
+
return p.catch(onRejected);
|
461
|
+
},
|
462
|
+
finally(onFinally) {
|
463
|
+
return p.finally(onFinally);
|
464
|
+
},
|
465
|
+
get stats() {
|
466
|
+
return stats;
|
467
|
+
},
|
468
|
+
abort() {
|
469
|
+
clearTimeout(timeoutRef);
|
470
|
+
shouldAbort = true;
|
471
|
+
stats.aborted = true;
|
472
|
+
return this;
|
473
|
+
}
|
474
|
+
};
|
475
|
+
return helper;
|
476
|
+
/**
|
477
|
+
* Function that iterates over the given datasource and start a bulk operation as soon
|
478
|
+
* as it reaches the configured bulk size. It's designed to use the Node.js asynchronous
|
479
|
+
* model at this maximum capacity, as it will collect the next body to send while there is
|
480
|
+
* a running http call. In this way, the CPU time will be used carefully.
|
481
|
+
* The objects will be serialized right away, to approximate the byte length of the body.
|
482
|
+
* It creates an array of strings instead of a ndjson string because the bulkOperation
|
483
|
+
* will navigate the body for matching failed operations with the original document.
|
484
|
+
*/
|
485
|
+
async function iterate() {
|
486
|
+
const { semaphore, finish } = buildSemaphore();
|
487
|
+
const startTime = Date.now();
|
488
|
+
const bulkBody = [];
|
489
|
+
let actionBody = '';
|
490
|
+
let payloadBody = '';
|
491
|
+
let chunkBytes = 0;
|
492
|
+
timeoutRef = setTimeout(onFlushTimeout, flushInterval); // eslint-disable-line
|
493
|
+
// @ts-expect-error datasoruce is an iterable
|
494
|
+
for await (const chunk of datasource) {
|
495
|
+
if (shouldAbort)
|
496
|
+
break;
|
497
|
+
timeoutRef.refresh();
|
498
|
+
const action = onDocument(chunk);
|
499
|
+
const operation = Array.isArray(action)
|
500
|
+
? Object.keys(action[0])[0]
|
501
|
+
: Object.keys(action)[0];
|
502
|
+
if (operation === 'index' || operation === 'create') {
|
503
|
+
actionBody = serializer.serialize(action);
|
504
|
+
payloadBody = typeof chunk === 'string' ? chunk : serializer.serialize(chunk);
|
505
|
+
chunkBytes += Buffer.byteLength(actionBody) + Buffer.byteLength(payloadBody);
|
506
|
+
bulkBody.push(actionBody, payloadBody);
|
507
|
+
}
|
508
|
+
else if (operation === 'update') {
|
509
|
+
// @ts-expect-error in case of update action is an array
|
510
|
+
actionBody = serializer.serialize(action[0]);
|
511
|
+
payloadBody = typeof chunk === 'string'
|
512
|
+
? `{"doc":${chunk}}`
|
513
|
+
// @ts-expect-error in case of update action is an array
|
514
|
+
: serializer.serialize({ doc: chunk, ...action[1] });
|
515
|
+
chunkBytes += Buffer.byteLength(actionBody) + Buffer.byteLength(payloadBody);
|
516
|
+
bulkBody.push(actionBody, payloadBody);
|
517
|
+
}
|
518
|
+
else if (operation === 'delete') {
|
519
|
+
actionBody = serializer.serialize(action);
|
520
|
+
chunkBytes += Buffer.byteLength(actionBody);
|
521
|
+
bulkBody.push(actionBody);
|
522
|
+
}
|
523
|
+
else {
|
524
|
+
clearTimeout(timeoutRef);
|
525
|
+
throw new ConfigurationError(`Bulk helper invalid action: '${operation}'`);
|
526
|
+
}
|
527
|
+
if (chunkBytes >= flushBytes) {
|
528
|
+
stats.bytes += chunkBytes;
|
529
|
+
const send = await semaphore();
|
530
|
+
send(bulkBody.slice());
|
531
|
+
bulkBody.length = 0;
|
532
|
+
chunkBytes = 0;
|
533
|
+
}
|
534
|
+
}
|
535
|
+
clearTimeout(timeoutRef);
|
536
|
+
// In some cases the previos http call does not have finished,
|
537
|
+
// or we didn't reach the flush bytes threshold, so we force one last operation.
|
538
|
+
if (!shouldAbort && chunkBytes > 0) {
|
539
|
+
const send = await semaphore();
|
540
|
+
stats.bytes += chunkBytes;
|
541
|
+
send(bulkBody);
|
542
|
+
}
|
543
|
+
await finish();
|
544
|
+
if (refreshOnCompletion !== false) {
|
545
|
+
await client.indices.refresh({
|
546
|
+
index: typeof refreshOnCompletion === 'string'
|
547
|
+
? refreshOnCompletion
|
548
|
+
: '_all'
|
549
|
+
}, reqOptions);
|
550
|
+
}
|
551
|
+
stats.time = Date.now() - startTime;
|
552
|
+
stats.total = stats.successful + stats.failed;
|
553
|
+
return stats;
|
554
|
+
async function onFlushTimeout() {
|
555
|
+
if (chunkBytes === 0)
|
556
|
+
return;
|
557
|
+
stats.bytes += chunkBytes;
|
558
|
+
const bulkBodyCopy = bulkBody.slice();
|
559
|
+
bulkBody.length = 0;
|
560
|
+
chunkBytes = 0;
|
561
|
+
try {
|
562
|
+
const send = await semaphore();
|
563
|
+
send(bulkBodyCopy);
|
564
|
+
}
|
565
|
+
catch (err) {
|
566
|
+
/* istanbul ignore next */
|
567
|
+
helper.abort(); // eslint-disable-line
|
568
|
+
}
|
569
|
+
}
|
570
|
+
}
|
571
|
+
// This function builds a semaphore using the concurrency
|
572
|
+
// options of the bulk helper. It is used inside the iterator
|
573
|
+
// to guarantee that no more than the number of operations
|
574
|
+
// allowed to run at the same time are executed.
|
575
|
+
// It returns a semaphore function which resolves in the next tick
|
576
|
+
// if we didn't reach the maximim concurrency yet, otherwise it returns
|
577
|
+
// a promise that resolves as soon as one of the running request has finshed.
|
578
|
+
// The semaphore function resolves a send function, which will be used
|
579
|
+
// to send the actual bulk request.
|
580
|
+
// It also returns a finish function, which returns a promise that is resolved
|
581
|
+
// when there are no longer request running. It rejects an error if one
|
582
|
+
// of the request has failed for some reason.
|
583
|
+
function buildSemaphore() {
|
584
|
+
let resolveSemaphore = null;
|
585
|
+
let resolveFinish = null;
|
586
|
+
let rejectFinish = null;
|
587
|
+
let error = null;
|
588
|
+
let running = 0;
|
589
|
+
return { semaphore, finish };
|
590
|
+
function finish() {
|
591
|
+
return new Promise((resolve, reject) => {
|
592
|
+
if (running === 0) {
|
593
|
+
if (error !== null) {
|
594
|
+
reject(error);
|
595
|
+
}
|
596
|
+
else {
|
597
|
+
resolve();
|
598
|
+
}
|
599
|
+
}
|
600
|
+
else {
|
601
|
+
resolveFinish = resolve;
|
602
|
+
rejectFinish = reject;
|
603
|
+
}
|
604
|
+
});
|
605
|
+
}
|
606
|
+
function semaphore() {
|
607
|
+
if (running < concurrency) {
|
608
|
+
running += 1;
|
609
|
+
return pImmediate(send);
|
610
|
+
}
|
611
|
+
else {
|
612
|
+
return new Promise((resolve, reject) => {
|
613
|
+
resolveSemaphore = resolve;
|
614
|
+
});
|
615
|
+
}
|
616
|
+
}
|
617
|
+
function send(bulkBody) {
|
618
|
+
/* istanbul ignore if */
|
619
|
+
if (running > concurrency) {
|
620
|
+
throw new Error('Max concurrency reached');
|
621
|
+
}
|
622
|
+
bulkOperation(bulkBody, err => {
|
623
|
+
running -= 1;
|
624
|
+
if (err != null) {
|
625
|
+
shouldAbort = true;
|
626
|
+
error = err;
|
627
|
+
}
|
628
|
+
if (resolveSemaphore !== null) {
|
629
|
+
running += 1;
|
630
|
+
resolveSemaphore(send);
|
631
|
+
resolveSemaphore = null;
|
632
|
+
}
|
633
|
+
else if (resolveFinish != null && rejectFinish != null && running === 0) {
|
634
|
+
if (error != null) {
|
635
|
+
rejectFinish(error);
|
636
|
+
}
|
637
|
+
else {
|
638
|
+
resolveFinish();
|
639
|
+
}
|
640
|
+
}
|
641
|
+
});
|
642
|
+
}
|
643
|
+
}
|
644
|
+
function bulkOperation(bulkBody, callback) {
|
645
|
+
let retryCount = retries;
|
646
|
+
let isRetrying = false;
|
647
|
+
// Instead of going full on async-await, which would make the code easier to read,
|
648
|
+
// we have decided to use callback style instead.
|
649
|
+
// This because every time we use async await, V8 will create multiple promises
|
650
|
+
// behind the scenes, making the code slightly slower.
|
651
|
+
tryBulk(bulkBody, retryDocuments);
|
652
|
+
function retryDocuments(err, bulkBody) {
|
653
|
+
if (err != null)
|
654
|
+
return callback(err);
|
655
|
+
if (shouldAbort)
|
656
|
+
return callback();
|
657
|
+
if (bulkBody.length > 0) {
|
658
|
+
if (retryCount > 0) {
|
659
|
+
isRetrying = true;
|
660
|
+
retryCount -= 1;
|
661
|
+
stats.retry += bulkBody.length;
|
662
|
+
setTimeout(tryBulk, wait, bulkBody, retryDocuments);
|
663
|
+
return;
|
664
|
+
}
|
665
|
+
for (let i = 0, len = bulkBody.length; i < len; i = i + 2) {
|
666
|
+
const operation = Object.keys(serializer.deserialize(bulkBody[i]))[0];
|
667
|
+
onDrop({
|
668
|
+
status: 429,
|
669
|
+
error: null,
|
670
|
+
operation: serializer.deserialize(bulkBody[i]),
|
671
|
+
// @ts-expect-error
|
672
|
+
document: operation !== 'delete'
|
673
|
+
? serializer.deserialize(bulkBody[i + 1])
|
674
|
+
/* istanbul ignore next */
|
675
|
+
: null,
|
676
|
+
retried: isRetrying
|
677
|
+
});
|
678
|
+
stats.failed += 1;
|
679
|
+
}
|
680
|
+
}
|
681
|
+
callback();
|
682
|
+
}
|
683
|
+
function tryBulk(bulkBody, callback) {
|
684
|
+
if (shouldAbort)
|
685
|
+
return callback(null, []);
|
686
|
+
client.bulk(Object.assign({}, bulkOptions, { body: bulkBody }), reqOptions)
|
687
|
+
.then(response => {
|
688
|
+
var _d, _e;
|
689
|
+
const result = response.body;
|
690
|
+
if (!result.errors) {
|
691
|
+
stats.successful += result.items.length;
|
692
|
+
for (const item of result.items) {
|
693
|
+
if (((_d = item.update) === null || _d === void 0 ? void 0 : _d.result) === 'noop') {
|
694
|
+
stats.noop++;
|
695
|
+
}
|
696
|
+
}
|
697
|
+
return callback(null, []);
|
698
|
+
}
|
699
|
+
const retry = [];
|
700
|
+
const { items } = result;
|
701
|
+
for (let i = 0, len = items.length; i < len; i++) {
|
702
|
+
const action = items[i];
|
703
|
+
const operation = Object.keys(action)[0];
|
704
|
+
// @ts-expect-error
|
705
|
+
const responseItem = action[operation];
|
706
|
+
(0, assert_1.default)(responseItem !== undefined, 'The responseItem is undefined, please file a bug report');
|
707
|
+
const indexSlice = operation !== 'delete' ? i * 2 : i;
|
708
|
+
if (responseItem.status >= 400) {
|
709
|
+
// 429 is the only staus code where we might want to retry
|
710
|
+
// a document, because it was not an error in the document itself,
|
711
|
+
// but the ES node were handling too many operations.
|
712
|
+
if (responseItem.status === 429) {
|
713
|
+
retry.push(bulkBody[indexSlice]);
|
714
|
+
/* istanbul ignore next */
|
715
|
+
if (operation !== 'delete') {
|
716
|
+
retry.push(bulkBody[indexSlice + 1]);
|
717
|
+
}
|
718
|
+
}
|
719
|
+
else {
|
720
|
+
onDrop({
|
721
|
+
status: responseItem.status,
|
722
|
+
error: (_e = responseItem.error) !== null && _e !== void 0 ? _e : null,
|
723
|
+
operation: serializer.deserialize(bulkBody[indexSlice]),
|
724
|
+
// @ts-expect-error
|
725
|
+
document: operation !== 'delete'
|
726
|
+
? serializer.deserialize(bulkBody[indexSlice + 1])
|
727
|
+
: null,
|
728
|
+
retried: isRetrying
|
729
|
+
});
|
730
|
+
stats.failed += 1;
|
731
|
+
}
|
732
|
+
}
|
733
|
+
else {
|
734
|
+
stats.successful += 1;
|
735
|
+
}
|
736
|
+
}
|
737
|
+
callback(null, retry);
|
738
|
+
})
|
739
|
+
.catch(err => {
|
740
|
+
callback(err, []);
|
741
|
+
});
|
742
|
+
}
|
743
|
+
}
|
744
|
+
}
|
745
|
+
}
|
746
|
+
exports.default = Helpers;
|
747
|
+
_a = kClient, _b = kMetaHeader, _c = kMaxRetries;
|
748
|
+
// Using a getter will improve the overall performances of the code,
|
749
|
+
// as we will reed the documents only if needed.
|
750
|
+
function addDocumentsGetter(result) {
|
751
|
+
Object.defineProperty(result, 'documents', {
|
752
|
+
get() {
|
753
|
+
var _d;
|
754
|
+
if (((_d = this.body.hits) === null || _d === void 0 ? void 0 : _d.hits) != null) {
|
755
|
+
// @ts-expect-error
|
756
|
+
return this.body.hits.hits.map(d => d._source);
|
757
|
+
}
|
758
|
+
return [];
|
759
|
+
}
|
760
|
+
});
|
761
|
+
}
|
762
|
+
function appendFilterPath(filter, params, force) {
|
763
|
+
if (params.filter_path !== undefined) {
|
764
|
+
params.filter_path += ',' + filter; // eslint-disable-line
|
765
|
+
}
|
766
|
+
else if (force) {
|
767
|
+
params.filter_path = filter;
|
768
|
+
}
|
769
|
+
}
|
770
|
+
function isReadableStream(obj) {
|
771
|
+
return obj != null && typeof obj.pipe === 'function';
|
772
|
+
}
|
773
|
+
function isAsyncIterator(obj) {
|
774
|
+
return (obj === null || obj === void 0 ? void 0 : obj[Symbol.asyncIterator]) != null;
|
775
|
+
}
|
776
|
+
//# sourceMappingURL=helpers.js.map
|