如何用ANTLR 4定义数组类型？

arrayType : typeName '[' ']' ;
typeName : 'int' | 'float' | 'string' | ... ;

ARRAY : 'array' ;

import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.*;

public class ArrayTypeParser {
    public static void main(String[] args) throws Exception {
        // 创建ANTLR输入流
        ANTLRInputStream input = new ANTLRInputStream("array[int][]");

        // 创建词法分析器
        ArrayTypeLexer lexer = new ArrayTypeLexer(input);

        // 创建词法记号流
        CommonTokenStream tokens = new CommonTokenStream(lexer);

        // 创建语法解析器
        ArrayTypeParser parser = new ArrayTypeParser(tokens);

        // 解析输入文本，获取语法树
        ParseTree tree = parser.arrayType();

        // 打印语法树
        System.out.println(tree.toStringTree(parser));
    }
}