1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.infra.binder.context.segment.select.projection.impl;
19
20 import lombok.EqualsAndHashCode;
21 import lombok.Getter;
22 import lombok.RequiredArgsConstructor;
23 import lombok.ToString;
24 import org.apache.shardingsphere.infra.binder.context.segment.select.projection.Projection;
25 import org.apache.shardingsphere.sql.parser.sql.common.value.identifier.IdentifierValue;
26
27 import java.util.Collection;
28 import java.util.LinkedList;
29 import java.util.Optional;
30
31
32
33
34 @RequiredArgsConstructor
35 @Getter
36 @EqualsAndHashCode
37 @ToString
38 public final class ShorthandProjection implements Projection {
39
40 private final IdentifierValue owner;
41
42 private final Collection<Projection> actualColumns;
43
44 @Override
45 public String getColumnName() {
46 return null == owner ? "*" : owner.getValue() + ".*";
47 }
48
49 @Override
50 public String getColumnLabel() {
51 return "*";
52 }
53
54 @Override
55 public String getExpression() {
56 return null == owner ? "*" : owner.getValue() + ".*";
57 }
58
59 @Override
60 public Optional<IdentifierValue> getAlias() {
61 return Optional.empty();
62 }
63
64
65
66
67
68
69 public Optional<IdentifierValue> getOwner() {
70 return Optional.ofNullable(owner);
71 }
72
73
74
75
76
77
78 public Collection<ColumnProjection> getColumnProjections() {
79 Collection<ColumnProjection> result = new LinkedList<>();
80 for (Projection each : actualColumns) {
81 if (each instanceof ColumnProjection) {
82 result.add((ColumnProjection) each);
83 }
84 }
85 return result;
86 }
87 }