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.parser.sql;
19  
20  import com.github.benmanes.caffeine.cache.LoadingCache;
21  import lombok.Getter;
22  import org.apache.shardingsphere.infra.database.core.type.DatabaseType;
23  import org.apache.shardingsphere.infra.parser.cache.SQLStatementCacheBuilder;
24  import org.apache.shardingsphere.sql.parser.api.CacheOption;
25  import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
26  
27  /**
28   * SQL statement parser engine.
29   */
30  public final class SQLStatementParserEngine {
31      
32      private final SQLStatementParserExecutor sqlStatementParserExecutor;
33      
34      private final LoadingCache<String, SQLStatement> sqlStatementCache;
35      
36      @Getter
37      private final CacheOption sqlStatementCacheOption;
38      
39      @Getter
40      private final CacheOption parseTreeCacheOption;
41      
42      public SQLStatementParserEngine(final DatabaseType databaseType, final CacheOption sqlStatementCacheOption, final CacheOption parseTreeCacheOption) {
43          sqlStatementParserExecutor = new SQLStatementParserExecutor(databaseType, parseTreeCacheOption);
44          sqlStatementCache = SQLStatementCacheBuilder.build(databaseType, sqlStatementCacheOption, parseTreeCacheOption);
45          this.sqlStatementCacheOption = sqlStatementCacheOption;
46          this.parseTreeCacheOption = parseTreeCacheOption;
47      }
48      
49      /**
50       * Parse to SQL statement.
51       *
52       * @param sql SQL to be parsed
53       * @param useCache whether to use cache
54       * @return SQL statement
55       */
56      public SQLStatement parse(final String sql, final boolean useCache) {
57          return useCache ? sqlStatementCache.get(sql) : sqlStatementParserExecutor.parse(sql);
58      }
59  }