From e39c9ede43e807ec55e6dbfbeb058ecdb1d459c7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E9=93=81?= <shaojin.wensj@alibaba-inc.com>
Date: Wed, 7 Oct 2020 00:17:24 +0800
Subject: [PATCH] wall filter add sqlite support

---
 .../druid/wall/spi/SQLiteWallProvider.java    | 38 +++++++++++++++++++
 .../druid/wall/spi/SQLiteWallVisitor.java     | 37 ++++++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100644 src/main/java/com/alibaba/druid/wall/spi/SQLiteWallProvider.java
 create mode 100644 src/main/java/com/alibaba/druid/wall/spi/SQLiteWallVisitor.java

diff --git a/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallProvider.java b/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallProvider.java
new file mode 100644
index 0000000000..fef97dc205
--- /dev/null
+++ b/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallProvider.java
@@ -0,0 +1,38 @@
+package com.alibaba.druid.wall.spi;
+
+import com.alibaba.druid.DbType;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlExportParameterVisitor;
+import com.alibaba.druid.sql.parser.SQLParserFeature;
+import com.alibaba.druid.sql.parser.SQLStatementParser;
+import com.alibaba.druid.sql.visitor.ExportParameterVisitor;
+import com.alibaba.druid.wall.WallConfig;
+import com.alibaba.druid.wall.WallProvider;
+import com.alibaba.druid.wall.WallVisitor;
+
+public class SQLiteWallProvider extends WallProvider {
+    public final static String DEFAULT_CONFIG_DIR = "META-INF/druid/wall/sqlite";
+
+    public SQLiteWallProvider(){
+        this(new WallConfig(DEFAULT_CONFIG_DIR));
+    }
+
+    public SQLiteWallProvider(WallConfig config){
+        super(config, DbType.sqlite);
+    }
+
+    @Override
+    public SQLStatementParser createParser(String sql) {
+        return new MySqlStatementParser(sql, SQLParserFeature.EnableSQLBinaryOpExprGroup);
+    }
+
+    @Override
+    public WallVisitor createWallVisitor() {
+        return new SQLiteWallVisitor(this);
+    }
+
+    @Override
+    public ExportParameterVisitor createExportParameterVisitor() {
+        return new MySqlExportParameterVisitor();
+    }
+}
diff --git a/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallVisitor.java b/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallVisitor.java
new file mode 100644
index 0000000000..4730a395a7
--- /dev/null
+++ b/src/main/java/com/alibaba/druid/wall/spi/SQLiteWallVisitor.java
@@ -0,0 +1,37 @@
+/*
+ * Copyright 1999-2017 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.wall.spi;
+
+import com.alibaba.druid.DbType;
+import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlASTVisitor;
+import com.alibaba.druid.wall.WallProvider;
+import com.alibaba.druid.wall.WallVisitor;
+
+public class SQLiteWallVisitor extends WallVisitorBase implements WallVisitor, MySqlASTVisitor {
+    public SQLiteWallVisitor(WallProvider provider){
+        super (provider);
+    }
+
+    @Override
+    public DbType getDbType() {
+        return DbType.postgresql;
+    }
+
+    public boolean visit(SQLIdentifierExpr x) {
+        return true;
+    }
+}