Examples with ReferenceMap org.graylog2.contentpacks.model.entities.references.ReferenceMap used on opensource projects

Example 1 with ReferenceMap

use of org.graylog2.contentpacks.model.entities.references.ReferenceMap in project graylog2-server by Graylog2.

the class InputFacade method exportNativeEntity.

@VisibleForTesting
Entity exportNativeEntity(InputWithExtractors inputWithExtractors, EntityDescriptorIds entityDescriptorIds) {
    final Input input = inputWithExtractors.input();
    // TODO: Create independent representation of entity?
    final Map<String, ValueReference> staticFields = input.getStaticFields().entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, kv -> ValueReference.of(kv.getValue())));
    final ReferenceMap configuration = toReferenceMap(input.getConfiguration());
    final List<ExtractorEntity> extractors = inputWithExtractors.extractors().stream().map(this::encodeExtractor).collect(Collectors.toList());
    final InputEntity inputEntity = InputEntity.create(ValueReference.of(input.getTitle()), configuration, staticFields, ValueReference.of(input.getType()), ValueReference.of(input.isGlobal()), extractors);
    final JsonNode data = objectMapper.convertValue(inputEntity, JsonNode.class);
    final Set<Constraint> constraints = versionConstraints(input);
    return EntityV1.builder().id(ModelId.of(entityDescriptorIds.getOrThrow(input.getId(), ModelTypes.INPUT_V1))).type(ModelTypes.INPUT_V1).data(data).constraints(ImmutableSet.copyOf(constraints)).build();
}

Example 2 with ReferenceMap

use of org.graylog2.contentpacks.model.entities.references.ReferenceMap in project graylog2-server by Graylog2.

the class ReferenceConverter method convert.

@Override
public Reference convert(JsonNode jsonNode) {
    if (jsonNode.isObject()) {
        final ImmutableSet<String> fieldNames = ImmutableSet.copyOf(jsonNode.fieldNames());
        if (fieldNames.equals(EXPECTED_FIELD_NAMES)) {
            // TODO: Possible to use ValueTypeDeserializer to avoid duplication?
            final String valueTypeText = jsonNode.path(ValueReference.FIELD_TYPE).asText();
            final ValueType valueType = ValueType.valueOf(valueTypeText.toUpperCase(Locale.ROOT));
            final JsonNode value = jsonNode.path(ValueReference.FIELD_VALUE);
            if (valueType == ValueType.BOOLEAN && value.isBoolean()) {
                return ValueReference.of(value.booleanValue());
            } else if (valueType == ValueType.DOUBLE && value.isDouble()) {
                return ValueReference.of(value.doubleValue());
            } else if (valueType == ValueType.FLOAT && value.isFloat()) {
                return ValueReference.of(value.floatValue());
            } else if (valueType == ValueType.INTEGER && value.isInt()) {
                return ValueReference.of(value.intValue());
            } else if (valueType == ValueType.LONG && (value.isLong() || value.isInt())) {
                // Jackson actually creates an int value for a small number so we check for both (long and int value) here
                return ValueReference.of(value.longValue());
            } else if (valueType == ValueType.STRING && value.isTextual()) {
                return ValueReference.of(value.textValue());
            } else if (valueType == ValueType.PARAMETER && value.isTextual()) {
                return ValueReference.createParameter(value.textValue());
            } else {
                return null;
            }
        } else {
            final ImmutableMap.Builder<String, Reference> map = ImmutableMap.builder();
            final Iterator<Map.Entry<String, JsonNode>> fields = jsonNode.fields();
            while (fields.hasNext()) {
                final Map.Entry<String, JsonNode> entry = fields.next();
                map.put(entry.getKey(), convert(entry.getValue()));
            }
            return new ReferenceMap(map.build());
        }
    } else if (jsonNode.isArray()) {
        final ImmutableList.Builder<Reference> list = ImmutableList.builder();
        for (JsonNode value : jsonNode) {
            list.add(convert(value));
        }
        return new ReferenceList(list.build());
    }
    return null;
}