es-painless-fields
Helpers for bulk update Elasticsearch documents by query using Painless scripts
Install
$ yarn add @shelf/es-painless-fields
Features & Motivation
The main purpose is to utilize _update_by_query
Elasticsearch API most efficiently. API is limited to updating
documents in-place by scripts, so you cannot rely on ES to replace document by passing partial parameters. This package
aims to ease partial bulk document updates.
- In-place set values to fields
- In-place setNotFlattened values to fields
- In-place unset values to fields
- In-place replace values in fields
- In-place increment values in fields
- In-place decrement values in fields
- In-place multiply values in fields
- In-place divide values in fields
- In-place updateObjectInArray - updates object in array
- In-place upsertObjectInArray - upserts object in array
- In-place removeObjectFromArray - removes object from array
- ... to be done
Usage
const esClient = require('elasticsearch').Client();
const painlessFields = require('@shelf/es-painless-fields');
const script = painlessFields.set({a: 1, b: 2});
esClient.updateByQuery({
conflicts: 'proceed',
body: {
query: {match_all: {}},
script,
},
});
API
.set(fieldsMap)
fieldsMap
Type: Object
Object fields which you would like to set. Example: {a: 1, b: 2}
Also can be in a flat form, like {'a.b.c': 1}
.setNotFlattened(fieldsMap, safeMode?: boolean = false)
fieldsMap
Type: Object
, Boolean
Object fields which you would like to set. Example: {a: 1, b: false, c: {x: [{o: null}], y: {z: "name"}}}
Array values will be fully overwritten by array values from fieldsMap
.
You can pass safeMode
as second argument to allow updating deeply nested objects, if they don't exist yet in the ES document.
For example, set the {meta:{a:1}}
when ES doc has no meta
object yet.
.unset(fields)
fields
Type: String[]
Array of field names which you would like to unset. Example: ['a', 'b'']
.increment(fieldsMap)
This library will handle the case if property did not yet exist. It will set the value to the incremented count.
fieldsMap
Type: Object
Object fields which you would like to increment. Example: {a: 1, b: 2}
Also, can be in a flat form, like {'a.b.c': 1}
.decrement(fieldsMap)
It will fallback to setting property value to 0 if it didn't exist yet.
fieldsMap
Type: Object
Object fields which you would like to decrement. Example: {a: 1, b: 2}
Also can be in a flat form, like {'a.b.c': 1}
.multiply(fieldsMap)
It will fallback to setting property value to 0 if it didn't exist yet.
fieldsMap
Type: Object
Object fields which you would like to multiply. Example: {a: 1, b: 2}
Also can be in a flat form, like {'a.b.c': 1}
.divide(fieldsMap)
It will fallback to setting property value to 0 if it didn't exist yet.
fieldsMap
Type: Object
Object fields which you would like to divide. Example: {a: 1, b: 2}
Also can be in a flat form, like {'a.b.c': 1}
.replace(fieldsReplacements)
fieldsReplacements
Type: Array
Array of objects describing what to replace. Example:
const fieldsReplacements = [
{field: 'a', pattern: 'foo', substring: 'bar'},
{field: 'b', pattern: 'hello', substring: 'world'},
];
Returns a script which replaces fields by pattern with substrings. Example:
{
"lang": "painless",
"source": "ctx._source.a = ctx._source.a.replace(params.patterns[0], params.substrings[0]); ctx._source.b = ctx._source.b.replace(params.patterns[1], params.substrings[1]);",
"params": {
"patterns": ["foo", "hello"],
"substrings": ["bar", "world"]
}
}
.replaceSubArray(fieldsReplacements)
fieldsReplacements
Type: Array
Array of objects describing what subarray to replace with new array. Example:
const fieldsReplacements = [
{field: 'a', subArray: ['1', '2'], newArray: ['10', '20']},
{field: 'b', subArray: ['3', '4'], newArray: ['30', '40']},
];
Returns a script which replaces fields by old subarray with new array. Example:
{
"lang": "painless",
"source": "for (int j=0;j<params.subArrays[0].length;j++) { if (ctx._source.a.contains(params.subArrays[0][j])) { ctx._source.a.remove(ctx._source.a.indexOf(params.subArrays[0][j])); } } ctx._source.a.addAll(params.newArrays[0]); for (int j=0;j<params.subArrays[1].length;j++) { if (ctx._source.b.contains(params.subArrays[1][j])) { ctx._source.b.remove(ctx._source.b.indexOf(params.subArrays[1][j])); } } ctx._source.b.addAll(params.newArrays[1]); ",
"params": {
"subArrays": [
["1", "2"],
["3", "4"]
],
"substrings": [
["10", "20"],
["30", "40"]
]
}
}
.removeFromArray(fieldsReplacements)
fieldsReplacements
Type: Array
Array of objects describing what items to remove from which array. Example:
const fieldsReplacements = [
{field: 'a', itemsToRemove: ['1', '2']},
{field: 'b', itemsToRemove: ['3', '4']},
];
Returns a script which removes items from array. Example:
{
"lang": "painless",
"params": {
"itemsToRemoveArrays": [
["1", "2"],
["3", "4"]
]
},
"source": "for (int j=0;j<params.itemsToRemoveArrays[0].length;j++) { if (ctx._source.a.contains(params.itemsToRemoveArrays[0][j])) { ctx._source.a.remove(ctx._source.a.indexOf(params.itemsToRemoveArrays[0][j])); } } for (int j=0;j<params.itemsToRemoveArrays[1].length;j++) { if (ctx._source.b.contains(params.itemsToRemoveArrays[1][j])) { ctx._source.b.remove(ctx._source.b.indexOf(params.itemsToRemoveArrays[1][j])); } }"
}
.updateObjectInArray(updateObjectInArrayParams)
updateObjectInArrayParams
Type: Object
arrayFieldName can be nested: 'data.film.actors'
Parameters required to update object's fields in array. Example:
const updateObjectInArrayParams = {
arrayFieldName: 'actors',
targetObject: {fieldName: 'id', fieldValue: 'actor-id-1'},
fieldsToUpdate: {name: 'Leonardo DiCaprio', hasOscar: true},
};
Returns a script which updates target object's fields in array. Example:
{
"lang": "painless",
"params": {
"fieldsToUpdate": {
"name": "Leonardo DiCaprio",
"hasOscar": true
},
"targetObject": {
"fieldName": "id",
"fieldValue": "actor-id-1"
}
},
"source": "if (ctx._source.actors != null) { def target = ctx._source.actors.find(objectInArray -> objectInArray[params.targetObject.fieldName] == params.targetObject.fieldValue); if (target != null) { for (key in params.fieldsToUpdate.keySet()) { def value = params.fieldsToUpdate[key]; if (target[key] != null && target[key] != value) { target[key] = value; } } } }"
}
.upsertObjectInArray(upsertObjectInArrayParams)
upsertObjectInArrayParams
Type: Object
If the target object exists, its fields will be updated based on fieldsToUpsert. If the target object does not exist, new object will be inserted in array based on fieldsToUpsert.
arrayFieldName can be nested: 'data.film.actors'
Parameters required to upsert object's fields in array. Example:
const upsertObjectInArrayParams = {
arrayFieldName: 'actors',
targetObject: {fieldName: 'id', fieldValue: 'actor-id-1'},
fieldsToUpsert: {name: 'Margot Robbie'},
};
Returns a script which upserts target object's fields in array. Example:
{
"lang": "painless",
"params": {
"fieldsToUpsert": {
"name": "Margot Robbie"
},
"targetObject": {
"fieldName": "id",
"fieldValue": "actor-id-1"
}
},
"source": "if (ctx._source.actors == null) { ctx._source.actors = []; } def target = ctx._source.actors.find(objectInArray -> objectInArray[params.targetObject.fieldName] == params.targetObject.fieldValue); if (target == null) { ctx._source.actors.add(params.fieldsToUpsert); } else { for (key in params.fieldsToUpsert.keySet()) { def value = params.fieldsToUpsert[key]; if (target[key] != null && target[key] != value) { target[key] = value; } } }"
}
.removeObjectFromArray(removeObjectFromArrayParams)
removeObjectFromArrayParams
Type: Object
arrayFieldName can be nested: 'data.film.actors'
Parameters required to remove object from array. Example:
const removeObjectFromArrayParams = {
arrayFieldName: 'actors',
targetObject: {fieldName: 'id', fieldValue: 'actor-id-1'},
};
Returns a script which removes target object from array. Example:
{
"lang": "painless",
"params": {
"targetObject": {
"fieldName": "id",
"fieldValue": "actor-id-1"
}
},
"source": "if (ctx._source.actors != null) { ctx._source.actors.removeIf(objectInArray -> objectInArray[params.targetObject.fieldName] == params.targetObject.fieldValue); }"
}
Publish
$ git checkout master
$ yarn version
$ yarn publish
$ git push origin master --tags
License
MIT © Shelf