Examples with JsonPath com.jayway.jsonpath.JsonPath used on opensource projects

Example 26 with JsonPath

use of com.jayway.jsonpath.JsonPath in project bender by Nextdoor.

the class JsonPathProvider method setField.

public static void setField(Object jsonObject, Object newVal, String pathStr) {
    JsonPath path = getPath(pathStr);
    path.set(jsonObject, newVal, CONFIG);
}

Example 27 with JsonPath

use of com.jayway.jsonpath.JsonPath in project druid by druid-io.

the class JSONPathParser method generateFieldPaths.

private Map<String, Pair<FieldType, JsonPath>> generateFieldPaths(List<FieldSpec> fieldSpecs) {
    Map<String, Pair<FieldType, JsonPath>> map = new LinkedHashMap<>();
    for (FieldSpec fieldSpec : fieldSpecs) {
        String fieldName = fieldSpec.getName();
        if (map.get(fieldName) != null) {
            throw new IllegalArgumentException("Cannot have duplicate field definition: " + fieldName);
        }
        JsonPath path = fieldSpec.getType() == FieldType.PATH ? JsonPath.compile(fieldSpec.getExpr()) : null;
        Pair<FieldType, JsonPath> pair = new Pair<>(fieldSpec.getType(), path);
        map.put(fieldName, pair);
    }
    return map;
}

Example 28 with JsonPath

use of com.jayway.jsonpath.JsonPath in project druid by druid-io.

the class JSONPathParser method parse.

/**
   * @param input JSON string. The root must be a JSON object, not an array.
   *              e.g., {"valid": "true"} and {"valid":[1,2,3]} are supported
   *              but [{"invalid": "true"}] and [1,2,3] are not.
   *
   * @return A map of field names and values
   */
@Override
public Map<String, Object> parse(String input) {
    try {
        Map<String, Object> map = new LinkedHashMap<>();
        Map<String, Object> document = mapper.readValue(input, new TypeReference<Map<String, Object>>() {
        });
        for (Map.Entry<String, Pair<FieldType, JsonPath>> entry : fieldPathMap.entrySet()) {
            String fieldName = entry.getKey();
            Pair<FieldType, JsonPath> pair = entry.getValue();
            JsonPath path = pair.rhs;
            Object parsedVal;
            if (pair.lhs == FieldType.ROOT) {
                parsedVal = document.get(fieldName);
            } else {
                parsedVal = path.read(document, jsonPathConfig);
            }
            if (parsedVal == null) {
                continue;
            }
            parsedVal = valueConversionFunction(parsedVal);
            map.put(fieldName, parsedVal);
        }
        if (useFieldDiscovery) {
            discoverFields(map, document);
        }
        return map;
    } catch (Exception e) {
        throw new ParseException(e, "Unable to parse row [%s]", input);
    }
}

Example 29 with JsonPath

use of com.jayway.jsonpath.JsonPath in project intellij-community by JetBrains.

the class JsonPathResponseHandler method extractRawValue.

@Nullable
private Object extractRawValue(@NotNull Selector selector, @NotNull String source) throws Exception {
    if (StringUtil.isEmpty(selector.getPath())) {
        return null;
    }
    JsonPath jsonPath = lazyCompile(selector.getPath());
    Object value;
    try {
        value = jsonPath.read(source);
    } catch (InvalidPathException e) {
        throw new Exception(String.format("JsonPath expression '%s' doesn't match", selector.getPath()), e);
    }
    if (value == null) {
        return null;
    }
    return value;
}

Example 30 with JsonPath

use of com.jayway.jsonpath.JsonPath in project nifi by apache.

the class EvaluateJsonPath method onTrigger.

@Override
public void onTrigger(final ProcessContext processContext, final ProcessSession processSession) throws ProcessException {
    FlowFile flowFile = processSession.get();
    if (flowFile == null) {
        return;
    }
    final ComponentLog logger = getLogger();
    DocumentContext documentContext;
    try {
        documentContext = validateAndEstablishJsonContext(processSession, flowFile);
    } catch (InvalidJsonException e) {
        logger.error("FlowFile {} did not have valid JSON content.", new Object[] { flowFile });
        processSession.transfer(flowFile, REL_FAILURE);
        return;
    }
    Set<Map.Entry<String, JsonPath>> attributeJsonPathEntries = attributeToJsonPathEntrySetQueue.poll();
    if (attributeJsonPathEntries == null) {
        attributeJsonPathEntries = processContext.getProperties().entrySet().stream().filter(e -> e.getKey().isDynamic()).collect(Collectors.toMap(e -> e.getKey().getName(), e -> JsonPath.compile(e.getValue()))).entrySet();
    }
    try {
        // We'll only be using this map if destinationIsAttribute == true
        final Map<String, String> jsonPathResults = destinationIsAttribute ? new HashMap<>(attributeJsonPathEntries.size()) : Collections.EMPTY_MAP;
        for (final Map.Entry<String, JsonPath> attributeJsonPathEntry : attributeJsonPathEntries) {
            final String jsonPathAttrKey = attributeJsonPathEntry.getKey();
            final JsonPath jsonPathExp = attributeJsonPathEntry.getValue();
            Object result;
            try {
                Object potentialResult = documentContext.read(jsonPathExp);
                if (returnType.equals(RETURN_TYPE_SCALAR) && !isJsonScalar(potentialResult)) {
                    logger.error("Unable to return a scalar value for the expression {} for FlowFile {}. Evaluated value was {}. Transferring to {}.", new Object[] { jsonPathExp.getPath(), flowFile.getId(), potentialResult.toString(), REL_FAILURE.getName() });
                    processSession.transfer(flowFile, REL_FAILURE);
                    return;
                }
                result = potentialResult;
            } catch (PathNotFoundException e) {
                if (pathNotFound.equals(PATH_NOT_FOUND_WARN)) {
                    logger.warn("FlowFile {} could not find path {} for attribute key {}.", new Object[] { flowFile.getId(), jsonPathExp.getPath(), jsonPathAttrKey }, e);
                }
                if (destinationIsAttribute) {
                    jsonPathResults.put(jsonPathAttrKey, StringUtils.EMPTY);
                    continue;
                } else {
                    processSession.transfer(flowFile, REL_NO_MATCH);
                    return;
                }
            }
            final String resultRepresentation = getResultRepresentation(result, nullDefaultValue);
            if (destinationIsAttribute) {
                jsonPathResults.put(jsonPathAttrKey, resultRepresentation);
            } else {
                flowFile = processSession.write(flowFile, out -> {
                    try (OutputStream outputStream = new BufferedOutputStream(out)) {
                        outputStream.write(resultRepresentation.getBytes(StandardCharsets.UTF_8));
                    }
                });
                processSession.getProvenanceReporter().modifyContent(flowFile, "Replaced content with result of expression " + jsonPathExp.getPath());
            }
        }
        // jsonPathResults map will be empty if this is false
        if (destinationIsAttribute) {
            flowFile = processSession.putAllAttributes(flowFile, jsonPathResults);
        }
        processSession.transfer(flowFile, REL_MATCH);
    } finally {
        attributeToJsonPathEntrySetQueue.offer(attributeJsonPathEntries);
    }
}