org.eclipse.ditto.placeholders.PipelineFunctionSplit Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ditto-placeholders Show documentation
Eclipse Ditto is a framework for creating and managing digital twins in the IoT.
The newest version!
/*
 * Copyright (c) 2022 Contributors to the Eclipse Foundation
 *
 * See the NOTICE file(s) distributed with this work for additional
 * information regarding copyright ownership.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package org.eclipse.ditto.placeholders;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import javax.annotation.concurrent.Immutable;

/**
 * Provides the {@code fn:split('separator')} function implementation.
 */
@Immutable
final class PipelineFunctionSplit implements PipelineFunction {

    private static final String FUNCTION_NAME = "split";

    private final PipelineFunctionParameterResolverFactory.SingleParameterResolver parameterResolver =
            PipelineFunctionParameterResolverFactory.forStringParameter();

    @Override
    public String getName() {
        return FUNCTION_NAME;
    }

    @Override
    public Signature getSignature() {
        return SplitFunctionSignature.INSTANCE;
    }

    @Override
    public PipelineElement apply(final PipelineElement value, final String paramsIncludingParentheses,
            final ExpressionResolver expressionResolver) {

        final String splitValue = parseAndResolve(paramsIncludingParentheses, expressionResolver);
        return PipelineElement.resolved(value.toStream()
                .flatMap(valueToSplit -> {
                    if (valueToSplit.contains(splitValue)) {
                        return Arrays.stream(valueToSplit.split(splitValue));
                    } else {
                        return Stream.of(valueToSplit);
                    }
                })
                .collect(Collectors.toList()));
    }

    private String parseAndResolve(final String paramsIncludingParentheses,
            final ExpressionResolver expressionResolver) {

        return parameterResolver.apply(paramsIncludingParentheses, expressionResolver, this)
                .findFirst()
                .orElseThrow(
                        () -> PlaceholderFunctionSignatureInvalidException.newBuilder(paramsIncludingParentheses, this)
                                .build());
    }

    /**
     * Describes the signature of the {@code split('separator')} function.
     */
    private static final class SplitFunctionSignature implements PipelineFunction.Signature {

        private static final SplitFunctionSignature INSTANCE = new SplitFunctionSignature();

        private final PipelineFunction.ParameterDefinition givenStringDescription;

        private SplitFunctionSignature() {
            givenStringDescription = new GivenStringParam();
        }

        @Override
        public List> getParameterDefinitions() {
            return Collections.singletonList(givenStringDescription);
        }

        @Override
        public String toString() {
            return renderSignature();
        }
    }

    /**
     * Describes the only param of the {@code split('separator')} function.
     */
    private static final class GivenStringParam implements ParameterDefinition {

        private GivenStringParam() {
        }

        @Override
        public String getName() {
            return "separator";
        }

        @Override
        public Class getType() {
            return String.class;
        }

        @Override
        public String getDescription() {
            return "Specifies the string to use in order to determine where to split";
        }
    }

}