-
Notifications
You must be signed in to change notification settings - Fork 215
/
PipelineFunctionSplit.java
125 lines (100 loc) · 4.04 KB
/
PipelineFunctionSplit.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
* 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.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) {
// Using the split function in a non-streaming way, returns the first element
return applyStreaming(value, paramsIncludingParentheses, expressionResolver).findFirst()
.orElse(PipelineElement.unresolved());
}
@Override
public Stream<PipelineElement> applyStreaming(final PipelineElement value, final String paramsIncludingParentheses,
final ExpressionResolver expressionResolver) {
final String splitValue = parseAndResolve(paramsIncludingParentheses, expressionResolver);
return value.toOptionalStream()
.flatMap(valueToSplit -> {
if (valueToSplit.contains(splitValue)) {
return Arrays.stream(valueToSplit.split(splitValue)).map(PipelineElement::resolved);
} else {
return Stream.of(PipelineElement.resolved(valueToSplit));
}
});
}
private String parseAndResolve(final String paramsIncludingParentheses,
final ExpressionResolver expressionResolver) {
return parameterResolver.apply(paramsIncludingParentheses, expressionResolver, this)
.toOptional()
.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<String> givenStringDescription;
private SplitFunctionSignature() {
givenStringDescription = new GivenStringParam();
}
@Override
public List<ParameterDefinition<?>> 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<String> {
private GivenStringParam() {
}
@Override
public String getName() {
return "separator";
}
@Override
public Class<String> getType() {
return String.class;
}
@Override
public String getDescription() {
return "Specifies the string to use in order to determine where to split";
}
}
}