1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.sql.parser.sql.common.segment.dml.item;
19
20 import lombok.Getter;
21 import lombok.Setter;
22 import org.apache.shardingsphere.sql.parser.sql.common.enums.AggregationType;
23 import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.ExpressionSegment;
24 import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.AliasAvailable;
25 import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.AliasSegment;
26 import org.apache.shardingsphere.sql.parser.sql.common.value.identifier.IdentifierValue;
27
28 import java.util.Collection;
29 import java.util.LinkedList;
30 import java.util.Optional;
31
32
33
34
35 @Getter
36 public class AggregationProjectionSegment implements ProjectionSegment, AliasAvailable, ExpressionSegment {
37
38 private final int startIndex;
39
40 private final int stopIndex;
41
42 private final AggregationType type;
43
44 private final String expression;
45
46 private final Collection<ExpressionSegment> parameters = new LinkedList<>();
47
48 @Setter
49 private AliasSegment alias;
50
51 public AggregationProjectionSegment(final int startIndex, final int stopIndex, final AggregationType type, final String expression) {
52 this.startIndex = startIndex;
53 this.stopIndex = stopIndex;
54 this.type = type;
55 this.expression = expression;
56 }
57
58 @Override
59 public String getColumnLabel() {
60 return getAliasName().orElse(expression);
61 }
62
63 @Override
64 public final Optional<String> getAliasName() {
65 return null == alias ? Optional.empty() : Optional.ofNullable(alias.getIdentifier().getValue());
66 }
67
68 @Override
69 public final Optional<IdentifierValue> getAlias() {
70 return Optional.ofNullable(alias).map(AliasSegment::getIdentifier);
71 }
72
73 @Override
74 public String getText() {
75 return expression;
76 }
77 }