Merge pull request #847 from wenshao/master

bug fixed for postgresql wall
This commit is contained in:
高铁
2015-06-02 09:23:21 +08:00
2 changed files with 3 additions and 2 deletions
@@ -115,6 +115,7 @@ import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleCreateSequenceStatement;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleMergeStatement;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleMultiInsertStatement;
import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGShowStatement;
import com.alibaba.druid.sql.dialect.sqlserver.ast.stmt.SQLServerCommitStatement;
import com.alibaba.druid.sql.dialect.sqlserver.ast.stmt.SQLServerExecStatement;
import com.alibaba.druid.sql.dialect.sqlserver.ast.stmt.SQLServerInsertStatement;
@@ -2452,7 +2453,7 @@ public class WallVisitorUtils {
allow = config.isDescribeAllow();
denyMessage = "describe not allow";
errorCode = ErrorCode.DESC_NOT_ALLOW;
} else if (x instanceof MySqlShowStatement) {
} else if (x instanceof MySqlShowStatement || x instanceof PGShowStatement) {
allow = config.isShowAllow();
denyMessage = "show not allow";
errorCode = ErrorCode.SHOW_NOT_ALLOW;
@@ -46,7 +46,7 @@ public class PGShowTest extends PGTest {
{
WallProvider provider = new PGWallProvider();
provider.check(sql);
Assert.assertTrue(provider.checkValid(sql));
}
// Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("reviews")));