-
Notifications
You must be signed in to change notification settings - Fork 215
/
PipelineFunctionSubstringBefore.java
121 lines (96 loc) · 3.93 KB
/
PipelineFunctionSubstringBefore.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
/*
* Copyright (c) 2021 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.Collections;
import java.util.List;
import java.util.stream.Stream;
import javax.annotation.concurrent.Immutable;
/**
* Provides the {@code fn:substring-before('...')} function implementation.
*/
@Immutable
final class PipelineFunctionSubstringBefore implements PipelineFunction {
private static final String FUNCTION_NAME = "substring-before";
private final PipelineFunctionParameterResolverFactory.SingleParameterResolver parameterResolver =
PipelineFunctionParameterResolverFactory.forStringParameter();
@Override
public String getName() {
return FUNCTION_NAME;
}
@Override
public Signature getSignature() {
return SubstringBeforeFunctionSignature.INSTANCE;
}
@Override
public PipelineElement apply(final PipelineElement value, final String paramsIncludingParentheses,
final ExpressionResolver expressionResolver) {
final String splitValue = parseAndResolve(paramsIncludingParentheses, expressionResolver);
return value.onResolved(previousStage -> {
if (previousStage.contains(splitValue)) {
return PipelineElement.resolved(previousStage.substring(0, previousStage.indexOf(splitValue)));
} else {
return PipelineElement.unresolved();
}
});
}
@Override
public Stream<PipelineElement> applyStreaming(final PipelineElement value, final String paramsIncludingParentheses,
final ExpressionResolver expressionResolver) {
return Stream.of(apply(value, paramsIncludingParentheses, expressionResolver));
}
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 substring-before('givenString')} function.
*/
private static final class SubstringBeforeFunctionSignature implements PipelineFunction.Signature {
private static final SubstringBeforeFunctionSignature INSTANCE = new SubstringBeforeFunctionSignature();
private final PipelineFunction.ParameterDefinition<String> givenStringDescription;
private SubstringBeforeFunctionSignature() {
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 substring-before('givenString')} function.
*/
private static final class GivenStringParam implements ParameterDefinition<String> {
private GivenStringParam() {
}
@Override
public String getName() {
return "givenString";
}
@Override
public Class<String> getType() {
return String.class;
}
@Override
public String getDescription() {
return "Specifies the string to use in order to determine the substring before the first occurrence of that given string";
}
}
}