UNPKG

@shelf/es-painless-fields

Version:

Helpers for bulk update Elasticsearch documents by query using Painless scripts

340 lines (239 loc) 9.27 kB
# es-painless-fields ![CircleCI](https://img.shields.io/circleci/project/github/shelfio/es-painless-fields.svg) ![](https://img.shields.io/badge/code_style-prettier-ff69b4.svg) ![node](https://img.shields.io/node/v/es-painless-fields.svg) > 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 ```js import es from 'elasticsearch'; import painlessFields from '@shelf/es-painless-fields'; const esClient = es.Client(); 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: ```js 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: ```json { "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: ```js 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: ```json { "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: ```js const fieldsReplacements = [ {field: 'a', itemsToRemove: ['1', '2']}, {field: 'b', itemsToRemove: ['3', '4']}, ]; ``` Returns a script which removes items from array. Example: ```json { "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: ```js 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: ```json { "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: ```js 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: ```json { "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: ```js const removeObjectFromArrayParams = { arrayFieldName: 'actors', targetObject: {fieldName: 'id', fieldValue: 'actor-id-1'}, }; ``` Returns a script which removes target object from array. Example: ```json { "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 ```sh $ git checkout master $ yarn version $ yarn publish $ git push origin master --tags ``` ## License MIT © [Shelf](https://shelf.io)