View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *     http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  
18  package org.apache.shardingsphere.infra.binder.segment.with;
19  
20  import lombok.AccessLevel;
21  import lombok.NoArgsConstructor;
22  import org.apache.shardingsphere.infra.binder.segment.expression.impl.SubquerySegmentBinder;
23  import org.apache.shardingsphere.infra.binder.segment.from.TableSegmentBinderContext;
24  import org.apache.shardingsphere.infra.binder.statement.SQLStatementBinderContext;
25  import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.complex.CommonTableExpressionSegment;
26  import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.subquery.SubquerySegment;
27  
28  import java.util.Map;
29  
30  /**
31   * Common table expression segment binder.
32   */
33  @NoArgsConstructor(access = AccessLevel.PRIVATE)
34  public final class CommonTableExpressionSegmentBinder {
35      
36      /**
37       * Bind common table expression segment with metadata.
38       *
39       * @param segment common table expression segment
40       * @param statementBinderContext statement binder context
41       * @param tableBinderContexts table binder contexts
42       * @return bounded common table expression segment
43       */
44      public static CommonTableExpressionSegment bind(final CommonTableExpressionSegment segment, final SQLStatementBinderContext statementBinderContext,
45                                                      final Map<String, TableSegmentBinderContext> tableBinderContexts) {
46          SubquerySegment boundedSubquerySegment = SubquerySegmentBinder.bind(segment.getSubquery(), statementBinderContext, tableBinderContexts);
47          CommonTableExpressionSegment result = new CommonTableExpressionSegment(segment.getStartIndex(), segment.getStopIndex(), segment.getIdentifier(), boundedSubquerySegment);
48          // TODO bind with columns
49          result.getColumns().addAll(segment.getColumns());
50          return result;
51      }
52  }