[shardingsphere] branch master updated: Fix `count mask rule` (#23213)

[jianglongtao]

This is an automated email from the ASF dual-hosted git repository.

jianglongtao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 6b9f510befc Fix `count mask rule` (#23213)
6b9f510befc is described below

commit 6b9f510befc1a41e7597ef50edfe382117f6c26f
Author: Zichao <57847965+qianyi951...@users.noreply.github.com>
AuthorDate: Sat Dec 31 16:13:36 2022 +1300

    Fix `count mask rule` (#23213)
    
    * Fix `count mask rule`
    
    * Fix `count mask rule`
---
 .../mask/distsql/parser/core/MaskDistSQLStatementVisitor.java       | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git 
a/features/mask/distsql/parser/src/main/java/org/apache/shardingsphere/mask/distsql/parser/core/MaskDistSQLStatementVisitor.java
 
b/features/mask/distsql/parser/src/main/java/org/apache/shardingsphere/mask/distsql/parser/core/MaskDistSQLStatementVisitor.java
index f972b450747..a63c48b0114 100644
--- 
a/features/mask/distsql/parser/src/main/java/org/apache/shardingsphere/mask/distsql/parser/core/MaskDistSQLStatementVisitor.java
+++ 
b/features/mask/distsql/parser/src/main/java/org/apache/shardingsphere/mask/distsql/parser/core/MaskDistSQLStatementVisitor.java
@@ -29,6 +29,7 @@ import 
org.apache.shardingsphere.distsql.parser.autogen.MaskDistSQLStatementPars
 import 
org.apache.shardingsphere.distsql.parser.autogen.MaskDistSQLStatementParser.PropertyContext;
 import 
org.apache.shardingsphere.distsql.parser.autogen.MaskDistSQLStatementParser.ShowMaskRulesContext;
 import 
org.apache.shardingsphere.distsql.parser.autogen.MaskDistSQLStatementParser.CountMaskRuleContext;
+import 
org.apache.shardingsphere.distsql.parser.autogen.MaskDistSQLStatementParser.DatabaseNameContext;
 import org.apache.shardingsphere.distsql.parser.segment.AlgorithmSegment;
 import org.apache.shardingsphere.mask.distsql.parser.segment.MaskColumnSegment;
 import org.apache.shardingsphere.mask.distsql.parser.segment.MaskRuleSegment;
@@ -106,4 +107,9 @@ public final class MaskDistSQLStatementVisitor extends 
MaskDistSQLStatementBaseV
         }
         return result;
     }
+
+    @Override
+    public ASTNode visitDatabaseName(final DatabaseNameContext ctx) {
+        return new DatabaseSegment(ctx.getStart().getStartIndex(), 
ctx.getStop().getStopIndex(), new IdentifierValue(ctx.getText()));
+    }
 }