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.encrypt.rewrite.condition.impl;
19  
20  import lombok.EqualsAndHashCode;
21  import lombok.Getter;
22  import lombok.ToString;
23  import org.apache.shardingsphere.encrypt.rewrite.condition.EncryptCondition;
24  import org.apache.shardingsphere.sql.parser.statement.core.segment.dml.column.ColumnSegment;
25  import org.apache.shardingsphere.sql.parser.statement.core.segment.dml.expr.ExpressionSegment;
26  import org.apache.shardingsphere.sql.parser.statement.core.segment.dml.expr.simple.LiteralExpressionSegment;
27  import org.apache.shardingsphere.sql.parser.statement.core.segment.dml.expr.simple.ParameterMarkerExpressionSegment;
28  
29  import java.util.LinkedHashMap;
30  import java.util.List;
31  import java.util.Map;
32  
33  /**
34   * Encrypt condition.
35   */
36  @Getter
37  @EqualsAndHashCode
38  @ToString
39  public final class EncryptInCondition implements EncryptCondition {
40      
41      private final ColumnSegment columnSegment;
42      
43      private final String tableName;
44      
45      private final int startIndex;
46      
47      private final int stopIndex;
48      
49      private final Map<Integer, Integer> positionIndexMap = new LinkedHashMap<>();
50      
51      private final Map<Integer, Object> positionValueMap = new LinkedHashMap<>();
52      
53      public EncryptInCondition(final ColumnSegment columnSegment, final String tableName, final int startIndex, final int stopIndex, final List<ExpressionSegment> expressionSegments) {
54          this.columnSegment = columnSegment;
55          this.tableName = tableName;
56          this.startIndex = startIndex;
57          this.stopIndex = stopIndex;
58          int count = 0;
59          for (ExpressionSegment each : expressionSegments) {
60              putPositionMap(count, each);
61              count++;
62          }
63      }
64      
65      private void putPositionMap(final int position, final ExpressionSegment expressionSegment) {
66          if (expressionSegment instanceof ParameterMarkerExpressionSegment) {
67              positionIndexMap.put(position, ((ParameterMarkerExpressionSegment) expressionSegment).getParameterMarkerIndex());
68          } else if (expressionSegment instanceof LiteralExpressionSegment) {
69              positionValueMap.put(position, ((LiteralExpressionSegment) expressionSegment).getLiterals());
70          }
71      }
72  }