1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.infra.binder.segment.from;
19
20 import com.cedarsoftware.util.CaseInsensitiveMap;
21 import lombok.RequiredArgsConstructor;
22 import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.item.ProjectionSegment;
23 import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.item.ShorthandProjectionSegment;
24
25 import java.util.Collection;
26 import java.util.Map;
27 import java.util.Optional;
28
29
30
31
32 @RequiredArgsConstructor
33 public final class SimpleTableSegmentBinderContext implements TableSegmentBinderContext {
34
35 private final Map<String, ProjectionSegment> columnLabelProjectionSegments;
36
37 public SimpleTableSegmentBinderContext(final Collection<ProjectionSegment> projectionSegments) {
38 columnLabelProjectionSegments = new CaseInsensitiveMap<>(projectionSegments.size(), 1F);
39 projectionSegments.forEach(each -> putColumnLabelProjectionSegments(each, columnLabelProjectionSegments));
40 }
41
42 private void putColumnLabelProjectionSegments(final ProjectionSegment projectionSegment, final Map<String, ProjectionSegment> columnLabelProjectionSegments) {
43 if (projectionSegment instanceof ShorthandProjectionSegment) {
44 ((ShorthandProjectionSegment) projectionSegment).getActualProjectionSegments().forEach(each -> columnLabelProjectionSegments.put(each.getColumnLabel(), each));
45 } else {
46 columnLabelProjectionSegments.put(projectionSegment.getColumnLabel(), projectionSegment);
47 }
48 }
49
50 @Override
51 public Optional<ProjectionSegment> findProjectionSegmentByColumnLabel(final String columnLabel) {
52 return Optional.ofNullable(columnLabelProjectionSegments.get(columnLabel));
53 }
54
55 @Override
56 public Collection<ProjectionSegment> getProjectionSegments() {
57 return columnLabelProjectionSegments.values();
58 }
59 }